From 11a3605c1b9543f22300924d291137e63e1a22ed Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Thu, 28 May 2020 11:30:05 +0200
Subject: [PATCH] refactor(connectionline): handle horizontal flows correctly
 closes #253

---
 src/components/ConnectionLine/index.tsx | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/components/ConnectionLine/index.tsx b/src/components/ConnectionLine/index.tsx
index c2c977ca..9eec2b54 100644
--- a/src/components/ConnectionLine/index.tsx
+++ b/src/components/ConnectionLine/index.tsx
@@ -1,7 +1,7 @@
 import React, { useEffect, useState, CSSProperties } from 'react';
 import cx from 'classnames';
 
-import { ElementId, Node, Transform, HandleElement } from '../../types';
+import { ElementId, Node, Transform, HandleElement, Position } from '../../types';
 
 interface ConnectionLineProps {
   connectionSourceId: ElementId;
@@ -61,9 +61,15 @@ export default ({
   let dAttr: string = '';
 
   if (connectionLineType === 'bezier') {
-    const yOffset = Math.abs(targetY - sourceY) / 2;
-    const centerY = targetY < sourceY ? targetY + yOffset : targetY - yOffset;
-    dAttr = `M${sourceX},${sourceY} C${sourceX},${centerY} ${targetX},${centerY} ${targetX},${targetY}`;
+    if (sourceHandle?.position === Position.Left || sourceHandle?.position === Position.Right) {
+      const xOffset = Math.abs(targetX - sourceX) / 2;
+      const centerX = targetX < sourceX ? targetX + xOffset : targetX - xOffset;
+      dAttr = `M${sourceX},${sourceY} C${centerX},${sourceY} ${centerX},${targetY} ${targetX},${targetY}`;
+    } else {
+      const yOffset = Math.abs(targetY - sourceY) / 2;
+      const centerY = targetY < sourceY ? targetY + yOffset : targetY - yOffset;
+      dAttr = `M${sourceX},${sourceY} C${sourceX},${centerY} ${targetX},${centerY} ${targetX},${targetY}`;
+    }
   } else {
     dAttr = `M${sourceX},${sourceY} ${targetX},${targetY}`;
   }