ai-chatbot/lib/request-error.ts

const RATE_LIMIT_PATTERNS = [
  /\btoo many requests\b/i,
  /\brate limit(?:ed|ing)?\b/i,
  /\bquota exceeded\b/i,
  /请求过于频繁/,
  /请求太频繁/,
  /访问过于频繁/,
  /频率过高/,
  /达到速率限制/,
  /限流/,
];
const PROVIDER_BALANCE_PATTERNS = [
  /余额(?:不足|不够|已?耗尽)/i,
  /额度(?:不足|不够|已?耗尽)/i,
  /预扣费额度(?:不足|失败)/i,
  /\binsufficient[ _-]?(?:balance|credits?|funds?|quota)\b/i,
  /\b(?:balance|credits?|funds?)\s+(?:is\s+)?(?:not enough|insufficient|exhausted)\b/i,
  /\b(?:exceeded_current_quota|pre_consume_token_quota_failed)\b/i,
  /\btoken quota is not enough\b/i,
  /\bquota is not enough\b/i,
  /\b(?:not enough credits?|credits? is not enough)\b/i,
  /\b(?:billing_hard_limit_reached|payment_required)\b/i,
];
const NETWORK_TRANSPORT_PATTERNS = [
  /\bfailed to fetch\b/i,
  /\bnetworkerror\b/i,
  /\bload failed\b/i,
  /network request failed/i,
  /having trouble sending your message/i,
  /check your internet connection/i,
];
const JSON_EMBEDDED_MESSAGE_PATTERN = /^[["{]/;

function isRecord(value: unknown): value is Record<string, unknown> {
  return typeof value === "object" && value !== null;
}

function normalizeTextCandidate(value: string) {
  return value.trim().replace(/^["']|["']$/g, "");
}

function collectTextCandidates(
  value: unknown,
  results: string[],
  seen = new WeakSet<object>(),
  depth = 0
) {
  if (depth > 4 || value == null) {
    return;
  }

  if (typeof value === "string") {
    const normalized = normalizeTextCandidate(value);
    if (normalized) {
      results.push(normalized);
    }
    return;
  }

  if (Array.isArray(value)) {
    for (const item of value) {
      collectTextCandidates(item, results, seen, depth + 1);
    }
    return;
  }

  if (!isRecord(value)) {
    return;
  }

  if (seen.has(value)) {
    return;
  }
  seen.add(value);

  const preferredKeys = [
    "message",
    "error",
    "detail",
    "cause",
    "statusText",
    "responseText",
    "body",
    "data",
  ];

  for (const key of preferredKeys) {
    if (key in value) {
      collectTextCandidates(value[key], results, seen, depth + 1);
    }
  }

  for (const [key, candidate] of Object.entries(value)) {
    if (preferredKeys.includes(key)) {
      continue;
    }
    collectTextCandidates(candidate, results, seen, depth + 1);
  }
}

function parseJsonEmbeddedMessage(value: string): string | null {
  const normalized = normalizeTextCandidate(value);
  if (!normalized || !JSON_EMBEDDED_MESSAGE_PATTERN.test(normalized)) {
    return null;
  }

  try {
    const parsed = JSON.parse(normalized);
    const messages: string[] = [];
    collectTextCandidates(parsed, messages);
    return messages.find(Boolean) || null;
  } catch {
    return null;
  }
}

export function getRequestErrorStatus(error: unknown): number | null {
  const candidates: unknown[] = [];

  if (isRecord(error)) {
    candidates.push(
      error.status,
      error.statusCode,
      isRecord(error.response) ? error.response.status : null,
      isRecord(error.cause) ? error.cause.status : null,
      isRecord(error.data) ? error.data.status : null,
      isRecord(error.data) ? error.data.statusCode : null
    );
  }

  for (const candidate of candidates) {
    const numeric = Number(candidate);
    if (Number.isFinite(numeric) && numeric >= 100) {
      return Math.round(numeric);
    }
  }

  return null;
}

export function getRequestErrorMessage(error: unknown): string | null {
  const candidates: string[] = [];

  if (error instanceof Error) {
    collectTextCandidates(
      {
        message: error.message,
        cause: error.cause,
        ...(isRecord(error) ? error : {}),
      },
      candidates
    );
  } else {
    collectTextCandidates(error, candidates);
  }

  for (const candidate of candidates) {
    const embeddedMessage = parseJsonEmbeddedMessage(candidate);
    const normalized = normalizeTextCandidate(embeddedMessage || candidate);
    if (normalized && normalized !== "[object Object]") {
      return normalized;
    }
  }

  return null;
}

function getRequestErrorText(error: unknown) {
  const candidates: string[] = [];
  if (error instanceof Error) {
    collectTextCandidates({ ...error, message: error.message, cause: error.cause }, candidates);
  } else {
    collectTextCandidates(error, candidates);
  }
  return candidates.join(" ").slice(0, 4000);
}

export function getGatewayErrorPageMessage(content: string): string | null {
  if (!/^\s*(?:<!doctype\s+html\b|<html\b)/i.test(content)) return null;
  if (!/\bcf-error-details\b/i.test(content)) return null;
  if (/\b524\b/.test(content)) {
    return "模型服务响应超时（HTTP 524），生成结果尚未确认。请先查看历史记录，避免重复提交。";
  }
  return "模型服务暂时不可用，请稍后再试或联系管理员。";
}

export function isGatewayResponseTimeout(error: unknown): boolean {
  return getRequestErrorStatus(error) === 524 ||
    /\bHTTP\s*524\b|\b524\s*:\s*A timeout occurred\b|Error code\s*524\b/i.test(getRequestErrorText(error));
}

function getPublicGatewayErrorMessage(error: unknown): string | null {
  if (isGatewayResponseTimeout(error)) {
    return "模型服务响应超时（HTTP 524），生成结果尚未确认。请先查看历史记录，避免重复提交。";
  }
  const message = getRequestErrorMessage(error);
  return message ? getGatewayErrorPageMessage(message) : null;
}

export function isProviderBalanceError(error: unknown): boolean {
  const message = getRequestErrorText(error);
  return PROVIDER_BALANCE_PATTERNS.some((pattern) => pattern.test(message));
}

export function isRateLimitError(error: unknown): boolean {
  // Some providers return HTTP 429 for an actual unpaid balance. The status
  // alone must not override an explicit billing error in the response body.
  if (isProviderBalanceError(error)) {
    return false;
  }
  if (getRequestErrorStatus(error) === 429) {
    return true;
  }

  const message = getRequestErrorText(error);
  if (!message) {
    return false;
  }

  return /(?:^|\D)429(?:\D|$)/.test(message) ||
    RATE_LIMIT_PATTERNS.some((pattern) => pattern.test(message));
}

export function getUserFacingImageGenerationErrorMessage(
  error: unknown,
  fallbackMessage = "图片生成失败，请稍后重试",
) {
  const gatewayMessage = getPublicGatewayErrorMessage(error);
  if (gatewayMessage) return gatewayMessage;
  if (isProviderBalanceError(error)) {
    return "图片服务商线路余额或额度不足，请切换模型或联系管理员；与您的积分余额无关。";
  }
  if (isRateLimitError(error)) {
    return "图片服务请求过于频繁，请稍后重试或切换模型。";
  }
  return getRequestErrorMessage(error) || fallbackMessage;
}

export function getFriendlyRateLimitMessage(
  customMessage = "当前请求过于频繁，请稍后再试"
) {
  return (
    normalizeTextCandidate(customMessage) || "当前请求过于频繁，请稍后再试"
  );
}

export function getUserFacingRequestErrorMessage(
  error: unknown,
  fallbackMessage = "请求失败，请稍后重试",
  options?: {
    rateLimitMessage?: string;
  }
) {
  const gatewayMessage = getPublicGatewayErrorMessage(error);
  if (gatewayMessage) return gatewayMessage;
  if (isProviderBalanceError(error)) {
    return "模型服务商线路余额或额度不足，请切换模型或联系管理员。";
  }
  if (isRateLimitError(error)) {
    return getFriendlyRateLimitMessage(options?.rateLimitMessage);
  }

  const message = getRequestErrorMessage(error);
  if (!message) {
    return fallbackMessage;
  }

  if (RATE_LIMIT_PATTERNS.some((pattern) => pattern.test(message))) {
    return getFriendlyRateLimitMessage(options?.rateLimitMessage);
  }

  if (NETWORK_TRANSPORT_PATTERNS.some((pattern) => pattern.test(message))) {
    return "请求发送失败，请检查网络或稍后重试。";
  }

  return message;
}