2022-05-03 05:35:02 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2022-05-13 12:50:25 +00:00
|
|
|
"io/fs"
|
|
|
|
"log"
|
2022-05-03 05:35:02 +00:00
|
|
|
"os"
|
2022-05-13 12:50:25 +00:00
|
|
|
"strings"
|
2022-05-03 05:35:02 +00:00
|
|
|
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
|
|
)
|
|
|
|
|
2022-05-12 07:11:26 +00:00
|
|
|
// Enum for the different launch types (browsers and CLI's)
|
2022-05-10 13:13:56 +00:00
|
|
|
const (
|
2022-05-12 07:11:26 +00:00
|
|
|
BrowserSessions int = iota
|
2022-05-10 13:13:56 +00:00
|
|
|
Commands
|
|
|
|
)
|
|
|
|
|
|
|
|
type sessionOrCommand struct {
|
2022-05-03 05:35:02 +00:00
|
|
|
displayString string
|
|
|
|
commandString string
|
|
|
|
}
|
|
|
|
|
2022-05-10 13:13:56 +00:00
|
|
|
type Choices [][]sessionOrCommand
|
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
type SelectionSet map[int]struct{}
|
|
|
|
|
2022-05-03 05:35:02 +00:00
|
|
|
type model struct {
|
2022-05-10 13:13:56 +00:00
|
|
|
activeSection int
|
|
|
|
cursor []int
|
|
|
|
choices Choices
|
2022-05-11 13:39:14 +00:00
|
|
|
selected []SelectionSet
|
2022-05-12 07:11:26 +00:00
|
|
|
status string
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
|
2022-05-12 07:11:26 +00:00
|
|
|
type statusMsg string
|
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
var browserSelection SelectionSet
|
|
|
|
var commandSelection SelectionSet
|
|
|
|
|
2022-05-10 13:13:56 +00:00
|
|
|
func getBrowserSessions() []sessionOrCommand {
|
2022-05-13 12:50:25 +00:00
|
|
|
// 1. List files in $XDG_DATA_HOME/qutebrowser/sessions/ (N.B.:
|
|
|
|
// UserConfigDir() in os)
|
|
|
|
userConfigDir, err := os.UserConfigDir()
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("Error finding user configuration directory: %v", err)
|
|
|
|
return []sessionOrCommand{}
|
|
|
|
}
|
|
|
|
log.Printf("INFO userConfigDir: %+v", userConfigDir)
|
|
|
|
fileSystem := os.DirFS(userConfigDir)
|
|
|
|
log.Printf("INFO fileSystem: %+v", fileSystem)
|
|
|
|
fileList, err := fs.ReadDir(fileSystem, "/qutebrowser/sessions")
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("Error reading browser sessions directory: %v", err)
|
|
|
|
return []sessionOrCommand{}
|
|
|
|
}
|
|
|
|
// 2. Exclude non-YAML files
|
|
|
|
// 3. Wrangle them into this struct array
|
|
|
|
result := make([]sessionOrCommand, len(fileList))
|
|
|
|
for _, entry := range fileList {
|
|
|
|
if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".yml") {
|
|
|
|
result = append(result, sessionOrCommand{displayString: entry.Name(), commandString: ""})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// log.Printf("result: %v", result)
|
|
|
|
return result
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func initialModel() model {
|
2022-05-11 13:39:14 +00:00
|
|
|
browserSelection = make(SelectionSet)
|
|
|
|
commandSelection = make(SelectionSet)
|
2022-05-03 05:35:02 +00:00
|
|
|
return model{
|
2022-05-10 13:13:56 +00:00
|
|
|
activeSection: BrowserSessions,
|
2022-05-11 13:39:14 +00:00
|
|
|
cursor: []int{0, 0},
|
2022-05-10 13:13:56 +00:00
|
|
|
choices: Choices{
|
|
|
|
getBrowserSessions(),
|
|
|
|
[]sessionOrCommand{{
|
|
|
|
displayString: "neomutt",
|
|
|
|
commandString: "neomutt",
|
|
|
|
}, {
|
|
|
|
displayString: "bottom",
|
|
|
|
commandString: "btm --group --battery --color gruvbox-light",
|
|
|
|
}},
|
|
|
|
},
|
2022-05-12 07:11:26 +00:00
|
|
|
// An array of maps which indicates which choices are selected.
|
|
|
|
// We're using the map like a mathematical set. The keys refer to
|
|
|
|
// the indexes of the `choices` slice, above.
|
2022-05-11 13:39:14 +00:00
|
|
|
selected: []SelectionSet{browserSelection, commandSelection},
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-05-12 07:11:26 +00:00
|
|
|
func launch() tea.Msg {
|
|
|
|
return statusMsg("This is a test launch message.")
|
|
|
|
}
|
|
|
|
|
2022-05-03 05:35:02 +00:00
|
|
|
func (m model) Init() tea.Cmd {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
|
|
switch msg := msg.(type) {
|
2022-05-12 07:11:26 +00:00
|
|
|
case statusMsg:
|
|
|
|
m.status = string(msg)
|
|
|
|
return m, tea.Quit
|
2022-05-03 05:35:02 +00:00
|
|
|
case tea.KeyMsg:
|
|
|
|
switch msg.String() {
|
|
|
|
case "ctrl+c", "q":
|
|
|
|
return m, tea.Quit
|
2022-05-11 13:39:14 +00:00
|
|
|
case "tab":
|
|
|
|
if m.activeSection == BrowserSessions {
|
|
|
|
m.activeSection = Commands
|
|
|
|
} else {
|
|
|
|
m.activeSection = BrowserSessions
|
|
|
|
}
|
2022-05-03 05:35:02 +00:00
|
|
|
case "up", "k":
|
2022-05-10 13:13:56 +00:00
|
|
|
if m.cursor[m.activeSection] > 0 {
|
|
|
|
m.cursor[m.activeSection]--
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
case "down", "j":
|
2022-05-11 13:39:14 +00:00
|
|
|
if m.cursor[m.activeSection] < len(m.choices[m.activeSection])-1 {
|
2022-05-10 13:13:56 +00:00
|
|
|
m.cursor[m.activeSection]++
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
2022-05-12 07:11:26 +00:00
|
|
|
case " ":
|
2022-05-11 13:39:14 +00:00
|
|
|
_, ok := m.selected[m.activeSection][m.cursor[m.activeSection]]
|
|
|
|
if ok {
|
|
|
|
delete(m.selected[m.activeSection], m.cursor[m.activeSection])
|
2022-05-03 05:35:02 +00:00
|
|
|
} else {
|
2022-05-11 13:39:14 +00:00
|
|
|
m.selected[m.activeSection][m.cursor[m.activeSection]] = struct{}{}
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
2022-05-12 07:11:26 +00:00
|
|
|
case "enter":
|
|
|
|
return m, launch
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return m, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m model) View() string {
|
|
|
|
s := "What should we buy at the market?\n\n"
|
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
for j := 0; j < 2; j++ {
|
|
|
|
for i, choice := range m.choices[j] {
|
|
|
|
cursor := " "
|
|
|
|
if m.cursor[j] == i {
|
|
|
|
cursor = ">"
|
|
|
|
}
|
2022-05-03 05:35:02 +00:00
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
checked := " "
|
|
|
|
if _, ok := m.selected[j][i]; ok {
|
|
|
|
checked = "x"
|
|
|
|
}
|
2022-05-03 05:35:02 +00:00
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.displayString)
|
|
|
|
}
|
|
|
|
s += "\n\n"
|
2022-05-03 05:35:02 +00:00
|
|
|
}
|
|
|
|
|
2022-05-11 13:39:14 +00:00
|
|
|
// s += fmt.Sprintf("\n%+v", m.selected) // debug
|
2022-05-12 07:11:26 +00:00
|
|
|
s += fmt.Sprintf("%s\n", m.status)
|
2022-05-03 05:35:02 +00:00
|
|
|
s += "\nPress q to quit.\n"
|
|
|
|
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
p := tea.NewProgram(initialModel())
|
|
|
|
if err := p.Start(); err != nil {
|
|
|
|
fmt.Printf("Alas, there's been an error: %v", err)
|
|
|
|
os.Exit(1)
|
|
|
|
}
|
|
|
|
}
|