package tray

import (
	"testing"
	"time"
)

type fakeMenuItem struct {
	clicked chan struct{}
	title   string
	tooltip string
}

func (i *fakeMenuItem) Clicked() <-chan struct{} {
	return i.clicked
}

func (i *fakeMenuItem) SetTitle(title string) {
	i.title = title
}

func (i *fakeMenuItem) SetTooltip(tooltip string) {
	i.tooltip = tooltip
}

type fakeBackend struct {
	icon        []byte
	tooltip     string
	items       map[string]*fakeMenuItem
	quitCalls   int
	registering bool
}

func (b *fakeBackend) Register(onReady func(), _ func()) {
	b.registering = true
	onReady()
}

func (b *fakeBackend) SetIcon(icon []byte) {
	b.icon = append([]byte(nil), icon...)
}

func (b *fakeBackend) SetTooltip(tooltip string) {
	b.tooltip = tooltip
}

func (b *fakeBackend) AddMenuItem(title, _ string) MenuItem {
	item := &fakeMenuItem{clicked: make(chan struct{}, 1), title: title}
	b.items[title] = item
	return item
}

func (b *fakeBackend) Quit() {
	b.quitCalls++
}

func waitFor(t *testing.T, signal <-chan struct{}) {
	t.Helper()
	select {
	case <-signal:
	case <-time.After(time.Second):
		t.Fatal("timed out waiting for tray action")
	}
}

func TestControllerInitializesTrayAndRoutesMenuActions(t *testing.T) {
	backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
	showCalls := make(chan struct{}, 1)
	quitCalls := make(chan struct{}, 1)
	controller := New(backend, []byte{1, 2, 3})

	controller.Start(Actions{
		Show: func() { showCalls <- struct{}{} },
		Quit: func() { quitCalls <- struct{}{} },
	})

	if !backend.registering || string(backend.icon) != string([]byte{1, 2, 3}) || backend.tooltip != "Verstak" {
		t.Fatalf("tray initialization = registering:%t icon:%v tooltip:%q", backend.registering, backend.icon, backend.tooltip)
	}
	showItem := backend.items["Show Verstak"]
	quitItem := backend.items["Quit"]
	if showItem == nil || quitItem == nil {
		t.Fatalf("tray menu = %#v, want Show Verstak and Quit", backend.items)
	}

	showItem.clicked <- struct{}{}
	waitFor(t, showCalls)
	quitItem.clicked <- struct{}{}
	waitFor(t, quitCalls)
}

func TestControllerStopsNativeTrayBackend(t *testing.T) {
	backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
	controller := New(backend, []byte{1})

	controller.Stop()

	if backend.quitCalls != 1 {
		t.Fatalf("backend quit calls = %d, want 1", backend.quitCalls)
	}
}

func TestControllerUsesAndUpdatesLocalizedLabels(t *testing.T) {
	backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
	controller := New(backend, []byte{1})
	controller.SetLabels(LabelsForPreference("ru"))
	controller.Start(Actions{})

	show := backend.items["Показать Верстак"]
	quit := backend.items["Выйти"]
	if show == nil || quit == nil {
		t.Fatalf("Russian tray menu = %#v", backend.items)
	}
	if show.tooltip != "Показать окно Верстака" || quit.tooltip != "Завершить Верстак" {
		t.Fatalf("Russian tray tooltips = show:%q quit:%q", show.tooltip, quit.tooltip)
	}

	controller.SetLabels(LabelsForPreference("en"))
	if show.title != "Show Verstak" || quit.title != "Quit" {
		t.Fatalf("English tray menu after update = show:%q quit:%q", show.title, quit.title)
	}
	if show.tooltip != "Show the Verstak window" || quit.tooltip != "Quit Verstak" {
		t.Fatalf("English tray tooltips after update = show:%q quit:%q", show.tooltip, quit.tooltip)
	}
}

func TestLabelsForSystemRussianLocale(t *testing.T) {
	labels := LabelsForPreference("system", "", "ru_RU.UTF-8", "en_US.UTF-8")
	if labels.ShowTitle != "Показать Верстак" || labels.QuitTitle != "Выйти" {
		t.Fatalf("system Russian labels = %#v", labels)
	}
	labels = LabelsForPreference("system", "C.UTF-8", "", "ru_RU.UTF-8")
	if labels.ShowTitle != "Show Verstak" || labels.QuitTitle != "Quit" {
		t.Fatalf("higher-priority system locale must win, got %#v", labels)
	}
}