#!/usr/bin/env node
import { existsSync } from "node:fs";
import { DatabaseSync } from "node:sqlite";
import { type BuildInfo, build, databasePath } from "./index.js";

const args = process.argv.slice(2);

if (args.includes("--help") || args.includes("-h")) {
  console.log(`Usage: critical [options]

Options:
  --output, -o <path>   Output database path (default: critical-packages.db)
  --skip-versions       Skip fetching version data (faster)
  --stats               Show database statistics
  --help, -h            Show this help message

Environment:
  ECOSYSTEMS_MAILTO     Contact address sent to the API. Without it the build runs on
                        the anonymous rate-limit tier (~5,000 requests/hour against
                        ~9,700 needed) and aborts once too many version fetches fail.

Examples:
  npx @ecosyste-ms/critical
  npx @ecosyste-ms/critical --skip-versions
  npx @ecosyste-ms/critical -o my-database.db
  npx @ecosyste-ms/critical --stats`);
  process.exit(0);
}

function getArg(flags: string[]): string | null {
  for (const flag of flags) {
    const idx = args.indexOf(flag);
    if (idx !== -1 && args[idx + 1]) {
      return args[idx + 1] ?? null;
    }
  }
  return null;
}

const dbPath = getArg(["--output", "-o"]) || "critical-packages.db";

if (args.includes("--stats")) {
  const path = existsSync(dbPath) ? dbPath : databasePath;
  if (!existsSync(path)) {
    console.error(`Database not found: ${path}`);
    process.exit(1);
  }
  const db = new DatabaseSync(path, { readOnly: true });
  const info = db.prepare("SELECT * FROM build_info WHERE id = 1").get() as
    | BuildInfo
    | undefined;
  const count = (table: string) =>
    Number(db.prepare(`SELECT COUNT(*) as count FROM ${table}`).get()?.count ?? 0);
  const packageCount = count("packages");
  const versionCount = count("versions");
  const advisoryCount = count("advisories");
  const ecosystems = db
    .prepare(`
    SELECT ecosystem, COUNT(*) as count
    FROM packages
    GROUP BY ecosystem
    ORDER BY count DESC
  `)
    .all() as unknown as { ecosystem: string; count: number }[];
  const severities = db
    .prepare(`
    SELECT severity, COUNT(*) as count
    FROM advisories
    GROUP BY severity
    ORDER BY count DESC
  `)
    .all() as unknown as { severity: string | null; count: number }[];

  console.log(`Database: ${path}`);
  console.log(`Built: ${info?.built_at || "unknown"}`);
  console.log(`\nPackages: ${packageCount}`);
  console.log(`Versions: ${versionCount}`);
  console.log(`Advisories: ${advisoryCount}`);
  console.log(`\nBy ecosystem:`);
  for (const e of ecosystems) {
    console.log(`  ${e.ecosystem}: ${e.count}`);
  }
  console.log(`\nBy severity:`);
  for (const s of severities) {
    console.log(`  ${s.severity || "unknown"}: ${s.count}`);
  }
  db.close();
  process.exit(0);
}

const skipVersions = args.includes("--skip-versions");

build({
  dbPath,
  fetchVersionsData: !skipVersions,
})
  .then(() => process.exit(0))
  .catch((err: unknown) => {
    console.error(err);
    process.exit(1);
  });