improve task status

mafredri · mafredri · commit 59c9c7e9c96c · 2025-08-22T08:31:51.000Z
diff --git a/coderd/aitasks.go b/coderd/aitasks.go
@@ -199,28 +199,39 @@ type tasksListResponse struct {
 }
 
 func mapTaskStatus(ws codersdk.Workspace) codersdk.TaskStatus {
-	if ws.LatestAppStatus != nil {
-		switch ws.LatestAppStatus.State {
-		case codersdk.WorkspaceAppStatusStateWorking:
-			return codersdk.TaskStatusWorking
-		case codersdk.WorkspaceAppStatusStateIdle:
-			return codersdk.TaskStatusIdle
-		case codersdk.WorkspaceAppStatusStateComplete:
-			return codersdk.TaskStatusCompleted
-		case codersdk.WorkspaceAppStatusStateFailure:
-			return codersdk.TaskStatusFailed
+	switch ws.LatestBuild.Status {
+	case codersdk.WorkspaceStatusPending:
+		return codersdk.TaskStatusPending
+
+	case codersdk.WorkspaceStatusStarting:
+		return codersdk.TaskStatusStarting
+
+	case codersdk.WorkspaceStatusRunning:
+		if ws.LatestAppStatus != nil {
+			switch ws.LatestAppStatus.State {
+			case codersdk.WorkspaceAppStatusStateWorking:
+				return codersdk.TaskStatusWorking
+			case codersdk.WorkspaceAppStatusStateIdle:
+				return codersdk.TaskStatusIdle
+			case codersdk.WorkspaceAppStatusStateComplete:
+				return codersdk.TaskStatusCompleted
+			case codersdk.WorkspaceAppStatusStateFailure:
+				return codersdk.TaskStatusFailed
+			}
 		}
-	}
+		return codersdk.TaskStatusStarting
+
+	case codersdk.WorkspaceStatusStopping, codersdk.WorkspaceStatusStopped:
+		return codersdk.TaskStatusStopping
+
+	case codersdk.WorkspaceStatusDeleting, codersdk.WorkspaceStatusDeleted:
+		return codersdk.TaskStatusDeleting
 
-	switch ws.LatestBuild.Status {
-	case codersdk.WorkspaceStatusPending, codersdk.WorkspaceStatusStarting, codersdk.WorkspaceStatusRunning:
-		return codersdk.TaskStatusWorking
-	case codersdk.WorkspaceStatusStopping, codersdk.WorkspaceStatusStopped, codersdk.WorkspaceStatusDeleting, codersdk.WorkspaceStatusDeleted:
-		return codersdk.TaskStatusCompleted
 	case codersdk.WorkspaceStatusFailed, codersdk.WorkspaceStatusCanceling, codersdk.WorkspaceStatusCanceled:
 		return codersdk.TaskStatusFailed
+
 	default:
-		return codersdk.TaskStatusWorking
+		return codersdk.TaskStatusPending
 	}
 }
 
diff --git a/codersdk/aitasks.go b/codersdk/aitasks.go
@@ -78,10 +78,12 @@ func (c *ExperimentalClient) CreateTask(ctx context.Context, user string, reques
 type TaskStatus string
 
 const (
-	TaskStatusQueued    TaskStatus = "queued"
+	TaskStatusPending   TaskStatus = "pending"
+	TaskStatusStarting  TaskStatus = "starting"
+	TaskStatusStopping  TaskStatus = "stopping"
+	TaskStatusDeleting  TaskStatus = "deleting"
 	TaskStatusWorking   TaskStatus = "working"
 	TaskStatusIdle      TaskStatus = "idle"
-	TaskStatusPaused    TaskStatus = "paused"
 	TaskStatusCompleted TaskStatus = "completed"
 	TaskStatusFailed    TaskStatus = "failed"
 )
@@ -105,7 +107,7 @@ type Task struct {
 	TemplateID     uuid.UUID     `json:"template_id" format:"uuid"`
 	WorkspaceID    uuid.NullUUID `json:"workspace_id" format:"uuid"`
 	Prompt         string        `json:"prompt"`
-	Status         TaskStatus    `json:"status" enum:"queued,working,idle,paused,completed,failed"`
+	Status         TaskStatus    `json:"status" enum:"pending,starting,stopping,deleting,working,idle,completed,failed"`
 	CreatedAt      time.Time     `json:"created_at" format:"date-time"`
 	UpdatedAt      time.Time     `json:"updated_at" format:"date-time"`
 }
