feat(hooks/atlas): use getWorkForSession in boulder lookups and session tracking

2026-05-11 13:37:19 +09:00
parent 42db7078af
commit 18af3d3617
4 changed files with 234 additions and 17 deletions
@@ -131,4 +131,77 @@ describe("resolveActiveBoulderSession", () => {
      rmSync(worktreeDirectory, { recursive: true, force: true })
    }
  })
+
+  test("uses work resolved by session id when works map is present", async () => {
+    // given
+    const legacyPlanPath = join(testDirectory, "legacy-plan.md")
+    const workAPlanPath = join(testDirectory, "work-a-plan.md")
+    const workBPlanPath = join(testDirectory, "work-b-plan.md")
+    writeFileSync(legacyPlanPath, "# Plan\n- [ ] Legacy\n", "utf-8")
+    writeFileSync(workAPlanPath, "# Plan\n- [ ] Work A\n", "utf-8")
+    writeFileSync(workBPlanPath, "# Plan\n- [x] Work B\n", "utf-8")
+
+    writeBoulderState(testDirectory, {
+      schema_version: 2,
+      active_work_id: "work-a",
+      active_plan: legacyPlanPath,
+      started_at: "2026-01-02T10:00:00Z",
+      session_ids: ["ses_legacy"],
+      plan_name: "legacy-plan",
+      works: {
+        "work-a": {
+          work_id: "work-a",
+          active_plan: workAPlanPath,
+          plan_name: "work-a-plan",
+          started_at: "2026-01-02T10:00:00Z",
+          session_ids: ["ses_work_a"],
+          status: "active",
+        },
+        "work-b": {
+          work_id: "work-b",
+          active_plan: workBPlanPath,
+          plan_name: "work-b-plan",
+          started_at: "2026-01-02T11:00:00Z",
+          session_ids: ["ses_work_b"],
+          status: "active",
+        },
+      },
+    })
+
+    // when
+    const result = await resolveActiveBoulderSession({
+      client: { session: { get: async () => ({ data: {} }) } } as never,
+      directory: testDirectory,
+      sessionID: "ses_work_b",
+    })
+
+    // then
+    expect(result).not.toBeNull()
+    expect(result?.boulderState.active_plan).toBe(workBPlanPath)
+    expect(result?.progress.isComplete).toBe(true)
+  })
+
+  test("falls back to top-level mirror when works map is missing", async () => {
+    // given
+    const legacyPlanPath = join(testDirectory, "legacy-only-plan.md")
+    writeFileSync(legacyPlanPath, "# Plan\n- [ ] Task 1\n", "utf-8")
+    writeBoulderState(testDirectory, {
+      active_plan: legacyPlanPath,
+      started_at: "2026-01-02T10:00:00Z",
+      session_ids: ["ses_legacy_only"],
+      plan_name: "legacy-only-plan",
+    })
+
+    // when
+    const result = await resolveActiveBoulderSession({
+      client: { session: { get: async () => ({ data: {} }) } } as never,
+      directory: testDirectory,
+      sessionID: "ses_legacy_only",
+    })
+
+    // then
+    expect(result).not.toBeNull()
+    expect(result?.boulderState.active_plan).toBe(legacyPlanPath)
+    expect(result?.progress.isComplete).toBe(false)
+  })
 })