#!/usr/bin/env node
/**
 * Vendored tree-sitter grammar update monitor.
 *
 * Checks each vendored grammar against its upstream source-of-origin and, for an
 * available AND ABI-compatible update, re-vendors the grammar source in place so
 * a PR can be opened. The version bump in vendor/<name>/package.json then triggers
 * .github/workflows/build-tree-sitter-prebuilds.yml, which cross-builds + ABI-
 * validates the prebuilds — so even an imperfect re-vendor can never silently
 * ship: its PR's CI goes red.
 *
 * The runtime is tree-sitter@0.25.1 (LANGUAGE_VERSION 13–15). Read the
 * candidate's generated ABI before updating; incompatible candidates are
 * reported rather than applied. Semantic compatibility is verified by PR CI.
 *
 * Usage:
 *   node update-vendored-grammars.mjs            # detect only → JSON report on stdout
 *   node update-vendored-grammars.mjs --apply X  # re-vendor grammar X in place
 *
 * Policy holds are recorded in the shared manifest. Swift is held at the
 * newest tested release that preserves declaration ownership during recovery.
 */
import { execFileSync } from 'node:child_process';
import fs from 'node:fs';
import os from 'node:os';
import path from 'node:path';
import { fileURLToPath, pathToFileURL } from 'node:url';

const __dirname = path.dirname(fileURLToPath(import.meta.url));
const REPO_ROOT = path.resolve(__dirname, '..', '..');
const VENDOR = path.join(REPO_ROOT, 'gitnexus', 'vendor');

const COMPATIBLE_ABI = new Set([13, 14, 15]); // tree-sitter@0.25.1 LANGUAGE_VERSION range

// Source-of-origin per grammar. npm grammars resolve `latest` via the registry;
// github grammars (no usable npm release) track the default branch HEAD. A `hold`
// reason makes a grammar report-only: updates are detected + surfaced but never
// auto-applied (for example, when newer upstream error recovery regresses ownership).
//
// The vendored set lives in .github/vendored-grammars.json — the SHARED source of
// truth this monitor and .github/scripts/check-tree-sitter-upgrade-readiness.py both
// read, so the two tree-sitter workflows can never disagree about which grammars are
// vendored or where their upstream lives. We reshape the manifest's
// `{ upstream: { npm | github } }` form into the flat `{ npm? , github? }` shape the
// rest of this script consumes. This is a local file read (import-safe, no network).
const MANIFEST = path.join(REPO_ROOT, '.github', 'vendored-grammars.json');
// `raw` is injectable for testing; production reads the manifest file.
function loadManifestGrammars(raw = null) {
  if (raw === null) {
    // Fail loud with a pointer, not a bare ENOENT/SyntaxError: this runs at import.
    try {
      raw = JSON.parse(fs.readFileSync(MANIFEST, 'utf8'));
    } catch (e) {
      throw new Error(
        `Could not load the vendored-grammars manifest at ${MANIFEST} ` +
          `(shared source of truth — see CONTRIBUTING.md → CI automation contracts): ${e.message}`,
      );
    }
  }
  return Object.fromEntries(
    Object.entries(raw.grammars || {}).map(([key, g]) => {
      if (!g.name)
        throw new Error(`manifest entry '${key}' is missing a 'name' field (${MANIFEST})`);
      // Defense-in-depth: `name` is joined into gitnexus/vendor/<name> paths (and
      // apply() WRITES there), so reject anything that isn't a plain grammar name
      // before it can traverse the filesystem (#2187).
      if (!/^tree-sitter-[a-z0-9-]+$/.test(g.name))
        throw new Error(
          `manifest entry '${key}' has an invalid grammar name '${g.name}' ` +
            `(must match tree-sitter-[a-z0-9-]+)`,
        );
      return [
        key,
        {
          name: g.name,
          ...(g.upstream?.npm ? { npm: g.upstream.npm } : {}),
          ...(g.upstream?.github ? { github: g.upstream.github } : {}),
          ...(g.hold ? { hold: g.hold } : {}),
        },
      ];
    }),
  );
}
const GRAMMARS = loadManifestGrammars();

