From ca33de0556b57591fda157cc3efa3de6e08bbc37 Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sun, 12 Dec 2021 22:54:25 +0100
Subject: [PATCH] fix(edges): add props to smoothstep edge explicitly

* plugin doesn't allow for nested/extended types currently

Signed-off-by: Braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 src/types/edge.ts | 49 ++++++++++++++++++++++++++++++++++++++++++-----
 src/types/node.ts |  7 ++++++-
 2 files changed, 50 insertions(+), 6 deletions(-)

diff --git a/src/types/edge.ts b/src/types/edge.ts
index 3d7fdac1..9a87b208 100644
--- a/src/types/edge.ts
+++ b/src/types/edge.ts
@@ -48,16 +48,21 @@ export interface Edge<T = any> extends Element<T> {
   updatable?: boolean
 }
 
-export interface GraphEdge<T = any> extends Edge<T> {
-  sourceNode: GraphNode
-  targetNode: GraphNode
+export interface GraphEdge<T = any, N = T> extends Edge<T> {
+  sourceNode: GraphNode<N>
+  targetNode: GraphNode<N>
   selected?: boolean
   z?: number
 }
 
 export interface EdgeProps<Data = any> {
   id: string
-  label?: string
+  label?:
+    | string
+    | {
+        props?: any
+        component: any
+      }
   type?: string
   data?: Data
   class?: string
@@ -87,7 +92,41 @@ export interface EdgeProps<Data = any> {
   markerEnd?: string
 }
 
-export interface SmoothStepEdgeProps<T = any> extends EdgeProps<T> {
+export interface SmoothStepEdgeProps<Data = any> extends EdgeProps<Data> {
+  id: string
+  label?:
+    | string
+    | {
+        props?: any
+        component: any
+      }
+  type?: string
+  data?: Data
+  class?: string
+  style?: CSSProperties
+  hidden?: boolean
+  sourceX: number
+  sourceY: number
+  targetX: number
+  targetY: number
+  selected?: boolean
+  sourcePosition: Position
+  targetPosition: Position
+  sourceHandleId?: string
+  targetHandleId?: string
+  source: string
+  target: string
+  sourceHandle?: string
+  targetHandle?: string
+  labelStyle?: any
+  labelShowBg?: boolean
+  labelBgStyle?: any
+  labelBgPadding?: [number, number]
+  labelBgBorderRadius?: number
+  animated?: boolean
+  updatable?: boolean
+  markerStart?: string
+  markerEnd?: string
   borderRadius?: number
 }
 
diff --git a/src/types/node.ts b/src/types/node.ts
index 312436a8..bd6fd76c 100644
--- a/src/types/node.ts
+++ b/src/types/node.ts
@@ -39,7 +39,12 @@ export interface GraphNode<T = any> extends Node<T> {
 }
 
 export interface NodeProps<T = any> {
-  label?: string
+  label?:
+    | string
+    | {
+        props?: any
+        component: any
+      }
   class?: string
   style?: CSSProperties
   hidden?: boolean