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.Check):
117				_, ok := m.selected[m.cursor]
118
119				if ok {
120					delete(m.selected, m.cursor)
121					m.tasks[m.cursor].Reopen()
122				} else {
123					m.selected[m.cursor] = struct{}{}
124					m.tasks[m.cursor].Complete()
125				}
126
127			case key.Matches(msg, m.keys.Sort):
128				m.tasks.Sort(todo.SortCompletedDateAsc, todo.SortDueDateAsc, todo.SortPriorityAsc)
129				m.selected = make(map[int]struct{})
130				for i, t := range m.tasks {
131					if t.Completed {
132						m.selected[i] = struct{}{}
133					}
134				}
135
136			case key.Matches(msg, m.keys.Clean):
137				m.selected = make(map[int]struct{})
138				doneTasks := m.tasks.Filter(todo.FilterCompleted)
139				doneFile, err := os.OpenFile(m.config.doneFilePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
140				if err != nil {
141					log.Fatal("Can't write done tasks to " + m.config.doneFilePath)
142				}
143				defer doneFile.Close()
144				if writeErr := doneTasks.WriteToFile(doneFile); writeErr != nil {
145					log.Fatal(writeErr)
146				}
147				m.tasks = m.tasks.Filter(todo.FilterNotCompleted)
148				m.selected = make(map[int]struct{})
149				for i, t := range m.tasks {
150					if t.Completed {
151						m.selected[i] = struct{}{}
152					}
153				}
154
155			// Interface
156			case key.Matches(msg, m.keys.Add):
157				m.interfaceState = Add
158
159			case key.Matches(msg, m.keys.SaveQuit):
160				m.tasks.WriteToPath(m.config.taskFilePath)
161				return m, tea.Quit
162
163			case key.Matches(msg, m.keys.Quit):
164				return m, tea.Quit
165
166			case key.Matches(msg, m.keys.Help):
167				m.help.ShowAll = !m.help.ShowAll
168
169			}
170
171			return m, nil
172
173		case Add:
174			switch msg.Type {
175			case tea.KeyCtrlC, tea.KeyEsc:
176				return m, tea.Quit
177			case tea.KeyEnter:
178				inputValue := m.textInput.Value()
179				if inputValue != "" {
180					newTask, _ := todo.ParseTask(inputValue)
181					m.tasks.AddTask(newTask)
182				}
183				m.textInput.Reset()
184				m.interfaceState = List
185			}
186
187			m.textInput, cmd = m.textInput.Update(msg)
188
189			return m, cmd
190
191		}
192	}
193
194	return m, nil
195}
196
197func (m model) View() string {
198	output := "\n"
199
200	switch m.interfaceState {
201
202	case List:
203		if len(m.tasks) > 0 {
204
205			for i, task := range m.tasks {
206
207				cursor := " "
208				if m.cursor == i {
209					cursor = "→"
210				}
211
212				// Render the row
213				_, checked := m.selected[i]
214				styles := NewTextStyle()
215				output += fmt.Sprintf("%s %s\n", cursor, styles.getTaskStyle(task, checked))
216			}
217
218		} else {
219			output += "No tasks in file"
220		}
221
222		output += "\n"
223		output += m.help.View(m.keys)
224	case Add:
225		output += fmt.Sprintf("Nouvelle tâche:\n\n%s", m.textInput.View())
226		output += "\n\n Press ESC to quit.\n"
227	}
228
229	return output
230
231}
232
233func main() {
234	p := tea.NewProgram(initialModel(), tea.WithAltScreen())
235	if _, err := p.Run(); err != nil {
236		fmt.Printf("Alas, there's been an error: %v", err)
237		os.Exit(1)
238	}
239}