scriptc/tests/harness/static-spatial.test.ts

import { spawnSync } from "node:child_process";
import { mkdtemp, mkdir, readFile, rm, writeFile } from "node:fs/promises";
import { join } from "node:path";
import { createRequire } from "node:module";
import { rollup } from "rollup";
import { expect, test } from "vitest";
import { compile } from "@scriptc/compiler";

const hasZig = spawnSync("zig", ["version"], { stdio: "ignore" }).status === 0;
const require = createRequire(import.meta.url);
const wasiRunner = `const {WASI}=require('node:wasi'); const {readFileSync}=require('node:fs');
const wasi=new WASI({version:'preview1',args:[process.argv[1]],returnOnExit:true});
const instance=new WebAssembly.Instance(new WebAssembly.Module(readFileSync(process.argv[1])),wasi.getImportObject());
process.exitCode=wasi.start(instance);`;
function run(command: string, args: string[]) {
  const result = spawnSync(command,args,{encoding:"utf8",timeout:30_000});
  if (result.error) throw result.error;
  const stderr = result.stderr.split("\n").filter(line => !line.startsWith("scriptc RC audit skipped:")).join("\n");
  return { stdout: result.stdout, stderr, status: result.status, signal: result.signal };
}

test.for(["native", "wasm32-wasi"])("published spatial queries and output arrays run statically on %s", async (target, context) => {
  if (target === "wasm32-wasi" && !hasZig) context.skip();
  const dir = await mkdtemp("/tmp/scriptc-spatial-");
  const previousTarget = process.env["SCRIPTC_TARGET"], previousCc = process.env["SCRIPTC_CC"];
  try {
    if(target === "wasm32-wasi") { process.env["SCRIPTC_TARGET"] = target; process.env["SCRIPTC_CC"] = "zigcc"; }
    const pkg = join(dir,"node_modules","spatial-fixture");
    await mkdir(pkg,{recursive:true});
    await writeFile(join(pkg,"package.json"),'{"name":"spatial-fixture","type":"module","main":"index.js"}');
    const entry = join(dir,"main.mjs");
    await writeFile(entry,'import "spatial-fixture";');
    for (const fixture of ["three/collision.mjs", "npm-static/array-output-parameters.mjs"]) {
      const source = join(import.meta.dirname,"../fixtures",fixture);
      const bundle = await rollup({ input:source, plugins:[{name:"three-cpu-entry",resolveId(id) {
        if(id === "three") return require.resolve("three/src/Three.Core.js");
        if(id.startsWith("three/")) return require.resolve(id);
        return null;
      }}] });
      try { await bundle.write({file:join(pkg,"index.js"),format:"es"}); }
      finally { await bundle.close(); }
      const wasm = target === "wasm32-wasi";
      const output = join(dir,wasm ? "program.wasm" : "program");
      const result = await compile(entry,{outDir:dir,outPath:output,backend:"llvm",dynamic:false,npmStatic:["spatial-fixture"],
        sanitize:!wasm && process.env["SCRIPTC_SAN"] === "1"});
      if(!result.ok) throw new Error(result.diagnostics.map(d=>`${d.code}: ${d.message}`).join("\n"));
      if(wasm) expect([...(await readFile(output)).subarray(0,4)]).toEqual([0,97,115,109]);
      const reference = run(process.execPath,["--no-warnings",source]);
      expect(reference.status).toBe(0);
      expect(run(process.execPath,["--no-warnings",entry])).toEqual(reference);
      expect(wasm ? run(process.execPath,["--no-warnings","-e",wasiRunner,output]) : run(output,[])).toEqual(reference);
    }
  } finally {
    if(previousTarget === undefined) delete process.env["SCRIPTC_TARGET"]; else process.env["SCRIPTC_TARGET"] = previousTarget;
    if(previousCc === undefined) delete process.env["SCRIPTC_CC"]; else process.env["SCRIPTC_CC"] = previousCc;
    await rm(dir,{recursive:true,force:true});
  }
});