fix: in the circuit when internal functions are not imported the impo…

…rt statement list is not updated
EYBlockchain · May 24, 2024 · 838924d · 838924d
1 parent 0415c4c
commit 838924d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/transformers/visitors/circuitInternalFunctionCallVisitor.ts b/src/transformers/visitors/circuitInternalFunctionCallVisitor.ts
@@ -125,6 +125,8 @@ const internalCallVisitor = {
                     state.state.newReturnParameterList.splice(nodeIndex,1);
                   }
                  })
+               } else if (childNode.nodeType === 'ImportStatementList'){
+                state.newImportStatementList = cloneDeep(childNode.imports);            
                }
              })
             if(state.circuitImport[index].isImported ==='true') {
@@ -458,6 +460,8 @@ const internalCallVisitor = {
                        childNode.body.postStatements.splice(!dupNode.index ? 0 : dupNode.index +1, 0, dupNode.node);
                     });
                     reorderBoilerPlate(childNode.body.postStatements);
+                  } else if (childNode.nodeType === 'ImportStatementList'){
+                    childNode.imports = [...new Set([...childNode.imports, ...state.newImportStatementList])];
                   }
 
                  })