refactor: wave 2 - split atlas, auto-update-checker, session-recovery, todo-enforcer, background-task hooks

- Extract atlas/ into 15 focused modules (hook, event handler, tool policies, types, etc.) - Split auto-update-checker into checker/ and hook/ subdirectories with single-purpose files - Decompose session-recovery into separate recovery strategy files per error type - Extract todo-continuation-enforcer from monolith to directory with dedicated modules - Split background-task/tools.ts into individual tool creator files - Extract command-executor, tmux-utils into focused sub-modules - Split config/schema.ts into domain-specific schema files - Decompose cli/config-manager.ts into focused modules - Rollback skill-mcp-manager, model-availability, index.ts splits that broke tests - Fix all import path depths for moved files (../../ -> ../../../) - Add explicit type annotations to resolve TS7006 implicit any errors Typecheck: 0 errors Tests: 2359 pass, 5 fail (all pre-existing)
2026-02-08 15:01:42 +09:00
parent df03300211
commit cea97f896b
158 changed files with 7806 additions and 7050 deletions
@@ -0,0 +1,158 @@
+import type { PluginInput } from "@opencode-ai/plugin"
+
+import type { BackgroundManager } from "../../features/background-agent"
+import { getMainSessionID, subagentSessions } from "../../features/claude-code-session-state"
+import type { ToolPermission } from "../../features/hook-message-injector"
+import { log } from "../../shared/logger"
+
+import {
+  ABORT_WINDOW_MS,
+  DEFAULT_SKIP_AGENTS,
+  HOOK_NAME,
+} from "./constants"
+import { isLastAssistantMessageAborted } from "./abort-detection"
+import { getIncompleteCount } from "./todo"
+import type { MessageInfo, ResolvedMessageInfo, Todo } from "./types"
+import type { SessionStateStore } from "./session-state"
+import { startCountdown } from "./countdown"
+
+export async function handleSessionIdle(args: {
+  ctx: PluginInput
+  sessionID: string
+  sessionStateStore: SessionStateStore
+  backgroundManager?: BackgroundManager
+  skipAgents?: string[]
+  isContinuationStopped?: (sessionID: string) => boolean
+}): Promise<void> {
+  const {
+    ctx,
+    sessionID,
+    sessionStateStore,
+    backgroundManager,
+    skipAgents = DEFAULT_SKIP_AGENTS,
+    isContinuationStopped,
+  } = args
+
+  log(`[${HOOK_NAME}] session.idle`, { sessionID })
+
+  const mainSessionID = getMainSessionID()
+  const isMainSession = sessionID === mainSessionID
+  const isBackgroundTaskSession = subagentSessions.has(sessionID)
+
+  if (mainSessionID && !isMainSession && !isBackgroundTaskSession) {
+    log(`[${HOOK_NAME}] Skipped: not main or background task session`, { sessionID })
+    return
+  }
+
+  const state = sessionStateStore.getState(sessionID)
+  if (state.isRecovering) {
+    log(`[${HOOK_NAME}] Skipped: in recovery`, { sessionID })
+    return
+  }
+
+  if (state.abortDetectedAt) {
+    const timeSinceAbort = Date.now() - state.abortDetectedAt
+    if (timeSinceAbort < ABORT_WINDOW_MS) {
+      log(`[${HOOK_NAME}] Skipped: abort detected via event ${timeSinceAbort}ms ago`, { sessionID })
+      state.abortDetectedAt = undefined
+      return
+    }
+    state.abortDetectedAt = undefined
+  }
+
+  const hasRunningBgTasks = backgroundManager
+    ? backgroundManager.getTasksByParentSession(sessionID).some((task: { status: string }) => task.status === "running")
+    : false
+
+  if (hasRunningBgTasks) {
+    log(`[${HOOK_NAME}] Skipped: background tasks running`, { sessionID })
+    return
+  }
+
+  try {
+    const messagesResp = await ctx.client.session.messages({
+      path: { id: sessionID },
+      query: { directory: ctx.directory },
+    })
+    const messages = (messagesResp as { data?: Array<{ info?: MessageInfo }> }).data ?? []
+    if (isLastAssistantMessageAborted(messages)) {
+      log(`[${HOOK_NAME}] Skipped: last assistant message was aborted (API fallback)`, { sessionID })
+      return
+    }
+  } catch (error) {
+    log(`[${HOOK_NAME}] Messages fetch failed, continuing`, { sessionID, error: String(error) })
+  }
+
+  let todos: Todo[] = []
+  try {
+    const response = await ctx.client.session.todo({ path: { id: sessionID } })
+    todos = (response.data ?? response) as Todo[]
+  } catch (error) {
+    log(`[${HOOK_NAME}] Todo fetch failed`, { sessionID, error: String(error) })
+    return
+  }
+
+  if (!todos || todos.length === 0) {
+    log(`[${HOOK_NAME}] No todos`, { sessionID })
+    return
+  }
+
+  const incompleteCount = getIncompleteCount(todos)
+  if (incompleteCount === 0) {
+    log(`[${HOOK_NAME}] All todos complete`, { sessionID, total: todos.length })
+    return
+  }
+
+  let resolvedInfo: ResolvedMessageInfo | undefined
+  let hasCompactionMessage = false
+  try {
+    const messagesResp = await ctx.client.session.messages({
+      path: { id: sessionID },
+    })
+    const messages = (messagesResp.data ?? []) as Array<{ info?: MessageInfo }>
+    for (let i = messages.length - 1; i >= 0; i--) {
+      const info = messages[i].info
+      if (info?.agent === "compaction") {
+        hasCompactionMessage = true
+        continue
+      }
+      if (info?.agent || info?.model || (info?.modelID && info?.providerID)) {
+        resolvedInfo = {
+          agent: info.agent,
+          model: info.model ?? (info.providerID && info.modelID ? { providerID: info.providerID, modelID: info.modelID } : undefined),
+          tools: info.tools as Record<string, ToolPermission> | undefined,
+        }
+        break
+      }
+    }
+  } catch (error) {
+    log(`[${HOOK_NAME}] Failed to fetch messages for agent check`, { sessionID, error: String(error) })
+  }
+
+  log(`[${HOOK_NAME}] Agent check`, { sessionID, agentName: resolvedInfo?.agent, skipAgents, hasCompactionMessage })
+
+  if (resolvedInfo?.agent && skipAgents.includes(resolvedInfo.agent)) {
+    log(`[${HOOK_NAME}] Skipped: agent in skipAgents list`, { sessionID, agent: resolvedInfo.agent })
+    return
+  }
+  if (hasCompactionMessage && !resolvedInfo?.agent) {
+    log(`[${HOOK_NAME}] Skipped: compaction occurred but no agent info resolved`, { sessionID })
+    return
+  }
+
+  if (isContinuationStopped?.(sessionID)) {
+    log(`[${HOOK_NAME}] Skipped: continuation stopped for session`, { sessionID })
+    return
+  }
+
+  startCountdown({
+    ctx,
+    sessionID,
+    incompleteCount,
+    total: todos.length,
+    resolvedInfo,
+    backgroundManager,
+    skipAgents,
+    sessionStateStore,
+  })
+}