From 3dadea457aa7b034041d0111573f5b23662719c1 Mon Sep 17 00:00:00 2001
From: Ian Gulliver <github@flamingcow.io>
Date: Thu, 3 Jul 2025 23:15:46 -0700
Subject: [PATCH] SetTaskSuccess should return the next task

---
 taskcp.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/taskcp.go b/taskcp.go
index 3b1bef9..d6657a6 100644
--- a/taskcp.go
+++ b/taskcp.go
@@ -79,16 +79,22 @@ func (p *Project) InsertTaskBefore(id string, instructions string, changeCallbac
 }
 
 func (p *Project) GetNextTask() *Task {
+	if p.NextTaskID == "" {
+		return nil
+	}
+
 	return p.Tasks[p.NextTaskID]
 }
 
-func (p *Project) SetTaskSuccess(id string, result string, notes string) {
+func (p *Project) SetTaskSuccess(id string, result string, notes string) *Task {
 	task := p.Tasks[id]
 	task.State = TaskStateSuccess
 	task.Result = result
 	task.Notes = notes
 	task.ChangeCallback(task)
 	p.NextTaskID = task.NextTaskID
+
+	return p.GetNextTask()
 }
 
 func (p *Project) SetTaskFailure(id string, error string, notes string) {