import { execFileSync } from "node:child_process";
import { createHash } from "node:crypto";
import { chmodSync, cpSync, mkdirSync, mkdtempSync, readFileSync, readdirSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { basename, dirname, join, relative, resolve } from "node:path";
import { fileURLToPath } from "node:url";

const repository = resolve(dirname(fileURLToPath(import.meta.url)), "..");
const packages = resolve(process.argv[2] ?? join(repository, "packages"));
const output = resolve(process.argv[3] ?? join(repository, ".scriptc/releases"));
mkdirSync(output, { recursive: true });
const compiler = JSON.parse(readFileSync(join(packages, "compiler/package.json"), "utf8"));
const runtimes = Object.keys(compiler.optionalDependencies).filter((name) => name.startsWith("@scriptc/runtime-"));
const sums = [];

for (const name of readdirSync(packages).filter((name) => name.startsWith("cli-")).sort()) {
  const source = join(packages, name);
  const identity = JSON.parse(readFileSync(join(source, "package.json"), "utf8"));
  if (identity.version !== compiler.version) throw new Error(`native CLI version mismatch: ${name}`);
  const stage = mkdtempSync(join(process.platform === "win32" ? tmpdir() : "/tmp", "scriptc-release-"));
  try {
    cpSync(join(source, "package.json"), join(stage, "package.json"));
    const distribution = join(stage, "dist");
    cpSync(join(source, "dist"), distribution, { recursive: true });
    const bin = join(distribution, "bin");
    const binary = join(bin, name.includes("win32-") ? "scriptc.exe" : "scriptc");
    const manifest = JSON.parse(readFileSync(binary + ".json", "utf8"));
    const hostPath = (value) => name.includes("win32-") ? value.replaceAll("\\", "/") : value;
    const packs = new Map();
    for (const packageName of runtimes) {
      const directory = packageName.replace("@scriptc/", "");
      const from = join(packages, directory);
      const to = join(distribution, "lib", directory);
      const runtime = JSON.parse(readFileSync(join(from, "runtime-pack.json"), "utf8"));
      if (runtime.version !== identity.version || runtime.package !== packageName) throw new Error(`runtime version mismatch: ${packageName}`);
      for (const member of ["package.json", "runtime-pack.json", "artifacts"]) {
        cpSync(join(from, member), join(to, member), { recursive: true });
      }
      packs.set(runtime.target.name, { target: runtime.target.name, path: relative(bin, to) });
    }
    manifest.runtime_packs = [...packs.values()];
    writeFileSync(binary + ".json", JSON.stringify(manifest, null, 2) + "\n");
    // GitHub artifact transport drops executable modes. Restore the native
    // tools before validating and creating the standalone archive.
    for (const path of [binary, resolve(bin, hostPath(manifest.ts7)), resolve(bin, hostPath(manifest.comptime)),
      join(resolve(bin, hostPath(manifest.llvm_package)), "bin", name.includes("win32-") ? "scriptc-llvm-codegen.exe" : "scriptc-llvm-codegen")]) {
      chmodSync(path, 0o755);
    }
    execFileSync(process.execPath, [join(repository, "scripts/verify-native-cli.mjs"), stage], { stdio: "inherit" });
    const archive = join(output, `scriptc-${identity.version}-${name.slice(4)}.tar.gz`);
    execFileSync("tar", ["-czf", archive, "-C", distribution, "."], { stdio: "inherit" });
    const digest = createHash("sha256").update(readFileSync(archive)).digest("hex");
    sums.push(`${digest}  ${basename(archive)}`);
    console.log(archive);
  } finally { rmSync(stage, { recursive: true, force: true }); }
}
if (sums.length === 0) throw new Error("no native CLI packages found");
writeFileSync(join(output, "SHA256SUMS"), sums.join("\n") + "\n");