OmniRoute/open-sse/services/autoCombo/autoPrefix.ts

export type AutoVariant = "coding" | "fast" | "cheap" | "offline" | "smart" | "lkgp" | "chaos";

export interface AutoPrefixParseResult {
  valid: boolean;
  variant?: AutoVariant;
  error?: string;
}

export const VALID_VARIANTS: AutoVariant[] = [
  "coding",
  "fast",
  "cheap",
  "offline",
  "smart",
  "lkgp",
  "chaos",
];

/**
 * Parses a model name to determine if it's an auto-prefixed model and extracts the variant.
 *
 * Examples:
 * - "auto"         -> { valid: true, variant: undefined } (default)
 * - "auto/coding"  -> { valid: true, variant: "coding" }
 * - "auto/lkgp"    -> { valid: true, variant: "lkgp" }
 * - "auto/"        -> { valid: true, variant: undefined } (default)
 * - "autocoding"   -> { valid: false, error: "Invalid auto prefix format" }
 * - "otherModel"   -> { valid: false, error: "Not an auto-prefixed model" }
 */
export function parseAutoPrefix(model: string | null | undefined): AutoPrefixParseResult {
  // Guard against null/undefined (called with non-string inputs)
  if (typeof model !== "string") {
    return { valid: false, error: "Not an auto-prefixed model" };
  }
  if (!model.startsWith("auto")) {
    return { valid: false, error: "Not an auto-prefixed model" };
  }

  const parts = model.split("/");

  if (parts.length === 1) {
    if (parts[0] === "auto") {
      return { valid: true, variant: undefined }; // Default auto
    } else {
      return { valid: false, error: "Invalid auto prefix format" };
    }
  }

  if (parts.length === 2) {
    if (parts[0] !== "auto") {
      return { valid: false, error: "Invalid auto prefix format" };
    }
    const variantStr: string = parts[1];
    if (variantStr === "" || VALID_VARIANTS.includes(variantStr as AutoVariant)) {
      return { valid: true, variant: variantStr === "" ? undefined : (variantStr as AutoVariant) };
    } else {
      return { valid: false, error: `Invalid auto variant: ${variantStr}` };
    }
  }

  return { valid: false, error: "Invalid auto prefix format" };
}