fix: support import attributes on vite

closes #23

Author: sxzz

src/index.ts

@@ -18,27 +18,26 @@ const unplugin: UnpluginInstance<Options | undefined, false> = createUnplugin(
       name: 'unplugin-raw',
       enforce: options.enforce,
 
-      resolveId:
-        meta.framework === 'rollup' || meta.framework === 'rolldown'
-          ? async function (this, id, importer, opt) {
-              const attributeType = (opt as any)?.attributes?.type
-              if (attributeType === 'text') {
-                id += `${id.includes('?') ? '&' : '?'}raw`
-              } else if (attributeType === 'bytes') {
-                id += `${id.includes('?') ? '&' : '?'}bytes`
-              }
-              if (!rawRE.test(id)) return
+      resolveId: ['rollup', 'rolldown', 'vite'].includes(meta.framework)
+        ? async function (this, id, importer, opt) {
+            const attributeType = (opt as any)?.attributes?.type
+            if (attributeType === 'text') {
+              id += `${id.includes('?') ? '&' : '?'}raw`
+            } else if (attributeType === 'bytes') {
+              id += `${id.includes('?') ? '&' : '?'}bytes`
+            }
+            if (!rawRE.test(id)) return
 
-              const file = cleanUrl(id)
-              const resolved = await (this as PluginContext).resolve(
-                file,
-                importer,
-              )
-              if (!resolved) return
+            const file = cleanUrl(id)
+            const resolved = await (this as PluginContext).resolve(
+              file,
+              importer,
+            )
+            if (!resolved) return
 
-              return id.replace(file, resolved.id)
-            }
-          : undefined,
+            return id.replace(file, resolved.id)
+          }
+        : undefined,
 
       load: {
         filter: { id: { include: rawRE } },

tests/__snapshots__/basic.test.ts.snap

@@ -49,3 +49,12 @@ var text4 = "export const msg = \\"hello with\\";\\n";
 console.log(text, text2, text3, text4);
 "
 `;
+
+exports[`vite 1`] = `
+"const text = 'export const msg = "hello ts";\\n';
+const text2 = 'export const msg = "hello js";\\n';
+const text3 = 'export const msg = /* @__PURE__ */ React.createElement("div", null, "hello jsx");\\n';
+const text4 = 'export const msg = "hello with";\\n';
+console.log(text, text2, text3, text4);
+"
+`;

tests/basic.test.ts

@@ -1,6 +1,7 @@
 import path from 'node:path'
-import { build } from 'esbuild'
-import { rollup } from 'rollup'
+import { build as esbuild } from 'esbuild'
+import { rollup, type Plugin, type RollupOutput } from 'rollup'
+import { build as vite } from 'vite'
 import { expect, test } from 'vitest'
 import Raw from '../src'
 
@@ -16,7 +17,7 @@ test('esbuild', async () => {
   console.log(text, text2, text3, text4, bytes1)
   `
 
-  const result = await build({
+  const result = await esbuild({
     stdin: {
       contents,
       resolveDir,
@@ -29,34 +30,49 @@ test('esbuild', async () => {
   expect(result.outputFiles[0].text).matchSnapshot()
 })
 
+const rollupCode = `
+import text from "./ts.ts?raw"
+import text2 from "./js.js?raw"
+import text3 from "./jsx.jsx?raw"
+import text4 from "./with.js" with { type: "text" }
+console.log(text, text2, text3, text4)
+`
+const entryFile = path.resolve(resolveDir, 'main.js')
+const rollupPlugin: Plugin = {
+  name: 'entry',
+  resolveId(id) {
+    if (id === entryFile) {
+      return entryFile
+    }
+  },
+  load(id) {
+    if (id === entryFile) {
+      return rollupCode
+    }
+  },
+}
+
 test('rollup', async () => {
-  const contents = `
-  import text from "./ts.ts?raw"
-  import text2 from "./js.js?raw"
-  import text3 from "./jsx.jsx?raw"
-  import text4 from "./with.js" with { type: "text" }
-  console.log(text, text2, text3, text4)
-  `
-  const entryFile = path.resolve(resolveDir, 'main.js')
   const bundle = await rollup({
     input: [entryFile],
-    plugins: [
-      Raw.rollup(),
-      {
-        name: 'entry',
-        resolveId(id) {
-          if (id === entryFile) {
-            return entryFile
-          }
-        },
-        load(id) {
-          if (id === entryFile) {
-            return contents
-          }
-        },
-      },
-    ],
+    plugins: [Raw.rollup(), rollupPlugin],
   })
   const result = await bundle.generate({ format: 'esm' })
   expect(result.output[0].code).matchSnapshot()
 })
+
+test('vite', async () => {
+  const output = await vite({
+    root: resolveDir,
+    plugins: [Raw.vite(), rollupPlugin],
+    build: {
+      rollupOptions: {
+        input: [entryFile],
+      },
+      minify: false,
+      write: false,
+    },
+    logLevel: 'silent',
+  })
+  expect((output as RollupOutput).output[0].code).matchSnapshot()
+})