motoko 3.1.3 → 3.2.0

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;
@@ -43,7 +49,7 @@ export function simplifyAST(ast: CompilerAST): AST {
         const node: Node = {
             ...(typeof subAst === 'string'
                 ? { name: subAst }
-                : simplifyAST(subAst as any as CompilerNode)),
+                : simplifyAST(subAst as any as CompilerNode, parent)),
             start: [+start.args[1], +start.args[2]],
             end: [+end.args[1], +end.args[2]],
         };
@@ -58,7 +64,7 @@ export function simplifyAST(ast: CompilerAST): AST {
         return {
             ...(typeof typeAst === 'string'
                 ? { name: typeAst }
-                : simplifyAST(typeAst)),
+                : simplifyAST(typeAst, parent)),
             type,
         };
     }
@@ -67,12 +73,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;
 }