package editor

import (
	"git.gophernest.net/azpect/TextEditor/internal/action"
	tea "github.com/charmbracelet/bubbletea"
)

func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {

	case tea.WindowSizeMsg:
		m.win_h = msg.Height
		m.win_w = msg.Width

	case tea.KeyMsg:
		if msg.String() == "ctrl+c" {
			return m, tea.Quit
		}

		switch m.mode {
		case action.NormalMode,
			action.VisualMode,
			action.VisualBlockMode,
			action.VisualLineMode:
			return m, m.input.Handle(&m, msg.String())

		// TODO: This should be handled elsewhere
		case action.InsertMode:
			key := msg.String()

			switch key {
			case "esc":
				if m.insertCount > 1 {
					m.replayInsert()
				}

				// Allow i to step back, but a to stay put
				if m.cursor.x > 0 {
					m.cursor.x--
				}
				m.mode = action.NormalMode
				m.insertCount = 0
				m.insertKeys = nil

			case "ctrl+c", "ctrl+d":
				return m, tea.Quit

			default:
				// Record and process
				m.insertKeys = append(m.insertKeys, key)
				m.processInsertKey(key)
			}
		case action.CommandMode:
			switch msg.String() {
			case "esc":
				m.mode = action.NormalMode
				m.command = ""

			default:
				m.command += msg.String()
			}
		}
	}

	return m, nil
}