import assert from "node:assert/strict";
import { createHash } from "node:crypto";
import { spawnSync } from "node:child_process";
import { cpSync, existsSync, mkdirSync, mkdtempSync, realpathSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { dirname, join } from "node:path";
import { fileURLToPath } from "node:url";
import test from "node:test";
import profiles from "../acpx-profiles.json" with { type: "json" };
import distributions from "../cursor-distributions.json" with { type: "json" };
import { verifyReleaseProviderPack } from "./release-provider-pack.mjs";
import { existingReleaseProviderPack, releaseProviderPackManifest } from "./stage-release-runner-binaries.mjs";

const revision = "a".repeat(40);
const digest = "sha256:" + "b".repeat(64);
const canonical = value => Array.isArray(value) ? `[${value.map(canonical).join(",")}]`
  : value && typeof value === "object" ? `{${Object.keys(value).sort().map(key => `${JSON.stringify(key)}:${canonical(value[key])}`).join(",")}}` : JSON.stringify(value);
function fixture() {
  const payload = { target: { platform: "linux", architecture: "x64" }, runnerSourceRevision: revision,
    pins: { acpx: profiles.acpxVersion },
    acpxProfileDigests: Object.fromEntries(["grok", "claude", "codex"].map(agent => [agent, profiles.profiles[agent].commandDigest])),
    providers: { cursor: { version: profiles.profiles.cursor.agentServerVersion, profileDigest: profiles.profiles.cursor.commandDigest,
      closureDigest: `sha256:${distributions.platforms["linux-x64"].closureSha256}`, qualification: "qualified",
      path: "provider-assets/cursor/linux-x64", sha256: digest } } };
  payload.providers.copilot = { version: profiles.profiles.copilot.agentServerVersion,
    profileDigest: profiles.profiles.copilot.commandDigest, closureDigest: "sha256:a3d8f4367cfa1694d79e3f8b7930b6fbfe42a229c272b375b11951d631db8d07",
    qualification: "qualified", path: "provider-assets/copilot/linux-x64", sha256: digest };
  return payload;
}
function seal(payload) { return { schema: "paperclip-runner/remote-provider-pack/v1", payload,
  digest: `sha256:${createHash("sha256").update(canonical(payload)).digest("hex")}` }; }

test("accepts a pack whose source and profiles match the assembled release", () => {
  const manifest = seal(fixture()); assert.equal(verifyReleaseProviderPack(manifest, revision), manifest);
});
for (const field of ["version", "profileDigest", "closureDigest", "qualification", "path"]) {
  test(`rejects an independently rehashed stale Cursor ${field}`, () => {
    const payload = fixture(); payload.providers.cursor[field] = field.endsWith("Digest") ? digest : "stale";
    assert.throws(() => verifyReleaseProviderPack(seal(payload), revision), /Cursor identity/);
  });
}
test("rejects omitted Cursor, conflicting legacy inventory, stale sources and forged payload digests", () => {
  const missing = fixture(); delete missing.providers.cursor;
  assert.throws(() => verifyReleaseProviderPack(seal(missing), revision), /Cursor identity/);
  const mixed = fixture(); mixed.candidateProviders = { cursor: { ...mixed.providers.cursor, profileDigest: digest } };
  assert.throws(() => verifyReleaseProviderPack(seal(mixed), revision), /Cursor identity/);
  const old = fixture(); old.runnerSourceRevision = "c".repeat(40);
  assert.throws(() => verifyReleaseProviderPack(seal(old), revision), /release source/);
  const forged = seal(fixture()); forged.digest = digest;
  assert.throws(() => verifyReleaseProviderPack(forged, revision), /payload digest/);
});

test("release assembly rejects rehashed stale or missing packs before staging any files", () => {
  const root = realpathSync(mkdtempSync(join(tmpdir(), "release-provider-pack-")));
  const source = dirname(dirname(fileURLToPath(import.meta.url)));
  try {
    // Run the actual assembler in an isolated layout so it cannot overwrite built artifacts.
    for (const file of ["scripts/stage-release-runner-binaries.mjs", "scripts/release-provider-pack.mjs", "src/live/runner-binary.ts", "acpx-profiles.json", "cursor-distributions.json", "release-runner-targets.json"]) {
      mkdirSync(dirname(join(root, file)), { recursive: true }); cpSync(join(source, file), join(root, file));
    }
    const platforms = {};
    for (const target of ["darwin-arm64", "darwin-x64", "linux-x64", "linux-arm64"]) {
      const bytes = Buffer.alloc(64);
      if (target.startsWith("linux-")) {
        bytes.set([0x7f, 0x45, 0x4c, 0x46, 2, 1]); bytes.writeUInt16LE(target === "linux-x64" ? 62 : 183, 18);
      } else {
        bytes.writeUInt32LE(0xfeedfacf, 0); bytes.writeUInt32LE(target === "darwin-arm64" ? 0x0100000c : 0x01000007, 4);
      }
      const path = join(root, target); writeFileSync(path, bytes);
      platforms[target] = { path, sha256: `sha256:${createHash("sha256").update(bytes).digest("hex")}` };
    }
    const payload = fixture(); payload.providers.cursor.profileDigest = digest;
    const bytes = JSON.stringify(seal(payload)), path = join(root, "provider-pack.json"); writeFileSync(path, bytes);
    const manifest = { sourceRevision: revision, platforms, remoteProviderPack: { path, sha256: `sha256:${createHash("sha256").update(bytes).digest("hex")}` } };
    for (const error of [/Cursor identity/, /requires the matching Linux remote provider pack/]) {
      writeFileSync(join(root, "release.json"), JSON.stringify(manifest));
      const result = spawnSync(process.execPath, [join(root, "scripts/stage-release-runner-binaries.mjs"), join(root, "release.json")], { encoding: "utf8" });
      assert.equal(result.status, 1); assert.match(result.stderr, error);
      assert.equal(existsSync(join(root, "dist")), false);
      delete manifest.remoteProviderPack;
    }
  } finally { rmSync(root, { recursive: true, force: true }); }
});


test("daemon-only assembly retains an existing same-source Linux provider-pack identity", () => {
  const root = realpathSync(mkdtempSync(join(tmpdir(), "release-provider-retained-")));
  try {
    assert.equal(existingReleaseProviderPack(root, revision), undefined);
    const file = join(root, "dist/remote-provider-packs/linux-x64/provider-pack.json");
    mkdirSync(dirname(file), { recursive: true });
    const pack = seal(fixture()); writeFileSync(file, JSON.stringify(pack));
    assert.deepEqual(existingReleaseProviderPack(root, revision), pack);
    assert.throws(() => existingReleaseProviderPack(root, "c".repeat(40)), /release source/);
    const wrongTarget = fixture(); wrongTarget.target = { platform: "darwin", architecture: "arm64" };
    writeFileSync(file, JSON.stringify(seal(wrongTarget)));
    assert.throws(() => existingReleaseProviderPack(root, revision), /Linux x64 target/);
  } finally { rmSync(root, { recursive: true, force: true }); }
});

for (const field of ["version", "profileDigest", "closureDigest", "qualification", "path", "sha256"]) {
  test(`rejects an independently rehashed stale Copilot ${field}`, () => {
    const payload = fixture(); payload.providers.copilot[field] = "stale";
    assert.throws(() => verifyReleaseProviderPack(seal(payload), revision), /Copilot identity/);
  });
}
test("rejects omitted Copilot and a duplicate candidate inventory", () => {
  const missing = fixture(); delete missing.providers.copilot;
  assert.throws(() => verifyReleaseProviderPack(seal(missing), revision), /Copilot identity/);
  const mixed = fixture(); mixed.candidateProviders = { copilot: mixed.providers.copilot };
  assert.throws(() => verifyReleaseProviderPack(seal(mixed), revision), /Copilot identity/);
});

function armFixture() {
  const payload = fixture();
  payload.target.architecture = "arm64";
  delete payload.providers;
  return payload;
}
test("ARM pack binds its own target and preserves other providers qualification limits", () => {
  const pack = seal(armFixture());
  assert.equal(verifyReleaseProviderPack(pack, revision, "linux-arm64"), pack);
  assert.throws(() => verifyReleaseProviderPack(pack, revision), /Linux x64 target/);
  assert.throws(() => verifyReleaseProviderPack(seal(fixture()), revision, "linux-arm64"), /linux-arm64 target/);
  for (const inventory of ["providers", "candidateProviders"]) {
    const payload = armFixture(); payload[inventory] = { pi: {} };
    assert.throws(() => verifyReleaseProviderPack(seal(payload), revision, "linux-arm64"), /must not expand/);
  }
});
test("both remote pack identities bind one source and remain separated during assembly", () => {
  const root = realpathSync(mkdtempSync(join(tmpdir(), "release-provider-targets-")));
  try {
    for (const [target, payload] of [["linux-x64", fixture()], ["linux-arm64", armFixture()]]) {
      const file = join(root, target, "provider-pack.json"); mkdirSync(dirname(file), { recursive: true });
      writeFileSync(file, JSON.stringify(seal(payload)));
    }
    const entries = releaseProviderPackManifest(root, revision);
    assert.deepEqual(Object.keys(entries), ["linux-x64", "linux-arm64"]);
    assert.notEqual(entries["linux-x64"].sha256, entries["linux-arm64"].sha256);
    assert.throws(() => releaseProviderPackManifest(root, "c".repeat(40)), /release source/);
    writeFileSync(join(root, "linux-arm64", "provider-pack.json"), JSON.stringify(seal(fixture())));
    assert.throws(() => releaseProviderPackManifest(root, revision), /linux-arm64 target/);
    rmSync(join(root, "linux-arm64"), { recursive: true });
    assert.throws(() => releaseProviderPackManifest(root, revision), /exactly both/);
  } finally { rmSync(root, { recursive: true, force: true }); }
});

test("frozen image validation uses release profiles rather than the newer workflow checkout", () => {
  const frozenProfiles = structuredClone(profiles);
  const frozenDistributions = structuredClone(distributions);
  frozenProfiles.acpxVersion = "frozen-acpx";
  for (const agent of ["claude", "codex", "grok", "cursor"]) frozenProfiles.profiles[agent].commandDigest = digest;
  frozenProfiles.profiles.cursor.agentServerVersion = "frozen-cursor";
  frozenDistributions.platforms["linux-x64"].closureSha256 = "c".repeat(64);
  const payload = fixture();
  payload.pins.acpx = frozenProfiles.acpxVersion;
  for (const agent of ["claude", "codex", "grok"]) payload.acpxProfileDigests[agent] = digest;
  Object.assign(payload.providers.cursor, { version: "frozen-cursor", profileDigest: digest, closureDigest: "sha256:" + "c".repeat(64) });
  const pack = seal(payload);
  const expected = { profiles: frozenProfiles, distributions: frozenDistributions };
  assert.throws(() => verifyReleaseProviderPack(pack, revision), /profiles/);
  assert.equal(verifyReleaseProviderPack(pack, revision, "linux-x64", expected), pack);
  payload.acpxProfileDigests.claude = "sha256:" + "d".repeat(64);
  assert.throws(() => verifyReleaseProviderPack(seal(payload), revision, "linux-x64", expected), /profiles/);
});

test("uses the frozen release's Copilot profile when the workflow checkout has advanced", () => {
  const releaseProfiles = structuredClone(profiles);
  releaseProfiles.profiles.copilot.commandDigest = "sha256:" + "e".repeat(64);
  const payload = fixture(); payload.providers.copilot.profileDigest = releaseProfiles.profiles.copilot.commandDigest;
  const manifest = seal(payload);
  assert.equal(verifyReleaseProviderPack(manifest, revision, "linux-x64", { profiles: releaseProfiles, distributions }), manifest);
  assert.throws(() => verifyReleaseProviderPack(manifest, revision), /Copilot identity/);
});