refactor(babel): use walkAST instead of @babel/traverse

Author: zhiyuanzmj

packages/babel/package.json

@@ -43,12 +43,13 @@
     "@babel/core": "catalog:",
     "@babel/parser": "catalog:",
     "@babel/plugin-syntax-jsx": "catalog:",
-    "@babel/traverse": "catalog:",
     "@babel/types": "catalog:",
     "@vue-jsx-vapor/compiler": "workspace:*",
+    "ast-kit": "catalog:",
     "source-map-js": "catalog:"
   },
   "devDependencies": {
+    "@babel/traverse": "catalog:",
     "@types/babel__core": "catalog:",
     "@types/babel__generator": "^7.27.0",
     "@types/babel__template": "^7.4.4",

packages/babel/src/transform.ts

@@ -1,13 +1,11 @@
 import { parse } from '@babel/parser'
-import _traverse, { type VisitNodeFunction } from '@babel/traverse'
 import { compile } from '@vue-jsx-vapor/compiler'
+import { walkAST } from 'ast-kit'
 import { SourceMapConsumer } from 'source-map-js'
 import { isJSXElement } from './utils'
 import type { Options } from '.'
-import type { JSXElement, JSXFragment } from '@babel/types'
-
-// @ts-ignore
-const traverse = (_traverse.default || _traverse) as typeof _traverse
+import type { VisitNodeFunction } from '@babel/traverse'
+import type { JSXElement, JSXFragment, Node } from '@babel/types'
 
 export const transformJSX: VisitNodeFunction<
   Options,
@@ -39,15 +37,19 @@ export const transformJSX: VisitNodeFunction<
 
   if (map) {
     const consumer = new SourceMapConsumer(map)
-    traverse(ast, {
-      Identifier({ node: id }) {
-        if (!id.loc) return
-        const originalLoc = consumer.originalPositionFor(id.loc.start)
-        if (originalLoc.column) {
-          id.loc.start.line = originalLoc.line
-          id.loc.start.column = originalLoc.column
-          id.loc.end.line = originalLoc.line
-          id.loc.end.column = originalLoc.column + id.name.length
+    walkAST<Node>(ast, {
+      enter(id) {
+        if (
+          (id.type === 'Identifier' || id.type === 'JSXIdentifier') &&
+          id.loc
+        ) {
+          const originalLoc = consumer.originalPositionFor(id.loc.start)
+          if (originalLoc.column) {
+            id.loc.start.line = originalLoc.line
+            id.loc.start.column = originalLoc.column
+            id.loc.end.line = originalLoc.line
+            id.loc.end.column = originalLoc.column + id.name.length
+          }
         }
       },
     })

packages/compiler/package.json

@@ -44,7 +44,7 @@
     "@babel/types": "catalog:",
     "@vue/compiler-dom": "catalog:",
     "@vue/shared": "catalog:",
-    "ast-kit": "^2.1.1",
+    "ast-kit": "catalog:",
     "source-map-js": "catalog:"
   }
 }

pnpm-lock.yaml

@@ -36,6 +36,7 @@ catalog:
   ts-macro: *ts-macro
   unplugin: ^2.3.5
   unplugin-utils: ^0.2.4
+  ast-kit: ^2.1.1
   vite: ^7.0.2
   vitest: ^3.2.4
   vue: https://pkg.pr.new/vue@13642