motoko 3.1.3 → 3.2.1

package/src/ast.ts

@@ -16,6 +16,7 @@ export interface Source {
 }
 
 export interface Node extends Partial<Source> {
+    parent?: Node | undefined;
     name: string;
     type?: string;
     doc?: string;
@@ -23,13 +24,18 @@ export interface Node extends Partial<Source> {
     args?: AST[];
 }
 
-export function simplifyAST(ast: CompilerNode): Node;
-export function simplifyAST(ast: CompilerAST[]): AST[];
-export function simplifyAST<T extends CompilerAST>(ast: T): T;
-
-export function simplifyAST(ast: CompilerAST): AST {
+export function simplifyAST(ast: CompilerNode, parent?: Node | undefined): Node;
+export function simplifyAST(
+    ast: CompilerAST[],
+    parent?: Node | undefined,
+): AST[];
+export function simplifyAST<T extends CompilerAST>(
+    ast: T,
+    parent?: Node | undefined,
+): T;
+export function simplifyAST(ast: CompilerAST, parent?: Node | undefined): AST {
     if (Array.isArray(ast)) {
-        return ast.map((a) => simplifyAST(a));
+        return ast.map((a) => simplifyAST(a, parent));
     }
     if (typeof ast !== 'object') {
         return ast;
@@ -40,13 +46,12 @@ export function simplifyAST(ast: CompilerAST): AST {
             CompilerSpan,
             CompilerAST,
         ];
-        const node: Node = {
-            ...(typeof subAst === 'string'
+        const node: Node =
+            typeof subAst === 'string'
                 ? { name: subAst }
-                : simplifyAST(subAst as any as CompilerNode)),
-            start: [+start.args[1], +start.args[2]],
-            end: [+end.args[1], +end.args[2]],
-        };
+                : simplifyAST(subAst as any as CompilerNode, parent);
+        node.start = [+start.args[1], +start.args[2]];
+        node.end = [+end.args[1], +end.args[2]];
         const file = start.args[0];
         if (file) {
             node.file = file;
@@ -58,7 +63,7 @@ export function simplifyAST(ast: CompilerAST): AST {
         return {
             ...(typeof typeAst === 'string'
                 ? { name: typeAst }
-                : simplifyAST(typeAst)),
+                : simplifyAST(typeAst, parent)),
             type,
         };
     }
@@ -67,12 +72,17 @@ export function simplifyAST(ast: CompilerAST): AST {
         return {
             ...(typeof docAst === 'string'
                 ? { name: docAst }
-                : simplifyAST(docAst)),
+                : simplifyAST(docAst, parent)),
             doc,
         };
     }
-    return {
+    const node: Node = {
         name: ast.name,
-        args: simplifyAST(ast.args),
     };
+    Object.defineProperty(node, 'parent', {
+        value: parent,
+        enumerable: false,
+    });
+    node.args = simplifyAST(ast.args, node);
+    return node;
 }