refactor(background-agent): split manager.ts into focused modules

Extract 30+ single-responsibility modules from manager.ts (1556 LOC): - task lifecycle: task-starter, task-completer, task-canceller, task-resumer - task queries: task-queries, task-poller, task-queue-processor - notifications: notification-builder, notification-tracker, parent-session-notifier - session handling: session-validator, session-output-validator, session-todo-checker - spawner: spawner/ directory with focused spawn modules - utilities: duration-formatter, error-classifier, message-storage-locator - result handling: result-handler-context, background-task-completer - shutdown: background-manager-shutdown, process-signal
2026-02-08 16:20:52 +09:00
parent c7122b4127
commit e3bd43ff64
43 changed files with 2753 additions and 2071 deletions
@@ -0,0 +1,117 @@
+import { log } from "../../shared"
+
+import type { BackgroundTask } from "./types"
+import type { LaunchInput } from "./types"
+import type { ConcurrencyManager } from "./concurrency"
+import type { OpencodeClient } from "./opencode-client"
+
+type QueueItem = { task: BackgroundTask; input: LaunchInput }
+
+export async function cancelBackgroundTask(args: {
+  taskId: string
+  options?: {
+    source?: string
+    reason?: string
+    abortSession?: boolean
+    skipNotification?: boolean
+  }
+  tasks: Map<string, BackgroundTask>
+  queuesByKey: Map<string, QueueItem[]>
+  completionTimers: Map<string, ReturnType<typeof setTimeout>>
+  idleDeferralTimers: Map<string, ReturnType<typeof setTimeout>>
+  concurrencyManager: ConcurrencyManager
+  client: OpencodeClient
+  cleanupPendingByParent: (task: BackgroundTask) => void
+  markForNotification: (task: BackgroundTask) => void
+  notifyParentSession: (task: BackgroundTask) => Promise<void>
+}): Promise<boolean> {
+  const {
+    taskId,
+    options,
+    tasks,
+    queuesByKey,
+    completionTimers,
+    idleDeferralTimers,
+    concurrencyManager,
+    client,
+    cleanupPendingByParent,
+    markForNotification,
+    notifyParentSession,
+  } = args
+
+  const task = tasks.get(taskId)
+  if (!task || (task.status !== "running" && task.status !== "pending")) {
+    return false
+  }
+
+  const source = options?.source ?? "cancel"
+  const abortSession = options?.abortSession !== false
+  const reason = options?.reason
+
+  if (task.status === "pending") {
+    const key = task.model
+      ? `${task.model.providerID}/${task.model.modelID}`
+      : task.agent
+    const queue = queuesByKey.get(key)
+    if (queue) {
+      const index = queue.findIndex((item) => item.task.id === taskId)
+      if (index !== -1) {
+        queue.splice(index, 1)
+        if (queue.length === 0) {
+          queuesByKey.delete(key)
+        }
+      }
+    }
+    log("[background-agent] Cancelled pending task:", { taskId, key })
+  }
+
+  task.status = "cancelled"
+  task.completedAt = new Date()
+  if (reason) {
+    task.error = reason
+  }
+
+  if (task.concurrencyKey) {
+    concurrencyManager.release(task.concurrencyKey)
+    task.concurrencyKey = undefined
+  }
+
+  const completionTimer = completionTimers.get(task.id)
+  if (completionTimer) {
+    clearTimeout(completionTimer)
+    completionTimers.delete(task.id)
+  }
+
+  const idleTimer = idleDeferralTimers.get(task.id)
+  if (idleTimer) {
+    clearTimeout(idleTimer)
+    idleDeferralTimers.delete(task.id)
+  }
+
+  cleanupPendingByParent(task)
+
+  if (abortSession && task.sessionID) {
+    client.session.abort({
+      path: { id: task.sessionID },
+    }).catch(() => {})
+  }
+
+  if (options?.skipNotification) {
+    log(`[background-agent] Task cancelled via ${source} (notification skipped):`, task.id)
+    return true
+  }
+
+  markForNotification(task)
+
+  try {
+    await notifyParentSession(task)
+    log(`[background-agent] Task cancelled via ${source}:`, task.id)
+  } catch (err) {
+    log("[background-agent] Error in notifyParentSession for cancelled task:", {
+      taskId: task.id,
+      error: err,
+    })
+  }
+
+  return true
+}