mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-02 22:42:14 +08:00
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/imroc/req/v3"
|
|
)
|
|
|
|
// PrivacyClientFactory creates an HTTP client for privacy API calls.
|
|
// Injected from repository layer to avoid import cycles.
|
|
type PrivacyClientFactory func(proxyURL string) (*req.Client, error)
|
|
|
|
const (
|
|
openAISettingsURL = "https://chatgpt.com/backend-api/settings/account_user_setting"
|
|
|
|
PrivacyModeTrainingOff = "training_off"
|
|
PrivacyModeFailed = "training_set_failed"
|
|
PrivacyModeCFBlocked = "training_set_cf_blocked"
|
|
)
|
|
|
|
// disableOpenAITraining calls ChatGPT settings API to turn off "Improve the model for everyone".
|
|
// Returns privacy_mode value: "training_off" on success, "cf_blocked" / "failed" on failure.
|
|
func disableOpenAITraining(ctx context.Context, clientFactory PrivacyClientFactory, accessToken, proxyURL string) string {
|
|
if accessToken == "" || clientFactory == nil {
|
|
return ""
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(ctx, 15*time.Second)
|
|
defer cancel()
|
|
|
|
client, err := clientFactory(proxyURL)
|
|
if err != nil {
|
|
slog.Warn("openai_privacy_client_error", "error", err.Error())
|
|
return PrivacyModeFailed
|
|
}
|
|
|
|
resp, err := client.R().
|
|
SetContext(ctx).
|
|
SetHeader("Authorization", "Bearer "+accessToken).
|
|
SetHeader("Origin", "https://chatgpt.com").
|
|
SetHeader("Referer", "https://chatgpt.com/").
|
|
SetQueryParam("feature", "training_allowed").
|
|
SetQueryParam("value", "false").
|
|
Patch(openAISettingsURL)
|
|
|
|
if err != nil {
|
|
slog.Warn("openai_privacy_request_error", "error", err.Error())
|
|
return PrivacyModeFailed
|
|
}
|
|
|
|
if resp.StatusCode == 403 || resp.StatusCode == 503 {
|
|
body := resp.String()
|
|
if strings.Contains(body, "cloudflare") || strings.Contains(body, "cf-") || strings.Contains(body, "Just a moment") {
|
|
slog.Warn("openai_privacy_cf_blocked", "status", resp.StatusCode)
|
|
return PrivacyModeCFBlocked
|
|
}
|
|
}
|
|
|
|
if !resp.IsSuccessState() {
|
|
slog.Warn("openai_privacy_failed", "status", resp.StatusCode, "body", truncate(resp.String(), 200))
|
|
return PrivacyModeFailed
|
|
}
|
|
|
|
slog.Info("openai_privacy_training_disabled")
|
|
return PrivacyModeTrainingOff
|
|
}
|
|
|
|
func truncate(s string, n int) string {
|
|
if len(s) <= n {
|
|
return s
|
|
}
|
|
return s[:n] + fmt.Sprintf("...(%d more)", len(s)-n)
|
|
}
|