inkos/packages/cli/src/commands/config.ts

import { Command } from "commander";
import { readFile, writeFile, mkdir } from "node:fs/promises";
import { join } from "node:path";
import { findProjectRoot, log, logError, GLOBAL_CONFIG_DIR, GLOBAL_ENV_PATH } from "../utils.js";
import { listModelsForService } from "@actalk/inkos-core";

export const configCommand = new Command("config")
  .description("Manage project configuration");

configCommand
  .command("set")
  .description("Set a configuration value")
  .argument("<key>", "Config key (e.g., llm.apiKey)")
  .argument("<value>", "Config value")
  .action(async (key: string, value: string) => {
    const root = findProjectRoot();
    const configPath = join(root, "inkos.json");

    try {
      const raw = await readFile(configPath, "utf-8");
      const config = JSON.parse(raw);

      const keys = key.split(".");

      const KNOWN_KEYS = new Set([
        "llm.provider", "llm.baseUrl", "llm.model", "llm.temperature",
        "llm.thinkingBudget", "llm.proxyUrl", "llm.apiFormat", "llm.stream",
        "inputGovernanceMode",
        "foundation.reviewRetries",
        "daemon.schedule.radarCron", "daemon.schedule.writeCron",
        "daemon.maxConcurrentBooks", "daemon.chaptersPerCycle",
        "daemon.retryDelayMs", "daemon.cooldownAfterChapterMs",
        "daemon.maxChaptersPerDay",
      ]);
      // Allow any key under llm.extra.* (passthrough to API)
      if (!KNOWN_KEYS.has(key) && !key.startsWith("llm.extra.")) {
        // Find closest match by edit distance on the last segment
        const candidates = [...KNOWN_KEYS];
        const inputParts = key.split(".");
        const samePrefixCandidates = candidates.filter(k => {
          const parts = k.split(".");
          return parts.length === inputParts.length && parts.slice(0, -1).join(".") === inputParts.slice(0, -1).join(".");
        });
        const editDist = (a: string, b: string): number => {
          const m = a.length, n = b.length;
          const dp = Array.from({ length: m + 1 }, (_, i) => Array.from({ length: n + 1 }, (_, j) => i === 0 ? j : j === 0 ? i : 0));
          for (let i = 1; i <= m; i++) for (let j = 1; j <= n; j++)
            dp[i]![j] = Math.min(dp[i-1]![j]! + 1, dp[i]![j-1]! + 1, dp[i-1]![j-1]! + (a[i-1] !== b[j-1] ? 1 : 0));
          return dp[m]![n]!;
        };
        const inputLast = inputParts[inputParts.length - 1]!;
        const suggestion = samePrefixCandidates
          .map(k => ({ k, d: editDist(k.split(".").pop()!, inputLast) }))
          .sort((a, b) => a.d - b.d)
          .find(x => x.d <= 3)?.k;
        logError(`Unknown config key "${key}".${suggestion ? ` Did you mean "${suggestion}"?` : ""}`);
        log(`Known keys: ${candidates.join(", ")}`);
        process.exit(1);
      }

      let target = config;
      for (let i = 0; i < keys.length - 1; i++) {
        const k = keys[i]!;
        if (!(k in target)) {
          target[k] = {};
        }
        target = target[k];
      }
      const finalKey = keys[keys.length - 1]!;
      // Auto-coerce types: numbers and booleans shouldn't be stored as strings
      if (/^\d+(\.\d+)?$/.test(value)) {
        target[finalKey] = parseFloat(value);
      } else if (value === "true") {
        target[finalKey] = true;
      } else if (value === "false") {
        target[finalKey] = false;
      } else {
        target[finalKey] = value;
      }

      await writeFile(configPath, JSON.stringify(config, null, 2), "utf-8");
      log(`Set ${key} = ${value}`);
    } catch (e) {
      logError(`Failed to update config: ${e}`);
      process.exit(1);
    }
  });

