import type { Model, Api } from "@earendil-works/pi-ai";

export type ModelOverride = Pick<
  Model<Api>,
  "reasoning" | "input" | "cost" | "contextWindow" | "maxTokens"
>;

export const findModelOverride = (id: string): ModelOverride => {
  const matched = overrides.find((m) => m.id.test(id));
  if (!matched) {
    throw new Error(`No model override found for id: ${id}`);
  }

  const { id: _, ...override } = matched;
  return override;
};

const overrides = [
  {
    id: /^composer-1\.5$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3.5, output: 17.5, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 4000,
  },
  {
    id: /^composer-2-fast$/,
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 1.5, output: 7.5, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 4000,
  },
  {
    id: /^composer-2$/,
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 0.5, output: 2.5, cacheRead: 0.2, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 4000,
  },
  {
    id: /^claude-sonnet-4-0$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
    contextWindow: 200000,
    maxTokens: 64000,
  },
  {
    // Cursor exposes a 1M Sonnet 4 family, but pi-mono does not have an exact canonical entry.
    // Reuse the official Sonnet 4 maxTokens value and Cursor pricing/context metadata.
    id: /^claude-sonnet-4-1m$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 6, output: 22.5, cacheRead: 0.6, cacheWrite: 7.5 },
    contextWindow: 1000000,
    maxTokens: 64000,
  },
  {
    id: /^claude-sonnet-4-5$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
    contextWindow: 200000,
    maxTokens: 64000,
  },
  {
    id: /^claude-sonnet-4-6$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
    contextWindow: 200000,
    maxTokens: 64000,
  },
  {
    id: /^claude-opus-4-5$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
    contextWindow: 200000,
    maxTokens: 64000,
  },
  {
    id: /^claude-opus-4-6$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
    contextWindow: 200000,
    maxTokens: 128000,
  },
  {
    // Cursor Opus 4.6 Max/fast pricing is 6x the base Opus 4.6 family.
    id: /^claude-opus-4-6-max$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 30, output: 150, cacheRead: 3, cacheWrite: 37.5 },
    contextWindow: 1000000,
    maxTokens: 128000,
  },
  {
    id: /^claude-opus-4-7-max$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 30, output: 150, cacheRead: 3, cacheWrite: 37.5 },
    contextWindow: 1000000,
    maxTokens: 128000,
  },
  {
    id: /^claude-opus-4-8$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
    contextWindow: 200000,
    maxTokens: 128000,
  },
  {
    id: /^claude-opus-4-8-max$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 30, output: 150, cacheRead: 3, cacheWrite: 37.5 },
    contextWindow: 1000000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5-mini$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.25, output: 2, cacheRead: 0.025, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.1$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.25, output: 10, cacheRead: 0.125, cacheWrite: 0 },
    contextWindow: 400000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.1-codex-max$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.25, output: 10, cacheRead: 0.125, cacheWrite: 0 },
    contextWindow: 1000000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.1-codex-max-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2.5, output: 20, cacheRead: 0.25, cacheWrite: 0 },
    contextWindow: 1000000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.1-codex-mini$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.25, output: 2, cacheRead: 0.025, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.2$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.75, output: 14, cacheRead: 0.175, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.2-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3.5, output: 28, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.2-codex$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.75, output: 14, cacheRead: 0.175, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.2-codex-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3.5, output: 28, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.3-codex$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.75, output: 14, cacheRead: 0.175, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.3-codex-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3.5, output: 28, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.3-codex-spark$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 1.75, output: 14, cacheRead: 0.175, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 32000,
  },
  {
    id: /^gpt-5\.4$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2.5, output: 15, cacheRead: 0.25, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.4-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 5, output: 30, cacheRead: 0.5, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.4-mini$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.75, output: 4.5, cacheRead: 0.075, cacheWrite: 0 },
    contextWindow: 400000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.4-nano$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.2, output: 1.25, cacheRead: 0.02, cacheWrite: 0 },
    contextWindow: 400000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.4-pro$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 5, output: 30, cacheRead: 0.5, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.4-pro-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 10, output: 60, cacheRead: 1.0, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.5$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 3.5, output: 17.5, cacheRead: 0.35, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.5-pro$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 7, output: 35, cacheRead: 0.7, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gpt-5\.5-fast$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 7, output: 35, cacheRead: 0.7, cacheWrite: 0 },
    contextWindow: 272000,
    maxTokens: 128000,
  },
  {
    id: /^gemini-3-flash-preview$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.5, output: 3, cacheRead: 0.05, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 65536,
  },
  {
    id: /^gemini-3-pro-preview$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 64000,
  },
  {
    id: /^gemini-3\.1-pro-preview$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 65536,
  },
  {
    id: /^gemini-3\.5-flash-preview$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.75, output: 4.5, cacheRead: 0.075, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 65536,
  },
  {
    id: /^gemini-3\.5-pro-preview$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2.5, output: 15, cacheRead: 0.25, cacheWrite: 0 },
    contextWindow: 200000,
    maxTokens: 65536,
  },
  {
    id: /^grok-4\.20-0309-non-reasoning$/,
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 2, output: 6, cacheRead: 0.2, cacheWrite: 0 },
    contextWindow: 2000000,
    maxTokens: 30000,
  },
  {
    id: /^grok-4\.20-0309-reasoning$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 2, output: 6, cacheRead: 0.2, cacheWrite: 0 },
    contextWindow: 2000000,
    maxTokens: 30000,
  },
  {
    id: /^kimi-k2\.5$/,
    reasoning: true,
    input: ["text", "image"],
    cost: { input: 0.6, output: 3, cacheRead: 0.1, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 65536,
  },
  {
    id: /^.*$/,
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 1.25, output: 6, cacheRead: 0.25, cacheWrite: 1.25 },
    contextWindow: 200000,
    maxTokens: 30000,
  },
] satisfies (ModelOverride & { id: RegExp })[];