refactor: reuse script ast for rewriting default (#136)

vitejs · Mar 28, 2023 · 9d84656 · 9d84656
1 parent ca139d8
commit 9d84656
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/packages/plugin-vue/src/main.ts b/packages/plugin-vue/src/main.ts
@@ -321,11 +321,16 @@ async function genScriptCode(
             ? (['typescript'] as const)
             : (['typescript', 'decorators-legacy'] as const)
           : []
-      scriptCode = options.compiler.rewriteDefault(
-        script.content,
-        '_sfc_main',
-        [...defaultPlugins, ...userPlugins],
-      )
+      const as = '_sfc_main'
+      if (options.compiler.rewriteDefaultAST && script.scriptAst) {
+        options.compiler.rewriteDefaultAST(script.scriptAst, script.s, as)
+      } else {
+        scriptCode = options.compiler.rewriteDefault(
+          script.content,
+          '_sfc_main',
+          [...defaultPlugins, ...userPlugins],
+        )
+      }
       map = script.map
     } else {
       if (script.src) {