configCommand
  .command("set-global")
  .description("Set global LLM config (~/.inkos/.env), shared by all projects")
  .requiredOption("--provider <provider>", "LLM provider (openai / anthropic)")
  .requiredOption("--base-url <url>", "API base URL")
  .requiredOption("--api-key <key>", "API key")
  .requiredOption("--model <model>", "Model name")
  .option("--temperature <n>", "Temperature")
  .option("--max-tokens <n>", "Max output tokens")
  .option("--thinking-budget <n>", "Anthropic thinking budget")
  .option("--api-format <format>", "API format (chat / responses)")
  .option("--lang <language>", "Default writing language: zh (Chinese) or en (English)")
  .action(async (opts) => {
    try {
      await mkdir(GLOBAL_CONFIG_DIR, { recursive: true });

      const lines = [
        "# InkOS Global LLM Configuration",
        `INKOS_LLM_PROVIDER=${opts.provider}`,
        `INKOS_LLM_BASE_URL=${opts.baseUrl}`,
        `INKOS_LLM_API_KEY=${opts.apiKey}`,
        `INKOS_LLM_MODEL=${opts.model}`,
      ];
      if (opts.temperature) lines.push(`INKOS_LLM_TEMPERATURE=${opts.temperature}`);
      if (opts.thinkingBudget) lines.push(`INKOS_LLM_THINKING_BUDGET=${opts.thinkingBudget}`);
      if (opts.apiFormat) lines.push(`INKOS_LLM_API_FORMAT=${opts.apiFormat}`);
      if (opts.lang) lines.push(`INKOS_DEFAULT_LANGUAGE=${opts.lang}`);

      await writeFile(GLOBAL_ENV_PATH, lines.join("\n") + "\n", "utf-8");
      log(`Global config saved to ${GLOBAL_ENV_PATH}`);
      log("All projects will use this config unless overridden by project .env");
    } catch (e) {
      logError(`Failed to set global config: ${e}`);
      process.exit(1);
    }
  });

configCommand
  .command("show-global")
  .description("Show global LLM config (~/.inkos/.env)")
  .action(async () => {
    try {
      const content = await readFile(GLOBAL_ENV_PATH, "utf-8");
      const masked = content.replace(
        /(INKOS_LLM_API_KEY=)(.{8})(.*)(.{4})/,
        "$1$2...$4",
      );
      log(masked);
    } catch {
      log("No global config found. Run 'inkos config set-global' to create one.");
    }
  });

configCommand
  .command("show")
  .description("Show current project configuration")
  .action(async () => {
    const root = findProjectRoot();
    const configPath = join(root, "inkos.json");

    try {
      const raw = await readFile(configPath, "utf-8");
      const config = JSON.parse(raw);
      // Mask API key
      if (config.llm?.apiKey) {
        const key = config.llm.apiKey;
        config.llm.apiKey = key.slice(0, 8) + "..." + key.slice(-4);
      }
      log(JSON.stringify(config, null, 2));
    } catch (e) {
      logError(`Failed to read config: ${e}`);
      process.exit(1);
    }
  });

const KNOWN_AGENTS = ["writer", "auditor", "reviser", "architect", "radar", "chapter-analyzer"] as const;
const ENV_VAR_NAME_PATTERN = /^[A-Za-z_][A-Za-z0-9_]*$/;

function validateApiKeyEnvName(value: string): string | undefined {
  if (ENV_VAR_NAME_PATTERN.test(value)) return undefined;
  if (/^(sk-|sess-|rk-|pk-)/i.test(value) || value.includes("://")) {
    return "--api-key-env expects an environment variable name like PACKY_API_KEY, not a raw API key or URL.";
  }
  return `--api-key-env expects an environment variable name like PACKY_API_KEY. "${value}" is not a valid env var name.`;
}

configCommand
  .command("set-model")
  .description("Set model override for a specific agent (with optional provider routing)")
  .argument("<agent>", `Agent name (${KNOWN_AGENTS.join(", ")})`)
  .argument("<model>", "Model name")
  .option("--base-url <url>", "API base URL (for different provider)")
  .option("--provider <provider>", "Provider type (openai / anthropic / custom)")
  .option("--api-key-env <envVar>", "Env variable name for API key (e.g., PACKYAPI_KEY)")
  .option("--stream", "Enable streaming (default)")
  .option("--no-stream", "Disable streaming")
  .action(async (agent: string, model: string, opts: { baseUrl?: string; provider?: string; apiKeyEnv?: string; stream?: boolean }) => {
    if (!KNOWN_AGENTS.includes(agent as typeof KNOWN_AGENTS[number])) {
      logError(`Unknown agent "${agent}". Valid agents: ${KNOWN_AGENTS.join(", ")}`);
      process.exit(1);
    }

    if (opts.apiKeyEnv) {
      const validationError = validateApiKeyEnvName(opts.apiKeyEnv);
      if (validationError) {
        logError(validationError);
        process.exit(1);
      }
    }

    const root = findProjectRoot();
    const configPath = join(root, "inkos.json");

    try {
      const raw = await readFile(configPath, "utf-8");
      const config = JSON.parse(raw);
      const overrides = config.modelOverrides ?? {};

      const hasProviderOpts = opts.baseUrl || opts.provider || opts.apiKeyEnv || opts.stream === false;
      if (hasProviderOpts) {
        const override: Record<string, unknown> = { model };
        if (opts.baseUrl) override.baseUrl = opts.baseUrl;
        if (opts.provider) override.provider = opts.provider;
        if (opts.apiKeyEnv) override.apiKeyEnv = opts.apiKeyEnv;
        if (opts.stream === false) override.stream = false;
        config.modelOverrides = { ...overrides, [agent]: override };
      } else {
        config.modelOverrides = { ...overrides, [agent]: model };
      }

      await writeFile(configPath, JSON.stringify(config, null, 2), "utf-8");
      log(`Model override: ${agent} → ${model}${opts.baseUrl ? ` (${opts.baseUrl})` : ""}`);
    } catch (e) {
      logError(`Failed to update config: ${e}`);
      process.exit(1);
    }
  });

