From d2efa2251bf0e8156a71260ba225998508a8bae7 Mon Sep 17 00:00:00 2001
From: BoykoAlex <aboyko@pivotal.io>
Date: Mon, 4 Nov 2019 11:32:59 -0500
Subject: [PATCH] Port type

---
 src/lib/shared/shapes.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/lib/shared/shapes.ts b/src/lib/shared/shapes.ts
index 18160c3..7d527cf 100644
--- a/src/lib/shared/shapes.ts
+++ b/src/lib/shared/shapes.ts
@@ -123,6 +123,7 @@ joint.shapes.flo.Node = joint.shapes.basic.Generic.extend({
         'stroke-width': 1
       },
       '.input-port': {
+        type: 'input',
         port: 'input',
         height: 8, width: 8,
         magnet: true,
@@ -132,6 +133,7 @@ joint.shapes.flo.Node = joint.shapes.basic.Generic.extend({
         'stroke-width': 1
       },
       '.output-port': {
+        type: 'output',
         port: 'output',
         height: 8, width: 8,
         magnet: true,
@@ -250,7 +252,7 @@ joint.shapes.flo.ElementView = joint.dia.ElementView.extend({
     linkView.notify('link:pointerdown', evt, x, y);
 
     /*** START MAIN DIFF ***/
-    const sourceOrTarget = $(magnet).attr('port') === 'input' ? 'source' : 'target';
+    const sourceOrTarget = $(magnet).attr('type') === 'input' ? 'source' : 'target';
     linkView.eventData(evt, linkView.startArrowheadMove(sourceOrTarget, { whenNotAllowed: 'remove' }));
     /*** END MAIN DIFF ***/