breakfast/main.go

191 lines
4.3 KiB
Go
Raw Permalink Normal View History

2022-05-03 05:35:02 +00:00
package main
2023-02-11 14:51:18 +00:00
// Testing out tea; really, git and ssh
2022-05-03 05:35:02 +00:00
import (
"fmt"
2022-05-13 12:50:25 +00:00
"io/fs"
"log"
2022-05-03 05:35:02 +00:00
"os"
2022-05-14 23:06:06 +00:00
"os/exec"
2022-05-13 12:50:25 +00:00
"strings"
"time"
2022-05-03 05:35:02 +00:00
tea "github.com/charmbracelet/bubbletea"
)
type commandChoice struct {
2022-05-03 05:35:02 +00:00
displayString string
command *exec.Cmd
2022-05-03 05:35:02 +00:00
}
type choices []commandChoice
2022-05-03 05:35:02 +00:00
type model struct {
cursor int
choices choices
selected map[int]struct{}
status string
2022-05-03 05:35:02 +00:00
}
2022-05-12 07:11:26 +00:00
type statusMsg string
var selection map[int]struct{}
func getBrowserSessions() []commandChoice {
2022-05-13 12:50:25 +00:00
userConfigDir, err := os.UserConfigDir()
if err != nil {
log.Printf("Error finding user configuration directory: %v", err)
return []commandChoice{}
2022-05-13 12:50:25 +00:00
}
log.Printf("userConfigDir: %+v", userConfigDir)
2022-05-13 12:50:25 +00:00
fileSystem := os.DirFS(userConfigDir)
log.Printf("fileSystem: %+v", fileSystem)
2022-05-13 23:00:36 +00:00
fileList, err := fs.ReadDir(fileSystem, "local/share/qutebrowser/sessions")
2022-05-13 12:50:25 +00:00
if err != nil {
log.Printf("Error reading browser sessions directory: %v", err)
return []commandChoice{}
2022-05-13 12:50:25 +00:00
}
result := make([]commandChoice, 0)
2022-05-13 12:50:25 +00:00
for _, entry := range fileList {
if !entry.IsDir() && strings.HasSuffix(entry.Name(), ".yml") {
log.Printf("%s", entry.Name())
result = append(result, commandChoice{
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")),
})
2022-05-13 12:50:25 +00:00
}
}
return result
2022-05-03 05:35:02 +00:00
}
func initialModel() model {
return model{
cursor: 0,
choices: append(
2022-05-10 13:13:56 +00:00
getBrowserSessions(),
commandChoice{
2022-05-10 13:13:56 +00:00
displayString: "bottom",
command: exec.Command("xterm", "-maximized", "-e", "btm", "--battery", "--color", "gruvbox-light"),
},
commandChoice{
displayString: "broot",
command: exec.Command("xterm", "-maximized", "-e", "broot"),
},
commandChoice{
2022-05-16 17:07:51 +00:00
displayString: "joplin",
command: exec.Command("xterm", "-maximized", "-e", "joplin"),
},
commandChoice{
2022-05-16 17:07:51 +00:00
displayString: "neomutt",
command: exec.Command("xterm", "-maximized", "-e", "neomutt"),
},
commandChoice{
2022-05-16 17:07:51 +00:00
displayString: "newsboat",
command: exec.Command("xterm", "-maximized", "-e", "newsboat"),
},
2022-05-20 13:00:37 +00:00
commandChoice{
displayString: "w3m",
command: exec.Command("xterm", "-maximized", "-e", "w3m", "https://languagehat.com/"),
},
),
selected: make(map[int]struct{}),
2022-05-03 05:35:02 +00:00
}
}
2022-05-14 23:06:06 +00:00
func launch(m model) tea.Cmd {
return func() tea.Msg {
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)
2022-05-14 23:06:06 +00:00
return result
}
2022-05-12 07:11:26 +00:00
}
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
case "up", "k":
if m.cursor > 0 {
m.cursor--
2022-05-03 05:35:02 +00:00
}
case "down", "j":
if m.cursor < len(m.choices)-1 {
m.cursor++
2022-05-03 05:35:02 +00:00
}
2022-05-12 07:11:26 +00:00
case " ":
_, ok := m.selected[m.cursor]
if ok {
delete(m.selected, m.cursor)
2022-05-03 05:35:02 +00:00
} else {
m.selected[m.cursor] = struct{}{}
2022-05-03 05:35:02 +00:00
}
2022-05-12 07:11:26 +00:00
case "enter":
2022-05-14 23:06:06 +00:00
return m, launch(m)
2022-05-03 05:35:02 +00:00
}
}
return m, nil
}
func (m model) View() string {
s := "\nLet's get started!\n\n"
2022-05-03 05:35:02 +00:00
for i, choice := range m.choices {
cursor := " "
if m.cursor == i {
cursor = ">"
}
2022-05-03 05:35:02 +00:00
checked := " "
if _, ok := m.selected[i]; ok {
checked = "x"
}
s += fmt.Sprintf("%s [%s] %s\n", cursor, checked, choice.displayString)
2022-05-03 05:35:02 +00:00
}
s += "\n\n"
2022-05-03 05:35:02 +00:00
2022-05-12 07:11:26 +00:00
s += fmt.Sprintf("%s\n", m.status)
2022-05-15 00:05:54 +00:00
s += "Press enter to launch.\n"
s += "Press q to quit.\n"
2022-05-03 05:35:02 +00:00
return s
}
func main() {
logfilePath := os.Getenv("BREAKFAST_LOG")
if logfilePath == "" {
logfilePath = fmt.Sprintf("/tmp/%s_breakfast.log", time.Now().Local().Format("20060102150405-0700"))
}
if _, err := tea.LogToFile(logfilePath, "DEBUG"); err != nil {
log.Fatal(err)
2022-05-18 12:50:11 +00:00
}
2022-05-03 05:35:02 +00:00
p := tea.NewProgram(initialModel())
if err := p.Start(); err != nil {
fmt.Printf("Alas, there's been an error: %v", err)
os.Exit(1)
}
}