From 410765b87b24ddab6ff28825f48532ff78188631 Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Wed, 26 Jun 2024 09:36:28 +0200
Subject: [PATCH] feat(core): allow string array or object as class to
 node/edge objects (#1502)

* feat(core): allow string array as class

* chore(changeset): add

* feat(core): allow record as class

* chore(changeset): add
---
 .changeset/thin-beans-wait.md    | 5 +++++
 .changeset/thirty-days-retire.md | 5 +++++
 packages/core/src/types/edge.ts  | 2 +-
 packages/core/src/types/node.ts  | 2 +-
 4 files changed, 12 insertions(+), 2 deletions(-)
 create mode 100644 .changeset/thin-beans-wait.md
 create mode 100644 .changeset/thirty-days-retire.md

diff --git a/.changeset/thin-beans-wait.md b/.changeset/thin-beans-wait.md
new file mode 100644
index 00000000..82633d86
--- /dev/null
+++ b/.changeset/thin-beans-wait.md
@@ -0,0 +1,5 @@
+---
+"@vue-flow/core": minor
+---
+
+Allow passing a Record<string, any> as class to node/edge objects
diff --git a/.changeset/thirty-days-retire.md b/.changeset/thirty-days-retire.md
new file mode 100644
index 00000000..003aa9ef
--- /dev/null
+++ b/.changeset/thirty-days-retire.md
@@ -0,0 +1,5 @@
+---
+"@vue-flow/core": minor
+---
+
+Allow passing an array of strings as class to node/edge objects
diff --git a/packages/core/src/types/edge.ts b/packages/core/src/types/edge.ts
index 535dd0b0..a08ca937 100644
--- a/packages/core/src/types/edge.ts
+++ b/packages/core/src/types/edge.ts
@@ -92,7 +92,7 @@ export interface DefaultEdge<
   /** Disable/enable deleting edge */
   deletable?: boolean
   /** Additional class names, can be a string or a callback returning a string (receives current flow element) */
-  class?: string | ClassFunc<GraphEdge<Data, CustomEvents>>
+  class?: string | string[] | Record<string, any> | ClassFunc<GraphEdge<Data, CustomEvents>>
   /** Additional styles, can be an object or a callback returning an object (receives current flow element) */
   style?: Styles | StyleFunc<GraphEdge<Data, CustomEvents>>
   /** Is edge hidden */
diff --git a/packages/core/src/types/node.ts b/packages/core/src/types/node.ts
index f098de62..582e1ef3 100644
--- a/packages/core/src/types/node.ts
+++ b/packages/core/src/types/node.ts
@@ -90,7 +90,7 @@ export interface Node<Data = ElementData, CustomEvents extends Record<string, Cu
   height?: number | string | HeightFunc
 
   /** Additional class names, can be a string or a callback returning a string (receives current flow element) */
-  class?: string | ClassFunc<GraphNode<Data, CustomEvents>>
+  class?: string | string[] | Record<string, any> | ClassFunc<GraphNode<Data, CustomEvents>>
   /** Additional styles, can be an object or a callback returning an object (receives current flow element) */
   style?: Styles | StyleFunc<GraphNode<Data, CustomEvents>>
   /** Is node hidden */