refactor: use @babel/types over estree

Author: antfu

README.md

@@ -81,7 +81,7 @@ mod.exports.default.foo.push("b");
 mod.exports.default.foo.unshift("a");
 
 // Generate code
-const { code, map } = generateCode(_module);
+const { code, map } = generateCode(mod);
 ```
 
 Generated code:

build.config.ts

@@ -1,5 +1,5 @@
 import { defineBuildConfig } from "unbuild";
 
 export default defineBuildConfig({
-  externals: ["estree", "@babel/types"],
+  externals: ["@babel/types"],
 });

package.json

@@ -30,7 +30,7 @@
   },
   "dependencies": {
     "@babel/parser": "^7.20.15",
-    "@types/estree": "^1.0.0",
+    "@babel/types": "^7.20.7",
     "recast": "^0.22.0"
   },
   "devDependencies": {

pnpm-lock.yaml

@@ -1,14 +1,26 @@
 import * as recast from "recast";
-import { ESNode, ParsedFileNode } from "../types";
+import { Program } from "@babel/types";
+import { ParsedFileNode } from "../types";
 import { ProxifiedModule } from "./types";
 import { createProxy, literalToAst, proxify } from "./_utils";
 
 export function proxifyModule<T>(ast: ParsedFileNode): ProxifiedModule<T> {
-  const root = ast.program as ESNode;
+  const root = ast.program;
   if (root.type !== "Program") {
     throw new Error(`Cannot proxify ${ast.type} as module`);
   }
 
+  return {
+    $ast: root,
+    $type: "module",
+    exports: createExportsProxy(root),
+    get imports() {
+      throw new Error("Not implemented");
+    },
+  } as any;
+}
+
+function createExportsProxy(root: Program) {
   const findExport = (key: string) => {
     const type =
       key === "default" ? "ExportDefaultDeclaration" : "ExportNamedDeclaration";
@@ -63,7 +75,7 @@ export function proxifyModule<T>(ast: ParsedFileNode): ProxifiedModule<T> {
     );
   };
 
-  const exportsProxy = createProxy(
+  return createProxy(
     root,
     {},
     {
@@ -103,13 +115,4 @@ export function proxifyModule<T>(ast: ParsedFileNode): ProxifiedModule<T> {
       },
     }
   );
-
-  return {
-    $ast: root,
-    $type: "module",
-    exports: exportsProxy,
-    get imports() {
-      throw new Error("Not implemented");
-    },
-  } as any;
 }

src/proxy/module.ts

@@ -10,7 +10,8 @@ export function proxifyObject<T>(node: ESNode): Proxified<T> {
   const getProp = (key: string | symbol) => {
     for (const prop of node.properties) {
       if ("key" in prop && "name" in prop.key && prop.key.name === key) {
-        return prop.value;
+        // TODO:
+        return (prop as any).value;
       }
     }
   };

src/proxy/object.ts

@@ -1,9 +1,9 @@
 import * as recast from "recast";
 
-import type { Node as ESNode } from "estree";
+import type { Program } from "@babel/types";
+export type { Node as ESNode } from "@babel/types";
 
 export * from "./proxy/types";
-export type { Node as ESNode } from "estree";
 
 export interface Loc {
   start?: { line?: number; column?: number; token?: number };
@@ -17,17 +17,9 @@ export interface Token {
   loc?: Loc;
 }
 
-export interface ProgramNode extends recast.types.ASTNode {
-  type: "Program";
-  body: ESNode[];
-  sourceType: "script" | "module";
-  loc: Loc;
-  errors: any[];
-}
-
 export interface ParsedFileNode extends recast.types.ASTNode {
   type: "file";
-  program: ProgramNode;
+  program: Program;
   name?: string;
   loc: Loc;
   comments: null | any;