OmniRoute/src/instrumentation.ts

/**
 * Next.js Instrumentation Hook
 *
 * Called once when the server starts (both dev and production).
 * Used to initialize graceful shutdown handlers, console log capture,
 * and compliance features (audit log table, expired log cleanup).
 *
 * @see https://nextjs.org/docs/app/building-your-application/optimizing/instrumentation
 */

function ensureSecrets(): void {
  // eslint-disable-next-line no-eval
  const crypto = eval("require")("crypto");

  if (!process.env.JWT_SECRET || process.env.JWT_SECRET.trim() === "") {
    const generated = crypto.randomBytes(48).toString("base64");
    process.env.JWT_SECRET = generated;
    console.log("[STARTUP] JWT_SECRET auto-generated (random 64-char secret)");
  }

  if (!process.env.API_KEY_SECRET || process.env.API_KEY_SECRET.trim() === "") {
    const generated = crypto.randomBytes(32).toString("hex");
    process.env.API_KEY_SECRET = generated;
    console.log("[STARTUP] API_KEY_SECRET auto-generated (random 64-char hex secret)");
  }
}

export async function register() {
  // Only run on the server (not during build or in Edge runtime)
  if (process.env.NEXT_RUNTIME === "nodejs") {
    ensureSecrets();
    // Console log file capture (must be first — before any logging occurs)
    const { initConsoleInterceptor } = await import("@/lib/consoleInterceptor");
    initConsoleInterceptor();

    const { initGracefulShutdown } = await import("@/lib/gracefulShutdown");
    initGracefulShutdown();

    const { initApiBridgeServer } = await import("@/lib/apiBridgeServer");
    initApiBridgeServer();

    // Quota cache: start background refresh for quota-aware account selection
    // Dynamic import required — quotaCache depends on better-sqlite3 (Node-only),
    // and instrumentation.ts is bundled for all runtimes including Edge.
    const { startBackgroundRefresh } = await import("@/domain/quotaCache");
    startBackgroundRefresh();
    console.log("[STARTUP] Quota cache background refresh started");

    // Model aliases: restore persisted custom aliases into in-memory state (#316)
    // Custom aliases are saved to settings.modelAliases on PUT /api/settings/model-aliases
    // but the in-memory _customAliases resets to {} on every restart — load them here.
    try {
      const { getSettings } = await import("@/lib/db/settings");
      const { setCustomAliases } = await import("@omniroute/open-sse/services/modelDeprecation.ts");
      const settings = await getSettings();
      if (settings.modelAliases) {
        const aliases =
          typeof settings.modelAliases === "string"
            ? JSON.parse(settings.modelAliases)
            : settings.modelAliases;
        if (aliases && typeof aliases === "object") {
          setCustomAliases(aliases);
          console.log(
            `[STARTUP] Restored ${Object.keys(aliases).length} custom model alias(es) from settings`
          );
        }
      }
    } catch (err: unknown) {
      const msg = err instanceof Error ? err.message : String(err);
      console.warn("[STARTUP] Could not restore model aliases:", msg);
    }

    // Compliance: Initialize audit_log table + cleanup expired logs
    try {
      const { initAuditLog, cleanupExpiredLogs } = await import("@/lib/compliance/index");
      initAuditLog();
      console.log("[COMPLIANCE] Audit log table initialized");

      const cleanup = cleanupExpiredLogs();
      if (cleanup.deletedUsage || cleanup.deletedCallLogs || cleanup.deletedAuditLogs) {
        console.log("[COMPLIANCE] Expired log cleanup:", cleanup);
      }
    } catch (err: unknown) {
      const msg = err instanceof Error ? err.message : String(err);
      console.warn("[COMPLIANCE] Could not initialize audit log:", msg);
    }
  }
}