refactor: parse it

Author: hi-ogawa

packages/plugin-rsc/src/plugin.ts

@@ -909,13 +909,10 @@ function scanBuildStripPlugin(): Plugin {
     name: 'rsc:scan-strip',
     apply: 'build',
     enforce: 'post',
-    transform(code, _id, _options) {
+    async transform(code, _id, _options) {
       if (!isScanBuild) return
-
-      const output = transformScanBuildStrip(code)
-      if (output) {
-        return { code: output, map: { mappings: '' } }
-      }
+      const output = await transformScanBuildStrip(code)
+      return { code: output, map: { mappings: '' } }
     },
   }
 }

packages/plugin-rsc/src/plugins/scan.test.ts

@@ -0,0 +1,25 @@
+import { describe, expect, it } from 'vitest'
+import { transformScanBuildStrip } from './scan'
+
+describe(transformScanBuildStrip, () => {
+  it('basic', async () => {
+    const input = `\
+import { a } from "a";
+import "b";
+import(String("c"))
+import.meta.glob("d", {
+  query: "?e",
+})
+import.meta.globee("d", { query: "?e" })
+export default "foo";
+`
+    expect(await transformScanBuildStrip(input)).toMatchInlineSnapshot(`
+      "import "a";
+      import "b";
+      import.meta.glob("d", {
+        query: "?e",
+      })
+      "
+    `)
+  })
+})

packages/plugin-rsc/src/plugins/scan.ts

@@ -1,19 +1,40 @@
 import * as esModuleLexer from 'es-module-lexer'
+import { parseAstAsync } from 'vite'
+import { walk } from 'estree-walker'
 
 // https://github.com/vitejs/vite/blob/86d2e8be50be535494734f9f5f5236c61626b308/packages/vite/src/node/plugins/importMetaGlob.ts#L113
 const importGlobRE = /\bimport\.meta\.glob(?:<\w+>)?\s*\(/g
 
-export function transformScanBuildStrip(code: string): string | undefined {
-  // bail out if import.meta.glob
-  // https://github.com/vitejs/rolldown-vite/issues/373
-  if (importGlobRE.test(code)) return
-
+export async function transformScanBuildStrip(code: string): Promise<string> {
   const [imports] = esModuleLexer.parse(code)
   let output = imports
     .map((e) => e.n && `import ${JSON.stringify(e.n)};\n`)
     .filter(Boolean)
     .join('')
-  output += 'module.exports = {};\n'
+
+  // preserve import.meta.glob for rolldown-vite
+  // https://github.com/vitejs/rolldown-vite/issues/373
+  if (importGlobRE.test(code)) {
+    const ast = await parseAstAsync(code)
+    walk(ast, {
+      enter(node) {
+        if (
+          node.type === 'CallExpression' &&
+          node.callee.type === 'MemberExpression' &&
+          node.callee.object.type === 'MetaProperty' &&
+          node.callee.object.meta.type === 'Identifier' &&
+          node.callee.object.meta.name === 'import' &&
+          node.callee.object.property.type === 'Identifier' &&
+          node.callee.object.property.name === 'meta' &&
+          node.callee.property.type === 'Identifier' &&
+          node.callee.property.name === 'glob'
+        ) {
+          output += code.slice(node.start, node.end) + '\n'
+        }
+      },
+    })
+    output += ''
+  }
 
   return output
 }