| 118 | } |
| 119 | |
| 120 | func (t *TaskTracker) MarkCurrentAs(status TaskStatus, errorMsg string) { |
| 121 | t.mu.Lock() |
| 122 | defer t.mu.Unlock() |
| 123 | |
| 124 | if t.plan == nil || len(t.plan.Tasks) == 0 { |
| 125 | return |
| 126 | } |
| 127 | |
| 128 | if t.plan.CurrentTask >= len(t.plan.Tasks) { |
| 129 | return |
| 130 | } |
| 131 | |
| 132 | task := t.plan.Tasks[t.plan.CurrentTask] |
| 133 | task.Status = status |
| 134 | task.Attempts++ |
| 135 | |
| 136 | switch status { |
| 137 | case TaskInProgress: |
| 138 | if task.StartedAt.IsZero() { |
| 139 | task.StartedAt = time.Now() |
| 140 | } |
| 141 | case TaskCompleted: |
| 142 | task.CompletedAt = time.Now() |
| 143 | t.plan.CurrentTask++ |
| 144 | case TaskFailed: |
| 145 | task.Error = errorMsg |
| 146 | t.plan.FailureCount++ |
| 147 | if t.plan.FailureCount >= 3 { |
| 148 | t.plan.NeedsReplan = true |
| 149 | } |
| 150 | } |
| 151 | |
| 152 | t.plan.UpdatedAt = time.Now() |
| 153 | } |
| 154 | |
| 155 | func (t *TaskTracker) GetCurrentTask() *Task { |
| 156 | t.mu.Lock() |