@addmaple/lz4 0.1.1 → 0.1.2

package/dist/node.d.ts

@@ -1,2 +1,5 @@
-export function init(imports?: WebAssembly.Imports): Promise<void>;
+export interface InitOptions {
+  backend?: 'auto' | 'simd' | 'base';
+}
+export function init(imports?: WebAssembly.Imports, opts?: InitOptions): Promise<void>;
 export * from "./custom.js";

package/dist/node.js

@@ -1,5 +1,5 @@
 import { setInstance, registerInit } from "./core.js";
-import { instantiateWithFallback } from "./util.js";
+import { instantiateWithBackend } from "./util.js";
 
 import { readFile } from "node:fs/promises";
 import { fileURLToPath } from "node:url";
@@ -7,20 +7,23 @@ import { fileURLToPath } from "node:url";
 const simdPath = fileURLToPath(new URL("./wasm/lz4.simd.wasm", import.meta.url));
 const basePath = fileURLToPath(new URL("./wasm/lz4.base.wasm", import.meta.url));
 
-async function getWasmBytes() {
-  const [simdBytes, baseBytes] = await Promise.all([
-    readFile(simdPath).catch(() => null),
-    readFile(basePath).catch(() => null)
-  ]);
-  return { simdBytes, baseBytes };
+async function getSimdBytes() {
+  return readFile(simdPath);
+}
+
+async function getBaseBytes() {
+  return readFile(basePath);
 }
 
 
 let _ready = null;
-export function init(imports = {}) {
-  return (_ready ??= (async () => {
-    const { simdBytes, baseBytes } = await getWasmBytes();
-    const { instance } = await instantiateWithFallback(simdBytes, baseBytes, imports);
+let _backend = null;
+export function init(imports = {}, opts = {}) {
+  const backend = opts.backend || 'auto';
+  if (_ready && _backend === backend) return _ready;
+  _backend = backend;
+  return (_ready = (async () => {
+    const { instance } = await instantiateWithBackend({ getSimdBytes, getBaseBytes, imports, backend });
     setInstance(instance);
   })());
 }

package/dist/util.js

@@ -12,3 +12,33 @@ export async function instantiateWithFallback(
     return { instance, backend: 'wasm' }
   }
 }
+
+export async function instantiateWithBackend({
+  getSimdBytes,
+  getBaseBytes,
+  imports,
+  backend = 'auto',
+}) {
+  if (backend === 'base') {
+    const baseBytes = await getBaseBytes()
+    const { instance } = await WebAssembly.instantiate(baseBytes, imports)
+    return { instance, backend: 'wasm' }
+  }
+
+  if (backend === 'simd') {
+    const simdBytes = await getSimdBytes()
+    const { instance } = await WebAssembly.instantiate(simdBytes, imports)
+    return { instance, backend: 'wasm-simd' }
+  }
+
+  // auto: try simd first, then fallback to baseline
+  try {
+    const simdBytes = await getSimdBytes()
+    const { instance } = await WebAssembly.instantiate(simdBytes, imports)
+    return { instance, backend: 'wasm-simd' }
+  } catch {
+    const baseBytes = await getBaseBytes()
+    const { instance } = await WebAssembly.instantiate(baseBytes, imports)
+    return { instance, backend: 'wasm' }
+  }
+}