// source-pin-contract: sandbox-wipe guard GUC identity; every guard swept is derived by walking the release schema (active baseline then forward migrations), wipe setters derived from engine sources
import assert from "node:assert/strict";
import { existsSync, readFileSync } from "node:fs";
import test from "node:test";
import { releaseSchemaFiles } from "../scripts/release-schema-sql.mjs";

test("the release schema starts from the active baseline named by the manifest", () => {
  // Installations apply the manifest's baseline and then the reviewed forward
  // migrations in ordinal order; anything else in the forward directory is an
  // unreviewed artifact, not a migration. Asserted as a shape rule, not an
  // inventory: a new reviewed migration with a valid ordinal passes without
  // re-pinning. Ordinal uniqueness and ordering live in migration-ordinals;
  // header spellings in the migration-headers gate.
  const manifest = JSON.parse(readFileSync("schema/migrations/baseline.json", "utf8")) as {
    filename: string;
  };
  assert.match(manifest.filename, /^baselines\/[a-z0-9_.-]+\.sql$/);
  assert.ok(
    existsSync(`schema/migrations/${manifest.filename}`),
    `the active baseline ${manifest.filename} must be present`,
  );
  const [root, ...forward] = releaseSchemaFiles().map((file) => file.filename);
  assert.equal(root, manifest.filename, "the release schema must begin with the active baseline");
  const malformed = forward.filter(
    (filename) => !/^generated\/\d{4}_[a-z0-9_]+\.sql$/.test(filename),
  );
  assert.deepEqual(
    malformed,
    [],
    `these are not reviewer-allocated forward migrations: ${malformed.join(", ")}`,
  );
});

test("every effective sandbox-wipe guard reads the GUC the wipe source sets", () => {
  // The wipe source (sandbox lifecycle) and the scratch-teardown path must
  // agree on one GUC, and every storage guard that honors a wipe must read
  // exactly that GUC — directly where it defines the mapping, through the
  // canonical helper everywhere else. A guard that trusts a raw GUC, or a
  // drifted name, silently stops honoring wipes (or honors forged ones).
  const schemaSources = releaseSchemaFiles().map(({ path }) => readFileSync(path, "utf8"));
  const lifecycleSource = readFileSync("engine/src/sandbox/lifecycle.ts", "utf8");
  const fixtureWipeSource = readFileSync("engine/src/testing/tenant-teardown.ts", "utf8");
  const setterMatch = lifecycleSource.match(
    /set_config\('([a-z0-9_.]+\.sandbox_wipe)', 'on', true\)/,
  );
  assert.ok(setterMatch, "sandbox lifecycle must set its wipe GUC explicitly");
  const wipeGuc = setterMatch[1]!;
  assert.match(
    fixtureWipeSource,
    new RegExp(`set_config\\('${wipeGuc.replaceAll(".", "\\.")}', 'on', true\\)`),
    "scratch teardown must use the same wipe GUC as sandbox lifecycle",
  );

  // Build the final function catalog in apply order: the baseline first, then
  // forward migrations replacing earlier bodies, so this checks the guards a
  // live database actually has — including ones a later migration could
  // reintroduce under a drifted name.
  const effectiveFunctions = new Map<string, string>();
  const functionDefinition =
    /CREATE(?: OR REPLACE)? FUNCTION public\.([a-z0-9_]+)\([^;]*?\)\s+RETURNS[\s\S]*?\s+AS (\$[a-z0-9_]*\$)([\s\S]*?)\2;/gi;
  for (const source of schemaSources) {
    for (const match of source.matchAll(functionDefinition)) {
      effectiveFunctions.set(match[1]!, match[3]!);
    }
  }
  const effectiveWipeBodies = new Map(
    [...effectiveFunctions].filter(([, body]) => body.includes("sandbox_wipe")),
  );
  assert.ok(effectiveWipeBodies.size > 0, "expected wipe-honoring guards in the effective catalog");
  for (const [functionName, body] of effectiveWipeBodies) {
    assert.doesNotMatch(body, /app\.sandbox_wipe/, `${functionName} retains the drifted GUC`);
    if (functionName === "openbooks_sandbox_wipe_allowed") {
      // The mapping definition itself: the one place allowed to read the raw
      // GUC, and it must be the canonical one.
      assert.match(
        body,
        new RegExp(`current_setting\\('${wipeGuc.replaceAll(".", "\\.")}'`),
        "the wipe helper must read the wipe source's GUC",
      );
      continue;
    }
    // Every other guard bypasses only through the helper, never by reading a
    // wipe GUC itself: a raw read trusts whatever the session claims, while
    // the helper is the authorized DELETE path the wipe source sets.
    assert.match(
      body,
      /openbooks_sandbox_wipe_allowed\(/,
      `${functionName} must bypass only through the canonical wipe helper`,
    );
    assert.doesNotMatch(
      body,
      /current_setting\([^)]*sandbox_wipe/i,
      `${functionName} must not trust a raw wipe GUC in its body`,
    );
  }
});