spacetea/mainscreen.go

package main

import (
	"fmt"
	"strconv"
	"time"

	sim "git.saintnet.tech/stryan/spacetea/simulator"
	"github.com/charmbracelet/bubbles/textinput"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

var style = lipgloss.NewStyle().
	Width(24).
	Align(lipgloss.Center).
	BorderStyle(lipgloss.NormalBorder()).
	BorderForeground(lipgloss.Color("63"))

type model struct {
	s     *sim.Simulator
	input textinput.Model
	next  string
}

type beat struct{}
type simCmd string

func heartbeat() tea.Cmd {
	return tea.Tick(time.Second*1, func(time.Time) tea.Msg {
		return beat{}
	})
}

func (m model) simCommand() tea.Msg {
	if m.next != "" {
		tmp := m.next
		m.next = ""
		return simCmd(tmp)
	}
	return nil
}
func initMainscreen() model {
	ti := textinput.New()
	ti.Placeholder = "input command"
	ti.CharLimit = 156
	ti.Width = 20

	return model{
		s:     simulator,
		input: ti,
	}
}
func (m model) Init() tea.Cmd {
	return heartbeat()
}

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd tea.Cmd
	switch msg := msg.(type) {
	case beat:
		//heartbeat
		return m, heartbeat()
	case placeMsg:
		simc := fmt.Sprintf("place %v", string(msg))
		m.s.Input(simc)
		return m, nil
	case craftMsg:
		simc := fmt.Sprintf("craft %v", string(msg))
		m.s.Input(simc)
		return m, nil
	case tea.KeyMsg:
		switch msg.Type {
		case tea.KeyCtrlC:
			m.s.Stop()
			return m, tea.Quit
		case tea.KeyEnter:
			m.s.Input(m.input.Value())
			m.input.Reset()
			return m, nil
		case tea.KeyRight:
			m.s.Input("right")
			return m, nil
		case tea.KeyLeft:
			m.s.Input("left")
			return m, nil
		case tea.KeyUp:
			m.s.Input("up")
			return m, nil
		case tea.KeyDown:
			m.s.Input("down")
			return m, nil
		case tea.KeyCtrlF:
			if m.input.Focused() {
				m.input.Blur()
			} else {
				m.input.Focus()
			}
			return m, nil
		case tea.KeyRunes:
			if !m.input.Focused() {
				switch msg.String() {
				case "g":
					m.s.Input("get")
				case "p":
					var res []sim.ItemEntry
					for _, k := range sim.GlobalItemList {
						if m.s.Player.Resources[k] != 0 {
							res = append(res, sim.Lookup(k))
						}
					}
					return newPlaceModel(res, nil), nil
				case "c":
					var res []sim.ItemEntry
					for _, k := range sim.GlobalTechList {
						if _, ok := m.s.Player.Techs[k]; ok {
							res = append(res, sim.LookupTech(k))
						}
					}
					return newCraftModel(res, nil), nil
				}
				return m, nil
			}
		}
	}
	m.input, cmd = m.input.Update(msg)
	return m, cmd

}

func (m model) View() string {
	var render string
	playerAndLog := lipgloss.JoinVertical(0, style.Render(fmt.Sprintf("%v", m.s.Player.String())), style.Render(fmt.Sprintf("%v", m.s.Player.Log())))
	placeAndInput := lipgloss.JoinVertical(0, style.Render(m.s.Place.String()), style.Render(m.input.View()))
	display := lipgloss.JoinHorizontal(0, placeAndInput, playerAndLog)
	render = fmt.Sprintf("%v\n%v\n", style.Render(fmt.Sprintf("Current Time: %v", strconv.Itoa(m.s.Time))), display)
	return render
}