package handlers_test

import (
	"slices"
	"testing"

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

func TestNewHelpHandler(t *testing.T) {
	h := handlers.NewHelpHandler()

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

	keys := h.KeysHandled()

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

func TestHelpHandler_Match(t *testing.T) {
	h := handlers.NewHelpHandler()

	tests := []struct {
		name     string
		mode     state.Mode
		key      string
		expected bool
	}{
		{
			name:     "Matches '?' in Navigation Mode",
			mode:     state.NavigationMode,
			key:      "?",
			expected: true,
		},
		{
			name:     "Matches '?' in Help Mode",
			mode:     state.HelpMode,
			key:      "?",
			expected: true,
		},
		{
			name:     "Ignores '?' in Input Mode",
			mode:     state.InputMode,
			key:      "?",
			expected: false,
		},
		{
			name:     "Ignores other keys in Navigation Mode",
			mode:     state.NavigationMode,
			key:      "q",
			expected: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			s := state.New(state.WithMode(tt.mode))
			msg := tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(tt.key)}

			if got := h.Match(s, msg); got != tt.expected {
				t.Errorf("Match() = %v, want %v", got, tt.expected)
			}
		})
	}
}

func TestHelpHandler_Handle(t *testing.T) {
	h := handlers.NewHelpHandler()

	t.Run("Toggles from Navigation to Help", func(t *testing.T) {
		s := state.New(state.WithMode(state.NavigationMode))
		msg := tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("?")}

		cmd := h.Handle(s, msg)

		if s.Mode() != state.HelpMode {
			t.Errorf("Expected mode to switch to HelpMode, got %v", s.Mode())
		}

		if cmd != nil {
			t.Error("Expected nil command, got non-nil")
		}
	})

	t.Run("Toggles from Help to Navigation", func(t *testing.T) {
		s := state.New(state.WithMode(state.HelpMode))
		msg := tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("?")}

		h.Handle(s, msg)

		if s.Mode() != state.NavigationMode {
			t.Errorf("Expected mode to switch to NavigationMode, got %v", s.Mode())
		}
	})
}