gui module
This commit is contained in:
Tyrel Souza
parent
64bcd68a4b
commit
881c6b4ded
|
|
@ -0,0 +1,190 @@
|
|||
package gui
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Mode int64
|
||||
|
||||
const (
|
||||
Select Mode = iota
|
||||
Create
|
||||
Draw
|
||||
)
|
||||
|
||||
type Card struct {
|
||||
Front string
|
||||
Back string
|
||||
LastCorrect time.Time
|
||||
}
|
||||
|
||||
type model struct {
|
||||
cards []Card // items on the to-do list
|
||||
cursor int // which to-do list item our cursor is pointing at
|
||||
selected map[int]struct{} // which to-do items are selected
|
||||
mode Mode
|
||||
}
|
||||
|
||||
func initialModel() model {
|
||||
return model{
|
||||
cards: []Card{
|
||||
{Front: "Hello (JP)", Back: "Konnichi wa"},
|
||||
{Front: "Hello (SP)", Back: "Hola"},
|
||||
{Front: "Hello (DE)", Back: "Guten Tag"},
|
||||
},
|
||||
|
||||
selected: make(map[int]struct{}),
|
||||
mode: Select,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *model) Init() tea.Cmd {
|
||||
// Just return `nil`, which means "no I/O right now, please."
|
||||
return nil
|
||||
}
|
||||
func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
if m.mode == Draw {
|
||||
return m.updateDraw(msg)
|
||||
}
|
||||
// Select is default mode
|
||||
return m.updateSelect(msg)
|
||||
}
|
||||
|
||||
func (m *model) updateDraw(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
|
||||
// Is it a key press?
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
|
||||
case "s":
|
||||
m.switchMode(Select)
|
||||
|
||||
// These keys should exit the program.
|
||||
case "ctrl+c", "q":
|
||||
return m, tea.Quit
|
||||
}
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (m *model) switchMode(mode Mode) {
|
||||
// TODO: initialization stuff
|
||||
switch mode {
|
||||
case Select:
|
||||
m.mode = Select
|
||||
case Draw:
|
||||
m.mode = Draw
|
||||
case Create:
|
||||
m.mode = Create
|
||||
}
|
||||
}
|
||||
|
||||
func (m *model) updateSelect(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
|
||||
// Is it a key press?
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
|
||||
// These keys should exit the program.
|
||||
case "ctrl+c", "q":
|
||||
return m, tea.Quit
|
||||
|
||||
// The "up" and "k" keys move the cursor up
|
||||
case "up", "k":
|
||||
if m.cursor > 0 {
|
||||
m.cursor--
|
||||
}
|
||||
|
||||
// The "down" and "j" keys move the cursor down
|
||||
case "down", "j":
|
||||
if m.cursor < len(m.cards)-1 {
|
||||
m.cursor++
|
||||
}
|
||||
case "d":
|
||||
m.switchMode(Draw)
|
||||
|
||||
// The "enter" key and the spacebar (a literal space) toggle
|
||||
// the selected state for the item that the cursor is pointing at.
|
||||
case "enter", " ":
|
||||
_, ok := m.selected[m.cursor]
|
||||
if ok {
|
||||
delete(m.selected, m.cursor)
|
||||
} else {
|
||||
m.selected[m.cursor] = struct{}{}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Return the updated model to the Bubble Tea runtime for processing.
|
||||
// Note that we're not returning a command.
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (m *model) cardListToString() string {
|
||||
s := ""
|
||||
for i, card := range m.cards {
|
||||
if _, ok := m.selected[i]; ok {
|
||||
s += fmt.Sprintf("%s [%s]\n", card.Front, card.Back)
|
||||
}
|
||||
}
|
||||
return s
|
||||
}
|
||||
func (m *model) View() string {
|
||||
if m.mode == Draw {
|
||||
return m.viewDraw()
|
||||
}
|
||||
// Select is default mode
|
||||
return m.viewSelect()
|
||||
}
|
||||
|
||||
func (m *model) viewDraw() string {
|
||||
s := "Draw Mode:\n\n"
|
||||
|
||||
s += m.cardListToString()
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
func (m *model) viewSelect() string {
|
||||
s := "Select cards to query:\n\n"
|
||||
|
||||
// Iterate over our choices
|
||||
for i, choice := range m.cards {
|
||||
|
||||
// Is the cursor pointing at this choice?
|
||||
cursor := " " // no cursor
|
||||
if m.cursor == i {
|
||||
cursor = ">" // cursor!
|
||||
}
|
||||
|
||||
// Is this choice selected?
|
||||
checked := " " // not selected
|
||||
if _, ok := m.selected[i]; ok {
|
||||
checked = "✓" // selected!
|
||||
}
|
||||
|
||||
// Render the row
|
||||
s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.Front)
|
||||
}
|
||||
|
||||
// The footer
|
||||
s += "\nPress c to Create (_unimplemented_).\nPress d to Draw.\nPress q to quit.\n"
|
||||
|
||||
// Send the UI for rendering
|
||||
return s
|
||||
}
|
||||
|
||||
func Run() {
|
||||
initial := initialModel()
|
||||
p := tea.NewProgram(&initial)
|
||||
if err := p.Start(); err != nil {
|
||||
fmt.Printf("Alas, there's been an error: %v", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
185
main.go
185
main.go
|
|
@ -1,190 +1,9 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
tea "github.com/charmbracelet/bubbletea"
|
||||
"os"
|
||||
"time"
|
||||
"gitea.tyrel.dev/tyrel/itor/gui"
|
||||
)
|
||||
|
||||
type Mode int64
|
||||
|
||||
const (
|
||||
Select Mode = iota
|
||||
Create
|
||||
Draw
|
||||
)
|
||||
|
||||
type Card struct {
|
||||
Front string
|
||||
Back string
|
||||
LastCorrect time.Time
|
||||
}
|
||||
|
||||
type model struct {
|
||||
cards []Card // items on the to-do list
|
||||
cursor int // which to-do list item our cursor is pointing at
|
||||
selected map[int]struct{} // which to-do items are selected
|
||||
mode Mode
|
||||
}
|
||||
|
||||
func initialModel() model {
|
||||
return model{
|
||||
cards: []Card{
|
||||
{Front: "Hello (JP)", Back: "Konnichi wa"},
|
||||
{Front: "Hello (SP)", Back: "Hola"},
|
||||
{Front: "Hello (DE)", Back: "Guten Tag"},
|
||||
},
|
||||
|
||||
selected: make(map[int]struct{}),
|
||||
mode: Select,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *model) Init() tea.Cmd {
|
||||
// Just return `nil`, which means "no I/O right now, please."
|
||||
return nil
|
||||
}
|
||||
func (m *model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
if m.mode == Draw {
|
||||
return m.updateDraw(msg)
|
||||
}
|
||||
// Select is default mode
|
||||
return m.updateSelect(msg)
|
||||
}
|
||||
|
||||
func (m *model) updateDraw(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
|
||||
// Is it a key press?
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
|
||||
case "s":
|
||||
m.switchMode(Select)
|
||||
|
||||
// These keys should exit the program.
|
||||
case "ctrl+c", "q":
|
||||
return m, tea.Quit
|
||||
}
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (m *model) switchMode(mode Mode) {
|
||||
// TODO: initialization stuff
|
||||
switch mode {
|
||||
case Select:
|
||||
m.mode = Select
|
||||
case Draw:
|
||||
m.mode = Draw
|
||||
case Create:
|
||||
m.mode = Create
|
||||
}
|
||||
}
|
||||
|
||||
func (m *model) updateSelect(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
|
||||
// Is it a key press?
|
||||
case tea.KeyMsg:
|
||||
switch msg.String() {
|
||||
|
||||
// These keys should exit the program.
|
||||
case "ctrl+c", "q":
|
||||
return m, tea.Quit
|
||||
|
||||
// The "up" and "k" keys move the cursor up
|
||||
case "up", "k":
|
||||
if m.cursor > 0 {
|
||||
m.cursor--
|
||||
}
|
||||
|
||||
// The "down" and "j" keys move the cursor down
|
||||
case "down", "j":
|
||||
if m.cursor < len(m.cards)-1 {
|
||||
m.cursor++
|
||||
}
|
||||
case "d":
|
||||
m.switchMode(Draw)
|
||||
|
||||
// The "enter" key and the spacebar (a literal space) toggle
|
||||
// the selected state for the item that the cursor is pointing at.
|
||||
case "enter", " ":
|
||||
_, ok := m.selected[m.cursor]
|
||||
if ok {
|
||||
delete(m.selected, m.cursor)
|
||||
} else {
|
||||
m.selected[m.cursor] = struct{}{}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Return the updated model to the Bubble Tea runtime for processing.
|
||||
// Note that we're not returning a command.
|
||||
return m, nil
|
||||
}
|
||||
|
||||
func (m *model) cardListToString() string {
|
||||
s := ""
|
||||
for i, card := range m.cards {
|
||||
if _, ok := m.selected[i]; ok {
|
||||
s += fmt.Sprintf("%s [%s]\n", card.Front, card.Back)
|
||||
}
|
||||
}
|
||||
return s
|
||||
}
|
||||
func (m *model) View() string {
|
||||
if m.mode == Draw {
|
||||
return m.viewDraw()
|
||||
}
|
||||
// Select is default mode
|
||||
return m.viewSelect()
|
||||
}
|
||||
|
||||
func (m *model) viewDraw() string {
|
||||
s := "Draw Mode:\n\n"
|
||||
|
||||
s += m.cardListToString()
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
func (m *model) viewSelect() string {
|
||||
s := "Select cards to query:\n\n"
|
||||
|
||||
// Iterate over our choices
|
||||
for i, choice := range m.cards {
|
||||
|
||||
// Is the cursor pointing at this choice?
|
||||
cursor := " " // no cursor
|
||||
if m.cursor == i {
|
||||
cursor = ">" // cursor!
|
||||
}
|
||||
|
||||
// Is this choice selected?
|
||||
checked := " " // not selected
|
||||
if _, ok := m.selected[i]; ok {
|
||||
checked = "✓" // selected!
|
||||
}
|
||||
|
||||
// Render the row
|
||||
s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.Front)
|
||||
}
|
||||
|
||||
// The footer
|
||||
s += "\nPress c to Create (_unimplemented_).\nPress d to Draw.\nPress q to quit.\n"
|
||||
|
||||
// Send the UI for rendering
|
||||
return s
|
||||
}
|
||||
|
||||
func main() {
|
||||
initial := initialModel()
|
||||
p := tea.NewProgram(&initial)
|
||||
if err := p.Start(); err != nil {
|
||||
fmt.Printf("Alas, there's been an error: %v", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
gui.Run()
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue