// Xai plugin module implements model id behavior.
export const XAI_OAUTH_AUTO_MODEL_ID = "auto";

export function isXaiGrok46ModelId(id: string): boolean {
  const normalized = normalizeXaiModelId(id.trim().toLowerCase());
  return normalized === "grok-4.6";
}

export function isXaiFrontierModelId(id: string): boolean {
  const normalized = normalizeXaiModelId(id.trim().toLowerCase());
  return (
    normalized === "grok-4.6" || normalized === "grok-4.5" || normalized.startsWith("grok-4.5-")
  );
}

export function resolveXaiOAuthAutoModelId(
  id: string,
  params?: Record<string, unknown> | null,
): string {
  if (id.trim().toLowerCase() !== XAI_OAUTH_AUTO_MODEL_ID) {
    return id;
  }
  const canonicalModelId = params?.canonicalModelId;
  return typeof canonicalModelId === "string" && canonicalModelId.trim()
    ? canonicalModelId.trim()
    : id;
}

export function normalizeXaiModelId(id: string): string {
  if (id === "grok-4.3-latest") {
    return "grok-4.3";
  }
  if (id === "grok-4.5-latest") {
    return "grok-4.5";
  }
  if (id === "grok-build-latest") {
    return "grok-4.5";
  }
  if (id === "grok-code-fast-1" || id === "grok-code-fast" || id === "grok-code-fast-1-0825") {
    return "grok-build-0.1";
  }
  if (id === "grok-4-fast-reasoning") {
    return "grok-4-fast";
  }
  if (id === "grok-4-1-fast-reasoning") {
    return "grok-4-1-fast";
  }
  return id;
}