feat: support binary server assets (#2107)

src/rollup/plugins/raw.ts

@@ -6,6 +6,8 @@ export interface RawOptions {
   extensions?: string[];
 }
 
+const HELPER_ID = "\0raw-helpers";
+
 export function raw(opts: RawOptions = {}): Plugin {
   const extensions = new Set([
     ".md",
@@ -18,9 +20,16 @@ export function raw(opts: RawOptions = {}): Plugin {
     ...(opts.extensions || []),
   ]);
 
+  // TODO: use ext=>mime
+  const isBinary = (id) => !extensions.has(extname(id));
+
   return {
     name: "raw",
     resolveId(id) {
+      if (id === HELPER_ID) {
+        return id;
+      }
+
       if (id[0] === "\0") {
         return;
       }
@@ -38,20 +47,45 @@ export function raw(opts: RawOptions = {}): Plugin {
       }
     },
     load(id) {
+      if (id === HELPER_ID) {
+        return getHelpers();
+      }
       if (id.startsWith("\0raw:")) {
         // this.addWatchFile(id.substring(5))
-        return fsp.readFile(id.slice(5), "utf8");
+        return fsp.readFile(id.slice(5), isBinary(id) ? "binary" : "utf8");
       }
     },
     transform(code, id) {
-      if (id.startsWith("\0raw:")) {
+      if (!id.startsWith("\0raw:")) {
+        return;
+      }
+      if (isBinary(id)) {
+        const serialized = Buffer.from(code, "binary").toString("base64");
+        return {
+          code: `// ROLLUP_NO_REPLACE \n import {base64ToUint8Array } from "${HELPER_ID}" \n export default base64ToUint8Array("${serialized}")`,
+          map: null,
+        };
+      } else {
         return {
-          code: `// ROLLUP_NO_REPLACE \n export default ${JSON.stringify(
-            code
-          )}`,
+          code: `// ROLLUP_NO_REPLACE \n export default ${JSON.stringify(code)}`,
           map: null,
         };
       }
     },
   };
 }
+
+function getHelpers() {
+  const js = String.raw;
+  return js`
+export function base64ToUint8Array(str) {
+  const data = atob(str);
+  const size = data.length;
+  const bytes = new Uint8Array(size);
+  for (let i = 0; i < size; i++) {
+    bytes[i] = data.charCodeAt(i);
+  }
+  return bytes;
+}
+  `;
+}

test/fixture/routes/assets/[id].ts

@@ -0,0 +1,17 @@
+export default eventHandler(async (event) => {
+  const serverAssets = useStorage("assets/server");
+  const id = event.context.params.id;
+
+  if (!(await serverAssets.hasItem(id))) {
+    throw createError({ message: `Asset ${id} not found`, statusCode: 404 });
+  }
+
+  const meta = (await serverAssets.getMeta(
+    event.context.params.id
+  )) as unknown as { type: string; etag: string; mtime: string };
+  meta.type && setResponseHeader(event, "content-type", meta.type);
+  meta.etag && setResponseHeader(event, "etag", meta.etag);
+  meta.mtime && setResponseHeader(event, "last-modified", meta.mtime);
+
+  return serverAssets.getItemRaw(event.context.params.id);
+});