package atlascloud

import (
	"context"
	"testing"

	"go-micro.dev/v5/ai"
)

func TestProvider_String(t *testing.T) {
	p := NewProvider()
	if p.String() != "atlascloud" {
		t.Errorf("Expected provider name 'atlascloud', got '%s'", p.String())
	}
}

func TestProvider_Init(t *testing.T) {
	p := NewProvider()

	err := p.Init(
		ai.WithModel("test-model"),
		ai.WithAPIKey("test-key"),
		ai.WithBaseURL("https://test.com"),
	)

	if err != nil {
		t.Fatalf("Init failed: %v", err)
	}

	opts := p.Options()
	if opts.Model != "test-model" {
		t.Errorf("Expected model 'test-model', got '%s'", opts.Model)
	}
	if opts.APIKey != "test-key" {
		t.Errorf("Expected API key 'test-key', got '%s'", opts.APIKey)
	}
	if opts.BaseURL != "https://test.com" {
		t.Errorf("Expected base URL 'https://test.com', got '%s'", opts.BaseURL)
	}
}

func TestProvider_Options(t *testing.T) {
	p := NewProvider(
		ai.WithModel("custom-model"),
		ai.WithAPIKey("my-key"),
	)

	opts := p.Options()
	if opts.Model != "custom-model" {
		t.Errorf("Expected model 'custom-model', got '%s'", opts.Model)
	}
	if opts.APIKey != "my-key" {
		t.Errorf("Expected API key 'my-key', got '%s'", opts.APIKey)
	}
}

func TestProvider_Defaults(t *testing.T) {
	p := NewProvider()

	opts := p.Options()
	if opts.Model != "deepseek-ai/DeepSeek-V3-0324" {
		t.Errorf("Expected default model 'deepseek-ai/DeepSeek-V3-0324', got '%s'", opts.Model)
	}
	if opts.BaseURL != "https://api.atlascloud.ai" {
		t.Errorf("Expected default base URL 'https://api.atlascloud.ai', got '%s'", opts.BaseURL)
	}
}

func TestProvider_Generate_NoAPIKey(t *testing.T) {
	p := NewProvider()

	req := &ai.Request{
		Prompt:       "Hello",
		SystemPrompt: "You are helpful",
	}

	_, err := p.Generate(context.Background(), req)
	if err == nil {
		t.Error("Expected error when API key is missing, got nil")
	}
}

func TestProvider_Stream_NotImplemented(t *testing.T) {
	p := NewProvider()

	req := &ai.Request{
		Prompt: "Hello",
	}

	_, err := p.Stream(context.Background(), req)
	if err == nil {
		t.Error("Expected error for unimplemented streaming, got nil")
	}
}

func TestProvider_Registration(t *testing.T) {
	m := ai.New("atlascloud", ai.WithAPIKey("test"))
	if m == nil {
		t.Fatal("ai.New('atlascloud') returned nil — provider not registered")
	}
	if m.String() != "atlascloud" {
		t.Errorf("Expected 'atlascloud', got '%s'", m.String())
	}
}

func TestProvider_ImageRegistration(t *testing.T) {
	ig := ai.NewImage("atlascloud", ai.WithAPIKey("test"))
	if ig == nil {
		t.Fatal("ai.NewImage('atlascloud') returned nil — image provider not registered")
	}
	if ig.String() != "atlascloud" {
		t.Errorf("Expected 'atlascloud', got '%s'", ig.String())
	}
}

func TestProvider_GenerateImage_NoAPIKey(t *testing.T) {
	p := NewProvider()
	_, err := p.GenerateImage(context.Background(), &ai.ImageRequest{Prompt: "a cat"})
	if err == nil {
		t.Error("Expected error when API key is missing, got nil")
	}
}

func TestProvider_ImplementsImageModel(t *testing.T) {
	var _ ai.ImageModel = (*Provider)(nil)
}

func TestProvider_VideoRegistration(t *testing.T) {
	vg := ai.NewVideo("atlascloud", ai.WithAPIKey("test"))
	if vg == nil {
		t.Fatal("ai.NewVideo('atlascloud') returned nil — video provider not registered")
	}
	if vg.String() != "atlascloud" {
		t.Errorf("Expected 'atlascloud', got '%s'", vg.String())
	}
}

func TestProvider_GenerateVideo_NoAPIKey(t *testing.T) {
	p := NewProvider()
	_, err := p.GenerateVideo(context.Background(), &ai.VideoRequest{Prompt: "a cat"})
	if err == nil {
		t.Error("Expected error when API key is missing, got nil")
	}
}

func TestProvider_ImplementsVideoModel(t *testing.T) {
	var _ ai.VideoModel = (*Provider)(nil)
}
