package plays

import (
	"fmt"
	"strings"

	tea "github.com/charmbracelet/bubbletea"
)

var (
	landingChoices = []string{"Login", "Register"}
)

type Landing struct {
	*Base

	choice string
	cursor int
}

func NewLanding(base *Base) *Landing {
	m := Landing{
		Base:   base,
		choice: "",
		cursor: 0,
	}
	return &m
}

func (m *Landing) Init() tea.Cmd {
	return tea.ClearScreen
}

func (m *Landing) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+l":
			return m, tea.ClearScreen

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

		case "enter":
			m.choice = landingChoices[m.cursor]
			return m, tea.Quit

		case "tab", "shift+tab", "up", "down":
			s := msg.String()

			if s == "up" || s == "shift+tab" {
				m.cursor--
			} else {
				m.cursor++
			}

			if m.cursor >= len(landingChoices) {
				m.cursor = 0
			} else if m.cursor < 0 {
				m.cursor = len(landingChoices) - 1
			}
		}
	}

	return m, nil
}

func (m *Landing) View() string {
	var b strings.Builder

	for i := 0; i < len(landingChoices); i++ {
		if m.cursor == i {
			fmt.Fprintf(&b, "(•) %s\n", landingChoices[i])
		} else {
			fmt.Fprintf(&b, "( ) %s\n", landingChoices[i])
		}
	}

	return b.String()
}

func (m *Landing) Next(queue *[]*tea.Program) error {
	switch m.choice {
	case "Login":
		*queue = append(*queue,
			tea.NewProgram(NewLogin(m.Base)))
	case "Register":
		*queue = append(*queue,
			tea.NewProgram(NewRegister(m.Base)))
	}
	return nil
}