team-api/relay/helper/error_message.go

package helper

import (
	"context"
	"encoding/json"
	"errors"
	"io"
	"net"
	"net/url"
	"regexp"
	"strings"
	"syscall"
)

// 用户可见错误消息中的敏感信息替换占位符。
const redactedPlaceholder = "[redacted]"

var (
	// redactURLPattern 匹配 http(s):// 开头的 URL（含 host、path、query）。
	redactURLPattern = regexp.MustCompile(`(?i)https?://[^\s"'<>]+`)
	// redactIPPattern 匹配 IPv4 地址。版本号偶有误伤，但在错误消息中无害。
	redactIPPattern = regexp.MustCompile(`\b(?:\d{1,3}\.){3}\d{1,3}\b`)
)

// UnwrapUpstreamErrorMessage 从「上游错误响应体原文」中提取人类可读的错误消息。
//
// 为什么需要：relay 各 adaptor 统一以 NewUpstreamError(status, string(body), nil)
// 上抛上游故障，Message 里是**整个上游错误响应体**。错误写出器若直接把它塞进
// 客户端错误体的 message 字段，客户端看到的就是一坨转义 JSON
// （`{"error":{"message":"{\"error\":{\"message\":\"rate limit exceeded\"...`）
// 而不是可读消息——对用户是噪音，对按消息匹配的客户端逻辑是破坏。
//
// 各协议错误信封形状不同，但消息位置固定：
//
//	OpenAI / Responses  {"error":{"message":"...","type":"..."}}
//	Claude              {"type":"error","error":{"type":"...","message":"..."}}
//	Gemini              {"error":{"code":N,"message":"...","status":"..."}}（偶为数组包装）
//	Ollama              {"error":"..."}  ← error 为字符串
//
// 非 JSON、或不是可识别的错误信封（例如网关自身的 "convert request (...)" 消息）
// 时原样返回，调用方可无条件套用。
func UnwrapUpstreamErrorMessage(raw string) string {
	trimmed := strings.TrimSpace(raw)
	if trimmed == "" || (!strings.HasPrefix(trimmed, "{") && !strings.HasPrefix(trimmed, "[")) {
		return raw // 非 JSON：网关自身消息等，原样返回
	}

	// Gemini 偶尔以单元素数组包装错误信封
	if strings.HasPrefix(trimmed, "[") {
		var arr []json.RawMessage
		if err := json.Unmarshal([]byte(trimmed), &arr); err != nil || len(arr) == 0 {
			return raw
		}
		trimmed = string(arr[0])
	}

	var envelope struct {
		Error json.RawMessage `json:"error"`
	}
	if err := json.Unmarshal([]byte(trimmed), &envelope); err != nil || len(envelope.Error) == 0 {
		return raw
	}

	// Ollama 形态：error 直接是字符串
	var asString string
	if err := json.Unmarshal(envelope.Error, &asString); err == nil {
		if asString != "" {
			return asString
		}
		return raw
	}

	var detail struct {
		Message string `json:"message"`
	}
	if err := json.Unmarshal(envelope.Error, &detail); err == nil && detail.Message != "" {
		return detail.Message
	}
	return raw
}

// SafeUpstreamErrorMessage 把可能包含上游渠道域名 / 内部传输细节的错误，
// 转换为对最终用户安全的消息文本。
//
// 设计原则：
//   - 传输层错误（client.Do 返回的 *url.Error）按类别归一化，保留
//     "超时 / 拒绝 / DNS / 重置" 等对用户有用的类别信息，但彻底抹掉
//     上游域名与 URL（这是本修复的核心目标）。
//   - 其余错误（业务错误、上游响应体解析错误等）保留原始消息，但做
//     URL / IP 抹除作为防御性兜底。
//
// 日志侧不受影响：调用方写日志时应使用原始 err（%v），保留完整诊断信息。
func SafeUpstreamErrorMessage(err error) string {
	if err == nil {
		return ""
	}

	// 传输层错误：http.Client.Do 失败时返回 *url.Error，其 Error() 形如
	// `Post "https://api.openai.com/v1/chat/completions": dial tcp: ...`，
	// 完整暴露上游域名，必须归一化。
	var urlErr *url.Error
	if errors.As(err, &urlErr) {
		return categorizeTransportError(urlErr)
	}

	// 非传输层错误：保留消息，但抹除其中可能残留的 URL / IP。
	return RedactMessage(err.Error())
}

// categorizeTransportError 按 *url.Error 内部成因归类为通用消息，不泄露域名。
func categorizeTransportError(urlErr *url.Error) string {
	inner := urlErr.Err

	// 客户端主动取消：不可重试，向用户说明即可
	if errors.Is(inner, context.Canceled) {
		return "请求已取消"
	}
	// 请求 / 响应超时（context 截止、i/o timeout、Client.Timeout 等）
	if errors.Is(inner, context.DeadlineExceeded) {
		return "请求超时，请重试"
	}
	var netErr net.Error
	if errors.As(inner, &netErr) && netErr.Timeout() {
		return "请求超时，请重试"
	}
	// 连接被上游拒绝（端口未开放 / 服务未启动）
	if errors.Is(inner, syscall.ECONNREFUSED) {
		return "服务暂时不可用，请稍后重试"
	}
	// DNS 解析失败：域名不存在或解析异常
	var dnsErr *net.DNSError
	if errors.As(inner, &dnsErr) {
		return "服务暂时不可用，请稍后重试"
	}
	// 连接被对端重置 / 提前关闭（GFW 干扰、上游瞬断的典型表现）
	if errors.Is(inner, syscall.ECONNRESET) ||
		errors.Is(inner, io.EOF) ||
		errors.Is(inner, io.ErrUnexpectedEOF) {
		return "连接中断，请重试"
	}
	// 其余无法归类的传输层错误
	return "服务暂时不可用，请稍后重试"
}

// RedactMessage 全量抹除字符串中的 URL / IP，替换为 [redacted]。
// 用作非传输层错误的防御性兜底（传输层域名已由 SafeUpstreamErrorMessage
// 的结构化分类兜住，此处只处理偶发含 URL/IP 的业务错误消息）。
func RedactMessage(msg string) string {
	if msg == "" {
		return msg
	}
	msg = redactURLPattern.ReplaceAllString(msg, redactedPlaceholder)
	msg = redactIPPattern.ReplaceAllString(msg, redactedPlaceholder)
	return msg
}