configCommand
  .command("remove-model")
  .description("Remove model override for a specific agent (falls back to default)")
  .argument("<agent>", "Agent name")
  .action(async (agent: string) => {
    const root = findProjectRoot();
    const configPath = join(root, "inkos.json");

    try {
      const raw = await readFile(configPath, "utf-8");
      const config = JSON.parse(raw);
      const overrides = config.modelOverrides;
      if (!overrides || !(agent in overrides)) {
        log(`No model override for "${agent}".`);
        return;
      }
      const { [agent]: _, ...rest } = overrides;
      config.modelOverrides = Object.keys(rest).length > 0 ? rest : undefined;
      await writeFile(configPath, JSON.stringify(config, null, 2), "utf-8");
      log(`Removed model override for ${agent}. Will use default model.`);
    } catch (e) {
      logError(`Failed to update config: ${e}`);
      process.exit(1);
    }
  });

configCommand
  .command("show-models")
  .description("Show model routing for all agents")
  .option("--json", "Output JSON")
  .action(async (opts) => {
    const root = findProjectRoot();
    const configPath = join(root, "inkos.json");

    try {
      const raw = await readFile(configPath, "utf-8");
      const config = JSON.parse(raw);
      const defaultModel = config.llm?.model ?? "(not set)";
      const overrides: Record<string, unknown> = config.modelOverrides ?? {};

      if (opts.json) {
        log(JSON.stringify({ defaultModel, overrides }, null, 2));
        return;
      }

      log(`Default model: ${defaultModel}\n`);
      if (Object.keys(overrides).length === 0) {
        log("No agent-specific overrides. All agents use the default model.");
        return;
      }
      log("Agent overrides:");
      for (const [agent, value] of Object.entries(overrides)) {
        if (typeof value === "string") {
          log(`  ${agent} → ${value}`);
        } else {
          const o = value as Record<string, unknown>;
          const parts = [o.model as string];
          if (o.baseUrl) parts.push(`@ ${o.baseUrl}`);
          if (o.stream === false) parts.push("[no-stream]");
          log(`  ${agent} → ${parts.join(" ")}`);
        }
      }
      log("");
      const usingDefault = KNOWN_AGENTS.filter((a) => !(a in overrides));
      if (usingDefault.length > 0) {
        log(`Using default: ${usingDefault.join(", ")}`);
      }
    } catch (e) {
      logError(`Failed to read config: ${e}`);
      process.exit(1);
    }
  });

// B17: list-models 命令 —— 列出指定 service 的可用模型（含元数据）
configCommand
  .command("list-models <service>")
  .description("List available models for a service (with maxOutput / contextWindow / abilities)")
  .option("--api-key <key>", "API Key (also reads from INKOS_LLM_API_KEY env)")
  .option("--base-url <url>", "Live /models probe baseUrl (for custom/newapi)")
  .option("--json", "Output as JSON")
  .action(async (service: string, opts: { apiKey?: string; baseUrl?: string; json?: boolean }) => {
    const apiKey = opts.apiKey ?? process.env.INKOS_LLM_API_KEY;
    const models = await listModelsForService(service, apiKey, opts.baseUrl);
    if (models.length === 0) {
      logError(`${service} 没有可用模型（可能需要 --api-key 和 --base-url）`);
      process.exit(1);
    }
    if (opts.json) {
      log(JSON.stringify(models, null, 2));
      return;
    }
    log(`${service}：${models.length} 个模型\n`);
    for (const m of models) {
      const maxOut = m.maxOutput ? `out=${m.maxOutput}` : "out=?";
      const ctx = m.contextWindow > 0 ? `ctx=${m.contextWindow}` : "ctx=?";
      log(`  ${m.id.padEnd(42)} ${maxOut.padEnd(14)} ${ctx}`);
    }
  });