fix error on ast state machine

better
checkmarble · Nov 21, 2024 · 9d98843 · 9d98843
1 parent ddff6fb
commit 9d98843
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/packages/app-builder/src/services/editor/ast-editor.tsx b/packages/app-builder/src/services/editor/ast-editor.tsx
@@ -91,10 +91,9 @@ export function useAstNodeEditor({
             name,
           };
           if (isUnaryMainAstOperatorFunction(name)) {
-            const children = newNode.children.slice(0, 1);
-            newNode.children = children;
+            newNode.children = newNChildren(1, newNode.children);
           } else if (isBinaryMainAstOperatorFunction(name)) {
-            newNode.children.push(NewUndefinedAstNode());
+            newNode.children = newNChildren(2, newNode.children);
           }
           set({
             rootAstNode: setAtPath(rootAstNode, path, newNode),
@@ -134,6 +133,13 @@ export function useAstNodeEditor({
   return store;
 }
 
+function newNChildren(n: number, currentChildren: AstNode[]) {
+  return Array.from(
+    { length: n },
+    (_, i) => currentChildren[i] ?? NewUndefinedAstNode(),
+  );
+}
+
 export function AstNodeEditorProvider({
   children,
   store,