From f30999ea7aad01823a0d33bb6bf25ae4f9f73507 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Mon, 11 May 2020 22:50:43 +0200
Subject: [PATCH] fix(edges): add step edge to standard types

---
 src/container/EdgeRenderer/utils.ts | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/src/container/EdgeRenderer/utils.ts b/src/container/EdgeRenderer/utils.ts
index b12278d1..0f36d6f8 100644
--- a/src/container/EdgeRenderer/utils.ts
+++ b/src/container/EdgeRenderer/utils.ts
@@ -2,27 +2,23 @@ import { ComponentType } from 'react';
 
 import StraightEdge from '../../components/Edges/StraightEdge';
 import BezierEdge from '../../components/Edges/BezierEdge';
+import StepEdge from '../../components/Edges/StepEdge';
 import wrapEdge from '../../components/Edges/wrapEdge';
 
 import { EdgeTypesType, EdgeCompProps } from '../../types';
 
 export function createEdgeTypes(edgeTypes: EdgeTypesType): EdgeTypesType {
   const standardTypes: EdgeTypesType = {
-    default: wrapEdge((edgeTypes.default || BezierEdge) as ComponentType<
-      EdgeCompProps
-    >),
-    straight: wrapEdge((edgeTypes.bezier || StraightEdge) as ComponentType<
-      EdgeCompProps
-    >),
+    default: wrapEdge((edgeTypes.default || BezierEdge) as ComponentType<EdgeCompProps>),
+    straight: wrapEdge((edgeTypes.bezier || StraightEdge) as ComponentType<EdgeCompProps>),
+    step: wrapEdge((edgeTypes.step || StepEdge) as ComponentType<EdgeCompProps>),
   };
 
   const wrappedTypes = {} as EdgeTypesType;
   const specialTypes: EdgeTypesType = Object.keys(edgeTypes)
-    .filter(k => !['default', 'bezier'].includes(k))
+    .filter((k) => !['default', 'bezier'].includes(k))
     .reduce((res, key) => {
-      res[key] = wrapEdge((edgeTypes[key] || BezierEdge) as ComponentType<
-        EdgeCompProps
-      >);
+      res[key] = wrapEdge((edgeTypes[key] || BezierEdge) as ComponentType<EdgeCompProps>);
 
       return res;
     }, wrappedTypes);