transform AST with source

Author: fisker

src/angular-parser.ts

@@ -61,6 +61,8 @@ function createAngularParseFunction<
       );
     }
 
+    console.log(result);
+
     return { result, comments, text };
   };
 }

src/ast-transform/index.ts

@@ -1 +1 @@
-export { transform } from './transform.ts';
+export { transform, transformAst } from './transform.ts';

src/ast-transform/transform-ast-with-source.ts

@@ -0,0 +1,8 @@
+import { type ASTWithSource } from '@angular/compiler';
+
+import { type Transformer } from './transform.ts';
+
+export const visitASTWithSource = (
+  node: ASTWithSource,
+  transformer: Transformer,
+) => transformer.transformChild(node.ast);

src/ast-transform/transform-parenthesized-expression.ts

@@ -1,6 +1,6 @@
 import { type ParenthesizedExpression } from '@angular/compiler';
 
-import { type Transformer } from '../transform.ts';
+import { type Transformer } from './transform.ts';
 
 export const visitParenthesizedExpression = (
   node: ParenthesizedExpression,

src/ast-transform/transform-unexpected-node.ts

@@ -1,12 +1,8 @@
-import { type ASTWithSource, type ImplicitReceiver } from '@angular/compiler';
+import { type ImplicitReceiver } from '@angular/compiler';
 
-function transformUnexpectedNode<T extends ASTWithSource | ImplicitReceiver>(
-  node: T,
-) {
+function transformUnexpectedNode<T extends ImplicitReceiver>(node: T) {
   throw new Error(`Unexpected node type '${node.constructor.name}'`);
 }
 
 // Handled in `./transform-member-expression.ts`
 export const visitImplicitReceiver = transformUnexpectedNode<ImplicitReceiver>;
-// Unreachable
-export const visitASTWithSource = transformUnexpectedNode<ASTWithSource>;

src/ast-transform/transform.ts

@@ -87,8 +87,12 @@ class Transformer extends Source {
   }
 }
 
-const transform = (node: angular.AST, text: string) => {
+const transformAst = (node: angular.AST, text: string) => {
   return Transformer.transform(node, text);
 };
 
-export { transform, Transformer };
+const transform = (ast: angular.ASTWithSource) => {
+  return Transformer.transform(ast, ast.source!);
+};
+
+export { transform, transformAst, Transformer };

src/ast-transform/transforms.ts

@@ -1,4 +1,5 @@
 export * from './transform-array-expression.ts';
+export * from './transform-ast-with-source.ts';
 export * from './transform-binary-expression.ts';
 export * from './transform-call-expression.ts';
 export * from './transform-chained-expression.ts';

src/transform-template-binding.ts

@@ -4,7 +4,7 @@ import {
   VariableBinding as NGVariableBinding,
 } from '@angular/compiler';
 
-import { transform as transformAst } from './ast-transform/index.ts';
+import { transformAst } from './ast-transform/index.ts';
 import { Source } from './source.ts';
 import type {
   NGMicrosyntax,

src/transform.ts

@@ -5,12 +5,8 @@ import type {
 import { transform as transformAst } from './ast-transform/index.ts';
 import { transform as transformTemplateBindings } from './transform-template-binding.ts';
 
-function transformAstResult({
-  result: { ast },
-  text,
-  comments,
-}: AstParseResult) {
-  return Object.assign(transformAst(ast, text), { comments });
+function transformAstResult({ result, comments }: AstParseResult) {
+  return Object.assign(transformAst(result), { comments });
 }
 
 function transformMicrosyntaxResult({