From 8e3f1bea410eca250b80e3d4b045f68278aa7f58 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Paul=20B=C3=BCtow?=
 <1224732+snonux@users.noreply.github.com>
Date: Fri, 20 Jun 2025 21:25:25 +0300
Subject: Fix undo for completed tasks

---
 internal/ui/table.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'internal/ui/table.go')

diff --git a/internal/ui/table.go b/internal/ui/table.go
index b175776..e293325 100644
--- a/internal/ui/table.go
+++ b/internal/ui/table.go
@@ -39,7 +39,7 @@ type Model struct {
 	filters []string
 	tasks   []task.Task
 
-	undoStack []int
+	undoStack []string
 
 	total      int
 	inProgress int
@@ -224,15 +224,20 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 				idStr := ansi.Strip(row[0])
 				if id, err := strconv.Atoi(idStr); err == nil {
 					task.Done(id)
-					m.undoStack = append(m.undoStack, id)
+					for _, tsk := range m.tasks {
+						if tsk.ID == id {
+							m.undoStack = append(m.undoStack, tsk.UUID)
+							break
+						}
+					}
 					m.reload()
 				}
 			}
 		case "U":
 			if n := len(m.undoStack); n > 0 {
-				id := m.undoStack[n-1]
+				uuid := m.undoStack[n-1]
 				m.undoStack = m.undoStack[:n-1]
-				task.SetStatus(id, "pending")
+				task.SetStatusUUID(uuid, "pending")
 				m.reload()
 			}
 		case "d":
-- 
cgit v1.2.3