bubbletea/examples/altscreen-toggle/main.go

80 lines
1.4 KiB
Go
Raw Normal View History

package main
import (
"fmt"
2023-09-14 18:00:38 -04:00
"log"
"os"
tea "github.com/charmbracelet/bubbletea"
"github.com/muesli/termenv"
)
var (
color = termenv.EnvColorProfile().Color
keyword = termenv.Style{}.Foreground(color("204")).Background(color("235")).Styled
2021-03-11 15:15:08 -05:00
help = termenv.Style{}.Foreground(color("241")).Styled
)
type model struct {
altscreen bool
quitting bool
}
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 msg.String() {
case "q", "ctrl+c", "esc":
m.quitting = true
return m, tea.Quit
case " ":
var cmd tea.Cmd
if m.altscreen {
cmd = tea.ExitAltScreen
} else {
cmd = tea.EnterAltScreen
}
m.altscreen = !m.altscreen
return m, cmd
}
}
return m, nil
}
func (m model) View() string {
if m.quitting {
return "Bye!\n"
}
const (
altscreenMode = " altscreen mode "
inlineMode = " inline mode "
)
2021-03-11 15:15:08 -05:00
var mode string
if m.altscreen {
mode = altscreenMode
} else {
mode = inlineMode
}
2021-03-11 15:15:08 -05:00
return fmt.Sprintf("\n\n You're in %s\n\n\n", keyword(mode)) +
help(" space: switch modes • q: exit\n")
}
func main() {
2023-09-14 18:00:38 -04:00
f, err := tea.LogToFile("log.txt", "")
if err != nil {
log.Fatal(err)
}
defer f.Close()
if _, err := tea.NewProgram(model{}).Run(); err != nil {
fmt.Println("Error running program:", err)
os.Exit(1)
}
}