import type { RuntimeDriverName } from "@pi-claude-code-agent/runtime";

export interface RuntimeModelCatalogEntry {
  id: string;
  name: string;
  contextWindow: number;
  maxTokens: number;
  reasoning: boolean;
  inputModalities: string[];
  inputCostPerMillion: number;
  outputCostPerMillion: number;
}

export interface RuntimeDriverModelCatalog {
  driver: RuntimeDriverName;
  provider: "anthropic" | "openai-codex" | "pi-ai" | "google-gemini";
  defaultModel: string;
  aliases: Record<string, string>;
  recommendations: RuntimeModelRecommendation[];
  cli: string;
  flag: string;
  source: string;
  models: RuntimeModelCatalogEntry[];
}

export interface RuntimeModelRecommendation {
  alias: string;
  model: string;
  useCase: string;
}

export interface RuntimeModelSelection {
  requestedModel?: string;
  runtimeModel?: string;
  note: string;
  alias?: string;
  entry?: RuntimeModelCatalogEntry;
}

const LANISTA_SOURCE = "lanista agents anthropic/codex";

export const RUNTIME_MODEL_CATALOGS: Record<string, RuntimeDriverModelCatalog> = {
  "claude-sdk": {
    driver: "claude-sdk",
    provider: "anthropic",
    defaultModel: "claude-opus-4-7",
    aliases: {
      haiku: "claude-haiku-4-5",
      opus: "claude-opus-4-7",
      sonnet: "claude-sonnet-4-6",
    },
    recommendations: [
      { alias: "opus", model: "claude-opus-4-7", useCase: "architecture, hard review, long-context planning" },
      { alias: "sonnet", model: "claude-sonnet-4-6", useCase: "coding, refactors, implementation reviews" },
      { alias: "haiku", model: "claude-haiku-4-5", useCase: "quick checks, cheap parallel workers, summaries" },
    ],
    cli: "claude-code",
    flag: "claude --model <id>",
    source: LANISTA_SOURCE,
    models: [
      { id: "claude-3-5-haiku-20241022", name: "Claude Haiku 3.5", contextWindow: 200000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 0.8, outputCostPerMillion: 4 },
      { id: "claude-3-5-haiku-latest", name: "Claude Haiku 3.5 (latest)", contextWindow: 200000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 0.8, outputCostPerMillion: 4 },
      { id: "claude-3-5-sonnet-20240620", name: "Claude Sonnet 3.5", contextWindow: 200000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-3-5-sonnet-20241022", name: "Claude Sonnet 3.5 v2", contextWindow: 200000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-3-7-sonnet-20250219", name: "Claude Sonnet 3.7", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-3-haiku-20240307", name: "Claude Haiku 3", contextWindow: 200000, maxTokens: 4096, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 0.25, outputCostPerMillion: 1.25 },
      { id: "claude-3-opus-20240229", name: "Claude Opus 3", contextWindow: 200000, maxTokens: 4096, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 15, outputCostPerMillion: 75 },
      { id: "claude-3-sonnet-20240229", name: "Claude Sonnet 3", contextWindow: 200000, maxTokens: 4096, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-haiku-4-5", name: "Claude Haiku 4.5 (latest)", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1, outputCostPerMillion: 5 },
      { id: "claude-haiku-4-5-20251001", name: "Claude Haiku 4.5", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1, outputCostPerMillion: 5 },
      { id: "claude-opus-4-0", name: "Claude Opus 4 (latest)", contextWindow: 200000, maxTokens: 32000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 15, outputCostPerMillion: 75 },
      { id: "claude-opus-4-1", name: "Claude Opus 4.1 (latest)", contextWindow: 200000, maxTokens: 32000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 15, outputCostPerMillion: 75 },
      { id: "claude-opus-4-1-20250805", name: "Claude Opus 4.1", contextWindow: 200000, maxTokens: 32000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 15, outputCostPerMillion: 75 },
      { id: "claude-opus-4-20250514", name: "Claude Opus 4", contextWindow: 200000, maxTokens: 32000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 15, outputCostPerMillion: 75 },
      { id: "claude-opus-4-5", name: "Claude Opus 4.5 (latest)", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 5, outputCostPerMillion: 25 },
      { id: "claude-opus-4-5-20251101", name: "Claude Opus 4.5", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 5, outputCostPerMillion: 25 },
      { id: "claude-opus-4-6", name: "Claude Opus 4.6", contextWindow: 1000000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 5, outputCostPerMillion: 25 },
      { id: "claude-opus-4-7", name: "Claude Opus 4.7", contextWindow: 1000000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 5, outputCostPerMillion: 25 },
      { id: "claude-sonnet-4-0", name: "Claude Sonnet 4 (latest)", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-sonnet-4-20250514", name: "Claude Sonnet 4", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-sonnet-4-5", name: "Claude Sonnet 4.5 (latest)", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-sonnet-4-5-20250929", name: "Claude Sonnet 4.5", contextWindow: 200000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
      { id: "claude-sonnet-4-6", name: "Claude Sonnet 4.6", contextWindow: 1000000, maxTokens: 64000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 3, outputCostPerMillion: 15 },
    ],
  },
  "codex-cli": {
    driver: "codex-cli",
    provider: "openai-codex",
    defaultModel: "gpt-5.5",
    aliases: {
      codex: "gpt-5.3-codex",
      mini: "gpt-5.4-mini",
      spark: "gpt-5.3-codex-spark",
    },
    recommendations: [
      { alias: "default", model: "gpt-5.5", useCase: "architecture, hard debugging, broad repo analysis" },
      { alias: "codex", model: "gpt-5.3-codex", useCase: "coding, tests, focused refactors" },
      { alias: "mini", model: "gpt-5.4-mini", useCase: "quick edits, cheap exploration, small reviews" },
      { alias: "spark", model: "gpt-5.3-codex-spark", useCase: "fast text-only worker tasks" },
    ],
    cli: "codex",
    flag: "codex --model <id>",
    source: LANISTA_SOURCE,
    models: [
      { id: "gpt-5.1", name: "GPT-5.1", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1.25, outputCostPerMillion: 10 },
      { id: "gpt-5.1-codex-max", name: "GPT-5.1 Codex Max", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1.25, outputCostPerMillion: 10 },
      { id: "gpt-5.1-codex-mini", name: "GPT-5.1 Codex Mini", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 0.25, outputCostPerMillion: 2 },
      { id: "gpt-5.2", name: "GPT-5.2", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1.75, outputCostPerMillion: 14 },
      { id: "gpt-5.2-codex", name: "GPT-5.2 Codex", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1.75, outputCostPerMillion: 14 },
      { id: "gpt-5.3-codex", name: "GPT-5.3 Codex", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 1.75, outputCostPerMillion: 14 },
      { id: "gpt-5.3-codex-spark", name: "GPT-5.3 Codex Spark", contextWindow: 128000, maxTokens: 128000, reasoning: true, inputModalities: ["text"], inputCostPerMillion: 3.5, outputCostPerMillion: 28 },
      { id: "gpt-5.4", name: "GPT-5.4", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 2.5, outputCostPerMillion: 15 },
      { id: "gpt-5.4-mini", name: "GPT-5.4 Mini", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 0.75, outputCostPerMillion: 4.5 },
      { id: "gpt-5.5", name: "GPT-5.5", contextWindow: 272000, maxTokens: 128000, reasoning: true, inputModalities: ["text", "image"], inputCostPerMillion: 5, outputCostPerMillion: 30 },
    ],
  },
  "agy": {
    driver: "agy",
    provider: "google-gemini",
    defaultModel: "gemini-3.5-flash",
    aliases: {
      flash: "gemini-3.5-flash",
      pro: "gemini-3.5-pro",
    },
    recommendations: [
      { alias: "flash", model: "gemini-3.5-flash", useCase: "fast iteration, cheap coding assistant" },
      { alias: "pro", model: "gemini-3.5-pro", useCase: "complex architecture and multi-file debugging" },
    ],
    cli: "agy",
    flag: "agy --model <id>",
    source: "lanista agents google",
    models: [
      { id: "gemini-3.5-flash", name: "Gemini 3.5 Flash", contextWindow: 1000000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 0.075, outputCostPerMillion: 0.3 },
      { id: "gemini-3.5-pro", name: "Gemini 3.5 Pro", contextWindow: 2000000, maxTokens: 8192, reasoning: false, inputModalities: ["text", "image"], inputCostPerMillion: 1.25, outputCostPerMillion: 5.0 },
    ],
  },
};

export function modelCatalogsForDriver(driver?: RuntimeDriverName): RuntimeDriverModelCatalog[] {
  if (driver) {
    return [catalogForDriver(driver)];
  }
  // claude-cli intentionally omitted: the driver is unstable in practice
  // (sticky-session / non-interactive output issues), so it is not part of
  // the recommended runtime surface. Callers that explicitly request it
  // via catalogForDriver("claude-cli") will throw.
  return [
    catalogForDriver("claude-sdk"),
    catalogForDriver("codex-cli"),
    catalogForDriver("pi-coding-agent"),
    catalogForDriver("agy"),
  ];
}

export function findRuntimeModel(driver: RuntimeDriverName, model: string): RuntimeModelCatalogEntry | undefined {
  const normalized = model.trim();
  return catalogForDriver(driver).models.find((entry) => entry.id === normalized);
}

export function resolveRuntimeModelSelection(driver: RuntimeDriverName, model?: string): RuntimeModelSelection {
  const catalog = catalogForDriver(driver);
  const requestedModel = model?.trim() || undefined;
  if (!requestedModel) {
    return {
      runtimeModel: undefined,
      note: `model default ${catalog.defaultModel}`,
    };
  }

  const aliasTarget = catalog.aliases[requestedModel.toLowerCase()];
  const runtimeModel = aliasTarget ?? requestedModel;
  const entry = findRuntimeModel(driver, runtimeModel);
  if (!entry) {
    return {
      requestedModel,
      runtimeModel,
      note: `model ${requestedModel} not in bundled ${catalog.provider} catalog; passing through to runtime`,
    };
  }

  const entryNote = `model ${entry.id} (${entry.name}, context window ${entry.contextWindow}, max output ${entry.maxTokens})`;
  if (aliasTarget) {
    return {
      requestedModel,
      runtimeModel,
      alias: requestedModel,
      entry,
      note: `model alias ${requestedModel} -> ${entryNote}`,
    };
  }

  return {
    requestedModel,
    runtimeModel,
    entry,
    note: entryNote,
  };
}

function catalogForDriver(driver: RuntimeDriverName): RuntimeDriverModelCatalog {
  if (driver === "claude-pty") {
    // Drives the real Claude Code TUI — same Anthropic models and
    // `claude --model <id>` flag as the SDK driver. Reuse that catalog.
    const base = RUNTIME_MODEL_CATALOGS["claude-sdk"]!;
    return { ...base, driver };
  }
  if (driver === "claude-cli") {
    throw new Error(
      "claude-cli driver is no longer supported by the runtime model catalog " +
        "(driver retired due to sticky-session / non-interactive output issues). " +
        "Use claude-sdk for Claude models.",
    );
  }
  if (driver === "pi-coding-agent") {
    // pi-coding-agent routes through pi-ai, which only proxies gpt-* models
    // in our environment. Inherit the codex-cli catalog so recommendations
    // and model entries advertise the actually-supported set.
    const base = RUNTIME_MODEL_CATALOGS["codex-cli"]!;
    return {
      ...base,
      driver,
      provider: "pi-ai",
      cli: "pi",
      flag: "pi --model <provider>/<id>",
      source: `${base.source}; shared with codex-cli catalog (pi-coding-agent routes through pi-ai to gpt-* models)`,
    };
  }
  return RUNTIME_MODEL_CATALOGS[driver]!;
}

export function describeModelSelection(driver: RuntimeDriverName, model?: string): string | undefined {
  return resolveRuntimeModelSelection(driver, model).note;
}