import { parse } from "jsr:@std/yaml@1";

export type BotProfile = {
  type: "claude" | "codex";
  model: string;
  token?: string;
};

const DEFAULT_MODELS: Record<BotProfile["type"], string> = {
  claude: "claude-sonnet-5",
  codex: "gpt-5.6-terra",
};

export function parseProfiles(text: string): BotProfile[] {
  const value: unknown = parse(text);
  if (!Array.isArray(value) || value.length === 0) {
    throw new Error("profiles must contain at least one profile");
  }
  return value.map((entry: unknown, index) => {
    const at = `profiles[${index}]`;
    if (typeof entry !== "object" || entry === null) {
      throw new Error(`${at}: expected a mapping`);
    }
    const { type, model, token } = entry as Record<string, unknown>;
    if (type !== "claude" && type !== "codex") {
      throw new Error(`${at}.type: expected "claude" or "codex"`);
    }
    if (model !== undefined && (typeof model !== "string" || model === "")) {
      throw new Error(`${at}.model: expected a model name`);
    }
    if (token !== undefined && (typeof token !== "string" || token === "")) {
      throw new Error(`${at}.token: expected a token`);
    }
    if (type === "claude" && token === undefined) {
      throw new Error(`${at}.token: required for Claude`);
    }
    return {
      type,
      model: model ?? DEFAULT_MODELS[type],
      ...(token && { token }),
    };
  });
}

function isUsageExhausted(error: unknown): boolean {
  return /(?:usage|rate) limit|hit your limit|insufficient_quota|exceeded your current quota|too many requests|(?:api error|http error|http|status(?: code)?)[ :]+4(?:02|29)\b/i
    .test(String(error));
}

export async function runProfiles<T>(
  profiles: BotProfile[],
  attempt: (profile: BotProfile) => Promise<T>,
): Promise<T> {
  for (const [index, profile] of profiles.entries()) {
    try {
      return await attempt(profile);
    } catch (error) {
      if (!isUsageExhausted(error) || index === profiles.length - 1) {
        throw error;
      }
      console.error(`${profile.type} usage exhausted; trying next profile`);
    }
  }
  throw new Error("profiles must contain at least one profile");
}

export function sanitizedEnvironment(
  environment: Record<string, string>,
): Record<string, string> {
  const result = { ...environment };
  for (
    const name of [
      "PROFILES",
      "REVIEWER_TOKEN",
      "BOT_TOKEN",
      "CLAUDE_CODE_OAUTH_TOKEN",
      "ANTHROPIC_API_KEY",
      "ANTHROPIC_AUTH_TOKEN",
      "OPENAI_API_KEY",
      "CODEX_API_KEY",
      "CODEX_ACCESS_TOKEN",
    ]
  ) {
    delete result[name];
  }
  return result;
}

export function agentEnvironment(
  environment: Record<string, string>,
  profile: BotProfile,
): Record<string, string> {
  const result = sanitizedEnvironment(environment);
  if (profile.token) {
    result[
      profile.type === "claude" ? "CLAUDE_CODE_OAUTH_TOKEN" : "OPENAI_API_KEY"
    ] = profile.token;
  }
  return result;
}