team-mode: auto-claim pending tasks on start

Author: code-yeongyu

src/features/team-mode/team-tasklist/update.test.ts

@@ -29,6 +29,29 @@ test("updateTaskStatus supports the one-way claim to complete flow", async () =>
   }
 })
 
+test("updateTaskStatus auto-claims when a member starts a pending task directly", async () => {
+  // given
+  const fixture = await createTasklistFixture()
+
+  try {
+    const task = await createTask(fixture.teamRunId, createTaskInput(), fixture.config)
+
+    // when
+    const inProgressTask = await updateTaskStatus(fixture.teamRunId, task.id, "in_progress", "member-a", fixture.config)
+    const loadedTask = await getTask(fixture.teamRunId, task.id, fixture.config)
+
+    // then
+    expect(inProgressTask.status).toBe("in_progress")
+    expect(inProgressTask.owner).toBe("member-a")
+    expect(typeof inProgressTask.claimedAt).toBe("number")
+    expect(loadedTask.status).toBe("in_progress")
+    expect(loadedTask.owner).toBe("member-a")
+    expect(typeof loadedTask.claimedAt).toBe("number")
+  } finally {
+    await fixture.cleanup()
+  }
+})
+
 test("updateTaskStatus rejects reverse transitions", async () => {
   // given
   const fixture = await createTasklistFixture()

src/features/team-mode/team-tasklist/update.ts

@@ -5,6 +5,7 @@ import { getTasksDir, resolveBaseDir } from "../team-registry"
 import { atomicWrite } from "../team-state-store/locks"
 import { TaskSchema } from "../types"
 import type { Task } from "../types"
+import { claimTask } from "./claim"
 import { getTask } from "./get"
 
 const ALLOWED_TRANSITIONS: Readonly<Record<Task["status"], ReadonlyArray<Task["status"]>>> = {
@@ -45,6 +46,11 @@ export async function updateTaskStatus(
 
   if (task.status === newStatus) return task
 
+  if (task.status === "pending" && newStatus === "in_progress") {
+    await claimTask(teamRunId, taskId, memberName, config)
+    return updateTaskStatus(teamRunId, taskId, newStatus, memberName, config)
+  }
+
   if (!isValidTransition(task.status, newStatus)) {
     throw new InvalidTaskTransitionError(task.status, newStatus)
   }