const sh = (cmd, args, opts = {}) =>
  execFileSync(cmd, args, { encoding: 'utf8', stdio: ['ignore', 'pipe', 'pipe'], ...opts }).trim();

const clean = (v) =>
  String(v || '')
    .replace(/^[v^~]/, '')
    .trim();

// Shared "is the candidate newer than what we ship?" check, used by BOTH detect()
// and apply() so they can never disagree. up.version is the comparable identity for
// both kinds: a plain semver for npm, and the `<base>-g<sha7>` provenance string for
// github (which apply() also writes to package.json). detect() previously compared
// the bare sha7 for github, so after the bot re-vendored a github grammar once it
// reported a perpetual false "update available" while apply() saw "already current"
// (#2187 review). Comparing up.version on both sides removes that asymmetry.
const isNewer = (up, have) => !have || up.version !== have;

// apply() throws this (instead of calling process.exit) so its error branches are
// exercisable in-process by tests; the CLI entrypoint maps `.code` back to the
// original exit code, keeping the monitor's subprocess contract identical (#2187).
class ApplyExit extends Error {
  constructor(message, code) {
    super(message);
    this.name = 'ApplyExit';
    this.code = code;
  }
}

function vendoredVersion(g) {
  const p = path.join(VENDOR, g.name, 'package.json');
  return clean(JSON.parse(fs.readFileSync(p, 'utf8')).version);
}

/** Resolve the upstream candidate: { version, ref, kind }. */
function resolveUpstream(g) {
  if (g.npm) {
    const version = clean(sh('npm', ['view', g.npm, 'version']));
    return { version, ref: version, kind: 'npm' };
  }
  // github: no reliable release tags here, so track the default branch HEAD sha.
  const meta = JSON.parse(sh('gh', ['api', `repos/${g.github}`]));
  const branch = meta.default_branch;
  const sha = JSON.parse(sh('gh', ['api', `repos/${g.github}/commits/${branch}`])).sha;
  // Version key: "<upstreamPkgVersion>-g<sha7>" — safeRef-compatible (no `+`,
  // which the build workflow's ref validator rejects) and changes on every commit.
  let base = '0.0.0';
  try {
    const pkg = JSON.parse(
      Buffer.from(
        JSON.parse(sh('gh', ['api', `repos/${g.github}/contents/package.json?ref=${sha}`])).content,
        'base64',
      ).toString('utf8'),
    );
    if (pkg.version) base = clean(pkg.version);
  } catch {
    /* no upstream package.json — base stays 0.0.0 */
  }
  return { version: `${base}-g${sha.slice(0, 7)}`, ref: sha, kind: 'github' };
}

/** Fetch the candidate source into a temp dir; return the package root. */
function fetchSource(g, ref) {
  const work = fs.mkdtempSync(
    path.join(os.tmpdir(), `revendor-${Object.keys(GRAMMARS).find((k) => GRAMMARS[k] === g)}-`),
  );
  if (g.npm) {
    sh('npm', ['pack', `${g.npm}@${ref}`, '--silent'], { cwd: work });
    const tgz = fs.readdirSync(work).find((f) => f.endsWith('.tgz'));
    sh('tar', ['xzf', tgz], { cwd: work });
    return path.join(work, 'package');
  }
  // github tarball at the resolved sha. Download + extract WITHOUT a shell
  // (no `bash -c`/redirect): `gh api` writes the binary tarball to stdout, which
  // we capture as a Buffer and write to a fixed path, then extract with execFile.
  // Avoids the shell-command-injection surface CodeQL flags when an API-derived
  // ref is interpolated into a `bash -c` string.
  const tgz = path.join(work, 'src.tgz');
  fs.writeFileSync(
    tgz,
    execFileSync('gh', ['api', `repos/${g.github}/tarball/${ref}`], {
      maxBuffer: 512 * 1024 * 1024,
    }),
  );
  sh('tar', ['xzf', tgz], { cwd: work });
  const dir = fs.readdirSync(work).find((f) => fs.statSync(path.join(work, f)).isDirectory());
  return path.join(work, dir);
}

