1package main
  2
  3import (
  4	"fmt"
  5	"log"
  6	"os"
  7	"strings"
  8
  9	todo "github.com/1set/todotxt"
 10	"github.com/charmbracelet/bubbles/help"
 11	"github.com/charmbracelet/bubbles/key"
 12	"github.com/charmbracelet/bubbles/textinput"
 13	tea "github.com/charmbracelet/bubbletea"
 14)
 15
 16type InterfaceState int
 17
 18const (
 19	List InterfaceState = iota
 20	Add
 21)
 22
 23type model struct {
 24	interfaceState InterfaceState
 25	config         config
 26	keys           keyMap
 27	help           help.Model
 28
 29	tasks    todo.TaskList
 30	cursor   int              // which to-do list item our cursor is pointing at
 31	selected map[int]struct{} // which to-do items are selected
 32
 33	textInput textinput.Model
 34}
 35
 36func initialModel() model {
 37
 38	config := NewConfig()
 39
 40	// New task input
 41	ti := textinput.New()
 42	ti.Focus()
 43	ti.CharLimit = 200
 44	ti.Width = 200
 45
 46	if tasklist, err := todo.LoadFromPath(config.taskFilePath); err != nil {
 47		log.Fatal(err)
 48		return model{}
 49	} else {
 50		tasks := tasklist
 51		tasks.Sort(todo.SortCompletedDateAsc, todo.SortDueDateAsc, todo.SortPriorityAsc)
 52		selected := make(map[int]struct{})
 53
 54		interfaceState := List
 55		if len(tasks) == 0 {
 56			interfaceState = Add
 57		}
 58
 59		for i, t := range tasks {
 60			if t.Completed {
 61				selected[i] = struct{}{}
 62			}
 63		}
 64
 65		return model{
 66			interfaceState: interfaceState,
 67			config:         config,
 68			tasks:          tasks,
 69			selected:       selected,
 70			textInput:      ti,
 71			keys:           keys,
 72			help:           help.New(),
 73		}
 74
 75	}
 76
 77}
 78
 79func (m model) Init() tea.Cmd {
 80	return textinput.Blink
 81}
 82
 83func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 84	var cmd tea.Cmd
 85
 86	switch msg := msg.(type) {
 87
 88	case tea.WindowSizeMsg:
 89		// If we set a width on the help menu it can gracefully truncate
 90		// its view as needed.
 91		m.help.Width = msg.Width
 92
 93	// Is it a key press?
 94	case tea.KeyMsg:
 95
 96		switch m.interfaceState {
 97
 98		case List:
 99
100			switch {
101			// Navigation
102			case key.Matches(msg, m.keys.Up):
103				if m.cursor > 0 {
104					m.cursor--
105				}
106
107			case key.Matches(msg, m.keys.Down):
108				if m.cursor < len(m.tasks)-1 {
109					m.cursor++
110				}
111
112			// Tasks management
113			case key.Matches(msg, m.keys.Priority):
114				m.tasks[m.cursor].Priority = strings.ToUpper(msg.String())
115
116			case key.Matches(msg, m.keys.ClearPriority):
117				m.tasks[m.cursor].Priority = ""
118
119			case key.Matches(msg, m.keys.Check):
120				_, ok := m.selected[m.cursor]
121
122				if ok {
123					delete(m.selected, m.cursor)
124					m.tasks[m.cursor].Reopen()
125				} else {
126					m.selected[m.cursor] = struct{}{}
127					m.tasks[m.cursor].Complete()
128				}
129
130			case key.Matches(msg, m.keys.Sort):
131				m.tasks.Sort(todo.SortCompletedDateAsc, todo.SortDueDateAsc, todo.SortPriorityAsc)
132				m.selected = make(map[int]struct{})
133				for i, t := range m.tasks {
134					if t.Completed {
135						m.selected[i] = struct{}{}
136					}
137				}
138
139			case key.Matches(msg, m.keys.Clean):
140				m.selected = make(map[int]struct{})
141				doneTasks := m.tasks.Filter(todo.FilterCompleted)
142				doneFile, err := os.OpenFile(m.config.doneFilePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
143				if err != nil {
144					log.Fatal("Can't write done tasks to " + m.config.doneFilePath)
145				}
146				defer doneFile.Close()
147				if writeErr := doneTasks.WriteToFile(doneFile); writeErr != nil {
148					log.Fatal(writeErr)
149				}
150				m.tasks = m.tasks.Filter(todo.FilterNotCompleted)
151				m.selected = make(map[int]struct{})
152				for i, t := range m.tasks {
153					if t.Completed {
154						m.selected[i] = struct{}{}
155					}
156				}
157
158			// Interface
159			case key.Matches(msg, m.keys.Add):
160				m.interfaceState = Add
161
162			case key.Matches(msg, m.keys.SaveQuit):
163				m.tasks.WriteToPath(m.config.taskFilePath)
164				return m, tea.Quit
165
166			case key.Matches(msg, m.keys.Quit):
167				return m, tea.Quit
168
169			case key.Matches(msg, m.keys.Help):
170				m.help.ShowAll = !m.help.ShowAll
171
172			}
173
174			return m, nil
175
176		case Add:
177			switch msg.Type {
178			case tea.KeyCtrlC, tea.KeyEsc:
179				return m, tea.Quit
180			case tea.KeyEnter:
181				inputValue := m.textInput.Value()
182				if inputValue != "" {
183					newTask, _ := todo.ParseTask(inputValue)
184					m.tasks.AddTask(newTask)
185				}
186				m.textInput.Reset()
187				m.interfaceState = List
188			}
189
190			m.textInput, cmd = m.textInput.Update(msg)
191
192			return m, cmd
193
194		}
195	}
196
197	return m, nil
198}
199
200func (m model) View() string {
201	output := "\n"
202
203	switch m.interfaceState {
204
205	case List:
206		if len(m.tasks) > 0 {
207
208			for i, task := range m.tasks {
209
210				cursor := " "
211				if m.cursor == i {
212					cursor = "→"
213				}
214
215				// Render the row
216				_, checked := m.selected[i]
217				styles := NewTextStyle()
218				output += fmt.Sprintf("%s %s\n", cursor, styles.getTaskStyle(task, checked))
219			}
220
221		} else {
222			output += "No tasks in file"
223		}
224
225		output += "\n"
226		output += m.help.View(m.keys)
227	case Add:
228		output += fmt.Sprintf("Nouvelle tâche:\n\n%s", m.textInput.View())
229		output += "\n\n Press ESC to quit.\n"
230	}
231
232	return output
233
234}
235
236func main() {
237	p := tea.NewProgram(initialModel(), tea.WithAltScreen())
238	if _, err := p.Run(); err != nil {
239		fmt.Printf("Alas, there's been an error: %v", err)
240		os.Exit(1)
241	}
242}