Compare commits

..

No commits in common. "main" and "003-get-qutebrowser-sessions" have entirely different histories.

2 changed files with 80 additions and 103 deletions

3
.gitignore vendored
View file

@ -17,6 +17,3 @@ breakfast
# Go workspace file # Go workspace file
go.work go.work
# Logs
logs/

158
main.go
View file

@ -1,115 +1,97 @@
package main package main
// Testing out tea; really, git and ssh
import ( import (
"fmt" "fmt"
"io/fs" "io/fs"
"log" "log"
"os" "os"
"os/exec"
"strings" "strings"
"time"
tea "github.com/charmbracelet/bubbletea" tea "github.com/charmbracelet/bubbletea"
) )
type commandChoice struct { // Enum for the different launch types (browsers and CLI's)
const (
BrowserSessions int = iota
Commands
)
type sessionOrCommand struct {
displayString string displayString string
command *exec.Cmd commandString string
} }
type choices []commandChoice type Choices [][]sessionOrCommand
type SelectionSet map[int]struct{}
type model struct { type model struct {
cursor int activeSection int
choices choices cursor []int
selected map[int]struct{} choices Choices
selected []SelectionSet
status string status string
} }
type statusMsg string type statusMsg string
var selection map[int]struct{} var browserSelection SelectionSet
var commandSelection SelectionSet
func getBrowserSessions() []commandChoice { func getBrowserSessions() []sessionOrCommand {
// 1. List files in $XDG_DATA_HOME/qutebrowser/sessions/ (N.B.:
// UserConfigDir() in os)
userConfigDir, err := os.UserConfigDir() userConfigDir, err := os.UserConfigDir()
if err != nil { if err != nil {
log.Printf("Error finding user configuration directory: %v", err) log.Printf("Error finding user configuration directory: %v", err)
return []commandChoice{} return []sessionOrCommand{}
} }
log.Printf("userConfigDir: %+v", userConfigDir) log.Printf("INFO userConfigDir: %+v", userConfigDir)
fileSystem := os.DirFS(userConfigDir) fileSystem := os.DirFS(userConfigDir)
log.Printf("fileSystem: %+v", fileSystem) log.Printf("INFO fileSystem: %+v", fileSystem)
fileList, err := fs.ReadDir(fileSystem, "local/share/qutebrowser/sessions") fileList, err := fs.ReadDir(fileSystem, "local/share/qutebrowser/sessions")
if err != nil { if err != nil {
log.Printf("Error reading browser sessions directory: %v", err) log.Printf("Error reading browser sessions directory: %v", err)
return []commandChoice{} return []sessionOrCommand{}
} }
result := make([]commandChoice, 0) // 2. Exclude non-YAML files
// 3. Wrangle them into this struct array
result := make([]sessionOrCommand, 0)
for _, entry := range fileList { for _, entry := range fileList {
if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".yml") { if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".yml") {
log.Printf("%s", entry.Name()) log.Printf("INFO %s", entry.Name())
result = append(result, commandChoice{ result = append(result, sessionOrCommand{displayString: strings.TrimSuffix(entry.Name(), ".yml"), commandString: ""})
displayString: fmt.Sprintf("Qutebrowser session: %s", strings.TrimSuffix(entry.Name(), ".yml")),
command: exec.Command("xterm", "-e", "qutebrowser", "--target", "window", "--restore", strings.TrimSuffix(entry.Name(), ".yml")),
})
} }
} }
// log.Printf("result: %v", result)
return result return result
} }
func initialModel() model { func initialModel() model {
browserSelection = make(SelectionSet)
commandSelection = make(SelectionSet)
return model{ return model{
cursor: 0, activeSection: BrowserSessions,
choices: append( cursor: []int{0, 0},
choices: Choices{
getBrowserSessions(), getBrowserSessions(),
commandChoice{ []sessionOrCommand{{
displayString: "bottom",
command: exec.Command("xterm", "-maximized", "-e", "btm", "--battery", "--color", "gruvbox-light"),
},
commandChoice{
displayString: "broot",
command: exec.Command("xterm", "-maximized", "-e", "broot"),
},
commandChoice{
displayString: "joplin",
command: exec.Command("xterm", "-maximized", "-e", "joplin"),
},
commandChoice{
displayString: "neomutt", displayString: "neomutt",
command: exec.Command("xterm", "-maximized", "-e", "neomutt"), commandString: "neomutt",
}, {
displayString: "bottom",
commandString: "btm --group --battery --color gruvbox-light",
}},
}, },
commandChoice{ // An array of maps which indicates which choices are selected.
displayString: "newsboat", // We're using the map like a mathematical set. The keys refer to
command: exec.Command("xterm", "-maximized", "-e", "newsboat"), // the indexes of the `choices` slice, above.
}, selected: []SelectionSet{browserSelection, commandSelection},
commandChoice{
displayString: "w3m",
command: exec.Command("xterm", "-maximized", "-e", "w3m", "https://languagehat.com/"),
},
),
selected: make(map[int]struct{}),
} }
} }
func launch(m model) tea.Cmd { func launch() tea.Msg {
return func() tea.Msg { return statusMsg("This is a test launch message.")
var result statusMsg
for q, item := range m.choices {
if _, ok := m.selected[q]; ok {
log.Printf("Launching: %v\n", item.displayString)
result += statusMsg(fmt.Sprintf("Launching command: %v\n", item.displayString))
err := item.command.Start()
if err != nil {
result += statusMsg(fmt.Sprintf("%v\n", err))
log.Fatalf("Error launching: %v\n", err)
}
}
}
time.Sleep(2000 * time.Millisecond)
return result
}
} }
func (m model) Init() tea.Cmd { func (m model) Init() tea.Cmd {
@ -125,23 +107,29 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
switch msg.String() { switch msg.String() {
case "ctrl+c", "q": case "ctrl+c", "q":
return m, tea.Quit return m, tea.Quit
case "tab":
if m.activeSection == BrowserSessions {
m.activeSection = Commands
} else {
m.activeSection = BrowserSessions
}
case "up", "k": case "up", "k":
if m.cursor > 0 { if m.cursor[m.activeSection] > 0 {
m.cursor-- m.cursor[m.activeSection]--
} }
case "down", "j": case "down", "j":
if m.cursor < len(m.choices)-1 { if m.cursor[m.activeSection] < len(m.choices[m.activeSection])-1 {
m.cursor++ m.cursor[m.activeSection]++
} }
case " ": case " ":
_, ok := m.selected[m.cursor] _, ok := m.selected[m.activeSection][m.cursor[m.activeSection]]
if ok { if ok {
delete(m.selected, m.cursor) delete(m.selected[m.activeSection], m.cursor[m.activeSection])
} else { } else {
m.selected[m.cursor] = struct{}{} m.selected[m.activeSection][m.cursor[m.activeSection]] = struct{}{}
} }
case "enter": case "enter":
return m, launch(m) return m, launch
} }
} }
@ -149,41 +137,33 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
} }
func (m model) View() string { func (m model) View() string {
s := "\nLet's get started!\n\n" s := "Let's get started!\n\n"
for i, choice := range m.choices { for j := 0; j < 2; j++ {
for i, choice := range m.choices[j] {
cursor := " " cursor := " "
if m.cursor == i { if m.cursor[j] == i {
cursor = ">" cursor = ">"
} }
checked := " " checked := " "
if _, ok := m.selected[i]; ok { if _, ok := m.selected[j][i]; ok {
checked = "x" checked = "x"
} }
s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.displayString) s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.displayString)
} }
s += "\n\n" s += "\n\n"
}
// s += fmt.Sprintf("\n%+v", m.selected) // debug
s += fmt.Sprintf("%s\n", m.status) s += fmt.Sprintf("%s\n", m.status)
s += "Press enter to launch.\n" s += "\nPress q to quit.\n"
s += "Press q to quit.\n"
return s return s
} }
func main() { func main() {
logfilePath := os.Getenv("BREAKFAST_LOG")
if logfilePath == "" {
logfilePath = fmt.Sprintf("/tmp/%s_breakfast.log", time.Now().Local().Format("20060102150405-0700"))
}
f, err := tea.LogToFile(logfilePath, "DEBUG")
if err != nil {
log.Fatal(err)
}
defer f.Close()
p := tea.NewProgram(initialModel()) p := tea.NewProgram(initialModel())
if err := p.Start(); err != nil { if err := p.Start(); err != nil {
fmt.Printf("Alas, there's been an error: %v", err) fmt.Printf("Alas, there's been an error: %v", err)