move recast

jycouet · Oct 26, 2024 · 33cdaa0 · 33cdaa0
1 parent db732f5
commit 33cdaa0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 15 deletions.
diff --git a/packages/internals/src/lib/ast/ast.ts b/packages/internals/src/lib/ast/ast.ts
@@ -1,4 +1,3 @@
-// import { prettyPrint } from 'recast'
 import { parse } from '@babel/parser'
 import * as recast from 'recast'
 import { prettyPrint as recastPrettyPrint } from 'recast'
@@ -83,4 +82,4 @@ export function parseTs(source: string | null) {
   return parsed
 }
 
-export const { visit } = recast.types
+export const { visit, builders } = recast.types
diff --git a/packages/internals/src/lib/index.ts b/packages/internals/src/lib/index.ts
@@ -7,5 +7,5 @@ export {
   getRelativePackagePath,
   findFileOrUp,
 } from './fs/fs.js'
-export { parseTs, extractHtmlElementAttr_Text, visit, prettyPrint } from './ast/ast.js'
+export { parseTs, extractHtmlElementAttr_Text, visit, prettyPrint, builders } from './ast/ast.js'
 export type { Statement } from './ast/ast.js'
diff --git a/packages/vite-plugin-stripper/src/lib/transformPackage.ts b/packages/vite-plugin-stripper/src/lib/transformPackage.ts
@@ -1,17 +1,8 @@
-import { parse } from '@babel/parser'
-import * as recast from 'recast'
-
-const {
-  visit,
-  types: { builders },
-} = recast
+import { builders, parseTs, prettyPrint, visit } from '@kitql/internals'
 
 export const removePackages = async (code: string, packages_to_strip: string[]) => {
   try {
-    const ast = parse(code, {
-      plugins: ['typescript', 'decorators-legacy', 'importAssertions'],
-      sourceType: 'module',
-    })
+    const ast = parseTs(code)
 
     const packages_striped: string[] = []
 
@@ -45,7 +36,7 @@ export const removePackages = async (code: string, packages_to_strip: string[])
     })
 
     return {
-      code: recast.print(ast).code,
+      code: prettyPrint(ast).code,
       info: packages_striped.map((pkg) => `Replaced import from '${pkg}'`),
     }
   } catch (error) {