circuitscript 0.1.31 → 0.1.33

package/dist/cjs/BaseVisitor.js

@@ -19,6 +19,7 @@ const PinDefinition_js_1 = require("./objects/PinDefinition.js");
 class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
     constructor(silent = false, onErrorHandler = null, environment) {
         super();
+        this.filePathStack = [];
         this.silent = false;
         this.printStream = [];
         this.printToConsole = true;
@@ -40,6 +41,7 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
         this.onImportFile = async (visitor, filePath, fileData, onErrorHandler) => {
             throw "Import file not implemented";
         };
+        this.refdesFileAnnotations = new Map();
         this.visitScript = async (ctx) => {
             this.log('===', 'start', '===');
             this.allowParseImports = true;
@@ -527,7 +529,18 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
             specificImports.push(...tmpSpecificImports);
         }
         const id = ctx._moduleName.text;
-        await this.handleImportFile(id, handling, true, ctx, specificImports);
+        const importedFile = await this.handleImportFile(id, handling, true, ctx, specificImports);
+        const ctxImportAnnotation = ctx.import_annotation_expr();
+        if (ctxImportAnnotation) {
+            const textValue = ctxImportAnnotation.getText().replace('#=', '');
+            const { importedModule } = importedFile;
+            if (textValue === 'annotate') {
+                importedModule.enableRefdesAnnotation = true;
+            }
+            else if (textValue === 'annotate-external') {
+                importedModule.enableRefdesAnnotationFile = true;
+            }
+        }
     }
     getReference(ctx) {
         const atomStr = ctx.getText();
@@ -578,6 +591,7 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
                             this.log('create new module context');
                             const importedModule = currentReference.rootValue;
                             const importedModuleContext = importedModule.context;
+                            this.enterFile(importedModule.moduleFilePath);
                             const newExecutor = this.handleEnterContext(this.getExecutor(), this.executionStack, importedModuleContext.name, ctx, {
                                 netNamespace: executor.netNamespace,
                                 namespace: importedModule.moduleNamespace
@@ -589,6 +603,7 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
                         if (isModuleFunction) {
                             this.log('pop module context scope');
                             this.handlePopContext(this.getExecutor(), this.executionStack, "", false);
+                            this.exitFile();
                         }
                         if ((0, utils_js_1.isReference)(functionResult)) {
                             currentReference = functionResult;
@@ -684,7 +699,8 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
         instance.ctxReferences.push({
             ctx,
             indexedStack,
-            creationFlag
+            creationFlag,
+            filePath: this.getCurrentFile(),
         });
         this.componentCtxLinks.set(ctx, instance);
     }
@@ -772,7 +788,7 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
                 hasParseError = importResult.hasParseError;
                 const importContext = executionStack.pop();
                 this.log(`import handling flag: ${importHandling}`);
-                importedModule = new types_js_1.ImportedModule(name, moduleNamespace, filePathUsed, importContext, importHandling, specificImports);
+                importedModule = new types_js_1.ImportedModule(name, moduleNamespace, filePathUsed, importResult.tree, importResult.tokens, importContext, importHandling, specificImports);
                 if (specificImports.length > 0) {
                     this.log('specific import: ' + specificImports.join(', '));
                 }
@@ -781,6 +797,7 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
                 importedModule.context.scope.modules.forEach((module, key) => {
                     scope.modules.set(key, module);
                 });
+                await this.checkModuleHasRefdesFile(filePathUsed);
             }
         }
         catch (err) {
@@ -813,6 +830,12 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
         this.importedFiles.push(newImportedFile);
         return newImportedFile;
     }
+    async checkModuleHasRefdesFile(filePath) {
+        return;
+    }
+    getRefdesFileAnnotation(filePath, startLine, startColumn, stopLine, stopColumn) {
+        return `${filePath}:${startLine}:${startColumn}:${stopLine}:${stopColumn}`;
+    }
     setupDefinedParameters(funcDefinedParameters, passedInParameters, executor) {
         for (let i = 0; i < funcDefinedParameters.length; i++) {
             const tmpFuncArg = funcDefinedParameters[i];
@@ -940,5 +963,16 @@ class BaseVisitor extends CircuitScriptVisitor_js_1.CircuitScriptVisitor {
             return (val instanceof ParamDefinition_js_1.NumericValue);
         }, 'numeric value');
     }
+    enterFile(filePath) {
+        this.log(`enter file: ${filePath}`);
+        this.filePathStack.push(filePath);
+    }
+    exitFile() {
+        this.log(`exit file: ${this.getCurrentFile()}`);
+        this.filePathStack.pop();
+    }
+    getCurrentFile() {
+        return this.filePathStack[this.filePathStack.length - 1];
+    }
 }
 exports.BaseVisitor = BaseVisitor;

