open-standards/wai/wai-web/demo/verify_generate_wasm.mjs

// The wasm32 generator and score synth equal the native pins. A page that makes a
// generated picture or a score's samples in the browser checks every byte it made
// against the SHA-256 the native reference wrote, so the binding it loads must make
// exactly those bytes:
//
//   - generateSpectral at every size tests/vectors/generate/ pins — sizes.json's two
//     sizes, the four listed sizes of instr-v0.json's object, and the first listed size
//     of every object instr-v0.json lists as accepted (one at each bound of form v0) —
//     gives the SHA-256 the vector states, made natively by `wai generate`; each
//     object names the committed model by its SHA-256, checked here;
//   - ScoreSink over every score-conformance vector gives the SHA-256 of its samples
//     (pcm_sha256: the i16 mixdown, little-endian, SPEC §7 Digests), its mixdown hash
//     and its score hash as expected_hashes.json states them, and the contract's
//     duration in samples.
//
//   node verify_generate_wasm.mjs [path/to/wai_web.js] [path/to/wai-rs/tests/vectors/generate] [path/to/score-conformance]
//
// Exit 0 iff every check passes.

import { readFileSync } from 'node:fs';
import { createHash } from 'node:crypto';
import { dirname, resolve } from 'node:path';
import { fileURLToPath, pathToFileURL } from 'node:url';

const here = dirname(fileURLToPath(import.meta.url));
const pkg = resolve(here, process.argv[2] ?? './pkg-node/wai_web.js');
const gen = resolve(process.argv[3] ?? resolve(here, '../../wai-rs/tests/vectors/generate'));
const scores = resolve(process.argv[4] ?? resolve(here, '../../score-conformance'));
const wai = await import(pathToFileURL(pkg).href);

const sha = (b) => createHash('sha256').update(b).digest('hex');
const json = (p) => JSON.parse(readFileSync(p, 'utf8'));
const model = new Uint8Array(readFileSync(`${gen}/generate_spectral_model.json`));
const modelSha = sha(model);

const checks = [];
const picture = (name, o, w, h, want) => checks.push([`${name} ${w}x${h}`, () => {
  const img = wai.generateSpectral(model, o.prompt, o.seed, o.steps, w, h);
  const rgb = img.rgb;
  img.free();
  return rgb.length === w * h * 3 ? sha(rgb) : `${rgb.length} bytes`;
}, want]);
const named = (name, o) => checks.push([`${name}: names the committed model`, () => o.model, `sha256:${modelSha}`]);

const sizes = json(`${gen}/sizes.json`);
checks.push(['sizes.json: the committed model', () => sizes.model_sha256, modelSha]);
for (const s of sizes.sizes) picture('sizes.json', sizes, s.w, s.h, s.sha256);

const v = json(`${gen}/instr-v0.json`);
const object = JSON.parse(v.object);
named('instr-v0.json', object);
for (const e of v.outputs) picture('instr-v0.json', object, e.w, e.h, e.sha256);
if (!(v.accepted?.length >= 12)) checks.push(['instr-v0.json lists an accepted object at each bound', () => v.accepted?.length ?? 0, '>= 12']);
for (const c of v.accepted ?? []) {
  const o = JSON.parse(c.object);
  named(`accepted (${c.why})`, o);
  for (const e of c.outputs) picture(`accepted (${c.why})`, o, e.w, e.h, e.sha256);
}

const expected = json(`${scores}/expected_hashes.json`);
for (const [file, want] of Object.entries(expected)) {
  checks.push([`score ${file}`, () => {
    const s = new wai.ScoreSink(new Uint8Array(readFileSync(`${scores}/vectors/${file}`)));
    const pcm = s.mixdown();
    const le = new Uint8Array(2 * pcm.length);
    const dv = new DataView(le.buffer);
    pcm.forEach((x, i) => dv.setInt16(2 * i, x, true));
    const got = [pcm.length, sha(le), s.mixdown_hash_hex(), s.score_hash_hex()].join(' ');
    s.free();
    return got;
  }, [want.duration, want.pcm_sha256, want.mixdown_hash, want.score_hash].join(' ')]);
}

let pass = true;
for (const [name, run, want] of checks) {
  let got;
  try { got = run(); } catch (e) { got = `threw: ${e?.message ?? e}`; }
  const ok = got === want;
  pass = pass && ok;
  console.log(`${ok ? 'PASS' : 'FAIL'}  ${name.padEnd(64)} ${String(got).slice(0, 16)}${ok ? '' : `  want ${want}`}`);
}
console.log(`${checks.length} checks`);
process.exit(pass ? 0 : 1);