85ed193ff0
- DoWithTLS 签名变更：(bool/profile) → (TLSMode, profile) - 所有调用方传入 account.GetTLSMode() 以支持 node/utls/off 三模式 - gateway_service.go、gemini_messages_compat、forward_as_* 全部更新 - claude_usage_service 的 ClaudeUsageFetchOptions 新增 TLSMode 字段 - 新增 decompressResponseBody（gzip/brotli/deflate）到 http_upstream.go - 新增 antigravity_privacy_service.go（setAntigravityPrivacy） - admin_service 新增 ForceOpenAIPrivacy/EnsureAntigravityPrivacy/ForceAntigravityPrivacy - antigravity.Client 新增 SetUserSettings/FetchUserInfo API
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"log/slog"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/antigravity"
|
||
)
|
||
|
||
const (
|
||
AntigravityPrivacySet = "privacy_set"
|
||
AntigravityPrivacyFailed = "privacy_set_failed"
|
||
)
|
||
|
||
// setAntigravityPrivacy 调用 Antigravity API 设置隐私并验证结果。
|
||
// 流程：
|
||
// 1. setUserSettings 清空设置 → 检查返回值 {\"userSettings\":{}}
|
||
// 2. fetchUserInfo 二次验证隐私是否已生效（需要 project_id）
|
||
//
|
||
// 返回 privacy_mode 值：\"privacy_set\" 成功，\"privacy_set_failed\" 失败，空串表示无法执行。
|
||
func setAntigravityPrivacy(ctx context.Context, accessToken, projectID, proxyURL string) string {
|
||
if accessToken == "" {
|
||
return ""
|
||
}
|
||
|
||
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
|
||
defer cancel()
|
||
|
||
client, err := antigravity.NewClient(proxyURL)
|
||
if err != nil {
|
||
slog.Warn("antigravity_privacy_client_error", "error", err.Error())
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
|
||
// 第 1 步：调用 setUserSettings，检查返回值
|
||
setResp, err := client.SetUserSettings(ctx, accessToken)
|
||
if err != nil {
|
||
slog.Warn("antigravity_privacy_set_failed", "error", err.Error())
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
if !setResp.IsSuccess() {
|
||
slog.Warn("antigravity_privacy_set_response_not_empty",
|
||
"user_settings", setResp.UserSettings,
|
||
)
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
|
||
// 第 2 步：调用 fetchUserInfo 二次验证隐私是否已生效
|
||
if strings.TrimSpace(projectID) == "" {
|
||
slog.Warn("antigravity_privacy_missing_project_id")
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
userInfo, err := client.FetchUserInfo(ctx, accessToken, projectID)
|
||
if err != nil {
|
||
slog.Warn("antigravity_privacy_verify_failed", "error", err.Error())
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
if !userInfo.IsPrivate() {
|
||
slog.Warn("antigravity_privacy_verify_not_private",
|
||
"user_settings", userInfo.UserSettings,
|
||
)
|
||
return AntigravityPrivacyFailed
|
||
}
|
||
|
||
slog.Info("antigravity_privacy_set_success")
|
||
return AntigravityPrivacySet
|
||
}
|
||
|
||
func applyAntigravityPrivacyMode(account *Account, mode string) {
|
||
if account == nil || strings.TrimSpace(mode) == "" {
|
||
return
|
||
}
|
||
extra := make(map[string]any, len(account.Extra)+1)
|
||
for k, v := range account.Extra {
|
||
extra[k] = v
|
||
}
|
||
extra["privacy_mode"] = mode
|
||
account.Extra = extra
|
||
}
|