From 452494851577d069fee907b30f4a8368f44a3e26 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sat, 8 Apr 2023 19:01:51 +0200
Subject: [PATCH] chore(core): cleanup types

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 packages/core/src/types/edge.ts  | 2 +-
 packages/core/src/types/store.ts | 2 +-
 packages/core/src/utils/drag.ts  | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/core/src/types/edge.ts b/packages/core/src/types/edge.ts
index 3e92e4ce..e989f006 100644
--- a/packages/core/src/types/edge.ts
+++ b/packages/core/src/types/edge.ts
@@ -172,7 +172,7 @@ export interface EdgeProps<Data = ElementData, CustomEvents = {}, Type extends s
   type: Type
   label?: string | VNode | Component<EdgeTextProps> | Object
   style?: CSSProperties
-  selected: boolean
+  selected?: boolean
   sourcePosition: Position
   targetPosition: Position
   sourceHandleId?: string
diff --git a/packages/core/src/types/store.ts b/packages/core/src/types/store.ts
index 54baa49d..9b9fa2d8 100644
--- a/packages/core/src/types/store.ts
+++ b/packages/core/src/types/store.ts
@@ -238,7 +238,7 @@ export interface Actions extends ViewportFunctions {
   /** apply translate extent to d3 */
   setTranslateExtent: (translateExtent: CoordinateExtent) => void
   /** apply extent to nodes */
-  setNodeExtent: (nodeExtent: CoordinateExtent) => void
+  setNodeExtent: (nodeExtent: CoordinateExtent | CoordinateExtentRange) => void
   /** enable/disable node interaction (dragging, selecting etc) */
   setInteractive: (isInteractive: boolean) => void
   /** set new state */
diff --git a/packages/core/src/utils/drag.ts b/packages/core/src/utils/drag.ts
index 60bb6d7a..a603542c 100644
--- a/packages/core/src/utils/drag.ts
+++ b/packages/core/src/utils/drag.ts
@@ -103,7 +103,7 @@ function getParentExtent(
 export function getExtent<T extends NodeDragItem | GraphNode>(
   item: T,
   onError: State['hooks']['error']['trigger'],
-  extent?: CoordinateExtent,
+  extent?: State['nodeExtent'],
   parent?: GraphNode,
 ) {
   let currentExtent = item.extent || extent
@@ -146,7 +146,7 @@ export function calcNextPosition(
   node: GraphNode | NodeDragItem,
   nextPosition: XYPosition,
   onError: State['hooks']['error']['trigger'],
-  nodeExtent?: CoordinateExtent,
+  nodeExtent?: State['nodeExtent'],
   parentNode?: GraphNode,
 ) {
   const extent = getExtent(node, onError, nodeExtent, parentNode)