1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
package main
import (
"encoding/csv"
"os"
"os/exec"
"strings"
tea "charm.land/bubbletea/v2"
)
type model struct {
choices [][]string
cursor int
selected map[int]struct{}
}
func initialModel() model {
out, err := exec.Command("hugo", "list", "drafts").Output()
if err != nil {
println("Error fetching Hugo drafts: Is this a Hugo site? Is Hugo installed?")
os.Exit(1)
}
csvReader := csv.NewReader(strings.NewReader(string(out)))
records, err := csvReader.ReadAll()
if err != nil {
println("Error parsing draft list.")
os.Exit(1)
}
if len(records) <= 1 {
println("No drafts found.")
os.Exit(1)
}
return model{
choices: records,
selected: make(map[int]struct{}),
}
}
func (m model) Init() tea.Cmd {
return nil
}
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
switch msg := msg.(type) {
case tea.KeyPressMsg:
switch msg.String() {
case "ctrl+c", "q":
return m, tea.Quit
case "up", "k":
if m.cursor > 0 {
m.cursor--
}
case "down", "j":
if m.cursor < len(m.choices)-1 {
m.cursor++
}
case "enter":
choice := m.choices[m.cursor]
path := choice[0]
return m, openEditor(path)
case "space":
_, ok := m.selected[m.cursor]
if ok {
delete(m.selected, m.cursor)
} else {
m.selected[m.cursor] = struct{}{}
}
}
case editorFinishedMsg:
return m, tea.Quit
}
return m, nil
}
|
