package tui

import (
	"fmt"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/gliderlabs/ssh"
)

type sessionState int

const (
	startState sessionState = iota
	errorState
	quittingState
	quitState
)

type stateMsg struct{ state sessionState }
type infoMsg struct{ text string }
type errMsg struct{ err error }

func (e errMsg) Error() string {
	return e.err.Error()
}

func SessionHandler(s ssh.Session) (tea.Model, error) {
	pty, changes, active := s.Pty()
	if !active {
		return nil, fmt.Errorf("you need to do this from a terminal with PTY support")
	}
	return NewModel(pty.Window.Width, pty.Window.Height, changes), nil
}

type Model struct {
	state         sessionState
	error         string
	info          string
	width         int
	height        int
	windowChanges <-chan ssh.Window
}

func NewModel(width int, height int, windowChanges <-chan ssh.Window) *Model {
	m := &Model{
		width:         width,
		height:        height,
		windowChanges: windowChanges,
	}
	m.state = startState
	return m
}

func (m *Model) Init() tea.Cmd {
	return tea.Batch(m.windowChangesCmd, tea.HideCursor)
}

func (m *Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	cmds := make([]tea.Cmd, 0)
	// Always allow state, error, info, window resize and quit messages
	switch msg := msg.(type) {
	case stateMsg:
		m.state = msg.state
	case tea.KeyMsg:
		switch msg.String() {
		case "q", "ctrl+c":
			return m, tea.Quit
		}
	case errMsg:
		m.error = msg.Error()
		m.state = errorState
		return m, nil
	case infoMsg:
		m.info = msg.text
	case windowMsg:
		cmds = append(cmds, m.windowChangesCmd)
	case tea.WindowSizeMsg:
		m.width = msg.Width
		m.height = msg.Height
	}
	return m, tea.Batch(cmds...)
}

func (m *Model) View() string {
	pad := 6
	h := headerStyle.Width(m.width - pad).Render("Smoothie")
	f := footerStyle.Render(m.info)
	s := ""
	content := ""
	switch m.state {
	case errorState:
		s += errorStyle.Render(fmt.Sprintf("Bummer: %s", m.error))
	default:
		s = normalStyle.Render(fmt.Sprintf("Doing something weird %d", m.state))
	}
	content = h + "\n" + s + "\n" + f
	return appBoxStyle.Render(content)
}