package services

import (
	"testing"

	"voicesnap/internal/config"
	"voicesnap/internal/model"
)

func TestEngineServiceDownloadStatusTracksCancelAndFinish(t *testing.T) {
	service := NewEngineService(nil)
	status := service.GetModelDownloadStatus()
	if status["active"].(bool) {
		t.Fatal("new service should not report an active model download")
	}

	_, finish, err := service.beginModelDownload("zipformer-ko")
	if err != nil {
		t.Fatal(err)
	}
	status = service.GetModelDownloadStatus()
	if !status["active"].(bool) {
		t.Fatal("active download was not reported")
	}
	if got := status["modelID"]; got != "zipformer-ko" {
		t.Fatalf("modelID = %v, want zipformer-ko", got)
	}
	if status["cancelling"].(bool) {
		t.Fatal("new download should not start in cancelling state")
	}

	service.updateModelDownloadProgress("zipformer-ko", 42.5, 425, 1000)
	status = service.GetModelDownloadStatus()
	if got := status["percent"]; got != 42.5 {
		t.Fatalf("percent = %v, want 42.5", got)
	}
	if got := status["downloaded"]; got != int64(425) {
		t.Fatalf("downloaded = %v, want 425", got)
	}
	if got := status["total"]; got != int64(1000) {
		t.Fatalf("total = %v, want 1000", got)
	}

	if !service.CancelModelDownload("zipformer-ko") {
		t.Fatal("expected active download to be cancelled")
	}
	status = service.GetModelDownloadStatus()
	if !status["cancelling"].(bool) {
		t.Fatal("cancelled download should report cancelling state until finish")
	}

	finish()
	status = service.GetModelDownloadStatus()
	if status["active"].(bool) {
		t.Fatal("finished download should clear active state")
	}
	if got := status["modelID"]; got != "" {
		t.Fatalf("modelID after finish = %v, want empty", got)
	}
}

func TestListModelOptionsForCantoneseOnlyOffersSenseVoiceYue(t *testing.T) {
	cfg := config.Default()
	cfg.LanguageMode = config.LanguageModeManual
	cfg.LanguageID = model.CantoneseLanguageID
	cfg.ModelSelectionMode = config.ModelSelectionModeAuto

	service := NewEngineService(cfg)
	options := service.ListModelOptions()
	if len(options) != 1 {
		t.Fatalf("options len = %d, want 1: %#v", len(options), options)
	}
	if got := options[0]["modelID"]; got != model.SenseVoiceYueModelID {
		t.Fatalf("modelID = %v, want %s", got, model.SenseVoiceYueModelID)
	}
	if got := options[0]["isDefault"]; got != true {
		t.Fatalf("isDefault = %v, want true", got)
	}
}

func TestListModelOptionsForLanguagePreviewsCantoneseWithoutPersisting(t *testing.T) {
	cfg := config.Default()
	cfg.LanguageMode = config.LanguageModeManual
	cfg.LanguageID = model.DefaultLanguageID
	service := NewEngineService(cfg)

	options := service.ListModelOptionsForLanguage(model.CantoneseLanguageID)
	if len(options) != 1 {
		t.Fatalf("options len = %d, want 1: %#v", len(options), options)
	}
	if got := options[0]["modelID"]; got != model.SenseVoiceYueModelID {
		t.Fatalf("modelID = %v, want %s", got, model.SenseVoiceYueModelID)
	}
	if got := options[0]["languageID"]; got != model.CantoneseLanguageID {
		t.Fatalf("languageID = %v, want %s", got, model.CantoneseLanguageID)
	}
	if cfg.LanguageID != model.DefaultLanguageID {
		t.Fatalf("cfg.LanguageID = %q, want unchanged %q", cfg.LanguageID, model.DefaultLanguageID)
	}
}

func TestListModelOptionsForLanguageDoesNotMarkSharedCurrentModelForPendingLanguage(t *testing.T) {
	cfg := config.Default()
	cfg.LanguageMode = config.LanguageModeManual
	cfg.LanguageID = model.DefaultLanguageID
	cfg.ModelSelectionMode = config.ModelSelectionModeManual
	cfg.SelectedModelID = model.XASRZhEn960ModelID
	service := NewEngineService(cfg)

	options := service.ListModelOptionsForLanguage(model.EnglishLanguageID)
	var found bool
	for _, option := range options {
		if option["modelID"] != model.XASRZhEn960ModelID {
			continue
		}
		found = true
		if got := option["isCurrent"]; got != false {
			t.Fatalf("isCurrent = %v, want false for pending language shared model", got)
		}
	}
	if !found {
		t.Fatalf("expected %s in English options: %#v", model.XASRZhEn960ModelID, options)
	}
}

func TestAllowedModelProfileForLanguageAcceptsCantoneseWithoutPersisting(t *testing.T) {
	cfg := config.Default()
	cfg.LanguageMode = config.LanguageModeManual
	cfg.LanguageID = model.DefaultLanguageID
	service := NewEngineService(cfg)

	profile, languageProfile, err := service.allowedModelProfileForLanguage(model.CantoneseLanguageID, model.SenseVoiceYueModelID)
	if err != nil {
		t.Fatalf("allowedModelProfileForLanguage returned error: %v", err)
	}
	if profile.ID != model.SenseVoiceYueModelID {
		t.Fatalf("profile id = %q, want %q", profile.ID, model.SenseVoiceYueModelID)
	}
	if languageProfile.ID != model.CantoneseLanguageID {
		t.Fatalf("language profile id = %q, want %q", languageProfile.ID, model.CantoneseLanguageID)
	}
	if _, err := service.allowedModelProfile(model.SenseVoiceYueModelID); err == nil {
		t.Fatal("expected current Chinese language to reject SenseVoice Yue")
	}
}

func TestAllowedModelProfileForLanguageAcceptsTargetLanguageUpgradeWithoutPersisting(t *testing.T) {
	cfg := config.Default()
	cfg.LanguageMode = config.LanguageModeManual
	cfg.LanguageID = model.DefaultLanguageID
	service := NewEngineService(cfg)

	profile, languageProfile, err := service.allowedModelProfileForLanguage(model.EnglishLanguageID, model.ParakeetModelID)
	if err != nil {
		t.Fatalf("allowedModelProfileForLanguage returned error: %v", err)
	}
	if profile.ID != model.ParakeetModelID {
		t.Fatalf("profile id = %q, want %q", profile.ID, model.ParakeetModelID)
	}
	if languageProfile.ID != model.EnglishLanguageID {
		t.Fatalf("language profile id = %q, want %q", languageProfile.ID, model.EnglishLanguageID)
	}
	if _, err := service.allowedModelProfile(model.ParakeetModelID); err == nil {
		t.Fatal("expected current Chinese language to reject Parakeet")
	}
}