feat(accounts): organize bulk account actions

Author: james-6-23

admin/batch_test_test.go

@@ -3,6 +3,8 @@ package admin
 import (
 	"net/http"
 	"testing"
+
+	"github.com/codex2api/auth"
 )
 
 func TestShouldMarkBatchTestAccountError(t *testing.T) {
@@ -52,3 +54,35 @@ func TestShouldMarkBatchTestAccountError(t *testing.T) {
 		})
 	}
 }
+
+func TestResolveBatchTestAccountsDefaultsToAllAccounts(t *testing.T) {
+	store := auth.NewStore(nil, nil, nil)
+	store.AddAccount(&auth.Account{DBID: 1, AccessToken: "token-1", Status: auth.StatusReady})
+	store.AddAccount(&auth.Account{DBID: 2, AccessToken: "token-2", Status: auth.StatusReady})
+
+	accounts, missing := resolveBatchTestAccounts(store, nil)
+	if missing != 0 {
+		t.Fatalf("missing = %d, want 0", missing)
+	}
+	if len(accounts) != 2 {
+		t.Fatalf("len(accounts) = %d, want 2", len(accounts))
+	}
+}
+
+func TestResolveBatchTestAccountsUsesSelectedIDs(t *testing.T) {
+	store := auth.NewStore(nil, nil, nil)
+	store.AddAccount(&auth.Account{DBID: 1, AccessToken: "token-1", Status: auth.StatusReady})
+	store.AddAccount(&auth.Account{DBID: 2, AccessToken: "token-2", Status: auth.StatusReady})
+
+	ids := []int64{2, 99, 2, 1}
+	accounts, missing := resolveBatchTestAccounts(store, &ids)
+	if missing != 1 {
+		t.Fatalf("missing = %d, want 1", missing)
+	}
+	if len(accounts) != 2 {
+		t.Fatalf("len(accounts) = %d, want 2", len(accounts))
+	}
+	if accounts[0].DBID != 2 || accounts[1].DBID != 1 {
+		t.Fatalf("account order = [%d, %d], want [2, 1]", accounts[0].DBID, accounts[1].DBID)
+	}
+}

admin/test_connection.go

@@ -378,11 +378,53 @@ func (h *Handler) connectionTestModel(ctx context.Context) string {
 	return "gpt-5.4"
 }
 
