package dashboard

import (
	"charm.land/bubbles/v2/key"
	tea "charm.land/bubbletea/v2"
	"github.com/LazyBachelor/LazyPM/pkg/tui/components"
	"github.com/LazyBachelor/LazyPM/pkg/tui/msgs"
)

type KeyMap struct {
	components.CommonKeyMap
	SwitchToKanbanBoard key.Binding
	Quit                key.Binding
	SelectIssue         key.Binding
	BackToList          key.Binding
	ScrollUp            key.Binding
	ScrollDown          key.Binding
	EditTitle           key.Binding
	EditDescription     key.Binding
	ChangeStatus        key.Binding
	ChangePriority      key.Binding
	ChangeType          key.Binding
	ChangeAssignee      key.Binding
	AddComment          key.Binding
	AddIssue            key.Binding
	DeleteIssue         key.Binding
}

var defaultDashboardKeyMap = KeyMap{
	CommonKeyMap: components.DefaultCommonKeyMap(),
	SwitchToKanbanBoard: key.NewBinding(
		key.WithKeys("v"),
		key.WithHelp("v", "switch to kanban")),
	EditTitle: key.NewBinding(
		key.WithKeys("e"),
		key.WithHelp("e", "edit title"),
	),
	EditDescription: key.NewBinding(
		key.WithKeys("d"),
		key.WithHelp("d", "edit description"),
	),
	ChangeStatus: key.NewBinding(
		key.WithKeys("s"),
		key.WithHelp("s", "change status"),
	),
	ChangePriority: key.NewBinding(
		key.WithKeys("p"),
		key.WithHelp("p", "change priority"),
	),
	ChangeType: key.NewBinding(
		key.WithKeys("t"),
		key.WithHelp("t", "change type"),
	),
	ChangeAssignee: key.NewBinding(
		key.WithKeys("A"),
		key.WithHelp("A", "change assignee"),
	),
	AddComment: key.NewBinding(
		key.WithKeys("c"),
		key.WithHelp("c", "add comment"),
	),
	AddIssue: key.NewBinding(
		key.WithKeys("a"),
		key.WithHelp("a", "add issue"),
	),
	DeleteIssue: key.NewBinding(
		key.WithKeys("x"),
		key.WithHelp("x", "delete issue"),
	),
}

func (m *Model) handleKeyPressMsg(msg tea.KeyPressMsg) tea.Cmd {
	var cmd tea.Cmd

	switch {
	case m.notInModalMsgWithKey(msg, m.keyMap.Help):
		m.helpBar.ToggleHelp()
		m.logAction("tui toggled help")

	case m.notInModalMsgWithKey(msg, m.keyMap.Quit):
		m.logAction("tui quit requested")
		return tea.Quit

	case m.notInModalMsgWithKey(msg, m.keyMap.SwitchToKanbanBoard):
		return func() tea.Msg { return msgs.SwitchToKanbanBoardMsg{} }

	case m.notInModalMsgWithKey(msg, m.keyMap.ScrollUp):
		m.issueDetail.ScrollUp(1)
		m.logAction("tui scrolled issue detail up")

	case m.notInModalMsgWithKey(msg, m.keyMap.ScrollDown):
		m.issueDetail.ScrollDown(1)
		m.logAction("tui scrolled issue detail down")

	case m.notInModalMsgWithKey(msg, m.keyMap.EditTitle):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startEditTitle(selected)
			m.logAction("tui started editing issue title")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.EditDescription):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startEditDescription(selected)
			m.logAction("tui started editing issue description")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.ChangeStatus):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startChooseStatus(selected)
			m.logAction("tui opened status picker")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.ChangePriority):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startChoosePriority(selected)
			m.logAction("tui opened priority picker")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.ChangeType):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startChooseType(selected)
			m.logAction("tui opened type picker")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.ChangeAssignee):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startEditAssignee(selected)
			m.logAction("tui started editing assignee")
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.AddComment):
		if selected := m.issueList.SelectedItem(); selected.ID != "" {
			cmd = m.startAddComment(selected)
		}

	case m.notInModalMsgWithKey(msg, m.keyMap.AddIssue):
		cmd = m.startCreateIssue()
		m.logAction("tui started creating issue")

	case m.notInModalMsgWithKey(msg, m.keyMap.DeleteIssue):
		fl := m.issueList
		if selected := fl.SelectedItem(); selected.ID != "" {
			cmd = m.startConfirmDelete(selected.ID, fl.Index())
			m.logAction("tui opened delete confirmation")
		}
	}

	return cmd
}

func (m *Model) notInModalMsgWithKey(msg tea.KeyPressMsg, keyBinding key.Binding) bool {
	return !m.IsInModal() && key.Matches(msg, keyBinding)
}