feat: implement ChatGPT owner demotion service with a FastAPI backend and a basic frontend, alongside updated project metadata.

backend/internal/api/error_cleaner.go

@@ -1,11 +1,15 @@
 package api
 
 import (
+	"encoding/json"
 	"fmt"
+	"io"
+	"net/http"
 	"strconv"
 	"sync"
 	"time"
 
+	"codex-pool/internal/config"
 	"codex-pool/internal/database"
 	"codex-pool/internal/logger"
 )
@@ -64,6 +68,128 @@ func StopErrorCleanerService() {
 	}
 }
 
+// getTotalAccountCount 获取 S2A 总账号数
+func getTotalAccountCount() (int, error) {
+	if config.Global == nil || config.Global.S2AApiBase == "" {
+		return 0, fmt.Errorf("S2A 配置未设置")
+	}
+
+	client := &http.Client{Timeout: 30 * time.Second}
+	url := fmt.Sprintf("%s/api/v1/admin/dashboard", config.Global.S2AApiBase)
+
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return 0, err
+	}
+	setS2AHeaders(req)
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return 0, err
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	var result struct {
+		Code int `json:"code"`
+		Data struct {
+			TotalAccounts int `json:"total_accounts"`
+		} `json:"data"`
+	}
+	if err := json.Unmarshal(body, &result); err != nil {
+		return 0, err
+	}
+
+	return result.Data.TotalAccounts, nil
+}
+
+// triggerAutoRefill 触发自动补号
+func triggerAutoRefill(count int) {
+	if database.Instance == nil {
+		return
+	}
+
+	// 检查是否有待处理的母号
+	pendingOwners, err := database.Instance.GetPendingOwners()
+	if err != nil || len(pendingOwners) == 0 {
+		logger.Warning("自动补号跳过: 无可用母号", "", "cleaner")
+		return
+	}
+
+	// 每个母号产生 4 个账号（默认），计算需要多少母号
+	membersPerTeam := 4
+	if val, _ := database.Instance.GetConfig("monitor_members_per_team"); val != "" {
+		if v, err := strconv.Atoi(val); err == nil && v > 0 {
+			membersPerTeam = v
+		}
+	}
+
+	// 计算需要处理的母号数量
+	needOwners := (count + membersPerTeam - 1) / membersPerTeam
+	if needOwners > len(pendingOwners) {
+		needOwners = len(pendingOwners)
+	}
+	if needOwners < 1 {
+		needOwners = 1
+	}
+
+	// 读取补号配置
+	concurrentTeams := 2
+	if val, _ := database.Instance.GetConfig("monitor_concurrent_teams"); val != "" {
+		if v, err := strconv.Atoi(val); err == nil && v > 0 {
+			concurrentTeams = v
+		}
+	}
+	s2aConcurrency := 2
+	if val, _ := database.Instance.GetConfig("monitor_s2a_concurrency"); val != "" {
+		if v, err := strconv.Atoi(val); err == nil && v > 0 {
+			s2aConcurrency = v
+		}
+	}
+	browserType := "chromedp"
+	if val, _ := database.Instance.GetConfig("monitor_browser_type"); val != "" {
+		browserType = val
+	}
+	useProxy := false
+	if val, _ := database.Instance.GetConfig("monitor_replenish_use_proxy"); val == "true" {
+		useProxy = true
+	}
+
+	proxy := ""
+	if useProxy {
+		proxy = config.Global.GetProxy()
+	}
+
+	logger.Info(fmt.Sprintf("自动补号启动: 处理 %d 个母号，预计补充 %d 个账号", needOwners, needOwners*membersPerTeam), "", "cleaner")
+
+	// 构建请求
+	owners := make([]TeamOwner, 0, needOwners)
+	for i := 0; i < needOwners && i < len(pendingOwners); i++ {
+		owners = append(owners, TeamOwner{
+			Email:     pendingOwners[i].Email,
+			Password:  pendingOwners[i].Password,
+			Token:     pendingOwners[i].Token,
+			AccountID: pendingOwners[i].AccountID,
+		})
+	}
+
+	req := TeamProcessRequest{
+		Owners:          owners,
+		MembersPerTeam:  membersPerTeam,
+		ConcurrentTeams: concurrentTeams,
+		ConcurrentS2A:   s2aConcurrency,
+		BrowserType:     browserType,
+		Headless:        true,
+		Proxy:           proxy,
+		IncludeOwner:    false,
+		ProcessCount:    needOwners,
+	}
+
+	// 异步启动处理（避免阻塞清理服务）
+	go runTeamProcess(req)
+}
+
 // checkAndCleanErrors 检查配置并清理错误账号
 func checkAndCleanErrors() {
 	if database.Instance == nil {
@@ -79,6 +205,13 @@ func checkAndCleanErrors() {
 		return
 	}
 
+	// 获取清理前的总账号数
+	totalBefore, err := getTotalAccountCount()
+	if err != nil {
+		logger.Warning(fmt.Sprintf("获取总账号数失败: %v", err), "", "cleaner")
+		totalBefore = 0
+	}
+
 	// 获取错误账号列表
 	errorAccounts, err := fetchAllErrorAccounts()
 	if err != nil {
@@ -110,6 +243,34 @@ func checkAndCleanErrors() {
 
 	lastCleanTime = time.Now()
 	logger.Success(fmt.Sprintf("定期清理错误账号完成: 成功=%d, 失败=%d, 总数=%d", success, failed, len(errorAccounts)), "", "cleaner")
+
+	// 检查是否需要触发自动补号
+	autoRefillEnabled := false
+	if val, _ := database.Instance.GetConfig("auto_refill_on_cleanup"); val == "true" {
+		autoRefillEnabled = true
+	}
+
+	if autoRefillEnabled && totalBefore > 0 && success > 0 {
+		// 阈值默认 25%
+		threshold := 25
+		if val, _ := database.Instance.GetConfig("auto_refill_threshold"); val != "" {
+			if v, err := strconv.Atoi(val); err == nil && v > 0 && v <= 100 {
+				threshold = v
+			}
+		}
+
+		// 计算是否达到阈值
+		thresholdCount := totalBefore * threshold / 100
+		if thresholdCount < 1 {
+			thresholdCount = 1
+		}
+
+		if success >= thresholdCount {
+			logger.Info(fmt.Sprintf("清理触发自动补号: 已删除 %d 个 (总数 %d，阈值 %d%%)，将补充 %d 个账号",
+				success, totalBefore, threshold, success), "", "cleaner")
+			triggerAutoRefill(success)
+		}
+	}
 }
 
 // GetCleanerStatus 获取清理服务状态
@@ -119,6 +280,9 @@ func GetCleanerStatus() map[string]interface{} {
 
 	enabled := false
 	interval := 3600
+	autoRefill := false
+	autoRefillThreshold := 25
+
 	if database.Instance != nil {
 		if val, _ := database.Instance.GetConfig("error_clean_enabled"); val == "true" {
 			enabled = true
@@ -128,12 +292,22 @@ func GetCleanerStatus() map[string]interface{} {
 				interval = v
 			}
 		}
+		if val, _ := database.Instance.GetConfig("auto_refill_on_cleanup"); val == "true" {
+			autoRefill = true
+		}
+		if val, _ := database.Instance.GetConfig("auto_refill_threshold"); val != "" {
+			if v, err := strconv.Atoi(val); err == nil {
+				autoRefillThreshold = v
+			}
+		}
 	}
 
 	return map[string]interface{}{
-		"running":         cleanerRunning,
-		"enabled":         enabled,
-		"interval":        interval,
-		"last_clean_time": lastCleanTime.Format(time.RFC3339),
+		"running":               cleanerRunning,
+		"enabled":               enabled,
+		"interval":              interval,
+		"last_clean_time":       lastCleanTime.Format(time.RFC3339),
+		"auto_refill":           autoRefill,
+		"auto_refill_threshold": autoRefillThreshold,
 	}
 }