package input

import (
	"testing"

	"github.com/jwijenbergh/puregotk/v4/gdk"
)

func TestParseKeyString_SingleKeys(t *testing.T) {
	tests := []struct {
		name   string
		input  string
		want   KeyBinding
		wantOk bool
	}{
		{
			name:   "escape",
			input:  "escape",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Escape), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "esc alias",
			input:  "esc",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Escape), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "return",
			input:  "return",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Return), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "enter alias",
			input:  "enter",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Return), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "tab",
			input:  "tab",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Tab), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "space",
			input:  "space",
			want:   KeyBinding{Keyval: uint(gdk.KEY_space), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "f5",
			input:  "f5",
			want:   KeyBinding{Keyval: uint(gdk.KEY_F5), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "f12",
			input:  "f12",
			want:   KeyBinding{Keyval: uint(gdk.KEY_F12), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "single letter",
			input:  "t",
			want:   KeyBinding{Keyval: uint('t'), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "single number",
			input:  "0",
			want:   KeyBinding{Keyval: uint(gdk.KEY_0), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "arrow left",
			input:  "left",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Left), Modifiers: ModNone},
			wantOk: true,
		},
		{
			name:   "arrow right",
			input:  "right",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Right), Modifiers: ModNone},
			wantOk: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, ok := ParseKeyString(tt.input)
			if ok != tt.wantOk {
				t.Errorf("ParseKeyString(%q) ok = %v, want %v", tt.input, ok, tt.wantOk)
				return
			}
			if !tt.wantOk {
				return
			}
			if got.Keyval != tt.want.Keyval {
				t.Errorf("ParseKeyString(%q) keyval = %d, want %d", tt.input, got.Keyval, tt.want.Keyval)
			}
			if got.Modifiers != tt.want.Modifiers {
				t.Errorf("ParseKeyString(%q) modifiers = %d, want %d", tt.input, got.Modifiers, tt.want.Modifiers)
			}
		})
	}
}

func TestParseKeyString_WithModifiers(t *testing.T) {
	tests := []struct {
		name   string
		input  string
		want   KeyBinding
		wantOk bool
	}{
		{
			name:   "ctrl+t",
			input:  "ctrl+t",
			want:   KeyBinding{Keyval: uint('t'), Modifiers: ModCtrl},
			wantOk: true,
		},
		{
			name:   "control+t",
			input:  "control+t",
			want:   KeyBinding{Keyval: uint('t'), Modifiers: ModCtrl},
			wantOk: true,
		},
		{
			name:   "shift+tab",
			input:  "shift+tab",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Tab), Modifiers: ModShift},
			wantOk: true,
		},
		{
			name:   "alt+left",
			input:  "alt+left",
			want:   KeyBinding{Keyval: uint(gdk.KEY_Left), Modifiers: ModAlt},
			wantOk: true,
		},
		{
			name:   "ctrl+shift+t",
			input:  "ctrl+shift+t",
			want:   KeyBinding{Keyval: uint('t'), Modifiers: ModCtrl | ModShift},
			wantOk: true,
		},
		{
			name:   "ctrl+r",
			input:  "ctrl+r",
			want:   KeyBinding{Keyval: uint('r'), Modifiers: ModCtrl},
			wantOk: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, ok := ParseKeyString(tt.input)
			if ok != tt.wantOk {
				t.Errorf("ParseKeyString(%q) ok = %v, want %v", tt.input, ok, tt.wantOk)
				return
			}
			if !tt.wantOk {
				return
			}
			if got.Keyval != tt.want.Keyval {
				t.Errorf("ParseKeyString(%q) keyval = %d, want %d", tt.input, got.Keyval, tt.want.Keyval)
			}
			if got.Modifiers != tt.want.Modifiers {
				t.Errorf("ParseKeyString(%q) modifiers = %d, want %d", tt.input, got.Modifiers, tt.want.Modifiers)
			}
		})
	}
}

func TestParseKeyString_Invalid(t *testing.T) {
	tests := []string{
		"",
		"ctrl+",
		"unknownkey",
		"ctrl+unknownkey",
	}

	for _, input := range tests {
		t.Run(input, func(t *testing.T) {
			_, ok := ParseKeyString(input)
			if ok {
				t.Errorf("ParseKeyString(%q) should have failed", input)
			}
		})
	}
}

