feat: support bytes for rollup

Author: sxzz

README.md

@@ -98,10 +98,12 @@ export interface Options {
 ```ts
 import text from './js.js?raw'
 import text2 from './jsx.jsx?raw'
+import bytes from './png.png?bytes'
 import text3 from './ts.ts?raw'
 
 // Import attributes (Rolldown doesn't support this syntax)
 import text4 from './with.js' with { type: 'text' }
+import bytes2 from './with.png' with { type: 'bytes' }
 ```
 
 ## Sponsors

src/index.ts

@@ -1,10 +1,12 @@
+import { Buffer } from 'node:buffer'
 import { createUnplugin, type UnpluginInstance } from 'unplugin'
 import { createFilter } from 'unplugin-utils'
 import { resolveOptions, type Options } from './core/options'
 import { transformRaw } from './core/transform'
 import type { PluginContext } from 'rollup'
 
 const rawRE = /[&?]raw(?:&|$)/
+const bytesRE = /[&?]bytes(?:&|$)/
 const postfixRE = /[#?].*$/s
 function cleanUrl(url: string) {
   return url.replace(postfixRE, '')
@@ -28,8 +30,9 @@ const unplugin: UnpluginInstance<Options | undefined, false> = createUnplugin(
               id += `${id.includes('?') ? '&' : '?'}raw`
             } else if (attributeType === 'bytes') {
               id += `${id.includes('?') ? '&' : '?'}bytes`
+            } else if (!rawRE.test(id) && !bytesRE.test(id)) {
+              return
             }
-            if (!rawRE.test(id)) return
 
             const file = cleanUrl(id)
             const resolved = await (this as PluginContext).resolve(
@@ -44,23 +47,28 @@ const unplugin: UnpluginInstance<Options | undefined, false> = createUnplugin(
 
       load: {
         filter: {
-          id: { include: rawRE },
+          id: { include: [rawRE, bytesRE] },
         },
         async handler(id) {
+          const isBytes = bytesRE.test(id)
           const file = cleanUrl(id)
           const context = this.getNativeBuildContext?.()
           const transform =
             context?.framework === 'esbuild'
               ? context.build.esbuild.transform
               : undefined
-          const contents = await transformRaw(
+          let contents = await transformRaw(
             file,
-            false,
+            isBytes,
             transform,
             transformFilter,
             options.transform ? options.transform.options : undefined,
           )
-          return contents as string
+          if (Buffer.isBuffer(contents)) {
+            contents = `export default new Uint8Array([${contents.join(', ')}])`
+          }
+
+          return contents
         },
       },
 

tests/__snapshots__/basic.test.ts.snap

@@ -65,7 +65,9 @@ var text3 = "export const msg = /* @__PURE__ */ React.createElement(\\"div\\", n
 
 var text4 = "export const msg = \\"hello with\\";\\n";
 
-console.log(text, text2, text3, text4);
+var bytes1 = new Uint8Array([97, 98, 99, 10]);
+
+console.log(text, text2, text3, text4, bytes1);
 "
 `;
 
@@ -74,6 +76,7 @@ exports[`vite 1`] = `
 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);
+const bytes1 = new Uint8Array([97, 98, 99, 10]);
+console.log(text, text2, text3, text4, bytes1);
 "
 `;

tests/basic.test.ts

@@ -40,7 +40,8 @@ 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)
+import bytes1 from "./abc.txt" with { type: "bytes" }
+console.log(text, text2, text3, text4, bytes1)
 `
 const entryFile = path.resolve(resolveDir, 'main.js')
 const rollupPlugin = (code: string): Plugin => ({