fix: delete connection error

retejs · Apr 10, 2024 · 1593247 · 1593247
1 parent da6cf1a
commit 1593247
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/src/presets/classic/index.ts b/src/presets/classic/index.ts
@@ -32,14 +32,19 @@ export function setup<Schemes extends BSchemes>(nodes: ItemDefinition<Schemes>[]
       key: 'delete',
       async handler() {
         const nodeId = context.id
-        const connections = editor.getConnections().filter(c => {
-          return c.source === nodeId || c.target === nodeId
-        })
 
-        for (const connection of connections) {
-          await editor.removeConnection(connection.id)
+        if (editor.getConnection(nodeId)) {
+          editor.removeConnection(nodeId)
+        } else {
+          const connections = editor.getConnections().filter(c => {
+            return c.source === nodeId || c.target === nodeId
+          })
+
+          for (const connection of connections) {
+            await editor.removeConnection(connection.id)
+          }
+          await editor.removeNode(nodeId)
         }
-        await editor.removeNode(nodeId)
       }
     }