func TestParseKeyString_CaseInsensitive(t *testing.T) {
	tests := []struct {
		input1 string
		input2 string
	}{
		{"ctrl+t", "CTRL+T"},
		{"Escape", "escape"},
		{"Tab", "TAB"},
		{"Ctrl+Shift+T", "ctrl+shift+t"},
	}

	for _, tt := range tests {
		t.Run(tt.input1+" vs "+tt.input2, func(t *testing.T) {
			got1, ok1 := ParseKeyString(tt.input1)
			got2, ok2 := ParseKeyString(tt.input2)

			if ok1 != ok2 {
				t.Errorf("ParseKeyString case sensitivity: %q=%v, %q=%v", tt.input1, ok1, tt.input2, ok2)
				return
			}

			if got1 != got2 {
				t.Errorf("ParseKeyString case sensitivity: %q=%v, %q=%v", tt.input1, got1, tt.input2, got2)
			}
		})
	}
}

func TestShortcutSet_Lookup(t *testing.T) {
	set := &ShortcutSet{
		Global: ShortcutTable{
			{uint(gdk.KEY_q), ModCtrl}: ActionQuit,
			{uint(gdk.KEY_l), ModCtrl}: ActionOpenOmnibox,
		},
		TabMode: ShortcutTable{
			{uint('n'), ModNone}:            ActionNewTab,
			{uint(gdk.KEY_Escape), ModNone}: ActionExitMode,
		},
		PaneMode: ShortcutTable{
			{uint('x'), ModNone}:            ActionClosePane,
			{uint(gdk.KEY_Escape), ModNone}: ActionExitMode,
		},
	}

	tests := []struct {
		name    string
		binding KeyBinding
		mode    Mode
		want    Action
		wantOk  bool
	}{
		{
			name:    "global in normal mode",
			binding: KeyBinding{uint(gdk.KEY_q), ModCtrl},
			mode:    ModeNormal,
			want:    ActionQuit,
			wantOk:  true,
		},
		{
			name:    "global in tab mode",
			binding: KeyBinding{uint(gdk.KEY_l), ModCtrl},
			mode:    ModeTab,
			want:    ActionOpenOmnibox,
			wantOk:  true,
		},
		{
			name:    "tab mode action",
			binding: KeyBinding{uint('n'), ModNone},
			mode:    ModeTab,
			want:    ActionNewTab,
			wantOk:  true,
		},
		{
			name:    "pane mode action",
			binding: KeyBinding{uint('x'), ModNone},
			mode:    ModePane,
			want:    ActionClosePane,
			wantOk:  true,
		},
		{
			name:    "escape exits tab mode",
			binding: KeyBinding{uint(gdk.KEY_Escape), ModNone},
			mode:    ModeTab,
			want:    ActionExitMode,
			wantOk:  true,
		},
		{
			name:    "escape exits pane mode",
			binding: KeyBinding{uint(gdk.KEY_Escape), ModNone},
			mode:    ModePane,
			want:    ActionExitMode,
			wantOk:  true,
		},
		{
			name:    "unknown key in normal mode",
			binding: KeyBinding{uint('z'), ModNone},
			mode:    ModeNormal,
			want:    "",
			wantOk:  false,
		},
		{
			name:    "tab action not in normal mode",
			binding: KeyBinding{uint('n'), ModNone},
			mode:    ModeNormal,
			want:    "",
			wantOk:  false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, ok := set.Lookup(tt.binding, tt.mode)
			if ok != tt.wantOk {
				t.Errorf("Lookup() ok = %v, want %v", ok, tt.wantOk)
				return
			}
			if got != tt.want {
				t.Errorf("Lookup() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestShouldAutoExitMode(t *testing.T) {
	exitActions := []Action{
		ActionNewTab,
		ActionCloseTab,
		ActionRenameTab,
		ActionSplitRight,
		ActionSplitLeft,
		ActionSplitUp,
		ActionSplitDown,
		ActionClosePane,
		ActionStackPane,
	}

	stayActions := []Action{
		ActionNextTab,
		ActionPreviousTab,
		ActionFocusRight,
		ActionFocusLeft,
		ActionGoBack,
		ActionZoomIn,
		ActionOpenOmnibox,
	}

	for _, action := range exitActions {
		if !ShouldAutoExitMode(action) {
			t.Errorf("ShouldAutoExitMode(%s) = false, want true", action)
		}
	}

	for _, action := range stayActions {
		if ShouldAutoExitMode(action) {
			t.Errorf("ShouldAutoExitMode(%s) = true, want false", action)
		}
	}
}