167 lines
3.1 KiB
Go
167 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/gdamore/tcell/v2"
|
|
"github.com/rivo/tview"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type mode string
|
|
|
|
const (
|
|
modeNormal mode = "normal"
|
|
modeFocus mode = "focus"
|
|
modeSearch mode = "search"
|
|
modeEx mode = "ex"
|
|
)
|
|
|
|
type UI struct {
|
|
Mode mode
|
|
App *tview.Application
|
|
Fields []field
|
|
|
|
// UI things
|
|
Pages *tview.Pages
|
|
TopFlex *tview.Flex
|
|
Field *tview.Box
|
|
FieldBar *tview.Flex
|
|
BottomBar *tview.Pages
|
|
ExInput *tview.InputField
|
|
ExOutput *tview.TextView
|
|
}
|
|
|
|
func (ui *UI) setMode(m mode) {
|
|
ui.Mode = m
|
|
switch ui.Mode {
|
|
case modeNormal:
|
|
ui.App.SetFocus(ui.TopFlex)
|
|
ui.ExInput.SetText("")
|
|
case modeEx:
|
|
ui.App.SetFocus(ui.ExInput)
|
|
}
|
|
}
|
|
|
|
func (ui *UI) handleInput(event *tcell.EventKey) *tcell.EventKey {
|
|
switch ui.Mode {
|
|
case modeNormal:
|
|
switch event.Rune() {
|
|
case ':':
|
|
ui.setMode(modeEx)
|
|
return nil
|
|
}
|
|
case modeEx:
|
|
case modeFocus:
|
|
case modeSearch:
|
|
default:
|
|
panic("mode?")
|
|
}
|
|
return event
|
|
}
|
|
|
|
func (ui *UI) handleExInput(key tcell.Key) {
|
|
text := ui.ExInput.GetText()
|
|
ui.setMode(modeNormal)
|
|
|
|
if key != tcell.KeyEnter {
|
|
return
|
|
}
|
|
|
|
switch text {
|
|
case "q", "quit":
|
|
ui.quit()
|
|
}
|
|
|
|
ui.ExOutput.SetText("")
|
|
fmt.Fprintf(ui.ExOutput, "did not understand '%s', sorry", text)
|
|
}
|
|
|
|
type field struct {
|
|
Name string
|
|
Selected bool
|
|
}
|
|
|
|
func NewUI() *UI {
|
|
app := tview.NewApplication()
|
|
ui := UI{
|
|
Mode: modeNormal,
|
|
App: app,
|
|
Fields: []field{{"scratch", true}, {"test", false}},
|
|
Pages: tview.NewPages(),
|
|
TopFlex: tview.NewFlex(),
|
|
Field: tview.NewBox(),
|
|
FieldBar: tview.NewFlex(),
|
|
BottomBar: tview.NewPages(),
|
|
ExInput: tview.NewInputField(),
|
|
ExOutput: tview.NewTextView(),
|
|
}
|
|
|
|
app.SetInputCapture(ui.handleInput)
|
|
|
|
ui.TopFlex.SetDirection(tview.FlexRow)
|
|
ui.TopFlex.AddItem(ui.Field, 0, 20, true)
|
|
ui.TopFlex.AddItem(ui.FieldBar, 0, 1, false)
|
|
ui.TopFlex.AddItem(ui.BottomBar, 0, 1, false)
|
|
|
|
ui.Pages.AddPage("main", ui.TopFlex, true, true)
|
|
|
|
ui.BottomBar.AddPage("output", ui.ExOutput, true, true)
|
|
ui.BottomBar.AddPage("input", ui.ExInput, true, false)
|
|
|
|
ui.ExInput.SetLabel(":")
|
|
ui.ExInput.SetDoneFunc(ui.handleExInput)
|
|
|
|
ui.ExOutput.SetMaxLines(1)
|
|
fmt.Fprintf(ui.ExOutput, "porphyry has started. :q to quit")
|
|
|
|
ui.FieldBar.SetDirection(tview.FlexColumn)
|
|
|
|
app.SetBeforeDrawFunc(func(_ tcell.Screen) bool {
|
|
// Handle field bar
|
|
ui.FieldBar.Clear()
|
|
|
|
for _, f := range ui.Fields {
|
|
t := tview.NewTextView().SetTextStyle(tcell.StyleDefault.Bold(f.Selected))
|
|
t.SetMaxLines(1)
|
|
t.SetBorder(true)
|
|
fmt.Fprintf(t, f.Name)
|
|
ui.FieldBar.AddItem(t, 0, 1, false)
|
|
}
|
|
|
|
// Handle ex mode prompt
|
|
|
|
if ui.Mode == modeEx {
|
|
ui.BottomBar.SwitchToPage("input")
|
|
} else {
|
|
ui.BottomBar.SwitchToPage("output")
|
|
}
|
|
return false
|
|
})
|
|
|
|
app.SetRoot(ui.Pages, true)
|
|
return &ui
|
|
}
|
|
|
|
func (ui *UI) quit() {
|
|
ui.App.Stop()
|
|
}
|
|
|
|
func _main() error {
|
|
cmd := &cobra.Command{
|
|
Use: "porphyry",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return NewUI().App.Run()
|
|
},
|
|
}
|
|
|
|
return cmd.Execute()
|
|
}
|
|
|
|
func main() {
|
|
if err := _main(); err != nil {
|
|
fmt.Fprintf(os.Stderr, err.Error()+"\n")
|
|
}
|
|
}
|