mirror of
https://github.com/CJackHwang/ds2api.git
synced 2026-05-05 00:45:29 +08:00
124 lines
3.7 KiB
Go
124 lines
3.7 KiB
Go
package admin
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"ds2api/internal/account"
|
|
"ds2api/internal/auth"
|
|
"ds2api/internal/config"
|
|
dsclient "ds2api/internal/deepseek/client"
|
|
adminaccounts "ds2api/internal/httpapi/admin/accounts"
|
|
adminconfig "ds2api/internal/httpapi/admin/configmgmt"
|
|
adminsettings "ds2api/internal/httpapi/admin/settings"
|
|
adminshared "ds2api/internal/httpapi/admin/shared"
|
|
)
|
|
|
|
var intFrom = adminshared.IntFrom
|
|
|
|
func toAccount(m map[string]any) config.Account { return adminshared.ToAccount(m) }
|
|
func fieldString(m map[string]any, key string) string {
|
|
return adminshared.FieldString(m, key)
|
|
}
|
|
func maskSecretPreview(secret string) string { return adminshared.MaskSecretPreview(secret) }
|
|
func boolFrom(v any) bool { return adminsettings.BoolFrom(v) }
|
|
|
|
func newAdminTestHandler(t *testing.T, raw string) *Handler {
|
|
t.Helper()
|
|
t.Setenv("DS2API_CONFIG_JSON", raw)
|
|
store := config.LoadStore()
|
|
return &Handler{
|
|
Store: store,
|
|
Pool: account.NewPool(store),
|
|
}
|
|
}
|
|
|
|
type testingDSMock struct {
|
|
loginToken string
|
|
deleteAllSessionsError error
|
|
deleteAllSessionsErrorOnce bool
|
|
sessionCount *dsclient.SessionStats
|
|
loginCalls int
|
|
deleteAllCalls int
|
|
}
|
|
|
|
func (m *testingDSMock) Login(_ context.Context, _ config.Account) (string, error) {
|
|
m.loginCalls++
|
|
if m.loginToken == "" {
|
|
return "token", nil
|
|
}
|
|
return m.loginToken, nil
|
|
}
|
|
|
|
func (m *testingDSMock) CreateSession(_ context.Context, _ *auth.RequestAuth, _ int) (string, error) {
|
|
return "session-id", nil
|
|
}
|
|
|
|
func (m *testingDSMock) GetPow(_ context.Context, _ *auth.RequestAuth, _ int) (string, error) {
|
|
return "pow", nil
|
|
}
|
|
|
|
func (m *testingDSMock) CallCompletion(_ context.Context, _ *auth.RequestAuth, _ map[string]any, _ string, _ int) (*http.Response, error) {
|
|
return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody}, nil
|
|
}
|
|
|
|
func (m *testingDSMock) DeleteAllSessionsForToken(_ context.Context, _ string) error {
|
|
m.deleteAllCalls++
|
|
if m.deleteAllSessionsError != nil {
|
|
err := m.deleteAllSessionsError
|
|
if m.deleteAllSessionsErrorOnce {
|
|
m.deleteAllSessionsError = nil
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *testingDSMock) GetSessionCountForToken(_ context.Context, _ string) (*dsclient.SessionStats, error) {
|
|
if m.sessionCount != nil {
|
|
return m.sessionCount, nil
|
|
}
|
|
return &dsclient.SessionStats{}, nil
|
|
}
|
|
|
|
func (h *Handler) configHandler() *adminconfig.Handler {
|
|
return &adminconfig.Handler{Store: h.Store, Pool: h.Pool, DS: h.DS, OpenAI: h.OpenAI, ChatHistory: h.ChatHistory}
|
|
}
|
|
|
|
func (h *Handler) settingsHandler() *adminsettings.Handler {
|
|
return &adminsettings.Handler{Store: h.Store, Pool: h.Pool, DS: h.DS, OpenAI: h.OpenAI, ChatHistory: h.ChatHistory}
|
|
}
|
|
|
|
func (h *Handler) getConfig(w http.ResponseWriter, r *http.Request) {
|
|
h.configHandler().GetConfig(w, r)
|
|
}
|
|
|
|
func (h *Handler) updateConfig(w http.ResponseWriter, r *http.Request) {
|
|
h.configHandler().UpdateConfig(w, r)
|
|
}
|
|
|
|
func (h *Handler) configImport(w http.ResponseWriter, r *http.Request) {
|
|
h.configHandler().ConfigImport(w, r)
|
|
}
|
|
|
|
func (h *Handler) batchImport(w http.ResponseWriter, r *http.Request) {
|
|
h.configHandler().BatchImport(w, r)
|
|
}
|
|
|
|
func (h *Handler) getSettings(w http.ResponseWriter, r *http.Request) {
|
|
h.settingsHandler().GetSettings(w, r)
|
|
}
|
|
|
|
func (h *Handler) updateSettings(w http.ResponseWriter, r *http.Request) {
|
|
h.settingsHandler().UpdateSettings(w, r)
|
|
}
|
|
|
|
func (h *Handler) updateSettingsPassword(w http.ResponseWriter, r *http.Request) {
|
|
h.settingsHandler().UpdateSettingsPassword(w, r)
|
|
}
|
|
|
|
func runAccountTestsConcurrently(accounts []config.Account, maxConcurrency int, testFn func(int, config.Account) map[string]any) []map[string]any {
|
|
return adminaccounts.RunAccountTestsConcurrently(accounts, maxConcurrency, testFn)
|
|
}
|