package handlers_test

import (
	"slices"
	"testing"

	"github.com/StevanFreeborn/term-tier/internal/core"
	"github.com/StevanFreeborn/term-tier/internal/handlers"
	"github.com/StevanFreeborn/term-tier/internal/state"
	tea "github.com/charmbracelet/bubbletea"
)

func TestNewSaveHandler(t *testing.T) {
	saver := &mockSaver{}

	h := handlers.NewSaveHandler(saver)

	if h.Description() != "Save tiers" {
		t.Errorf("Expected description 'Save tiers', got %q", h.Description())
	}

	keys := h.KeysHandled()

	if !slices.Contains(keys, "ctrl+s") {
		t.Errorf("Expected keys ['ctrl+s'], got %v", keys)
	}
}

func TestSaveHandler_Match(t *testing.T) {
	saver := &mockSaver{}
	h := handlers.NewSaveHandler(saver)

	tests := []struct {
		name     string
		mode     state.Mode
		msg      tea.KeyMsg
		expected bool
	}{
		{
			name:     "Matches ctrl+s in NavigationMode",
			mode:     state.NavigationMode,
			msg:      tea.KeyMsg{Type: tea.KeyCtrlS},
			expected: true,
		},
		{
			name:     "Ignores ctrl+s in InputMode",
			mode:     state.InputMode,
			msg:      tea.KeyMsg{Type: tea.KeyCtrlS},
			expected: false,
		},
		{
			name:     "Ignores other keys",
			mode:     state.NavigationMode,
			msg:      tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("s")},
			expected: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s := state.New(state.WithMode(tt.mode))
			if got := h.Match(s, tt.msg); got != tt.expected {
				t.Errorf("Match() = %v, want %v", got, tt.expected)
			}
		})
	}
}

func TestSaveHandler_Handle(t *testing.T) {
	t.Run("Successfully saves data", func(t *testing.T) {
		saver := &mockSaver{}

		h := handlers.NewSaveHandler(saver)

		tiers := []core.Tier{{Name: "S Tier", Items: []core.Item{"Godzilla"}}}

		s := state.New(
			state.WithMode(state.NavigationMode),
			state.WithTiers(tiers),
		)

		msg := tea.KeyMsg{Type: tea.KeyCtrlS}
		cmd := h.Handle(s, msg)

		if s.Mode() != state.InputMode {
			t.Fatal("Expected state to switch to InputMode")
		}

		if cmd == nil {
			t.Error("Expected command from Handle (blink)")
		}

		filename := "my_list.json"

		for _, r := range filename {
			s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
		}

		s.SubmitInput()

		if len(saver.savedData.Tiers) != 1 {
			t.Error("Expected saver to receive 1 tier")
		}

		expectedStatusMsg := "Saved to " + filename

		if s.StatusMsg() != expectedStatusMsg {
			t.Errorf("Expected status message %q, got %q", expectedStatusMsg, s.StatusMsg())
		}
	})

	t.Run("Handles save errors", func(t *testing.T) {
		saver := &mockSaver{shouldError: true}

		h := handlers.NewSaveHandler(saver)
		s := state.New(state.WithMode(state.NavigationMode))

		h.Handle(s, tea.KeyMsg{Type: tea.KeyCtrlS})

		for _, r := range "bad_perms.json" {
			s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
		}

		s.SubmitInput()

		expectedPrefix := "Error:"

		if len(s.StatusMsg()) < len(expectedPrefix) || s.StatusMsg()[:len(expectedPrefix)] != expectedPrefix {
			t.Errorf("Expected error status, got %q", s.StatusMsg())
		}
	})
}