refactor(core): use flowElements type as arg for removeSelectedElements

e2e/cypress/component/1-store/elements/removeSelectedElements.cy.ts

@@ -1,4 +1,4 @@
-import type { GraphEdge, GraphNode } from '@vue-flow/core'
+import type { Elements } from '@vue-flow/core'
 import { isNode, useVueFlow } from '@vue-flow/core'
 import { getElements } from '../../../utils'
 
@@ -16,26 +16,23 @@ describe('Store Action: `removeSelectedElements`', () => {
     })
   })
 
-  it('removes passed elements from selected elements in store', () => {
+  it('removes passed elements from selected elements in store', async () => {
     randomNumber = Math.floor(Math.random() * [...nodes, ...edges].length)
     randomNumber2 = Math.floor(Math.random() * randomNumber)
     store.addSelectedElements(Array.from({ length: randomNumber }, (_, i) => store.getElements.value[i]))
     store.removeSelectedElements(
-      Array.from({ length: randomNumber2 }, (_, i) => store.getElements.value[i]).reduce(
-        (acc, curr) => {
-          if (isNode(curr)) {
-            acc.nodes.push(curr)
-          } else {
-            acc.edges.push(curr)
-          }
+      Array.from({ length: randomNumber2 }, (_, i) => store.getElements.value[i]).reduce((acc, curr) => {
+        if (isNode(curr)) {
+          acc.push(curr)
+        } else {
+          acc.push(curr)
+        }
 
-          return acc
-        },
-        { nodes: [] as GraphNode[], edges: [] as GraphEdge[] },
-      ),
+        return acc
+      }, [] as Elements),
     )
 
-    cy.tryAssertion(() => {
+    await cy.tryAssertion(() => {
       expect(store.getSelectedElements.value).to.have.length(randomNumber - randomNumber2)
     })
   })

packages/core/src/types/store.ts

@@ -171,7 +171,7 @@ export interface Actions extends ViewportFunctions {
   /** manually unselect nodes and remove from state */
   removeSelectedNodes: (nodes: GraphNode[]) => void
   /** unselect selected elements (if none are passed, all elements are unselected) */
-  removeSelectedElements: (elements?: { nodes?: GraphNode[]; edges?: GraphEdge[] }) => void
+  removeSelectedElements: (elements?: Elements) => void
   /** apply min zoom value to d3 */
   setMinZoom: (zoom: number) => void
   /** apply max zoom value to d3 */

packages/core/src/utils/node.ts

@@ -37,6 +37,6 @@ export const handleNodeClick = (
   if (!node.selected) {
     addSelectedNodes([node])
   } else if (node.selected && multiSelectionActive) {
-    removeSelectedElements({ nodes: [node] })
+    removeSelectedElements([node])
   }
 }