import test from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { promisify } from "node:util";
import { execFile as callback } from "node:child_process";
import { planPinUpgrades, applyPinUpgrades, type UpgradeEffects } from "../src/capability-language/pin-upgrades.js";
const execFile = promisify(callback);

test(
  "real jj snapshots and immutable Git publication propagate a changed interface into the root",
  { skip: !process.env.QX_CHECK_GENERATOR },
  async (context) => {
    const workbench = await fs.mkdtemp(path.join(os.tmpdir(), "qx-real-upgrade-"));
    context.after(() => fs.rm(workbench, { recursive: true, force: true }));
    // Exercise the actual effects with local bare remotes, without external writes.
    const environment = {
      GIT_CONFIG_COUNT: "1",
      GIT_CONFIG_KEY_0: `url.file://${workbench}/remotes/.insteadOf`,
      GIT_CONFIG_VALUE_0: "https://upgrade.test/",
      QUIXOS_JJ_NO_CHECKPOINT: "1",
      QUIXOS_CHECK_GENERATOR: process.env.QX_CHECK_GENERATOR!,
    };
    const previous = Object.fromEntries(Object.keys(environment).map((key) => [key, process.env[key]]));
    Object.assign(process.env, environment);
    context.after(() => {
      for (const [key, value] of Object.entries(previous))
        if (value === undefined) delete process.env[key];
        else process.env[key] = value;
    });
    const run = async (cwd: string, command: string, args: string[]) =>
      (await execFile(command, args, { cwd })).stdout.trim();
    await fs.mkdir(path.join(workbench, "remotes"));
    const nodes = [];
    for (const [kind, directory, remote] of [
      ["interface", "resources/Named", "named.git"],
      ["workspace", "root", "workspace.git"],
    ] as const) {
      const root = path.join(workbench, directory);
      await fs.mkdir(root, { recursive: true });
      await run(workbench, "git", ["init", "--bare", path.join(workbench, "remotes", remote)]);
      await run(root, "jj", ["git", "init", "--colocate"]);
      await run(root, "git", ["remote", "add", "origin", `https://upgrade.test/${remote}`]);
      await fs.writeFile(path.join(root, ".gitignore"), ".quixos/\n");
      const child = nodes[0];
      await fs.writeFile(
        path.join(root, "quixos.lock"),
        `quixos-lock version 1 { quixos source { repository "https://upgrade.test/quixos.git"; commit "${"a".repeat(40)}"; } ${child ? `interface Named source { repository "${child.source.repository}"; commit "${child.source.commit}"; }` : ""} }`,
      );
      await fs.writeFile(
        path.join(root, `${kind}.qx`),
        kind === "interface"
          ? 'interface Named id "interface:named" revision "interface:named@1" {}'
          : `workspace W id "workspace:w" revision "workspace:w@1" commit "${"a".repeat(40)}" { import interface Named; atom A id "atom:a"; }`,
      );
      await run(root, "jj", ["describe", "-m", "Initial source"]);
      const commit = await run(root, "jj", ["log", "--no-graph", "-r", "@", "-T", "commit_id"]);
      await run(root, "git", ["push", "origin", `${commit}:refs/tags/quixos-reachability/${commit}`]);
      nodes.push({ kind, directory, source: { repository: `https://upgrade.test/${remote}`, commit } });
    }
    await fs.mkdir(path.join(workbench, ".quixos"));
    await fs.writeFile(path.join(workbench, ".quixos/resource-graph.json"), JSON.stringify({ resources: [nodes[0]] }));
    await fs.appendFile(path.join(workbench, nodes[0].directory, "interface.qx"), "\n// incremental author edit\n");
    const plan = await planPinUpgrades(workbench, { nodes, bootstrap: true });
    const result = await applyPinUpgrades(plan);
    assert.equal(result.activated, false);
    const graph = JSON.parse(await fs.readFile(path.join(workbench, ".quixos/resource-graph.json"), "utf8"));
    const commit = graph.resources[0].source.commit;
    assert.notEqual(commit, nodes[0].source.commit);
    assert.match(await fs.readFile(path.join(workbench, "root/quixos.lock"), "utf8"), new RegExp(commit));
    assert.match(
      await run(workbench, "git", ["--git-dir", path.join(workbench, "remotes/named.git"), "show-ref"]),
      new RegExp(commit),
    );
  },
);

