import assert from 'node:assert/strict';
import {spawnSync} from 'node:child_process';
import {cpSync, mkdirSync, mkdtempSync, readFileSync, writeFileSync} from 'node:fs';
import {join, relative, resolve} from 'node:path';
import {hashAt} from './boot-contract-report.mjs';

assert.equal(process.argv.length, 3,
  'usage: node scripts/os-acceptance/waitset-readiness.mjs <K1345-matrix-report.json>');
const root = process.cwd();
const predecessorPath = relative(root, resolve(process.argv[2]));
assert.match(predecessorPath,
  /^reports\/os-implementation\/K1345-waitset-el0-matrix-check-[A-Za-z0-9_-]+\/report\.json$/);
const predecessor = JSON.parse(readFileSync(join(root, predecessorPath), 'utf8'));
assert.equal(predecessor.status, 'K1345_WAITSET_EL0_DEVELOPMENT_MATRIX_VALIDATED');
assert.equal(predecessor.guestAccepted, true);
const directory = mkdtempSync(join(root,
  'reports/os-implementation/K1350-waitset-readiness-local-preflight-'));
const workspace = join(directory, 'workspace');
const source = 'os-waitset/readiness/waitset.rs';
const tests = 'kernel/tests/os/waitset_readiness.rs';
mkdirSync(join(workspace, 'os-waitset/readiness'), {recursive: true});
mkdirSync(join(workspace, 'kernel/tests/os'), {recursive: true});
cpSync(join(root, source), join(workspace, source));
cpSync(join(root, tests), join(workspace, tests));
const artifact = path => ({path: relative(root, path),
  sha256: hashAt(root, relative(root, path))});
const run = (name, command, args) => {
  const answer = spawnSync(command, args, {cwd: root, encoding: 'utf8',
    timeout: 600000, maxBuffer: 64 * 1024 * 1024});
  const log = join(directory, name + '.log');
  writeFileSync(log, (answer.stdout ?? '') + (answer.stderr ?? ''));
  return {exitCode: answer.status, error: answer.error?.message ?? null,
    log: artifact(log)};
};
const binary = join(directory, 'waitset-readiness-test');
const compile = run('host-compile', 'rustc', ['--edition=2024', '--test',
  join(workspace, tests), '-o', binary]);
assert.equal(compile.exitCode, 0,
  readFileSync(join(root, compile.log.path), 'utf8'));
const green = run('host-tests', binary, ['--nocapture']);
const greenText = readFileSync(join(root, green.log.path), 'utf8');
assert.equal(green.exitCode, 0, greenText);
for (const name of ['level_triggered_poll_repeats_until_source_clears',
  'readiness_filters_unrelated_bits_and_reuse_starts_clear',
  'readiness_denials_leave_snapshot_unchanged'])
  assert.match(greenText, new RegExp(`test ${name} \\.\\.\\. ok`));
assert.match(greenText, /3 passed; 0 failed/);

const wrapper = join(workspace, 'os-waitset/readiness/no_std_check.rs');
writeFileSync(wrapper, '#![no_std]\n#[path = "waitset.rs"]\nmod waitset;\n');
const target = run('aarch64-no-std-check', 'rustc', ['--edition=2024',
  '--crate-type=lib', '--target=aarch64-unknown-none', '--emit=metadata',
  wrapper, '-o', join(directory, 'waitset-readiness-aarch64.rmeta')]);
assert.equal(target.exitCode, 0,
  readFileSync(join(root, target.log.path), 'utf8'));

const copiedSource = join(workspace, source);
const original = readFileSync(copiedSource, 'utf8');
const guard = 'let bits = self.levels[slot] & interest.bits();';
assert.equal(original.split(guard).length, 2);
writeFileSync(copiedSource, original.replace(guard,
  'let bits = self.levels[slot] & interest.bits(); self.levels[slot] = 0;'));
const mutantCompile = run('consume-on-poll-mutant-compile', 'rustc',
  ['--edition=2024', '--test', join(workspace, tests),
    '-o', join(directory, 'waitset-readiness-mutant-test')]);
assert.equal(mutantCompile.exitCode, 0,
  readFileSync(join(root, mutantCompile.log.path), 'utf8'));
const red = run('consume-on-poll-mutant',
  join(directory, 'waitset-readiness-mutant-test'),
  ['level_triggered_poll_repeats_until_source_clears', '--exact', '--nocapture']);
const redText = readFileSync(join(root, red.log.path), 'utf8');
assert.notEqual(red.exitCode, 0);
assert.match(redText, /level_triggered_poll_repeats_until_source_clears \.\.\. FAILED/);
writeFileSync(copiedSource, original);

const report = {schema: 'turkos-k1350-waitset-readiness-local-preflight/v1',
  status: 'K1346_K1349_LOCAL_VALIDATED',
  guestAccepted: false, productAccepted: false,
  predecessor: artifact(join(root, predecessorPath)),
  sources: {contract: artifact(join(root,
    'os-waitset/readiness/CONTRACT.md')),
  model: artifact(join(root, source)), tests: artifact(join(root, tests)),
  harness: artifact(join(root, 'scripts/os-acceptance/waitset-readiness.mjs'))},
  tests: {compile, green, target, mutantCompile, red,
    greenCount: 3, mutationRejected: true},
  scope: 'Local level-triggered waitset readiness model: repeated polls preserve a source level, clearing removes it, unrelated bits are filtered, reuse starts empty, owner and invalid-level denials leave snapshots unchanged, and AArch64 no_std compilation plus a consume-on-poll mutant are checked. Native EL0 six-profile integration remains K1350.'};
const reportPath = join(directory, 'report.json');
writeFileSync(reportPath, JSON.stringify(report, null, 2) + '\n');
console.log(JSON.stringify({status: report.status,
  report: relative(root, reportPath), green: report.tests.greenCount}));