feat(atlas): add background task session tracking with retry scheduling

- Add background-launch-session-tracking.ts to persist delegated sessions - Add task-context.ts for task context resolution utilities - Modify idle-event.ts to schedule retries when background tasks are running - Update tool-execute-after.ts to integrate session tracking - Add comprehensive tests for background task retry behavior 🤖 Generated with assistance of OhMyOpenCode
2026-04-05 14:59:03 +09:00
parent 90407a9789
commit ccfc54ab38
6 changed files with 454 additions and 53 deletions
@@ -0,0 +1,63 @@
+import type { PluginInput } from "@opencode-ai/plugin"
+import { appendSessionId, type BoulderState, upsertTaskSessionState } from "../../features/boulder-state"
+import { log } from "../../shared/logger"
+import { HOOK_NAME } from "./hook-name"
+import { extractSessionIdFromOutput, validateSubagentSessionId } from "./subagent-session-id"
+import { resolveTaskContext } from "./task-context"
+import type { PendingTaskRef, ToolExecuteAfterInput, ToolExecuteAfterOutput } from "./types"
+
+export async function syncBackgroundLaunchSessionTracking(input: {
+  ctx: PluginInput
+  boulderState: BoulderState | null
+  toolInput: ToolExecuteAfterInput
+  toolOutput: ToolExecuteAfterOutput
+  pendingTaskRef: PendingTaskRef | undefined
+  metadataSessionId?: string
+}): Promise<void> {
+  const { ctx, boulderState, toolInput, toolOutput, pendingTaskRef, metadataSessionId } = input
+  if (!boulderState) {
+    return
+  }
+
+  if (toolInput.sessionID && !boulderState.session_ids.includes(toolInput.sessionID)) {
+    appendSessionId(ctx.directory, toolInput.sessionID)
+  }
+
+  const extractedSessionId = metadataSessionId ?? extractSessionIdFromOutput(toolOutput.output)
+  const lineageSessionIDs = toolInput.sessionID && !boulderState.session_ids.includes(toolInput.sessionID)
+    ? [...boulderState.session_ids, toolInput.sessionID]
+    : boulderState.session_ids
+  const subagentSessionId = await validateSubagentSessionId({
+    client: ctx.client,
+    sessionID: extractedSessionId,
+    lineageSessionIDs,
+  })
+
+  if (!subagentSessionId) {
+    return
+  }
+
+  appendSessionId(ctx.directory, subagentSessionId)
+
+  const { currentTask, shouldSkipTaskSessionUpdate } = resolveTaskContext(
+    pendingTaskRef,
+    boulderState.active_plan,
+  )
+
+  if (currentTask && !shouldSkipTaskSessionUpdate) {
+    upsertTaskSessionState(ctx.directory, {
+      taskKey: currentTask.key,
+      taskLabel: currentTask.label,
+      taskTitle: currentTask.title,
+      sessionId: subagentSessionId,
+      agent: typeof toolOutput.metadata?.agent === "string" ? toolOutput.metadata.agent : undefined,
+      category: typeof toolOutput.metadata?.category === "string" ? toolOutput.metadata.category : undefined,
+    })
+  }
+
+  log(`[${HOOK_NAME}] Background launch session tracked`, {
+    sessionID: toolInput.sessionID,
+    subagentSessionId,
+    taskKey: currentTask?.key,
+  })
+}