feat: support custom modifiers for glob imports

packages/playground/glob-import/__tests__/glob-import.spec.ts

@@ -66,6 +66,18 @@ const rawResult = {
   }
 }
 
+const customModifierResult = {
+  '/dir/alias.js': {
+    exportNames: ['default']
+  },
+  '/dir/foo.js': {
+    exportNames: ['msg']
+  },
+  '/dir/index.js': {
+    exportNames: ['modules', 'globWithAlias']
+  }
+}
+
 test('should work', async () => {
   expect(await page.textContent('.result')).toBe(
     JSON.stringify(allResult, null, 2)
@@ -81,6 +93,12 @@ test('import glob raw', async () => {
   )
 })
 
+test('custom modifer', async () => {
+  expect(await page.textContent('.custom-modifier')).toBe(
+    JSON.stringify(customModifierResult, null, 2)
+  )
+})
+
 if (!isBuild) {
   test('hmr for adding/removing files', async () => {
     addFile('dir/a.js', '')

packages/playground/glob-import/index.html

@@ -1,6 +1,7 @@
 <pre class="result"></pre>
 <pre class="result-node_modules"></pre>
 <pre class="globraw"></pre>
+<pre class="custom-modifier"></pre>
 
 <script type="module" src="./dir/index.js"></script>
 <script type="module">
@@ -52,3 +53,14 @@
     2
   )
 </script>
+
+<script type="module">
+  const modulesMeta = import.meta.globEager('/dir/*.js', {
+    as: 'meta'
+  })
+  document.querySelector('.custom-modifier').textContent = JSON.stringify(
+    modulesMeta,
+    null,
+    2
+  )
+</script>

packages/playground/glob-import/package.json

@@ -7,5 +7,8 @@
     "build": "vite build",
     "debug": "node --inspect-brk ../../vite/bin/vite",
     "preview": "vite preview"
+  },
+  "dependencies": {
+    "es-module-lexer": "^0.9.3"
   }
 }

packages/playground/glob-import/vite.config.ts

@@ -1,10 +1,39 @@
 import path from 'path'
-import { defineConfig } from 'vite'
+import { defineConfig, Plugin } from 'vite'
+import { init, parse } from 'es-module-lexer'
 
 export default defineConfig({
   resolve: {
     alias: {
       '@dir': path.resolve(__dirname, './dir/')
     }
-  }
+  },
+  plugins: [customModifierPlugin()]
 })
+
+function customModifierPlugin() {
+  const metaRE = /(\?|&)meta(?:&|$)/
+  return {
+    name: 'custom-modifier',
+    async transform(src, id) {
+      if (!metaRE.test(id)) {
+        return
+      }
+
+      const exportNames = await getExportNames(src)
+
+      const code = `export const exportNames = ${JSON.stringify(exportNames)};`
+
+      return {
+        code,
+        map: { mappings: '' }
+      }
+    }
+  } as Plugin
+}
+
+async function getExportNames(src: string): Promise<readonly string[]> {
+  await init
+  const exportNames = parse(src)[1]
+  return exportNames
+}

packages/vite/src/node/importGlob.ts

@@ -17,7 +17,6 @@ import {
 } from './utils'
 import type { RollupError } from 'rollup'
 import type { Logger } from '.'
-import colors from 'picocolors'
 
 interface GlobParams {
   base: string
@@ -136,17 +135,16 @@ export async function transformImportGlob(
       options?.assert?.type === 'raw'
     ) {
       logger.warn(
-        colors.yellow(
-          colors.bold(
-            "(!) Use `import.meta.globEager('/dir/*.js', { as: 'raw' })` instead of `import.meta.globEager('/dir/*.js', { assert: { type: 'raw' } })` (it will be deprecated in Vite 3.0)."
-          )
-        )
+        "Use `import.meta.globEager('/dir/*.js', { as: 'raw' })` instead of `import.meta.globEager('/dir/*.js', { assert: { type: 'raw' } })` (it will be deprecated in Vite 3.0)."
       )
       entries += ` ${JSON.stringify(file)}: ${JSON.stringify(
         await fsp.readFile(path.join(base, file), 'utf-8')
       )},`
     } else {
-      const importeeUrl = isCSSRequest(importee) ? `${importee}?used` : importee
+      let importeeUrl = isCSSRequest(importee) ? `${importee}?used` : importee
+      if (options?.as) {
+        importeeUrl = `${importee}?${options.as}`
+      }
       if (isEager) {
         const identifier = `__glob_${importIndex}_${i}`
         // css imports injecting a ?used query to export the css string