-// BatchTest 批量测试所有账号连接
+type batchTestRequest struct {
+	IDs *[]int64 `json:"ids"`
+}
+
+func resolveBatchTestAccounts(store *auth.Store, ids *[]int64) ([]*auth.Account, int) {
+	if store == nil {
+		return nil, 0
+	}
+	if ids == nil {
+		return store.Accounts(), 0
+	}
+
+	accounts := make([]*auth.Account, 0, len(*ids))
+	missing := 0
+	seen := make(map[int64]struct{}, len(*ids))
+	for _, id := range *ids {
+		if _, ok := seen[id]; ok {
+			continue
+		}
+		seen[id] = struct{}{}
+		acc := store.FindByID(id)
+		if acc == nil {
+			missing++
+			continue
+		}
+		accounts = append(accounts, acc)
+	}
+	return accounts, missing
+}
+
+// BatchTest 批量测试账号连接；未传 ids 时测试所有账号，传 ids 时仅测试指定账号。
 // POST /api/admin/accounts/batch-test
 func (h *Handler) BatchTest(c *gin.Context) {
-	accounts := h.store.Accounts()
-	if len(accounts) == 0 {
+	var req batchTestRequest
+	if c.Request.Body != nil && c.Request.ContentLength != 0 {
+		if err := c.ShouldBindJSON(&req); err != nil {
+			writeError(c, http.StatusBadRequest, "请求格式错误")
+			return
+		}
+	}
+	if req.IDs != nil && len(*req.IDs) == 0 {
+		writeError(c, http.StatusBadRequest, "请提供要测试的账号 ID 列表")
+		return
+	}
+
+	accounts, missingCount := resolveBatchTestAccounts(h.store, req.IDs)
+	if len(accounts) == 0 && missingCount == 0 {
 		c.JSON(http.StatusOK, gin.H{"total": 0, "success": 0, "failed": 0, "banned": 0, "rate_limited": 0})
 		return
 	}
@@ -393,7 +435,7 @@ func (h *Handler) BatchTest(c *gin.Context) {
 
 	var (
 		successCount   int64
-		failedCount    int64
+		failedCount    = int64(missingCount)
 		bannedCount    int64
 		rateLimitCount int64
 		wg             sync.WaitGroup
@@ -457,7 +499,7 @@ func (h *Handler) BatchTest(c *gin.Context) {
 	wg.Wait()
 
 	c.JSON(http.StatusOK, gin.H{
-		"total":        len(accounts),
+		"total":        len(accounts) + missingCount,
 		"success":      successCount,
 		"failed":       failedCount,
 		"banned":       bannedCount,

frontend/src/api.ts

@@ -274,8 +274,11 @@ export const api = {
     request<PromptFilterRulesResponse>('/prompt-filter/rules'),
   getModels: () => request<ModelsResponse>('/models'),
   syncModels: () => request<ModelSyncResponse>('/models/sync', { method: 'POST' }),
-  batchTestAccounts: () =>
-    request<{ total: number; success: number; failed: number; banned: number; rate_limited: number }>('/accounts/batch-test', { method: 'POST' }),
+  batchTestAccounts: (ids?: number[]) =>
+    request<{ total: number; success: number; failed: number; banned: number; rate_limited: number }>('/accounts/batch-test', {
+      method: 'POST',
+      body: ids ? JSON.stringify({ ids }) : undefined,
+    }),
   cleanBanned: () =>
     request<{ message: string; cleaned: number }>('/accounts/clean-banned', { method: 'POST' }),
   cleanRateLimited: () =>

frontend/src/locales/en.json

@@ -146,7 +146,11 @@
     "batchTest": "Batch Test",
     "batchTesting": "Testing…",
     "batchRefresh": "Batch Refresh",
+    "refreshTokens": "Refresh Tokens",
     "batchDelete": "Batch Delete",
+    "maintenanceActions": "Maintenance",
+    "cleanupActions": "Cleanup",
+    "dataActions": "Import / Export",
     "cleanBanned": "Clear Banned",
     "cleanRateLimited": "Clear Rate-Limited",
     "cleanError": "Clear Errors",
@@ -384,6 +388,11 @@
     "lockFailed": "Operation failed: {{error}}",
     "lockHint": "Locked accounts are skipped by auto-cleanup",
     "unlockHint": "Unlocked accounts will be subject to auto-cleanup",
+    "lockSubscriptionAccounts": "Lock Paid Accounts",
+    "lockingSubscriptionAccounts": "Locking…",
+    "lockSubscriptionAccountsHint": "Lock currently unlocked paid accounts in the account pool ({{count}} pending)",
+    "lockSubscriptionAccountsDone": "Paid account lock complete: {{success}} succeeded, {{fail}} failed",
+    "noSubscriptionAccountsToLock": "No paid accounts need locking",
     "filterDisabled": "Disabled",
     "filterLocked": "Locked",
     "batchEnableDone": "Batch enable complete: {{success}} succeeded, {{fail}} failed",

frontend/src/locales/zh.json

@@ -146,7 +146,11 @@
     "batchTest": "批量测试",
     "batchTesting": "测试中…",
     "batchRefresh": "批量刷新",
+    "refreshTokens": "刷新令牌",
     "batchDelete": "批量删除",
+    "maintenanceActions": "账号维护",
+    "cleanupActions": "清理",
+    "dataActions": "导入导出",
     "cleanBanned": "清理封禁",
     "cleanRateLimited": "清理限流",
     "cleanError": "清理错误",
@@ -384,6 +388,11 @@
     "lockFailed": "操作失败：{{error}}",
     "lockHint": "锁定后，自动清理将跳过此账号",
     "unlockHint": "解锁后，账号将正常参与自动清理",
+    "lockSubscriptionAccounts": "一键锁定订阅账号",
+    "lockingSubscriptionAccounts": "锁定中…",
+    "lockSubscriptionAccountsHint": "锁定当前账号池中未锁定的订阅账号（{{count}} 个待锁定）",
+    "lockSubscriptionAccountsDone": "订阅账号锁定完成：成功 {{success}}，失败 {{fail}}",
+    "noSubscriptionAccountsToLock": "没有需要锁定的订阅账号",
     "filterDisabled": "已禁用",
     "filterLocked": "已锁定",
     "batchEnableDone": "批量启用完成：成功 {{success}}，失败 {{fail}}",