#!/usr/bin/env node
/* eslint-disable no-console */
import { spawn } from "node:child_process";
import { createHash } from "node:crypto";
import { mkdir, readdir, readFile, stat, writeFile } from "node:fs/promises";
import path from "node:path";
import { fileURLToPath } from "node:url";
import { markFailedBaselineGates } from "./baseline-gate-exit.mjs";

const __dirname = path.dirname(fileURLToPath(import.meta.url));
const repoRoot = path.resolve(__dirname, "..");
export const reportPath = path.join(repoRoot, ".tc/baselines/mfa-scope/baseline-health.md");

const ignoredDirectoryNames = new Set([
  ".git",
  ".cache",
  ".next",
  ".react-router",
  ".tc",
  ".turbo",
  ".vite",
  "build",
  "cache",
  "coverage",
  "dist",
  "generated",
  "node_modules",
]);

const ignoredFileNames = new Set(["package-lock.json", "pnpm-lock.yaml", "yarn.lock"]);

export const mfaScopeMfaScopeInventoryScanRoots = ["apps/web/app", "prisma/schema", "docs/security", "scripts", "package.json"];

export const mfaScopeMfaScopeInventoryPatterns = [
  { id: "mfa", pattern: /\b(?:MFA|multi-factor|multifactor|multi factor|two-factor|2FA)\b/gi },
  { id: "totp", pattern: /\b(?:TOTP|time-based one-time|authenticator app|one-time password|otp)\b/gi },
  { id: "recovery-code", pattern: /\b(?:recovery code|backup code|scratch code|emergency code)\b/gi },
  { id: "step-up", pattern: /\b(?:step-up|step up|reauth(?:enticate|entication)?|elevated auth|privilege escalation)\b/gi },
  { id: "admin-enforcement", pattern: /\b(?:admin enforcement|administrator enforcement|admin required|require mfa|enforce mfa|mandatory mfa)\b/gi },
  { id: "auth-session", pattern: /\b(?:auth session|session token|session cookie|session invalidation|remembered device|trusted device)\b/gi },
  { id: "security-doc", pattern: /\b(?:security doc|security documentation|threat model|risk assessment|security checklist|security policy)\b/gi },
];

export const gates = [
  { id: "typecheck", command: "pnpm", args: ["run", "typecheck"] },
  { id: "lint", command: "pnpm", args: ["run", "lint"] },
  { id: "ci-baseline", command: "pnpm", args: ["run", "verify:ci-baseline"] },
  { id: "mfaScope-mfa-scope-inventory", type: "mfaScope-mfa-scope-inventory" },
];

function shellCommand(gate) {
  if (gate.type === "mfaScope-mfa-scope-inventory") return "read-only MFA Scope MFA scope inventory";
  return [gate.command, ...gate.args].join(" ");
}

function outputFingerprint(output) {
  const normalized = output
    .split(/\r?\n/)
    .map((line) => line.trim())
    .filter(Boolean)
    .slice(-40)
    .join("\n");
  return { hash: createHash("sha256").update(output).digest("hex").slice(0, 12), excerpt: normalized.slice(0, 1200) || "(no output)" };
}

function parseNumber(value) {
  const parsed = Number.parseInt(value, 10);
  return Number.isFinite(parsed) ? parsed : 0;
}

function conservativeFailureCount(count, output) {
  return count > 0 ? count : output.trim() ? 1 : 0;
}

