package channel

import (
	"opencatd-open/internal/store"
	"testing"
)

func TestChannelFormatsEffective(t *testing.T) {
	tests := []struct {
		name     string
		channel  store.Channel
		expected []string
	}{
		{
			name: "anthropic default",
			channel: store.Channel{
				Provider: store.ChannelProviderAnthropic,
			},
			expected: []string{store.FormatMessages},
		},
		{
			name: "openai default",
			channel: store.Channel{
				Provider: store.ChannelProviderOpenAI,
			},
			expected: []string{store.FormatChat, store.FormatResponses},
		},
		{
			name: "compatible default",
			channel: store.Channel{
				Provider: store.ChannelProviderCompatible,
			},
			expected: []string{store.FormatChat},
		},
		{
			name: "custom formats override",
			channel: store.Channel{
				Provider: store.ChannelProviderOpenAI,
				Formats:  []string{store.FormatChat},
			},
			expected: []string{store.FormatChat},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := tt.channel.FormatsEffective()
			if len(result) != len(tt.expected) {
				t.Errorf("FormatsEffective() returned %d formats, want %d", len(result), len(tt.expected))
				return
			}
			for i, f := range result {
				if f != tt.expected[i] {
					t.Errorf("FormatsEffective()[%d] = %q, want %q", i, f, tt.expected[i])
				}
			}
		})
	}
}

func TestChannelUpstreamURL(t *testing.T) {
	tests := []struct {
		name     string
		channel  store.Channel
		proto    string
		path     string
		expected string
	}{
		{
			name: "basic openai",
			channel: store.Channel{
				BaseURL: "https://api.openai.com",
			},
			proto:    "chat",
			path:     "/chat/completions",
			expected: "https://api.openai.com/v1/chat/completions",
		},
		{
			name: "with trailing slash",
			channel: store.Channel{
				BaseURL: "https://api.openai.com/",
			},
			proto:    "chat",
			path:     "/chat/completions",
			expected: "https://api.openai.com/v1/chat/completions",
		},
		{
			name: "with version segment",
			channel: store.Channel{
				BaseURL: "https://api.openai.com/v1",
			},
			proto:    "chat",
			path:     "/chat/completions",
			expected: "https://api.openai.com/v1/chat/completions",
		},
		{
			name: "custom base URL per protocol",
			channel: store.Channel{
				BaseURL:  "https://default.openai.com",
				BaseURLs: map[string]string{"chat": "https://chat.openai.com"},
			},
			proto:    "chat",
			path:     "/chat/completions",
			expected: "https://chat.openai.com/v1/chat/completions",
		},
		{
			name: "empty base",
			channel: store.Channel{
				BaseURL: "",
			},
			proto:    "chat",
			path:     "/chat/completions",
			expected: "/chat/completions",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := tt.channel.UpstreamURL(tt.proto, tt.path)
			if result != tt.expected {
				t.Errorf("UpstreamURL() = %q, want %q", result, tt.expected)
			}
		})
	}
}