From a55e33f91fa41448ba79feb10c9dff6880ab1d42 Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sun, 21 Nov 2021 21:50:49 +0100
Subject: [PATCH] feat: add draghandle prop to node

* specify a drag handle on the node

Signed-off-by: Braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 src/components/Edges/EdgeText.vue    |  8 ++++----
 src/components/Nodes/NodeWrapper.vue |  1 +
 src/types/components.ts              | 18 +++++++++++++++++-
 src/types/node.ts                    |  1 +
 4 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/src/components/Edges/EdgeText.vue b/src/components/Edges/EdgeText.vue
index 3cdaf865..e2334dd0 100644
--- a/src/components/Edges/EdgeText.vue
+++ b/src/components/Edges/EdgeText.vue
@@ -1,14 +1,14 @@
 <script lang="ts" setup>
-import { CSSProperties, HTMLAttributes } from 'vue'
+import { Component, CSSProperties, DefineComponent } from 'vue'
 
-interface EdgeTextProps extends HTMLAttributes {
+interface EdgeTextProps {
   x: number
   y: number
   label?:
     | string
     | {
-        component: any
-        props?: any
+        component: Component | DefineComponent
+        props?: Record<string, any>
       }
   labelStyle?: CSSProperties
   labelShowBg?: boolean
diff --git a/src/components/Nodes/NodeWrapper.vue b/src/components/Nodes/NodeWrapper.vue
index 000f549b..76bdfce9 100644
--- a/src/components/Nodes/NodeWrapper.vue
+++ b/src/components/Nodes/NodeWrapper.vue
@@ -160,6 +160,7 @@ export default {
 <template>
   <DraggableCore
     cancel=".nodrag"
+    :handle="props.node.dragHandle"
     :disabled="!draggable"
     :scale="scale"
     :grid="props.snapGrid"
diff --git a/src/types/components.ts b/src/types/components.ts
index dcac0529..d5b97074 100644
--- a/src/types/components.ts
+++ b/src/types/components.ts
@@ -1,4 +1,4 @@
-import { CSSProperties } from 'vue'
+import { Component, CSSProperties, DefineComponent } from 'vue'
 import { BackgroundVariant, Dimensions, ElementId, FitViewParams, Position, XYPosition } from './flow'
 import { Connection } from './connection'
 import { Node } from './node'
@@ -56,3 +56,19 @@ export interface MiniMapNodeProps {
   strokeColor?: string
   strokeWidth?: number
 }
+
+export interface EdgeTextProps {
+  x: number
+  y: number
+  label?:
+    | string
+    | {
+        component: Component | DefineComponent
+        props?: Record<string, any>
+      }
+  labelStyle?: CSSProperties
+  labelShowBg?: boolean
+  labelBgStyle?: any
+  labelBgPadding?: [number, number]
+  labelBgBorderRadius?: number
+}
diff --git a/src/types/node.ts b/src/types/node.ts
index 75c0dc78..774e350f 100644
--- a/src/types/node.ts
+++ b/src/types/node.ts
@@ -26,6 +26,7 @@ export interface Node<T = any> {
   draggable?: boolean
   selectable?: boolean
   connectable?: boolean
+  dragHandle?: string
 }
 
 export interface GraphNode<T = any> extends Node<T> {