| package admin |
|
|
| import ( |
| "context" |
| "strings" |
| "sync" |
| "time" |
|
|
| "github.com/Wei-Shaw/sub2api/internal/service" |
| ) |
|
|
| type stubAdminService struct { |
| users []service.User |
| apiKeys []service.APIKey |
| groups []service.Group |
| accounts []service.Account |
| proxies []service.Proxy |
| proxyCounts []service.ProxyWithAccountCount |
| redeems []service.RedeemCode |
| createdAccounts []*service.CreateAccountInput |
| createdProxies []*service.CreateProxyInput |
| updatedProxyIDs []int64 |
| updatedProxies []*service.UpdateProxyInput |
| testedProxyIDs []int64 |
| createAccountErr error |
| updateAccountErr error |
| bulkUpdateAccountErr error |
| checkMixedErr error |
| lastMixedCheck struct { |
| accountID int64 |
| platform string |
| groupIDs []int64 |
| } |
| mu sync.Mutex |
| } |
|
|
| func newStubAdminService() *stubAdminService { |
| now := time.Now().UTC() |
| user := service.User{ |
| ID: 1, |
| Email: "user@example.com", |
| Role: service.RoleUser, |
| Status: service.StatusActive, |
| CreatedAt: now, |
| UpdatedAt: now, |
| } |
| apiKey := service.APIKey{ |
| ID: 10, |
| UserID: user.ID, |
| Key: "sk-test", |
| Name: "test", |
| Status: service.StatusActive, |
| CreatedAt: now, |
| UpdatedAt: now, |
| } |
| group := service.Group{ |
| ID: 2, |
| Name: "group", |
| Platform: service.PlatformAnthropic, |
| Status: service.StatusActive, |
| CreatedAt: now, |
| UpdatedAt: now, |
| } |
| account := service.Account{ |
| ID: 3, |
| Name: "account", |
| Platform: service.PlatformAnthropic, |
| Type: service.AccountTypeOAuth, |
| Status: service.StatusActive, |
| CreatedAt: now, |
| UpdatedAt: now, |
| } |
| proxy := service.Proxy{ |
| ID: 4, |
| Name: "proxy", |
| Protocol: "http", |
| Host: "127.0.0.1", |
| Port: 8080, |
| Status: service.StatusActive, |
| CreatedAt: now, |
| UpdatedAt: now, |
| } |
| redeem := service.RedeemCode{ |
| ID: 5, |
| Code: "R-TEST", |
| Type: service.RedeemTypeBalance, |
| Value: 10, |
| Status: service.StatusUnused, |
| CreatedAt: now, |
| } |
| return &stubAdminService{ |
| users: []service.User{user}, |
| apiKeys: []service.APIKey{apiKey}, |
| groups: []service.Group{group}, |
| accounts: []service.Account{account}, |
| proxies: []service.Proxy{proxy}, |
| proxyCounts: []service.ProxyWithAccountCount{{Proxy: proxy, AccountCount: 1}}, |
| redeems: []service.RedeemCode{redeem}, |
| } |
| } |
|
|
| func (s *stubAdminService) ListUsers(ctx context.Context, page, pageSize int, filters service.UserListFilters) ([]service.User, int64, error) { |
| return s.users, int64(len(s.users)), nil |
| } |
|
|
| func (s *stubAdminService) GetUser(ctx context.Context, id int64) (*service.User, error) { |
| for i := range s.users { |
| if s.users[i].ID == id { |
| return &s.users[i], nil |
| } |
| } |
| user := service.User{ID: id, Email: "user@example.com", Status: service.StatusActive} |
| return &user, nil |
| } |
|
|
| func (s *stubAdminService) CreateUser(ctx context.Context, input *service.CreateUserInput) (*service.User, error) { |
| user := service.User{ID: 100, Email: input.Email, Status: service.StatusActive} |
| return &user, nil |
| } |
|
|
| func (s *stubAdminService) UpdateUser(ctx context.Context, id int64, input *service.UpdateUserInput) (*service.User, error) { |
| user := service.User{ID: id, Email: "updated@example.com", Status: service.StatusActive} |
| return &user, nil |
| } |
|
|
| func (s *stubAdminService) DeleteUser(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) UpdateUserBalance(ctx context.Context, userID int64, balance float64, operation string, notes string) (*service.User, error) { |
| user := service.User{ID: userID, Balance: balance, Status: service.StatusActive} |
| return &user, nil |
| } |
|
|
| func (s *stubAdminService) GetUserAPIKeys(ctx context.Context, userID int64, page, pageSize int) ([]service.APIKey, int64, error) { |
| return s.apiKeys, int64(len(s.apiKeys)), nil |
| } |
|
|
| func (s *stubAdminService) GetUserUsageStats(ctx context.Context, userID int64, period string) (any, error) { |
| return map[string]any{"user_id": userID}, nil |
| } |
|
|
| func (s *stubAdminService) ListGroups(ctx context.Context, page, pageSize int, platform, status, search string, isExclusive *bool) ([]service.Group, int64, error) { |
| return s.groups, int64(len(s.groups)), nil |
| } |
|
|
| func (s *stubAdminService) GetAllGroups(ctx context.Context) ([]service.Group, error) { |
| return s.groups, nil |
| } |
|
|
| func (s *stubAdminService) GetAllGroupsByPlatform(ctx context.Context, platform string) ([]service.Group, error) { |
| return s.groups, nil |
| } |
|
|
| func (s *stubAdminService) GetGroup(ctx context.Context, id int64) (*service.Group, error) { |
| group := service.Group{ID: id, Name: "group", Status: service.StatusActive} |
| return &group, nil |
| } |
|
|
| func (s *stubAdminService) CreateGroup(ctx context.Context, input *service.CreateGroupInput) (*service.Group, error) { |
| group := service.Group{ID: 200, Name: input.Name, Status: service.StatusActive} |
| return &group, nil |
| } |
|
|
| func (s *stubAdminService) UpdateGroup(ctx context.Context, id int64, input *service.UpdateGroupInput) (*service.Group, error) { |
| group := service.Group{ID: id, Name: input.Name, Status: service.StatusActive} |
| return &group, nil |
| } |
|
|
| func (s *stubAdminService) DeleteGroup(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) GetGroupAPIKeys(ctx context.Context, groupID int64, page, pageSize int) ([]service.APIKey, int64, error) { |
| return s.apiKeys, int64(len(s.apiKeys)), nil |
| } |
|
|
| func (s *stubAdminService) GetGroupRateMultipliers(_ context.Context, _ int64) ([]service.UserGroupRateEntry, error) { |
| return nil, nil |
| } |
|
|
| func (s *stubAdminService) ClearGroupRateMultipliers(_ context.Context, _ int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) BatchSetGroupRateMultipliers(_ context.Context, _ int64, _ []service.GroupRateMultiplierInput) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) ListAccounts(ctx context.Context, page, pageSize int, platform, accountType, status, search string, groupID int64) ([]service.Account, int64, error) { |
| return s.accounts, int64(len(s.accounts)), nil |
| } |
|
|
| func (s *stubAdminService) GetAccount(ctx context.Context, id int64) (*service.Account, error) { |
| account := service.Account{ID: id, Name: "account", Status: service.StatusActive} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) GetAccountsByIDs(ctx context.Context, ids []int64) ([]*service.Account, error) { |
| out := make([]*service.Account, 0, len(ids)) |
| for _, id := range ids { |
| account := service.Account{ID: id, Name: "account", Status: service.StatusActive} |
| out = append(out, &account) |
| } |
| return out, nil |
| } |
|
|
| func (s *stubAdminService) CreateAccount(ctx context.Context, input *service.CreateAccountInput) (*service.Account, error) { |
| s.mu.Lock() |
| s.createdAccounts = append(s.createdAccounts, input) |
| s.mu.Unlock() |
| if s.createAccountErr != nil { |
| return nil, s.createAccountErr |
| } |
| account := service.Account{ID: 300, Name: input.Name, Status: service.StatusActive} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) UpdateAccount(ctx context.Context, id int64, input *service.UpdateAccountInput) (*service.Account, error) { |
| if s.updateAccountErr != nil { |
| return nil, s.updateAccountErr |
| } |
| account := service.Account{ID: id, Name: input.Name, Status: service.StatusActive} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) DeleteAccount(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) RefreshAccountCredentials(ctx context.Context, id int64) (*service.Account, error) { |
| account := service.Account{ID: id, Name: "account", Status: service.StatusActive} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) ClearAccountError(ctx context.Context, id int64) (*service.Account, error) { |
| account := service.Account{ID: id, Name: "account", Status: service.StatusActive} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) SetAccountError(ctx context.Context, id int64, errorMsg string) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) SetAccountSchedulable(ctx context.Context, id int64, schedulable bool) (*service.Account, error) { |
| account := service.Account{ID: id, Name: "account", Status: service.StatusActive, Schedulable: schedulable} |
| return &account, nil |
| } |
|
|
| func (s *stubAdminService) BulkUpdateAccounts(ctx context.Context, input *service.BulkUpdateAccountsInput) (*service.BulkUpdateAccountsResult, error) { |
| if s.bulkUpdateAccountErr != nil { |
| return nil, s.bulkUpdateAccountErr |
| } |
| return &service.BulkUpdateAccountsResult{Success: len(input.AccountIDs), Failed: 0, SuccessIDs: input.AccountIDs}, nil |
| } |
|
|
| func (s *stubAdminService) CheckMixedChannelRisk(ctx context.Context, currentAccountID int64, currentAccountPlatform string, groupIDs []int64) error { |
| s.lastMixedCheck.accountID = currentAccountID |
| s.lastMixedCheck.platform = currentAccountPlatform |
| s.lastMixedCheck.groupIDs = append([]int64(nil), groupIDs...) |
| return s.checkMixedErr |
| } |
|
|
| func (s *stubAdminService) ListProxies(ctx context.Context, page, pageSize int, protocol, status, search string) ([]service.Proxy, int64, error) { |
| search = strings.TrimSpace(strings.ToLower(search)) |
| filtered := make([]service.Proxy, 0, len(s.proxies)) |
| for _, proxy := range s.proxies { |
| if protocol != "" && proxy.Protocol != protocol { |
| continue |
| } |
| if status != "" && proxy.Status != status { |
| continue |
| } |
| if search != "" { |
| name := strings.ToLower(proxy.Name) |
| host := strings.ToLower(proxy.Host) |
| if !strings.Contains(name, search) && !strings.Contains(host, search) { |
| continue |
| } |
| } |
| filtered = append(filtered, proxy) |
| } |
| return filtered, int64(len(filtered)), nil |
| } |
|
|
| func (s *stubAdminService) ListProxiesWithAccountCount(ctx context.Context, page, pageSize int, protocol, status, search string) ([]service.ProxyWithAccountCount, int64, error) { |
| return s.proxyCounts, int64(len(s.proxyCounts)), nil |
| } |
|
|
| func (s *stubAdminService) GetAllProxies(ctx context.Context) ([]service.Proxy, error) { |
| return s.proxies, nil |
| } |
|
|
| func (s *stubAdminService) GetAllProxiesWithAccountCount(ctx context.Context) ([]service.ProxyWithAccountCount, error) { |
| return s.proxyCounts, nil |
| } |
|
|
| func (s *stubAdminService) GetProxy(ctx context.Context, id int64) (*service.Proxy, error) { |
| for i := range s.proxies { |
| proxy := s.proxies[i] |
| if proxy.ID == id { |
| return &proxy, nil |
| } |
| } |
| proxy := service.Proxy{ID: id, Name: "proxy", Status: service.StatusActive} |
| return &proxy, nil |
| } |
|
|
| func (s *stubAdminService) GetProxiesByIDs(ctx context.Context, ids []int64) ([]service.Proxy, error) { |
| if len(ids) == 0 { |
| return []service.Proxy{}, nil |
| } |
| out := make([]service.Proxy, 0, len(ids)) |
| seen := make(map[int64]struct{}, len(ids)) |
| for _, id := range ids { |
| seen[id] = struct{}{} |
| } |
| for i := range s.proxies { |
| proxy := s.proxies[i] |
| if _, ok := seen[proxy.ID]; ok { |
| out = append(out, proxy) |
| } |
| } |
| return out, nil |
| } |
|
|
| func (s *stubAdminService) CreateProxy(ctx context.Context, input *service.CreateProxyInput) (*service.Proxy, error) { |
| s.mu.Lock() |
| s.createdProxies = append(s.createdProxies, input) |
| s.mu.Unlock() |
| proxy := service.Proxy{ID: 400, Name: input.Name, Status: service.StatusActive} |
| return &proxy, nil |
| } |
|
|
| func (s *stubAdminService) UpdateProxy(ctx context.Context, id int64, input *service.UpdateProxyInput) (*service.Proxy, error) { |
| s.mu.Lock() |
| s.updatedProxyIDs = append(s.updatedProxyIDs, id) |
| s.updatedProxies = append(s.updatedProxies, input) |
| s.mu.Unlock() |
| proxy := service.Proxy{ID: id, Name: input.Name, Status: service.StatusActive} |
| return &proxy, nil |
| } |
|
|
| func (s *stubAdminService) DeleteProxy(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) BatchDeleteProxies(ctx context.Context, ids []int64) (*service.ProxyBatchDeleteResult, error) { |
| return &service.ProxyBatchDeleteResult{DeletedIDs: ids}, nil |
| } |
|
|
| func (s *stubAdminService) GetProxyAccounts(ctx context.Context, proxyID int64) ([]service.ProxyAccountSummary, error) { |
| return []service.ProxyAccountSummary{{ID: 1, Name: "account"}}, nil |
| } |
|
|
| func (s *stubAdminService) CheckProxyExists(ctx context.Context, host string, port int, username, password string) (bool, error) { |
| return false, nil |
| } |
|
|
| func (s *stubAdminService) TestProxy(ctx context.Context, id int64) (*service.ProxyTestResult, error) { |
| s.mu.Lock() |
| s.testedProxyIDs = append(s.testedProxyIDs, id) |
| s.mu.Unlock() |
| return &service.ProxyTestResult{Success: true, Message: "ok"}, nil |
| } |
|
|
| func (s *stubAdminService) CheckProxyQuality(ctx context.Context, id int64) (*service.ProxyQualityCheckResult, error) { |
| return &service.ProxyQualityCheckResult{ |
| ProxyID: id, |
| Score: 95, |
| Grade: "A", |
| Summary: "通过 5 项，告警 0 项，失败 0 项，挑战 0 项", |
| PassedCount: 5, |
| WarnCount: 0, |
| FailedCount: 0, |
| ChallengeCount: 0, |
| CheckedAt: time.Now().Unix(), |
| Items: []service.ProxyQualityCheckItem{ |
| {Target: "base_connectivity", Status: "pass", Message: "ok"}, |
| {Target: "openai", Status: "pass", HTTPStatus: 401}, |
| {Target: "anthropic", Status: "pass", HTTPStatus: 401}, |
| {Target: "gemini", Status: "pass", HTTPStatus: 200}, |
| {Target: "sora", Status: "pass", HTTPStatus: 401}, |
| }, |
| }, nil |
| } |
|
|
| func (s *stubAdminService) ListRedeemCodes(ctx context.Context, page, pageSize int, codeType, status, search string) ([]service.RedeemCode, int64, error) { |
| return s.redeems, int64(len(s.redeems)), nil |
| } |
|
|
| func (s *stubAdminService) GetRedeemCode(ctx context.Context, id int64) (*service.RedeemCode, error) { |
| code := service.RedeemCode{ID: id, Code: "R-TEST", Status: service.StatusUnused} |
| return &code, nil |
| } |
|
|
| func (s *stubAdminService) GenerateRedeemCodes(ctx context.Context, input *service.GenerateRedeemCodesInput) ([]service.RedeemCode, error) { |
| return s.redeems, nil |
| } |
|
|
| func (s *stubAdminService) DeleteRedeemCode(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) BatchDeleteRedeemCodes(ctx context.Context, ids []int64) (int64, error) { |
| return int64(len(ids)), nil |
| } |
|
|
| func (s *stubAdminService) ExpireRedeemCode(ctx context.Context, id int64) (*service.RedeemCode, error) { |
| code := service.RedeemCode{ID: id, Code: "R-TEST", Status: service.StatusUsed} |
| return &code, nil |
| } |
|
|
| func (s *stubAdminService) GetUserBalanceHistory(ctx context.Context, userID int64, page, pageSize int, codeType string) ([]service.RedeemCode, int64, float64, error) { |
| return s.redeems, int64(len(s.redeems)), 100.0, nil |
| } |
|
|
| func (s *stubAdminService) UpdateGroupSortOrders(ctx context.Context, updates []service.GroupSortOrderUpdate) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) AdminUpdateAPIKeyGroupID(ctx context.Context, keyID int64, groupID *int64) (*service.AdminUpdateAPIKeyGroupIDResult, error) { |
| for i := range s.apiKeys { |
| if s.apiKeys[i].ID == keyID { |
| k := s.apiKeys[i] |
| if groupID != nil { |
| if *groupID == 0 { |
| k.GroupID = nil |
| } else { |
| gid := *groupID |
| k.GroupID = &gid |
| } |
| } |
| return &service.AdminUpdateAPIKeyGroupIDResult{APIKey: &k}, nil |
| } |
| } |
| return nil, service.ErrAPIKeyNotFound |
| } |
|
|
| func (s *stubAdminService) ResetAccountQuota(ctx context.Context, id int64) error { |
| return nil |
| } |
|
|
| func (s *stubAdminService) EnsureOpenAIPrivacy(ctx context.Context, account *service.Account) string { |
| return "" |
| } |
|
|
| func (s *stubAdminService) ReplaceUserGroup(ctx context.Context, userID, oldGroupID, newGroupID int64) (*service.ReplaceUserGroupResult, error) { |
| return &service.ReplaceUserGroupResult{MigratedKeys: 0}, nil |
| } |
|
|
| |
| var _ service.AdminService = (*stubAdminService)(nil) |
|
|
