Improve performance and bundled size

src/main.test.ts

@@ -1,15 +1,32 @@
 import arrayBuffer from "./typescript.svg?arraybuffer";
 import uint8array from "./typescript.svg?uint8array";
+import arrayBufferB64 from "./typescript.svg?arraybuffer&base64";
+import uint8arrayB64 from "./typescript.svg?uint8array&base64";
+import crypto from "node:crypto";
 
 import { describe, expect, test } from "vitest";
 
+const hash = (buffer: ArrayBuffer) => crypto.createHash("md5").update(Buffer.from(buffer)).digest("hex");
+
 describe("test plugin", () => {
   test("?arraybuffer", () => {
     expect(arrayBuffer[Symbol.toStringTag]).toBe("ArrayBuffer");
     expect(arrayBuffer instanceof ArrayBuffer).toBe(true);
+    expect(hash(arrayBuffer)).toEqual("7167f7caac27a336c58b0c16cc5003d7");
   });
   test("?uint8array", () => {
     expect(uint8array[Symbol.toStringTag]).toBe("Uint8Array");
     expect(uint8array instanceof Uint8Array).toBe(true);
+    expect(hash(uint8array.buffer)).toEqual("7167f7caac27a336c58b0c16cc5003d7");
+  });
+  test("?arraybuffer&base64", () => {
+    expect(arrayBufferB64[Symbol.toStringTag]).toBe("ArrayBuffer");
+    expect(arrayBufferB64 instanceof ArrayBuffer).toBe(true);
+    expect(hash(arrayBufferB64)).toEqual("7167f7caac27a336c58b0c16cc5003d7");
+  });
+  test("?uint8array&base64", () => {
+    expect(uint8arrayB64[Symbol.toStringTag]).toBe("Uint8Array");
+    expect(uint8arrayB64 instanceof Uint8Array).toBe(true);
+    expect(hash(uint8arrayB64.buffer)).toEqual("7167f7caac27a336c58b0c16cc5003d7");
   });
 });

src/main.ts

@@ -7,7 +7,12 @@ export default function vitePluginArraybuffer(): PluginOption {
   return {
     name: "vite-plugin-arraybuffer",
     resolveId(id) {
-      if (id.endsWith("?arraybuffer") || id.endsWith("?uint8array")) {
+      if (
+        id.endsWith("?arraybuffer") ||
+        id.endsWith("?uint8array") ||
+        id.endsWith("?arraybuffer&base64") ||
+        id.endsWith("?uint8array&base64")
+      ) {
         return id
       }
 
@@ -30,6 +35,48 @@ export default function vitePluginArraybuffer(): PluginOption {
           ${new Uint8Array(await promises.readFile(file)).join(',')}
         ])`;
       }
+      if (id.endsWith("?arraybuffer&base64")) {
+        const file = id.slice(0, -12)
+        this.addWatchFile(file)
+
+        const buffer = await promises.readFile(file)
+        const b64 = buffer.toString("base64")
+
+        return `
+          function toUint8(b64) {
+            let bin = atob(b64);
+            let len = bin.length;
+            let bytes = new Uint8Array(len);
+            for (let i = 0; i < len; i++) {
+              bytes[i] = bin.charCodeAt(i);
+            }
+            return bytes;
+          }
+
+          export default toUint8("${b64}").buffer
+        `
+      }
+      if (id.endsWith("?uint8array&base64")) {
+        const file = id.slice(0, -11)
+        this.addWatchFile(file)
+
+        const buffer = await promises.readFile(file)
+        const b64 = buffer.toString("base64")
+
+        return `
+          function toUint8(b64) {
+            let bin = atob(b64);
+            let len = bin.length;
+            let bytes = new Uint8Array(len);
+            for (let i = 0; i < len; i++) {
+              bytes[i] = bin.charCodeAt(i);
+            }
+            return bytes;
+          }
+
+          export default toUint8("${b64}")
+        `
+      }
 
       return;
     },

src/types.d.ts

@@ -9,3 +9,15 @@ declare module "*?uint8array" {
 
   export default value;
 }
+
+declare module "*?arraybuffer&base64" {
+  const value: ArrayBuffer;
+
+  export default value;
+}
+
+declare module "*?uint8array&base64" {
+  const value: Uint8Array;
+
+  export default value;
+}