export function failureCount(gateId, stdout, stderr, exitCode = 1) {
  if (exitCode === 0 || gateId === "mfaScope-mfa-scope-inventory") return 0;
  const output = `${stdout}\n${stderr}`;
  if (gateId === "typecheck") return conservativeFailureCount((output.match(/error TS\d+/g) ?? []).length, output);
  if (gateId === "lint") {
    const problemsMatch = output.match(/✖\s+(\d+)\s+problems?\s+\((\d+)\s+errors?/i);
    const compactMatch = output.match(/(\d+)\s+errors?/i);
    return conservativeFailureCount(problemsMatch ? parseNumber(problemsMatch[2]) : compactMatch ? parseNumber(compactMatch[1]) : 0, output);
  }
  if (gateId === "ci-baseline") {
    const summaryMatch = output.match(/Summary:\s+\d+\s+passed,\s+\d+\s+observed,\s+(\d+)\s+failed/i);
    return conservativeFailureCount(summaryMatch ? parseNumber(summaryMatch[1]) : 0, output);
  }
  return output.trim() ? 1 : 0;
}

function runShellGate(gate) {
  return new Promise((resolve) => {
    const startedAt = Date.now();
    const child = spawn(gate.command, gate.args, { cwd: repoRoot, env: { ...process.env, CI: "1" }, shell: false });
    let stdout = "";
    let stderr = "";
    child.stdout.on("data", (chunk) => {
      stdout += chunk;
    });
    child.stderr.on("data", (chunk) => {
      stderr += chunk;
    });
    child.on("error", (error) => {
      stderr += `\n${error.message}`;
    });
    child.on("close", (code) => {
      const output = `${stdout}\n${stderr}`;
      resolve({
        id: gate.id,
        command: shellCommand(gate),
        exitCode: code ?? 1,
        durationMs: Date.now() - startedAt,
        failureCount: failureCount(gate.id, stdout, stderr, code ?? 1),
        fingerprint: outputFingerprint(output),
      });
    });
  });
}

function isIgnoredDirectory(name, relativePath) {
  return ignoredDirectoryNames.has(name) || ignoredDirectoryNames.has(relativePath) || name.endsWith(".cache");
}

function isIgnoredFile(name) {
  return ignoredFileNames.has(name) || name.endsWith(".generated.ts") || name.endsWith(".generated.tsx") || name.endsWith(".min.js") || name.endsWith(".map");
}

async function rootExists(root, rootRelativePath) {
  try {
    await stat(path.join(root, rootRelativePath));
    return true;
  } catch (error) {
    if (error?.code === "ENOENT") return false;
    throw error;
  }
}

async function collectFiles(root, rootRelativePath) {
  const absoluteRoot = path.join(root, rootRelativePath);
  const rootStat = await stat(absoluteRoot);
  if (rootStat.isFile()) return [absoluteRoot];
  const files = [];
  async function walk(directory) {
    const entries = await readdir(directory, { withFileTypes: true });
    for (const entry of entries) {
      const absolutePath = path.join(directory, entry.name);
      const relativePath = path.relative(root, absolutePath).split(path.sep).join("/");
      if (entry.isDirectory()) {
        if (isIgnoredDirectory(entry.name, relativePath)) continue;
        await walk(absolutePath);
      } else if (entry.isFile() && /\.(?:cjs|cts|js|json|jsx|mdx?|mjs|mts|prisma|ts|tsx)$/.test(entry.name) && !isIgnoredFile(entry.name)) {
        files.push(absolutePath);
      }
    }
  }
  await walk(absoluteRoot);
  return files;
}

export async function runMfaScopeMfaScopeInventory(root = repoRoot) {
  const existingRoots = [];
  const missingRoots = [];
  for (const rootRelativePath of mfaScopeMfaScopeInventoryScanRoots) {
    if (await rootExists(root, rootRelativePath)) existingRoots.push(rootRelativePath);
    else missingRoots.push(rootRelativePath);
  }
  const files = (await Promise.all(existingRoots.map((rootRelativePath) => collectFiles(root, rootRelativePath)))).flat();
  const counts = new Map(mfaScopeMfaScopeInventoryPatterns.map(({ id }) => [id, 0]));
  const matches = [];
  for (const file of files) {
    const relativeFile = path.relative(root, file).split(path.sep).join("/");
    const lines = (await readFile(file, "utf8")).split(/\r?\n/);
    lines.forEach((line, index) => {
      for (const { id, pattern } of mfaScopeMfaScopeInventoryPatterns) {
        pattern.lastIndex = 0;
        const found = line.match(pattern);
        if (!found) continue;
        counts.set(id, (counts.get(id) ?? 0) + found.length);
        for (const match of found) {
          matches.push(`${relativeFile}:${index + 1}: [${id}] ${match}: ${line.trim()}`);
        }
      }
    });
  }
  return { roots: existingRoots, missingRoots, filesScanned: files.length, counts: Object.fromEntries(counts.entries()), totalMatches: matches.length, matches };
}

async function runMfaScopeMfaScopeInventoryGate(gate) {
  const startedAt = Date.now();
  const inventory = await runMfaScopeMfaScopeInventory(repoRoot);
  const outputLines = [
    `MFA Scope MFA scope inventory scanned roots: ${inventory.roots.join(", ") || "(none)"}`,
    `MFA Scope MFA scope inventory missing roots skipped: ${inventory.missingRoots.join(", ") || "(none)"}`,
    `files scanned: ${inventory.filesScanned}`,
    ...Object.entries(inventory.counts).map(([id, count]) => `${id}: ${count}`),
    `total matches: ${inventory.totalMatches}`,
    ...inventory.matches.slice(0, 240),
  ];
  if (inventory.matches.length > 240) outputLines.push(`... ${inventory.matches.length - 240} additional matches omitted`);
  const output = outputLines.join("\n");
  return { id: gate.id, command: shellCommand(gate), exitCode: 0, durationMs: Date.now() - startedAt, failureCount: 0, inventory, fingerprint: outputFingerprint(output) };
}

async function runGate(gate) {
  if (gate.type === "mfaScope-mfa-scope-inventory") return runMfaScopeMfaScopeInventoryGate(gate);
  return runShellGate(gate);
}

function renderInventoryCounts(result) {
  if (!result.inventory) return [];
  return [
    "- Inventory Roots: " + (result.inventory.roots.join(", ") || "(none)"),
    "- Missing Roots Skipped: " + (result.inventory.missingRoots.join(", ") || "(none)"),
    `- Files Scanned: ${result.inventory.filesScanned}`,
    "- Inventory Counts:",
    ...Object.entries(result.inventory.counts).map(([id, count]) => `  - ${id}: ${count}`),
    `- Total Inventory Matches: ${result.inventory.totalMatches}`,
  ];
}

export function renderReport(results) {
  const lines = [
    "# MFA Scope Baseline Health",
    "",
    `Generated: ${new Date().toISOString()}`,
    "",
    "This report captures non-E2E baseline gates for MFA Scope baseline. The runner exits 0 when every gate was attempted and this report was written, even if individual gates fail.",
    "The shell gates run with CI=1 and fingerprint TypeScript, ESLint, and CI baseline output. The MFA Scope MFA scope inventory gate is read-only and fingerprints MFA/TOTP/recovery-code/step-up/admin-enforcement/auth-session/security-doc references across the configured roots while reporting missing roots as skipped.",
    "",
  ];

  for (const result of results) {
    lines.push(
      `## ${result.id}`,
      "",
      `- Command: \`${result.command}\``,
      `- Exitcode: ${result.exitCode}`,
      `- Failure Count: ${result.failureCount}`,
      `- Duration: ${result.durationMs}ms`,
      `- Output Fingerprint: ${result.fingerprint.hash}`,
      ...renderInventoryCounts(result),
      "",
      "```text",
      result.fingerprint.excerpt,
      "```",
      "",
    );
  }
  return `${lines.join("\n")}\n`;
}

async function writeFailureReport(error) {
  await mkdir(path.dirname(reportPath), { recursive: true });
  const output = error?.stack ?? String(error);
  await writeFile(
    reportPath,
    renderReport([{ id: "runner-error", command: "node scripts/verify-mfa-scope-baseline.mjs", exitCode: 1, durationMs: 0, failureCount: 1, fingerprint: outputFingerprint(output) }]),
    "utf8",
  );
}

async function main() {
  const results = [];
  for (const gate of gates) {
    console.log(`Running ${gate.id}: ${shellCommand(gate)}`);
    results.push(await runGate(gate));
  }
  await mkdir(path.dirname(reportPath), { recursive: true });
  await writeFile(reportPath, renderReport(results), "utf8");
  console.log(`Wrote ${path.relative(repoRoot, reportPath)}`);
  markFailedBaselineGates(results);
}

if (process.argv[1] === fileURLToPath(import.meta.url)) {
  main().catch(async (error) => {
    console.error(error);
    try {
      await writeFailureReport(error);
    } finally {
      process.exit(1);
    }
  });
}