test("pin upgrades publish children before parent locks and resume without republishing completed nodes", async (context) => {
  const workbench = await fs.mkdtemp(path.join(os.tmpdir(), "qx-upgrade-test-"));
  context.after(() => fs.rm(workbench, { recursive: true, force: true }));
  const from = "a".repeat(40),
    to = "b".repeat(40),
    framework = "c".repeat(40);
  const sources = [
    {
      kind: "workspace" as const,
      directory: "root",
      source: { repository: "https://example.test/workspace.git", commit: from },
    },
    {
      kind: "interface" as const,
      directory: "resources/Named",
      source: { repository: "https://example.test/named.git", commit: from },
    },
  ];
  const lock = `quixos-lock version 1 { quixos source { repository "https://example.test/quixos.git"; commit "${framework}"; }`;
  for (const node of sources) {
    const directory = path.join(workbench, node.directory);
    await fs.mkdir(directory, { recursive: true });
    await execFile("git", ["-C", directory, "init"]);
    await execFile("git", ["-C", directory, "remote", "add", "origin", node.source.repository]);
    await fs.writeFile(path.join(directory, ".gitignore"), ".quixos/\n");
    await fs.writeFile(
      path.join(directory, "quixos.lock"),
      lock +
        (node.kind === "workspace"
          ? ` interface Named source { repository "${sources[1].source.repository}"; commit "${from}"; }`
          : "") +
        " }",
    );
  }
  await fs.writeFile(
    path.join(workbench, "resources/Named/interface.qx"),
    'interface Named id "interface:named" revision "interface:named@1" { value name id "member:name" : string { get id "op:get"; } }',
  );
  await fs.writeFile(
    path.join(workbench, "root/workspace.qx"),
    `workspace W id "workspace:w" revision "workspace:w@1" commit "${from}" { import interface Named; atom A id "atom:a"; }`,
  );
  const snapshotMap = path.join(workbench, "snapshots.json");
  await fs.writeFile(
    snapshotMap,
    JSON.stringify({
      resources: [
        {
          kind: "interface",
          repository: sources[1].source.repository,
          commit: to,
          directory: path.join(workbench, "resources/Named"),
        },
      ],
    }),
  );
  const oldMap = process.env.QUIXOS_SNAPSHOT_MAP;
  process.env.QUIXOS_SNAPSHOT_MAP = snapshotMap;
  context.after(() => {
    if (oldMap === undefined) delete process.env.QUIXOS_SNAPSHOT_MAP;
    else process.env.QUIXOS_SNAPSHOT_MAP = oldMap;
  });
  const plan = await planPinUpgrades(workbench, { nodes: sources });
  await fs.mkdir(path.join(workbench, ".quixos"), { recursive: true });
  await fs.writeFile(
    path.join(workbench, ".quixos/resource-graph.json"),
    JSON.stringify({ resources: [{ ...sources[1], source: { resolver: "git", ...sources[1].source } }] }),
  );
  assert.deepEqual(
    plan.nodes.map((node) => node.directory),
    ["resources/Named", "root"],
  );
  let fail = true;
  const published: string[] = [];
  const effects: UpgradeEffects = {
    check: async (node) => {
      if (node.kind === "workspace" && fail) throw new Error("refactor required");
    },
    snapshot: async () => to,
    publish: async (root) => {
      published.push(path.relative(workbench, root));
    },
  };
  await assert.rejects(() => applyPinUpgrades(plan, undefined, effects), /refactor required/);
  assert.deepEqual(published, ["resources/Named"]);
  assert.match(await fs.readFile(path.join(workbench, "root/quixos.lock"), "utf8"), new RegExp(to));
  const id = (await fs.readdir(path.join(workbench, ".quixos/upgrades")))
    .find((name) => name.endsWith(".json"))!
    .slice(0, -5);
  fail = false;
  await fs.appendFile(path.join(workbench, "root/workspace.qx"), "\n// explicit refactor\n");
  await assert.rejects(() => applyPinUpgrades(plan, id, effects), /accept-edits/);
  const result = await applyPinUpgrades(plan, id, effects, { acceptEdits: true });
  assert.equal(result.activated, false);
  assert.deepEqual(published, ["resources/Named", "root"]);
  const graph = JSON.parse(await fs.readFile(path.join(workbench, ".quixos/resource-graph.json"), "utf8"));
  assert.equal(graph.resources.length, 1);
  assert.equal(graph.resources[0].source.commit, to);
  const next = await planPinUpgrades(workbench, {
    nodes: [sources[0], { ...sources[1], source: graph.resources[0].source }],
  });
  assert.deepEqual(next.nodes.find((node) => node.kind === "workspace")!.dependencies, ["resources/Named"]);
});