/** Read parser.c's LANGUAGE_VERSION (ABI). Prefer the ABI-14 default parser.c. */
function readAbi(srcRoot) {
  const candidates = ['src/parser.c', 'parser.c'];
  for (const rel of candidates) {
    const p = path.join(srcRoot, rel);
    if (!fs.existsSync(p)) continue;
    // Read only the head — the #define is near the top.
    const head = fs.readFileSync(p, 'utf8').slice(0, 4000);
    const m = head.match(/#define\s+LANGUAGE_VERSION\s+(\d+)/);
    if (m) return Number(m[1]);
  }
  return null; // unknown (e.g. parser.c only generated at build time)
}

// `deps` injects the network/filesystem seams (vendoredVersion / resolveUpstream /
// fetchSource / readAbi) so the classification logic — newer-detection, the ABI
// gate, and the policy-hold gate — can be unit-tested offline with fixtures, never
// touching live npm/GitHub. Production passes nothing and gets the real functions.
function detect(deps = {}) {
  const getVendored = deps.vendoredVersion || vendoredVersion;
  const resolveUp = deps.resolveUpstream || resolveUpstream;
  const fetchSrc = deps.fetchSource || fetchSource;
  const readAbiFn = deps.readAbi || readAbi;
  const report = [];
  for (const [key, g] of Object.entries(GRAMMARS)) {
    const have = getVendored(g);
    let up;
    try {
      up = resolveUp(g);
    } catch (err) {
      report.push({ grammar: key, error: String(err.message || err) });
      continue;
    }
    const newer = isNewer(up, have);
    let abi = null;
    if (newer) {
      try {
        abi = readAbiFn(fetchSrc(g, up.ref));
      } catch {
        /* fetch/abi best-effort; null = unknown */
      }
    }
    report.push({
      grammar: key,
      vendored: have,
      upstream: up.version,
      ref: up.ref,
      kind: up.kind,
      update: newer,
      abi,
      abiCompatible: abi == null ? null : COMPATIBLE_ABI.has(abi),
      hold: g.hold || null,
      // Auto-appliable only when there's an update, the ABI is known-compatible,
      // AND the grammar is not on a policy hold (e.g. swift).
      applicable: newer && abi != null && COMPATIBLE_ABI.has(abi) && !g.hold,
    });
  }
  return report;
}

const copyFile = (srcRoot, dest, rel) => {
  const from = path.join(srcRoot, rel);
  if (!fs.existsSync(from)) return false;
  const to = path.join(dest, rel);
  fs.mkdirSync(path.dirname(to), { recursive: true });
  fs.copyFileSync(from, to);
  return true;
};

/**
 * Re-vendor one grammar in place from its ABI-compatible upstream candidate.
 * Copies ONLY the generated source-build + runtime files; deliberately KEEPS the
 * GitNexus-hardened binding.gyp (Windows cflags, target_name), README (vendor
 * notice), LICENSE, and prebuilds/ (the build workflow refreshes those). Bumps the
 * stripped vendor package.json version + provenance — never re-introduces
 * scripts/dependencies (#836/#1728). Returns the new version.
 *
 * opts.dryRun resolves + ABI-validates the candidate but writes NOTHING — it logs
 * what it would re-vendor and returns the version, so the flow can be rehearsed
 * (locally or in CI) without mutating gitnexus/vendor/. opts.deps injects the
 * network/fs seams for offline testing (same shape as detect()).
 */
function apply(key, opts = {}) {
  const dryRun = opts.dryRun || false;
  const deps = opts.deps || {};
  const getVendored = deps.vendoredVersion || vendoredVersion;
  const resolveUp = deps.resolveUpstream || resolveUpstream;
  const fetchSrc = deps.fetchSource || fetchSource;
  const readAbiFn = deps.readAbi || readAbi;
  const g = GRAMMARS[key];
  if (!g) throw new ApplyExit(`unknown grammar '${key}'`, 2);
  if (g.hold)
    throw new ApplyExit(
      `${key}: report-only (${g.hold}); not auto-applied. Re-vendor manually if intended.`,
      3,
    );
  const have = getVendored(g);
  const up = resolveUp(g);
  const newer = isNewer(up, have);
  if (!newer) {
    // Already current: nothing to apply. Return (exit 0 via the CLI) — NOT an error.
    console.error(`${key}: already current (${have}); nothing to apply.`);
    return have;
  }
  const srcRoot = fetchSrc(g, up.ref);
  const abi = readAbiFn(srcRoot);
  if (abi == null || !COMPATIBLE_ABI.has(abi))
    throw new ApplyExit(
      `${key}: candidate ${up.version} is ABI ${abi ?? 'unknown'} — not tree-sitter@0.25.1 ` +
        `compatible (need 13/14/15); refusing to re-vendor. Handle manually.`,
      3,
    );

  if (dryRun) {
    console.log(
      `${key}: [dry-run] would re-vendor ${g.name} → ${up.version} (ABI ${abi}); no files written.`,
    );
    return up.version;
  }

  const dest = path.join(VENDOR, g.name);
  // The source-build inputs + runtime entrypoints that change between versions.
  // Keep the synchronous CommonJS wrapper: upstream C now uses top-level await.
  // binding.gyp / bindings/node/index.* / README / LICENSE / prebuilds are preserved.
  for (const rel of [
    'src/parser.c',
    'src/scanner.c',
    'src/node-types.json',
    'src/tree_sitter/alloc.h',
    'src/tree_sitter/array.h',
    'src/tree_sitter/parser.h',
    'bindings/node/binding.cc',
  ]) {
    copyFile(srcRoot, dest, rel);
  }

  const pkgPath = path.join(dest, 'package.json');
  const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf8'));
  pkg.version = up.version;
  if (up.kind === 'github') pkg._upstreamCommit = up.ref;
  else delete pkg._upstreamCommit;
  pkg._vendoredBy =
    `gitnexus - re-vendored from ${g.npm ? `npm ${g.npm}@${up.version}` : `${g.github}@${up.ref}`} ` +
    `by grammar-update-monitor on ABI ${abi}. Source-build inputs (parser.c/scanner.c/src/) refreshed; ` +
    `the GitNexus-hardened binding.gyp + synchronous loader + vendor README + prebuilds are preserved (prebuilds are ` +
    `rebuilt by build-tree-sitter-prebuilds.yml on this version change). No scripts/dependencies here ` +
    `(#836/#1728).`;
  fs.writeFileSync(pkgPath, JSON.stringify(pkg, null, 2) + '\n');

  console.log(`${key}: re-vendored ${g.name} → ${up.version} (ABI ${abi}).`);
  return up.version;
}

// Run the CLI only when invoked directly (not when imported by a test) — detect()
// makes live network calls, so importing must be side-effect-free.
const isMain = process.argv[1] && import.meta.url === pathToFileURL(process.argv[1]).href;
if (isMain) {
  const args = process.argv.slice(2);
  const dryRun = args.includes('--dry-run');
  if (args[0] === '--apply') {
    // `--apply <grammar> [--dry-run]` — --dry-run previews without writing.
    // Map apply()'s thrown ApplyExit back to the original exit codes (0/2/3) so
    // the monitor workflow's subprocess (which only distinguishes zero vs non-zero)
    // sees identical behavior.
    try {
      apply(args[1], { dryRun });
    } catch (e) {
      console.error(e.message);
      process.exit(e instanceof ApplyExit ? e.code : 1);
    }
  } else {
    process.stdout.write(JSON.stringify(detect(), null, 2) + '\n');
  }
}

export {
  detect,
  apply,
  resolveUpstream,
  readAbi,
  vendoredVersion,
  loadManifestGrammars,
  GRAMMARS,
  COMPATIBLE_ABI,
};