Carlos Florêncio
41 lines
1.6 KiB
TypeScript
41 lines
1.6 KiB
TypeScript
import type { TaskState, TaskSessionState } from "@/lib/types/http";
|
|
|
|
// Single source of truth for bucketing a task into review / in_progress /
|
|
// backlog. Used by the sidebar sort in task-switcher and the per-item
|
|
// spinner icon in task-item so the two views always agree.
|
|
|
|
const REVIEW_STATES = new Set<TaskSessionState>([
|
|
"WAITING_FOR_INPUT",
|
|
"COMPLETED",
|
|
"FAILED",
|
|
"CANCELLED",
|
|
]);
|
|
const IN_PROGRESS_STATES = new Set<TaskSessionState>(["RUNNING"]);
|
|
const TASK_STATE_REVIEW = new Set<TaskState | undefined>(["REVIEW", "COMPLETED"]);
|
|
const TASK_STATE_IN_PROGRESS = new Set<TaskState | undefined>(["IN_PROGRESS", "SCHEDULING"]);
|
|
|
|
export type TaskBucket = "review" | "in_progress" | "backlog";
|
|
|
|
export function classifyTask(
|
|
sessionState: TaskSessionState | undefined,
|
|
taskState?: TaskState,
|
|
): TaskBucket {
|
|
if (!sessionState) {
|
|
if (TASK_STATE_REVIEW.has(taskState)) return "review";
|
|
if (TASK_STATE_IN_PROGRESS.has(taskState)) return "in_progress";
|
|
return "backlog";
|
|
}
|
|
// CREATED/STARTING are transient session states while the agent process is
|
|
// booting or reattaching. Defer to the workflow task state so an
|
|
// IN_PROGRESS task keeps showing activity, while a REVIEW task does not
|
|
// flicker out of the review bucket during auto-resume.
|
|
if ((sessionState === "CREATED" || sessionState === "STARTING") && taskState) {
|
|
if (TASK_STATE_REVIEW.has(taskState)) return "review";
|
|
if (TASK_STATE_IN_PROGRESS.has(taskState)) return "in_progress";
|
|
return "backlog";
|
|
}
|
|
if (REVIEW_STATES.has(sessionState)) return "review";
|
|
if (IN_PROGRESS_STATES.has(sessionState)) return "in_progress";
|
|
return "backlog";
|
|
}
|