Add projects

2026-08-24 00:10:41 +02:00 · 2026-08-24 00:10:41 +02:00 · 8b607dd700
commit 8b607dd700
parent 2d3a9ad623
1802 changed files with 503346 additions and 2 deletions
--- a/backend/internal/api/handlers/management/config_apikey_disable_test.go
+++ b/backend/internal/api/handlers/management/config_apikey_disable_test.go
@ -0,0 +1,162 @@
+package management
+
+import (
+	"testing"
+
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
+	"github.com/router-for-me/CLIProxyAPI/v7/internal/watcher/synthesizer"
+	coreauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"
+)
+
+func TestSetConfigAPIKeyExcludedAll(t *testing.T) {
+	gotDisable := setConfigAPIKeyExcludedAll([]string{"gpt-5"}, true)
+	if len(gotDisable) != 2 || gotDisable[0] != "gpt-5" || gotDisable[1] != "*" {
+		t.Fatalf("unexpected disable list: %#v", gotDisable)
+	}
+	gotEnable := setConfigAPIKeyExcludedAll([]string{"gpt-5", "*"}, false)
+	if len(gotEnable) != 1 || gotEnable[0] != "gpt-5" {
+		t.Fatalf("unexpected enable list: %#v", gotEnable)
+	}
+}
+
+func TestToggleConfigAPIKeyExcludedAll_XAI(t *testing.T) {
+	cfg := &config.Config{
+		XAIKey: []config.XAIKey{{
+			APIKey:  "xai-test",
+			BaseURL: "https://api.x.ai/v1",
+		}},
+	}
+	idGen := synthesizer.NewStableIDGenerator()
+	authID, _ := idGen.Next("xai:apikey", "xai-test", "https://api.x.ai/v1", "", "", "")
+	auth := &coreauth.Auth{
+		ID:       authID,
+		Provider: "xai",
+		Attributes: map[string]string{
+			"api_key":  "xai-test",
+			"base_url": "https://api.x.ai/v1",
+			"source":   "config:xai[abc]",
+		},
+	}
+
+	handled, errToggle := toggleConfigAPIKeyExcludedAll(cfg, auth, true)
+	if errToggle != nil || !handled {
+		t.Fatalf("toggle disable: handled=%v err=%v", handled, errToggle)
+	}
+	if len(cfg.XAIKey[0].ExcludedModels) != 1 || cfg.XAIKey[0].ExcludedModels[0] != "*" {
+		t.Fatalf("excluded-models = %#v, want [*]", cfg.XAIKey[0].ExcludedModels)
+	}
+}
+
+func TestToggleConfigAPIKeyExcludedAll_Codex(t *testing.T) {
+	cfg := &config.Config{
+		CodexKey: []config.CodexKey{{
+			APIKey:  "sk-test",
+			BaseURL: "https://example.com/v1",
+		}},
+	}
+	idGen := synthesizer.NewStableIDGenerator()
+	authID, _ := idGen.Next("codex:apikey", "sk-test", "https://example.com/v1", "", "", "")
+	auth := &coreauth.Auth{
+		ID:       authID,
+		Provider: "codex",
+		Attributes: map[string]string{
+			"api_key":  "sk-test",
+			"base_url": "https://example.com/v1",
+			"source":   "config:codex[abc]",
+		},
+	}
+
+	handled, err := toggleConfigAPIKeyExcludedAll(cfg, auth, true)
+	if err != nil || !handled {
+		t.Fatalf("toggle disable: handled=%v err=%v", handled, err)
+	}
+	if len(cfg.CodexKey[0].ExcludedModels) != 1 || cfg.CodexKey[0].ExcludedModels[0] != "*" {
+		t.Fatalf("expected excluded-models [*], got %#v", cfg.CodexKey[0].ExcludedModels)
+	}
+
+	handled, err = toggleConfigAPIKeyExcludedAll(cfg, auth, false)
+	if err != nil || !handled {
+		t.Fatalf("toggle enable: handled=%v err=%v", handled, err)
+	}
+	if len(cfg.CodexKey[0].ExcludedModels) != 0 {
+		t.Fatalf("expected excluded-models cleared, got %#v", cfg.CodexKey[0].ExcludedModels)
+	}
+}
+
+func TestToggleConfigAPIKeyExcludedAll_Vertex_NoBaseURL(t *testing.T) {
+	cfg := &config.Config{
+		VertexCompatAPIKey: []config.VertexCompatKey{{
+			APIKey: "vertex-key-only",
+		}},
+	}
+	idGen := synthesizer.NewStableIDGenerator()
+	authID, _ := idGen.Next("vertex:apikey", "vertex-key-only", "", "")
+	auth := &coreauth.Auth{
+		ID:       authID,
+		Provider: "vertex",
+		Attributes: map[string]string{
+			"auth_kind": "apikey",
+			"api_key":   "vertex-key-only",
+			"source":    "config:vertex[xyz]",
+		},
+	}
+
+	handled, errToggle := toggleConfigAPIKeyExcludedAll(cfg, auth, true)
+	if errToggle != nil || !handled {
+		t.Fatalf("toggle disable: handled=%v err=%v", handled, errToggle)
+	}
+	if len(cfg.VertexCompatAPIKey[0].ExcludedModels) != 1 || cfg.VertexCompatAPIKey[0].ExcludedModels[0] != "*" {
+		t.Fatalf("excluded-models = %#v, want [*]", cfg.VertexCompatAPIKey[0].ExcludedModels)
+	}
+}
+
+func TestToggleConfigAPIKeyExcludedAll_EmptyKeyWithBaseURL(t *testing.T) {
+	cfg := &config.Config{
+		ClaudeKey: []config.ClaudeKey{{
+			APIKey:  "",
+			BaseURL: "https://custom-claude.example.com",
+		}},
+		GeminiKey: []config.GeminiKey{{
+			APIKey:  "   ",
+			BaseURL: "https://custom-gemini.example.com",
+		}},
+	}
+	idGen := synthesizer.NewStableIDGenerator()
+	claudeID, _ := idGen.Next("claude:apikey", "", "https://custom-claude.example.com", "", "", "")
+	geminiID, _ := idGen.Next("gemini:apikey", "", "https://custom-gemini.example.com", "", "", "")
+
+	claudeAuth := &coreauth.Auth{
+		ID:       claudeID,
+		Provider: "claude",
+		Attributes: map[string]string{
+			"auth_kind": "apikey",
+			"base_url":  "https://custom-claude.example.com",
+			"source":    "config:claude[abc]",
+		},
+	}
+	geminiAuth := &coreauth.Auth{
+		ID:       geminiID,
+		Provider: "gemini",
+		Attributes: map[string]string{
+			"auth_kind": "apikey",
+			"base_url":  "https://custom-gemini.example.com",
+			"source":    "config:gemini[def]",
+		},
+	}
+
+	handled, err := toggleConfigAPIKeyExcludedAll(cfg, claudeAuth, true)
+	if err != nil || !handled {
+		t.Fatalf("toggle claude: handled=%v err=%v", handled, err)
+	}
+	if len(cfg.ClaudeKey[0].ExcludedModels) != 1 || cfg.ClaudeKey[0].ExcludedModels[0] != "*" {
+		t.Fatalf("claude excluded-models = %#v, want [*]", cfg.ClaudeKey[0].ExcludedModels)
+	}
+
+	handled, err = toggleConfigAPIKeyExcludedAll(cfg, geminiAuth, true)
+	if err != nil || !handled {
+		t.Fatalf("toggle gemini: handled=%v err=%v", handled, err)
+	}
+	if len(cfg.GeminiKey[0].ExcludedModels) != 1 || cfg.GeminiKey[0].ExcludedModels[0] != "*" {
+		t.Fatalf("gemini excluded-models = %#v, want [*]", cfg.GeminiKey[0].ExcludedModels)
+	}
+}