#!/usr/bin/env node

import { createHash } from "node:crypto";
import { spawnSync } from "node:child_process";
import { copyFile, lstat, mkdir, mkdtemp, readFile, rm, writeFile } from "node:fs/promises";
import { tmpdir } from "node:os";
import { dirname, join, resolve } from "node:path";
import { fileURLToPath } from "node:url";

const scriptDir = dirname(fileURLToPath(import.meta.url));
const workspaceRoot = resolve(scriptDir, "../../..");
const engineRoot = resolve(
  process.env.NODEDC_ENGINE_SOURCE_ROOT || join(workspaceRoot, "NODEDC_ENGINE_INFRA"),
);
const artifactDir = resolve(
  process.env.NODEDC_DEPLOY_ARTIFACT_DIR || resolve(scriptDir, "../deploy-artifacts"),
);
const [patchId = "engine-mcp-execution-plan-sandbox-runtime-20260724-045", ...extra] =
  process.argv.slice(2);

if (
  extra.length
  || !/^engine-mcp-execution-plan-sandbox-runtime-\d{8}-\d{3}$/.test(patchId)
) {
  throw new Error(
    "usage: build-engine-mcp-execution-plan-sandbox-runtime-artifact.mjs "
      + "[engine-mcp-execution-plan-sandbox-runtime-YYYYMMDD-NNN]",
  );
}

const expectedSha256 = Object.freeze({
  "nodedc-source/server/l2ExecutionPlan/compiler.js":
    "64f5196a83018505c6dac77a0f8674c27941257a874eed9b024c1c94f169be2b",
  "nodedc-source/server/deployTransitions/executionPlanSandboxRuntimeV4.json":
    "a4692643afb86dfeeee6ca24aefcc181913e715eff38fa158667d2f10b901847",
});
const files = Object.freeze(Object.keys(expectedSha256));
const artifact = join(artifactDir, `nodedc-${patchId}.tgz`);
const checksum = `${artifact}.sha256`;
const stage = await mkdtemp(join(tmpdir(), "nodedc-engine-execution-plan-sandbox-runtime-"));

await assertFresh(artifact);
await assertExactSources();

try {
  for (const relativePath of files) {
    const destination = join(stage, "payload", relativePath);
    await mkdir(dirname(destination), { recursive: true });
    await copyFile(join(engineRoot, relativePath), destination);
  }
  await writeFile(
    join(stage, "manifest.env"),
    `id=${patchId}\ncomponent=engine\ntype=app-overlay\n`,
    "utf8",
  );
  await writeFile(join(stage, "files.txt"), `${files.join("\n")}\n`, "utf8");
  await mkdir(artifactDir, { recursive: true });
  run("python3", ["-c", canonicalTarScript(), artifact, stage]);
  const sha256 = digest(await readFile(artifact));
  await writeFile(checksum, `${sha256}  ${artifact.split("/").at(-1)}\n`, "utf8");
  console.log(JSON.stringify({
    ok: true,
    patchId,
    artifact,
    checksum,
    sha256,
    services: ["nodedc-backend"],
    mcpVersion: "0.11.0",
    executionPlanCompilerVersion: "1.4.0",
    n8nVersion: "2.3.2",
    codeNodeMode: "runOnceForAllItems",
    sandboxGlobalDependencies: [],
    responseByteBudget: "bounded-pure-javascript-utf8",
    rawProviderPayloadAtPublish: "forbidden",
    engineProviderHardcode: false,
    n8nCoreChanged: false,
    l1Changed: false,
    credentialsChanged: false,
    files,
  }, null, 2));
} finally {
  await rm(stage, { recursive: true, force: true });
}

async function assertExactSources() {
  for (const [relativePath, expected] of Object.entries(expectedSha256)) {
    const sourcePath = join(engineRoot, relativePath);
    const info = await lstat(sourcePath);
    if (!info.isFile() || info.isSymbolicLink()) {
      throw new Error(`engine_execution_plan_sandbox_runtime_source_unsafe:${relativePath}`);
    }
    const actual = digest(await readFile(sourcePath));
    if (actual !== expected) {
      throw new Error(
        `engine_execution_plan_sandbox_runtime_target_mismatch:${relativePath}:`
          + `expected=${expected}:actual=${actual}`,
      );
    }
  }

  const compiler = await readFile(
    join(engineRoot, "nodedc-source/server/l2ExecutionPlan/compiler.js"),
    "utf8",
  );
  const descriptor = JSON.parse(await readFile(
    join(
      engineRoot,
      "nodedc-source/server/deployTransitions/executionPlanSandboxRuntimeV4.json",
    ),
    "utf8",
  ));
  if (
    !compiler.includes("const encodedBytes = (value) => {")
    || compiler.includes("new TextEncoder().encode(JSON.stringify(value ?? null))")
    || !compiler.includes("return extracted.map((source) => ({ json:")
    || descriptor?.id !== "engine-mcp-l2-execution-plan-sandbox-runtime-v4"
    || descriptor?.predecessor !== "engine-mcp-l1-credential-provenance-v2"
    || descriptor?.runtimeCompatibility?.sandboxGlobalDependencies?.length !== 0
    || descriptor?.runtimeCompatibility?.responseByteBudget
      !== "bounded-pure-javascript-utf8"
    || descriptor?.dataBoundary?.rawProviderPayloadAtPublish !== "forbidden"
    || descriptor?.engineProviderHardcode !== false
    || descriptor?.n8nCoreChanged !== false
  ) {
    throw new Error("engine_execution_plan_sandbox_runtime_boundary_invalid");
  }
}

async function assertFresh(path) {
  try {
    await lstat(path);
  } catch (error) {
    if (error?.code === "ENOENT") return;
    throw error;
  }
  throw new Error("artifact_already_exists");
}

function canonicalTarScript() {
  return [
    "import gzip,io,pathlib,sys,tarfile",
    "root=pathlib.Path(sys.argv[2])",
    "with open(sys.argv[1],'xb') as out:",
    " with gzip.GzipFile(filename='',mode='wb',fileobj=out,compresslevel=9,mtime=0) as gz:",
    "  with tarfile.open(fileobj=gz,mode='w',format=tarfile.PAX_FORMAT) as tar:",
    "   for top in ('manifest.env','files.txt','payload'):",
    "    p=root/top; paths=[p]+(sorted(p.rglob('*')) if p.is_dir() else [])",
    "    for x in paths:",
    "     info=tar.gettarinfo(str(x),arcname=x.relative_to(root).as_posix())",
    "     info.uid=info.gid=0; info.uname=info.gname='root'; info.mtime=0; info.mode=0o755 if info.isdir() else 0o644",
    "     with (open(x,'rb') if info.isfile() else io.BytesIO()) as src: tar.addfile(info,src if info.isfile() else None)",
  ].join("\n");
}

function digest(value) {
  return createHash("sha256").update(value).digest("hex");
}

function run(command, args) {
  const result = spawnSync(command, args, {
    encoding: "utf8",
    maxBuffer: 128 * 1024 * 1024,
  });
  if (result.status !== 0) {
    throw new Error(`${command}_failed:${result.stderr || result.stdout}`);
  }
}
