twink/tui/app.go

package tui

import (
	"fmt"
	"strings"

	tea "github.com/charmbracelet/bubbletea"
)

type (
	AppModel struct{
		Size Size
		Header HeaderModel
		Screen tea.Model
		Screens []tea.Model
	}
)

func NewApp() AppModel {
	screens := []tea.Model{
		NewHomeScreen(),
		NewNotificationScreen(),
	}
	return AppModel{
		Header: NewHeader(),
		Screen: screens[0],
		Screens: screens,
	}
}

func (m AppModel) Init() tea.Cmd {
	return nil
}

func (m AppModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd, headerCmd, screenCmd tea.Cmd

	switch msg := msg.(type) {
	case tea.WindowSizeMsg:
		m.SetSize(msg.Width, msg.Height)
	case SwitchHomeScreenMsg:
		m.Screen = m.Screens[0]
	case SwitchNotificationScreenMsg:
		m.Screen = m.Screens[1]
	}

	m.Header, headerCmd = m.Header.Update(msg)
	m.Screen, screenCmd = m.Screen.Update(msg)

	return m, tea.Batch(cmd, headerCmd, screenCmd)
}

func (m *AppModel) SetSize(width, height int) {
	m.Size = NewSize(width, height)
	m.Header.SetWidth(width)
	screenHeight := height - m.Header.Size.Height
	switch screen := m.Screen.(type) {
	case Sized:
		m.Screen = screen.SetSize(width, screenHeight)
	case SizedWidth:
		m.Screen = screen.SetWidth(width)
	case SizedHeight:
		m.Screen = screen.SetHeight(screenHeight)
	}
}

func (m AppModel) View() string {
	return strings.Join([]string{
		fmt.Sprintf("%d x %d", m.Size.Width, m.Size.Height),
		m.Header.View(),
		m.Screen.View(),
	}, "\n")
}