fix(vite): compatible with legacy Vite on transformIndexHtml, fix #3357

antfu · antfu · commit 49b3451093c1 · 2023-11-21T12:19:32.000+01:00
diff --git a/packages/vite/src/modes/global/build.ts b/packages/vite/src/modes/global/build.ts
@@ -85,6 +85,11 @@ export function GlobalModeBuildPlugin(ctx: UnocssPluginContext<VitePluginConfig>
         handler(code, { filename }) {
           tasks.push(extract(code, filename))
         },
+        // Compatibility with Legacy Vite
+        enforce: 'pre',
+        transform(code, { filename }) {
+          tasks.push(extract(code, filename))
+        },
       },
       resolveId(id, importer) {
         const entry = resolveId(id)
diff --git a/packages/vite/src/modes/global/dev.ts b/packages/vite/src/modes/global/dev.ts
@@ -137,6 +137,12 @@ export function GlobalModeDevPlugin({ uno, tokens, tasks, flushTasks, affectedMo
           setWarnTimer()
           tasks.push(extract(code, filename))
         },
+        // Compatibility with Legacy Vite
+        enforce: 'pre',
+        transform(code, { filename }) {
+          setWarnTimer()
+          tasks.push(extract(code, filename))
+        },
       },
       resolveId(id) {
         const entry = resolveId(id)
diff --git a/packages/vite/src/transformers.ts b/packages/vite/src/transformers.ts
@@ -3,19 +3,26 @@ import type { UnocssPluginContext } from '@unocss/core'
 import { applyTransformers } from '../../shared-integration/src/transformers'
 
 export function createTransformerPlugins(ctx: UnocssPluginContext): Plugin[] {
-  const enforces = ['default', 'pre', 'post'] as const
-  return enforces.map((enforce): Plugin => ({
-    name: `unocss:transformers:${enforce}`,
-    enforce: enforce === 'default' ? undefined : enforce,
-    transform(code, id) {
-      return applyTransformers(ctx, code, id, enforce)
-    },
-    transformIndexHtml: {
-      order: enforce === 'default' ? undefined : enforce,
-      handler(code) {
-        return applyTransformers(ctx, code, 'index.html', enforce)
-          .then(t => t?.code)
+  const orders = ['default', 'pre', 'post'] as const
+  return orders.map((_order): Plugin => {
+    const order = _order === 'default' ? undefined : _order
+    const htmlHandler = (code: string) => {
+      return applyTransformers(ctx, code, 'index.html', order)
+        .then(t => t?.code)
+    }
+    return {
+      name: `unocss:transformers:${order}`,
+      enforce: order,
+      transform(code, id) {
+        return applyTransformers(ctx, code, id, order)
       },
-    },
-  }))
+      transformIndexHtml: {
+        order,
+        handler: htmlHandler,
+        // Compatibility with Legacy Vite
+        enforce: order,
+        transform: htmlHandler,
+      },
+    } satisfies Plugin
+  })
 }