package/dist/cjs/RefdesAnnotationVisitor.js

@@ -10,8 +10,7 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
         this.resultText = '';
         this.debug = false;
         this.visitScript = async (ctx) => {
-            const result = this.runExpressions(this.getExecutor(), ctx.expression());
-            this.setResult(ctx, result);
+            this.runExpressions(this.getExecutor(), ctx.expression());
             this.getExecutor().closeOpenPathBlocks();
             this.resultText = this.generateModifiedText();
         };
@@ -43,8 +42,10 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
             });
             if (allRefdes.length > 0) {
                 const originalText = this.getOriginalText(ctx);
-                const annotation = ' #= ' + allRefdes.join(',');
-                this.modifications.set(ctx, originalText + annotation);
+                this.modifications.set(ctx, {
+                    originalText,
+                    refdes: allRefdes,
+                });
             }
         };
         this.visitFunction_def_expr = (ctx) => {
@@ -77,10 +78,10 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
             const instance = this.componentCtxLinks.get(ctx);
             const alreadyHaveRefdesAnnotation = instance.assignedRefDes !== null ?
                 (this.addedRefdesAnnotations.indexOf(instance.assignedRefDes) !== -1) : false;
-            if (!instance.hasParam('refdes')
+            const { forceSaveRefdesAnnotation: forceSaveRefdes } = instance;
+            if (!alreadyHaveRefdesAnnotation && (forceSaveRefdes || (!instance.hasParam('refdes')
                 && instance.placeHolderRefDes === null
-                && instance.assignedRefDes
-                && !alreadyHaveRefdesAnnotation) {
+                && instance.assignedRefDes))) {
                 let useRefDes = instance.assignedRefDes;
                 let isPlaceholderRefdes = false;
                 const { ctxReferences } = instance;
@@ -106,13 +107,25 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
         const refdes = this.generateRefdesAnnotationComment(ctx);
         if (refdes !== null) {
             const originalText = this.getOriginalText(ctx);
-            const annotation = ' #= ' + refdes;
-            this.modifications.set(ctx, originalText + annotation);
+            this.modifications.set(ctx, {
+                originalText,
+                refdes: [refdes]
+            });
         }
     }
     getOutput() {
         return this.resultText;
     }
+    getOutputForExternalRefdesFile() {
+        const result = [];
+        this.modifications.forEach((modification, ctx) => {
+            const { line: startLine, column: startColumn } = ctx.start;
+            const { line: stopLine, column: stopColumn } = ctx.stop;
+            const joinedRefdes = modification.refdes.join(',');
+            result.push(`${startLine}:${startColumn}:${stopLine}:${stopColumn}:${joinedRefdes}`);
+        });
+        return result;
+    }
     generateModifiedText() {
         const output = [];
         const allTokens = this.tokenStream.getTokens();
@@ -140,7 +153,7 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
                         output.push(this.sourceText.substring(lastSourcePos, token.start));
                     }
                     if (this.modifications.has(ctx)) {
-                        output.push(this.modifications.get(ctx));
+                        output.push(this.generateReplacementText(this.modifications.get(ctx)));
                         this.markTokensAsProcessed(ctx, processedTokens);
                         if (ctx.stop) {
                             lastSourcePos = ctx.stop.stop + 1;
@@ -193,5 +206,9 @@ class RefdesAnnotationVisitor extends BaseVisitor_js_1.BaseVisitor {
             console.log(...message);
         }
     }
+    generateReplacementText(modification) {
+        const joinedRefdes = modification.refdes.join(', ');
+        return `${modification.originalText} #= ${joinedRefdes}`;
+    }
 }
 exports.RefdesAnnotationVisitor = RefdesAnnotationVisitor;