daedalOS/components/apps/Terminal/loadWapm.ts

import { type WASIBindings } from "wasi-js";
import { config } from "components/apps/Terminal/config";
import { type LocalEcho } from "components/apps/Terminal/types";

type WASIError = Error & {
  code: number;
};

let bindings: WASIBindings | null;

const loadWapm = async (
  commandArgs: string[],
  localEcho: LocalEcho,
  wasmFile?: Buffer
): Promise<void> => {
  const { fetchCommandFromWAPM } = await import("@wasmer/wasm-terminal");
  const { lowerI64Imports } = await import("@wasmer/wasm-transformer");
  const { default: WASI } = await import("wasi-js");

  try {
    const wasmBinary =
      wasmFile || (await fetchCommandFromWAPM({ args: commandArgs }));

    if (
      wasmBinary.length < 1024 &&
      new TextDecoder().decode(wasmBinary).includes("NoSuchKey")
    ) {
      throw new Error(`command not found ${commandArgs[0]}`);
    }

    const moduleResponse = await lowerI64Imports(wasmBinary);

    if (moduleResponse !== undefined && moduleResponse instanceof Uint8Array) {
      bindings ||= (await import("wasi-js/dist/bindings/browser")).default;

      const wasmModule = await WebAssembly.compile(moduleResponse);
      const wasi = new WASI({
        args: commandArgs,
        bindings,
        env: {
          COLUMNS: config.cols?.toString(),
          LINES: config.rows?.toString(),
        },
        sendStderr: (buffer: Uint8Array) => localEcho?.print(buffer.toString()),
        sendStdout: (buffer: Uint8Array) => localEcho?.print(buffer.toString()),
      });
      const instance = await WebAssembly.instantiate(
        wasmModule,
        wasi.getImports(wasmModule) as WebAssembly.Imports
      );

      wasi.start(instance);
    }
  } catch (error) {
    const { message } = error as WASIError;

    localEcho?.println(message);
  }
};

export default loadWapm;