vite-plugin-binary/dist/main.js

Vite plugin for importing file in Uint8Array format

// src/main.ts
import { promises } from "fs";

// src/encode.ts
import { gzipSync } from "zlib";
var z85 = charsetToMap(`0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ.-:+=^!/*?&<>()[]{}@%$#`);
function charsetToMap(charset) {
  const ui8a = new Uint8Array(85);
  for (let i = 0; i < 85; i++) {
    ui8a[i] = charset.charAt(i).charCodeAt(0);
  }
  return ui8a;
}
function encodeBinary(ui8a, gzip) {
  if (gzip) {
    ui8a = gzipSync(ui8a);
  }
  const remain = ui8a.length % 4;
  const last5Length = remain ? remain + 1 : 0;
  const length = Math.ceil(ui8a.length * 5 / 4);
  const target = new Uint8Array(length);
  const dw = new DataView(ui8a.buffer, ui8a.byteOffset, ui8a.byteLength);
  const to = Math.trunc(ui8a.length / 4);
  for (let i = 0; i < to; i++) {
    let num = dw.getUint32(4 * i);
    for (let k = 4; k >= 0; k--) {
      target[k + i * 5] = z85[num % 85];
      num = Math.trunc(num / 85);
    }
  }
  if (remain) {
    const lastPartIndex = Math.trunc(ui8a.length / 4) * 4;
    const lastPart = Uint8Array.from([...ui8a.slice(lastPartIndex), 0, 0, 0]);
    const offset = target.length - last5Length - 1;
    const dw2 = new DataView(lastPart.buffer);
    let num = dw2.getUint32(0);
    for (let i = 4; i >= 0; i--) {
      const value = z85[num % 85];
      num = Math.trunc(num / 85);
      if (i < last5Length) {
        const index = offset + i + 1;
        target[index] = value;
      }
    }
  }
  return new TextDecoder().decode(target);
}

// src/main.ts
var decodeBinaryRaw = `const z85 = charsetToMap('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ.-:+=^!/*?&<>()[]{}@%$#')
const pow2 = 7225
const pow3 = 614125
const pow4 = 52200625

function charsetToMap(charset) {
  const ui8a = new Uint8Array(85)
  for (let i = 0; i < 85; i++) {
    ui8a[i] = charset.charAt(i).charCodeAt(0)
  }
  return ui8a
}

function getReverseMap(mapOrig) {
  const revMap = new Uint8Array(128)
  for (const [num, charCode] of Object.entries(mapOrig)) {
    revMap[charCode] = Number.parseInt(num)
  }
  return revMap
}

export default function decodeBinary(base85) {
  const revMap = getReverseMap(z85)

  const base85ab = new TextEncoder().encode(base85)
  const pad = (5 - (base85ab.length % 5)) % 5

  const ints = new Uint8Array((Math.ceil(base85ab.length / 5) * 4) - pad)
  let dw = new DataView(ints.buffer)
  let i = 0
  for (; i < base85ab.length / 5 - 1; i++) {
    const c1 = revMap[base85ab[i * 5 + 4]]
    const c2 = revMap[base85ab[i * 5 + 3]] * 85
    const c3 = revMap[base85ab[i * 5 + 2]] * pow2
    const c4 = revMap[base85ab[i * 5 + 1]] * pow3
    const c5 = revMap[base85ab[i * 5]] * pow4
    dw.setUint32(i * 4, c1 + c2 + c3 + c4 + c5)
  }

  const lCh = z85[z85.length - 1]
  const lastPart = new Uint8Array([...base85ab.slice(i * 5), lCh, lCh, lCh, lCh])
  dw = new DataView(lastPart.buffer)
  const c1 = revMap[lastPart[4]]
  const c2 = revMap[lastPart[3]] * 85
  const c3 = revMap[lastPart[2]] * pow2
  const c4 = revMap[lastPart[1]] * pow3
  const c5 = revMap[lastPart[0]] * pow4
  dw.setUint32(0, c1 + c2 + c3 + c4 + c5)
  for (let j = 0; j < 4 - pad; j++) {
    ints[i * 4 + j] = lastPart[j]
  }

  return ints
}
`;
function vitePluginBinary(args) {
  return {
    name: "vite-plugin-binary",
    resolveId(id) {
      if (id === "virtual:decode-binary") {
        return id;
      }
      return null;
    },
    load(id) {
      if (id === "virtual:decode-binary") {
        return decodeBinaryRaw;
      }
      return null;
    },
    async transform(_src, id) {
      if (id.endsWith("?binary")) {
        const file = id.slice(0, -7);
        this.addWatchFile(file);
        const buffer = await promises.readFile(file);
        const b64 = encodeBinary(buffer, args === void 0 || args.gzip);
        return {
          code: `import decodeBinary from 'virtual:decode-binary'
export default decodeBinary("${b64}")`,
          map: { mappings: "" }
        };
      }
      return null;
    }
  };
}
export {
  vitePluginBinary as default
};