Bubbletea many views

main.go

package main

import (
	"fmt"
	"os"

	tea "github.com/charmbracelet/bubbletea"
)

func main() {
	err := tea.NewProgram(initialModel(), tea.WithAltScreen()).Start()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		os.Exit(1)
	}
}

func initialModel() model {
	return model{
		currentView: "A",
	}
}

type model struct {
	currentView string
	viewA       viewA
	viewB       viewB
}

type (
	msgWentToA struct{}
	msgWentToB struct{}
)

func cmdGoToA() tea.Msg { return msgWentToA{} }
func cmdGoToB() tea.Msg { return msgWentToB{} }

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

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch {
		case msg.String() == "ctrl+c":
			return m, tea.Quit
		}
	case msgWentToA:
		m.currentView = "A"
		return m, nil
	case msgWentToB:
		m.currentView = "B"
		return m, nil
	}

	switch m.currentView {
	case "A":
		var cmd tea.Cmd
		m.viewA, cmd = m.viewA.Update(msg)
		return m, cmd
	case "B":
		var cmd tea.Cmd
		m.viewB, cmd = m.viewB.Update(msg)
		return m, cmd
	}

	return m, nil
}

func (m model) View() string {
	switch m.currentView {
	case "A":
		return m.viewA.View()
	case "B":
		return m.viewB.View()
	default:
		return ""
	}
}

view_a.go

package main

import (
	"strings"

	tea "github.com/charmbracelet/bubbletea"
)

type viewA struct{}

func (m viewA) Update(msg tea.Msg) (viewA, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch {
		case strings.ToUpper(msg.String()) == "B":
			return m, cmdGoToB
		}
	}
	return m, nil
}

func (m viewA) View() string {
	return "Viewing A...\nPress B to switch to B view."
}

view_b.go

package main

import (
	"strings"

	tea "github.com/charmbracelet/bubbletea"
)

type viewB struct{}

func (m viewB) Update(msg tea.Msg) (viewB, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch {
		case strings.ToUpper(msg.String()) == "A":
			return m, cmdGoToA
		}
	}
	return m, nil
}

func (m viewB) View() string {
	return "Viewing B...\nPress A to switch to A view."
}