a9518cc5be
新增 codex_cli_only 开关并默认关闭，关闭时完全绕过限制逻辑。 在 OpenAI 网关引入统一检测入口，集中判定账号类型、开关与客户端族。 开启后仅放行 codex_cli_rs、codex_vscode、codex_app 客户端家族。 补充后端判定与网关分支测试，并在前端创建/编辑页增加开关配置与回显。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package openai
|
|
|
|
import "strings"
|
|
|
|
// CodexCLIUserAgentPrefixes matches Codex CLI User-Agent patterns
|
|
// Examples: "codex_vscode/1.0.0", "codex_cli_rs/0.1.2"
|
|
var CodexCLIUserAgentPrefixes = []string{
|
|
"codex_vscode/",
|
|
"codex_cli_rs/",
|
|
}
|
|
|
|
// CodexOfficialClientUserAgentPrefixes matches Codex 官方客户端家族 User-Agent 前缀。
|
|
// 该列表仅用于 OpenAI OAuth `codex_cli_only` 访问限制判定。
|
|
var CodexOfficialClientUserAgentPrefixes = []string{
|
|
"codex_cli_rs/",
|
|
"codex_vscode/",
|
|
"codex_app/",
|
|
}
|
|
|
|
// IsCodexCLIRequest checks if the User-Agent indicates a Codex CLI request
|
|
func IsCodexCLIRequest(userAgent string) bool {
|
|
ua := strings.ToLower(strings.TrimSpace(userAgent))
|
|
if ua == "" {
|
|
return false
|
|
}
|
|
for _, prefix := range CodexCLIUserAgentPrefixes {
|
|
normalizedPrefix := strings.ToLower(strings.TrimSpace(prefix))
|
|
if normalizedPrefix == "" {
|
|
continue
|
|
}
|
|
// 优先前缀匹配；若 UA 被网关/代理拼接为复合字符串时，退化为包含匹配。
|
|
if strings.HasPrefix(ua, normalizedPrefix) || strings.Contains(ua, normalizedPrefix) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// IsCodexOfficialClientRequest checks if the User-Agent indicates a Codex 官方客户端请求。
|
|
// 与 IsCodexCLIRequest 解耦，避免影响历史兼容逻辑。
|
|
func IsCodexOfficialClientRequest(userAgent string) bool {
|
|
ua := strings.ToLower(strings.TrimSpace(userAgent))
|
|
if ua == "" {
|
|
return false
|
|
}
|
|
for _, prefix := range CodexOfficialClientUserAgentPrefixes {
|
|
normalizedPrefix := strings.ToLower(strings.TrimSpace(prefix))
|
|
if normalizedPrefix == "" {
|
|
continue
|
|
}
|
|
// 优先前缀匹配；若 UA 被网关/代理拼接为复合字符串时，退化为包含匹配。
|
|
if strings.HasPrefix(ua, normalizedPrefix) || strings.Contains(ua, normalizedPrefix) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|