fix: variables check (#2849)

* fix: variables check * remove log
labring · Oct 1, 2024 · fc6f28f · fc6f28f
1 parent d4b99dd
commit fc6f28f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/packages/global/core/workflow/utils.ts b/packages/global/core/workflow/utils.ts
@@ -298,7 +298,8 @@ export const formatEditorVariablePickerIcon = (
 };
 
 export const isReferenceValue = (value: any, nodeIds: string[]): boolean => {
-  return Array.isArray(value) && value.length === 2 && nodeIds.includes(value[0]);
+  const validIdList = [VARIABLE_NODE_ID, ...nodeIds];
+  return Array.isArray(value) && value.length === 2 && validIdList.includes(value[0]);
 };
 
 export const getElseIFLabel = (i: number) => {

diff --git a/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/hooks/useWorkflow.tsx b/projects/app/src/pages/app/detail/components/WorkflowComponents/Flow/hooks/useWorkflow.tsx
@@ -423,8 +423,8 @@ export const useWorkflow = () => {
     }
 
     // If the node has child nodes, remove the child nodes
-    if (nodes.some((n) => n.data.parentNodeId === node.id)) {
-      const childNodes = nodes.filter((n) => n.data.parentNodeId === node.id);
+    const childNodes = nodes.filter((n) => n.data.parentNodeId === node.id);
+    if (childNodes.length > 0) {
       const childNodeIds = childNodes.map((n) => n.id);
       const childNodesChange = childNodes.map((node) => ({
         ...change,
@@ -440,14 +440,14 @@ export const useWorkflow = () => {
             !childNodeIds.includes(edge.target)
         )
       );
-      return;
     }
 
     setEdges((state) =>
       state.filter((edge) => edge.source !== change.id && edge.target !== change.id)
     );
+    onNodesChange([change]);
 
-    return true;
+    return;
   });
   const handleSelectNode = useMemoizedFn((change: NodeSelectionChange) => {
     // If the node is not selected and the Ctrl key is pressed, select the node
@@ -517,9 +517,8 @@ export const useWorkflow = () => {
       if (change.type === 'remove') {
         const node = nodes.find((n) => n.id === change.id);
         // 如果删除失败，则不继续执行
-        if (node && !handleRemoveNode(change, node)) {
-          return;
-        }
+        node && handleRemoveNode(change, node);
+        return;
       } else if (change.type === 'select') {
         handleSelectNode(change);
       } else if (change.type === 'position') {