b20e142249
- 新增 header_util.go，通过 setHeaderRaw/getHeaderRaw/addHeaderRaw 绕过 Go 的 canonical-case 规范化，保持真实 Claude CLI 抓包的请求头大小写 （如 "x-app" 而非 "X-App"，"X-Stainless-OS" 而非 "X-Stainless-Os"） - 新增管理后台开关：指纹统一化（默认开启）和 metadata 透传（默认关闭）， 使用 atomic.Value + singleflight 缓存模式，60s TTL - 调试日志从控制台 body 打印升级为文件级完整快照 （按真实 wire 顺序输出 headers + 格式化 JSON body + 上下文元数据） - 恢复 accept-encoding 到白名单，在 http_upstream.go 新增 decompressResponseBody 处理 gzip/brotli/deflate 解压（Go 显式设置 Accept-Encoding 时不会自动解压） - OAuth 服务 axios UA 从 1.8.4 更新至 1.13.6 - 测试断言改用 getHeaderRaw 适配 raw header 存储方式
248 lines
11 KiB
Go
248 lines
11 KiB
Go
package dto
|
||
|
||
import (
|
||
"encoding/json"
|
||
"strings"
|
||
)
|
||
|
||
// CustomMenuItem represents a user-configured custom menu entry.
|
||
type CustomMenuItem struct {
|
||
ID string `json:"id"`
|
||
Label string `json:"label"`
|
||
IconSVG string `json:"icon_svg"`
|
||
URL string `json:"url"`
|
||
Visibility string `json:"visibility"` // "user" or "admin"
|
||
SortOrder int `json:"sort_order"`
|
||
}
|
||
|
||
// CustomEndpoint represents an admin-configured API endpoint for quick copy.
|
||
type CustomEndpoint struct {
|
||
Name string `json:"name"`
|
||
Endpoint string `json:"endpoint"`
|
||
Description string `json:"description"`
|
||
}
|
||
|
||
// SystemSettings represents the admin settings API response payload.
|
||
type SystemSettings struct {
|
||
RegistrationEnabled bool `json:"registration_enabled"`
|
||
EmailVerifyEnabled bool `json:"email_verify_enabled"`
|
||
RegistrationEmailSuffixWhitelist []string `json:"registration_email_suffix_whitelist"`
|
||
PromoCodeEnabled bool `json:"promo_code_enabled"`
|
||
PasswordResetEnabled bool `json:"password_reset_enabled"`
|
||
FrontendURL string `json:"frontend_url"`
|
||
InvitationCodeEnabled bool `json:"invitation_code_enabled"`
|
||
TotpEnabled bool `json:"totp_enabled"` // TOTP 双因素认证
|
||
TotpEncryptionKeyConfigured bool `json:"totp_encryption_key_configured"` // TOTP 加密密钥是否已配置
|
||
|
||
SMTPHost string `json:"smtp_host"`
|
||
SMTPPort int `json:"smtp_port"`
|
||
SMTPUsername string `json:"smtp_username"`
|
||
SMTPPasswordConfigured bool `json:"smtp_password_configured"`
|
||
SMTPFrom string `json:"smtp_from_email"`
|
||
SMTPFromName string `json:"smtp_from_name"`
|
||
SMTPUseTLS bool `json:"smtp_use_tls"`
|
||
|
||
TurnstileEnabled bool `json:"turnstile_enabled"`
|
||
TurnstileSiteKey string `json:"turnstile_site_key"`
|
||
TurnstileSecretKeyConfigured bool `json:"turnstile_secret_key_configured"`
|
||
|
||
LinuxDoConnectEnabled bool `json:"linuxdo_connect_enabled"`
|
||
LinuxDoConnectClientID string `json:"linuxdo_connect_client_id"`
|
||
LinuxDoConnectClientSecretConfigured bool `json:"linuxdo_connect_client_secret_configured"`
|
||
LinuxDoConnectRedirectURL string `json:"linuxdo_connect_redirect_url"`
|
||
|
||
SiteName string `json:"site_name"`
|
||
SiteLogo string `json:"site_logo"`
|
||
SiteSubtitle string `json:"site_subtitle"`
|
||
APIBaseURL string `json:"api_base_url"`
|
||
ContactInfo string `json:"contact_info"`
|
||
DocURL string `json:"doc_url"`
|
||
HomeContent string `json:"home_content"`
|
||
HideCcsImportButton bool `json:"hide_ccs_import_button"`
|
||
PurchaseSubscriptionEnabled bool `json:"purchase_subscription_enabled"`
|
||
PurchaseSubscriptionURL string `json:"purchase_subscription_url"`
|
||
SoraClientEnabled bool `json:"sora_client_enabled"`
|
||
CustomMenuItems []CustomMenuItem `json:"custom_menu_items"`
|
||
CustomEndpoints []CustomEndpoint `json:"custom_endpoints"`
|
||
|
||
DefaultConcurrency int `json:"default_concurrency"`
|
||
DefaultBalance float64 `json:"default_balance"`
|
||
DefaultSubscriptions []DefaultSubscriptionSetting `json:"default_subscriptions"`
|
||
|
||
// Model fallback configuration
|
||
EnableModelFallback bool `json:"enable_model_fallback"`
|
||
FallbackModelAnthropic string `json:"fallback_model_anthropic"`
|
||
FallbackModelOpenAI string `json:"fallback_model_openai"`
|
||
FallbackModelGemini string `json:"fallback_model_gemini"`
|
||
FallbackModelAntigravity string `json:"fallback_model_antigravity"`
|
||
|
||
// Identity patch configuration (Claude -> Gemini)
|
||
EnableIdentityPatch bool `json:"enable_identity_patch"`
|
||
IdentityPatchPrompt string `json:"identity_patch_prompt"`
|
||
|
||
// Ops monitoring (vNext)
|
||
OpsMonitoringEnabled bool `json:"ops_monitoring_enabled"`
|
||
OpsRealtimeMonitoringEnabled bool `json:"ops_realtime_monitoring_enabled"`
|
||
OpsQueryModeDefault string `json:"ops_query_mode_default"`
|
||
OpsMetricsIntervalSeconds int `json:"ops_metrics_interval_seconds"`
|
||
|
||
MinClaudeCodeVersion string `json:"min_claude_code_version"`
|
||
MaxClaudeCodeVersion string `json:"max_claude_code_version"`
|
||
|
||
// 分组隔离
|
||
AllowUngroupedKeyScheduling bool `json:"allow_ungrouped_key_scheduling"`
|
||
|
||
// Backend Mode
|
||
BackendModeEnabled bool `json:"backend_mode_enabled"`
|
||
|
||
// Gateway forwarding behavior
|
||
EnableFingerprintUnification bool `json:"enable_fingerprint_unification"`
|
||
EnableMetadataPassthrough bool `json:"enable_metadata_passthrough"`
|
||
}
|
||
|
||
type DefaultSubscriptionSetting struct {
|
||
GroupID int64 `json:"group_id"`
|
||
ValidityDays int `json:"validity_days"`
|
||
}
|
||
|
||
type PublicSettings struct {
|
||
RegistrationEnabled bool `json:"registration_enabled"`
|
||
EmailVerifyEnabled bool `json:"email_verify_enabled"`
|
||
RegistrationEmailSuffixWhitelist []string `json:"registration_email_suffix_whitelist"`
|
||
PromoCodeEnabled bool `json:"promo_code_enabled"`
|
||
PasswordResetEnabled bool `json:"password_reset_enabled"`
|
||
InvitationCodeEnabled bool `json:"invitation_code_enabled"`
|
||
TotpEnabled bool `json:"totp_enabled"` // TOTP 双因素认证
|
||
TurnstileEnabled bool `json:"turnstile_enabled"`
|
||
TurnstileSiteKey string `json:"turnstile_site_key"`
|
||
SiteName string `json:"site_name"`
|
||
SiteLogo string `json:"site_logo"`
|
||
SiteSubtitle string `json:"site_subtitle"`
|
||
APIBaseURL string `json:"api_base_url"`
|
||
ContactInfo string `json:"contact_info"`
|
||
DocURL string `json:"doc_url"`
|
||
HomeContent string `json:"home_content"`
|
||
HideCcsImportButton bool `json:"hide_ccs_import_button"`
|
||
PurchaseSubscriptionEnabled bool `json:"purchase_subscription_enabled"`
|
||
PurchaseSubscriptionURL string `json:"purchase_subscription_url"`
|
||
CustomMenuItems []CustomMenuItem `json:"custom_menu_items"`
|
||
CustomEndpoints []CustomEndpoint `json:"custom_endpoints"`
|
||
LinuxDoOAuthEnabled bool `json:"linuxdo_oauth_enabled"`
|
||
SoraClientEnabled bool `json:"sora_client_enabled"`
|
||
BackendModeEnabled bool `json:"backend_mode_enabled"`
|
||
Version string `json:"version"`
|
||
}
|
||
|
||
// SoraS3Settings Sora S3 存储配置 DTO（响应用，不含敏感字段）
|
||
type SoraS3Settings struct {
|
||
Enabled bool `json:"enabled"`
|
||
Endpoint string `json:"endpoint"`
|
||
Region string `json:"region"`
|
||
Bucket string `json:"bucket"`
|
||
AccessKeyID string `json:"access_key_id"`
|
||
SecretAccessKeyConfigured bool `json:"secret_access_key_configured"`
|
||
Prefix string `json:"prefix"`
|
||
ForcePathStyle bool `json:"force_path_style"`
|
||
CDNURL string `json:"cdn_url"`
|
||
DefaultStorageQuotaBytes int64 `json:"default_storage_quota_bytes"`
|
||
}
|
||
|
||
// SoraS3Profile Sora S3 存储配置项 DTO（响应用，不含敏感字段）
|
||
type SoraS3Profile struct {
|
||
ProfileID string `json:"profile_id"`
|
||
Name string `json:"name"`
|
||
IsActive bool `json:"is_active"`
|
||
Enabled bool `json:"enabled"`
|
||
Endpoint string `json:"endpoint"`
|
||
Region string `json:"region"`
|
||
Bucket string `json:"bucket"`
|
||
AccessKeyID string `json:"access_key_id"`
|
||
SecretAccessKeyConfigured bool `json:"secret_access_key_configured"`
|
||
Prefix string `json:"prefix"`
|
||
ForcePathStyle bool `json:"force_path_style"`
|
||
CDNURL string `json:"cdn_url"`
|
||
DefaultStorageQuotaBytes int64 `json:"default_storage_quota_bytes"`
|
||
UpdatedAt string `json:"updated_at"`
|
||
}
|
||
|
||
// ListSoraS3ProfilesResponse Sora S3 配置列表响应
|
||
type ListSoraS3ProfilesResponse struct {
|
||
ActiveProfileID string `json:"active_profile_id"`
|
||
Items []SoraS3Profile `json:"items"`
|
||
}
|
||
|
||
// OverloadCooldownSettings 529过载冷却配置 DTO
|
||
type OverloadCooldownSettings struct {
|
||
Enabled bool `json:"enabled"`
|
||
CooldownMinutes int `json:"cooldown_minutes"`
|
||
}
|
||
|
||
// StreamTimeoutSettings 流超时处理配置 DTO
|
||
type StreamTimeoutSettings struct {
|
||
Enabled bool `json:"enabled"`
|
||
Action string `json:"action"`
|
||
TempUnschedMinutes int `json:"temp_unsched_minutes"`
|
||
ThresholdCount int `json:"threshold_count"`
|
||
ThresholdWindowMinutes int `json:"threshold_window_minutes"`
|
||
}
|
||
|
||
// RectifierSettings 请求整流器配置 DTO
|
||
type RectifierSettings struct {
|
||
Enabled bool `json:"enabled"`
|
||
ThinkingSignatureEnabled bool `json:"thinking_signature_enabled"`
|
||
ThinkingBudgetEnabled bool `json:"thinking_budget_enabled"`
|
||
}
|
||
|
||
// BetaPolicyRule Beta 策略规则 DTO
|
||
type BetaPolicyRule struct {
|
||
BetaToken string `json:"beta_token"`
|
||
Action string `json:"action"`
|
||
Scope string `json:"scope"`
|
||
ErrorMessage string `json:"error_message,omitempty"`
|
||
}
|
||
|
||
// BetaPolicySettings Beta 策略配置 DTO
|
||
type BetaPolicySettings struct {
|
||
Rules []BetaPolicyRule `json:"rules"`
|
||
}
|
||
|
||
// ParseCustomMenuItems parses a JSON string into a slice of CustomMenuItem.
|
||
// Returns empty slice on empty/invalid input.
|
||
func ParseCustomMenuItems(raw string) []CustomMenuItem {
|
||
raw = strings.TrimSpace(raw)
|
||
if raw == "" || raw == "[]" {
|
||
return []CustomMenuItem{}
|
||
}
|
||
var items []CustomMenuItem
|
||
if err := json.Unmarshal([]byte(raw), &items); err != nil {
|
||
return []CustomMenuItem{}
|
||
}
|
||
return items
|
||
}
|
||
|
||
// ParseUserVisibleMenuItems parses custom menu items and filters out admin-only entries.
|
||
func ParseUserVisibleMenuItems(raw string) []CustomMenuItem {
|
||
items := ParseCustomMenuItems(raw)
|
||
filtered := make([]CustomMenuItem, 0, len(items))
|
||
for _, item := range items {
|
||
if item.Visibility != "admin" {
|
||
filtered = append(filtered, item)
|
||
}
|
||
}
|
||
return filtered
|
||
}
|
||
|
||
// ParseCustomEndpoints parses a JSON string into a slice of CustomEndpoint.
|
||
// Returns empty slice on empty/invalid input.
|
||
func ParseCustomEndpoints(raw string) []CustomEndpoint {
|
||
raw = strings.TrimSpace(raw)
|
||
if raw == "" || raw == "[]" {
|
||
return []CustomEndpoint{}
|
||
}
|
||
var items []CustomEndpoint
|
||
if err := json.Unmarshal([]byte(raw), &items); err != nil {
|
||
return []CustomEndpoint{}
|
||
}
|
||
return items
|
||
}
|