package llmutils

import (
	"os"
	"testing"

	"codeberg.org/snonux/hexai/internal/appconfig"
)

func TestNewClientFromApp_Ollama(t *testing.T) {
	cfg := appconfig.App{Provider: "ollama"}
	c, err := NewClientFromApp(cfg)
	if err != nil || c == nil {
		t.Fatalf("ollama client failed: %v %v", c, err)
	}
}

func TestNewClientFromApp_OpenAI_WithKey(t *testing.T) {
	t.Setenv("HEXAI_OPENAI_API_KEY", "test-key")
	cfg := appconfig.App{Provider: "openai"}
	c, err := NewClientFromApp(cfg)
	if err != nil || c == nil {
		t.Fatalf("openai client failed: %v %v", c, err)
	}
	// ensure env override precedence
	_ = os.Unsetenv("OPENAI_API_KEY")
}

func TestCanonicalProvider(t *testing.T) {
	if got := CanonicalProvider("  OpenRouter "); got != "openrouter" {
		t.Fatalf("CanonicalProvider(openrouter) = %q", got)
	}
	if got := CanonicalProvider(" "); got != "openai" {
		t.Fatalf("CanonicalProvider(empty) = %q", got)
	}
}

func TestDefaultModelForProvider(t *testing.T) {
	cfg := appconfig.App{
		OpenAIModel:     "gpt-4.1",
		OpenRouterModel: "openrouter/auto",
		OllamaModel:     "qwen3",
		AnthropicModel:  "claude",
	}
	if got := DefaultModelForProvider(cfg, "openai"); got != "gpt-4.1" {
		t.Fatalf("openai model = %q", got)
	}
	if got := DefaultModelForProvider(cfg, "openrouter"); got != "openrouter/auto" {
		t.Fatalf("openrouter model = %q", got)
	}
	if got := DefaultModelForProvider(cfg, "ollama"); got != "qwen3" {
		t.Fatalf("ollama model = %q", got)
	}
	if got := DefaultModelForProvider(cfg, "anthropic"); got != "claude" {
		t.Fatalf("anthropic model = %q", got)
	}
}

func TestConfigForProvider(t *testing.T) {
	base := appconfig.App{
		Provider:       "openai",
		OpenAIModel:    "gpt-4.1",
		OllamaModel:    "qwen3",
		AnthropicModel: "claude",
	}
	got := ConfigForProvider(base, "ollama", "qwen3-coder")
	if got.Provider != "ollama" {
		t.Fatalf("provider = %q", got.Provider)
	}
	if got.OllamaModel != "qwen3-coder" {
		t.Fatalf("ollama model = %q", got.OllamaModel)
	}
	if got.OpenAIModel != "gpt-4.1" {
		t.Fatalf("openai model unexpectedly changed: %q", got.OpenAIModel)
	}
}