pi-crew 0.2.3 → 0.2.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (348) hide show
  1. package/AGENTS.md +57 -32
  2. package/CHANGELOG.md +466 -448
  3. package/LICENSE +21 -21
  4. package/NOTICE.md +16 -16
  5. package/README.md +323 -323
  6. package/docs/FEATURE_INTAKE.md +126 -0
  7. package/docs/HARNESS.md +86 -0
  8. package/docs/HARNESS_BACKLOG.md +41 -0
  9. package/docs/TEST_MATRIX.md +49 -0
  10. package/docs/actions-reference.md +595 -595
  11. package/docs/architecture.md +180 -180
  12. package/docs/code-review-2026-05-11.md +592 -592
  13. package/docs/commands-reference.md +347 -347
  14. package/docs/comparison-pi-subagents-vs-pi-crew.md +303 -0
  15. package/docs/decisions/0001-durable-state.md +41 -0
  16. package/docs/decisions/0002-child-process-for-async.md +42 -0
  17. package/docs/decisions/0003-depth-guard.md +36 -0
  18. package/docs/decisions/0004-execfile-over-exec.md +34 -0
  19. package/docs/decisions/0005-no-parameter-properties.md +49 -0
  20. package/docs/decisions/0006-publish-bundled-esm.md +63 -0
  21. package/docs/decisions/0007-active-run-binary-index.md +54 -0
  22. package/docs/decisions/0008-child-pi-warm-pool.md +61 -0
  23. package/docs/decisions/README.md +23 -0
  24. package/docs/followup-review-round4-2026-05-13.md +107 -0
  25. package/docs/implementation-plan-top3.md +333 -0
  26. package/docs/live-mailbox-runtime.md +36 -36
  27. package/docs/next-upgrade-roadmap.md +808 -808
  28. package/docs/oh-my-pi-research.md +509 -0
  29. package/docs/perf/baseline-2026-05.md +113 -0
  30. package/docs/perf/final-report-2026-05.md +206 -0
  31. package/docs/perf/sprint-1-report.md +71 -0
  32. package/docs/perf/sprint-2-report.md +81 -0
  33. package/docs/perf/sprint-2.5-report.md +53 -0
  34. package/docs/perf/sprint-3-report.md +36 -0
  35. package/docs/perf/sprint-4-report.md +47 -0
  36. package/docs/perf/sprint-5-report.md +51 -0
  37. package/docs/perf/sprint-6-report.md +94 -0
  38. package/docs/perf/sprint-7-report.md +74 -0
  39. package/docs/perf/upgrade-plan-2026-05.md +147 -0
  40. package/docs/pi-subagents3-deep-analysis.md +508 -0
  41. package/docs/product/README.md +31 -0
  42. package/docs/product/platform.md +27 -0
  43. package/docs/product/runtime-safety.md +37 -0
  44. package/docs/product/team-run.md +39 -0
  45. package/docs/product/team-tool.md +37 -0
  46. package/docs/publishing.md +65 -65
  47. package/docs/resource-formats.md +134 -134
  48. package/docs/runtime-analysis-child-vs-live.md +171 -0
  49. package/docs/runtime-flow.md +148 -148
  50. package/docs/runtime-migration-in-process-analysis.md +250 -0
  51. package/docs/stories/README.md +30 -0
  52. package/docs/stories/backlog.md +36 -0
  53. package/docs/templates/decision.md +27 -0
  54. package/docs/templates/story.md +44 -0
  55. package/docs/templates/validation-report.md +32 -0
  56. package/docs/usage.md +238 -238
  57. package/index.ts +7 -6
  58. package/install.mjs +65 -65
  59. package/package.json +107 -100
  60. package/schema.json +222 -222
  61. package/skills/child-pi-spawning/SKILL.md +213 -0
  62. package/skills/context-artifact-hygiene/SKILL.md +32 -0
  63. package/skills/event-log-tracing/SKILL.md +299 -0
  64. package/skills/git-master/SKILL.md +225 -24
  65. package/skills/live-agent-lifecycle/SKILL.md +192 -0
  66. package/skills/mailbox-interactive/SKILL.md +300 -19
  67. package/skills/model-routing-context/SKILL.md +94 -0
  68. package/skills/multi-perspective-review/SKILL.md +88 -0
  69. package/skills/read-only-explorer/SKILL.md +250 -26
  70. package/skills/safe-bash/SKILL.md +307 -21
  71. package/skills/verification-before-done/SKILL.md +11 -2
  72. package/skills/widget-rendering/SKILL.md +258 -0
  73. package/skills/workspace-isolation/SKILL.md +202 -0
  74. package/skills/worktree-isolation/SKILL.md +202 -18
  75. package/src/adapters/claude-adapter.ts +25 -25
  76. package/src/adapters/codex-adapter.ts +21 -21
  77. package/src/adapters/cursor-adapter.ts +17 -17
  78. package/src/adapters/export-util.ts +137 -137
  79. package/src/adapters/index.ts +15 -15
  80. package/src/adapters/registry.ts +18 -18
  81. package/src/adapters/types.ts +23 -23
  82. package/src/agents/agent-config.ts +38 -38
  83. package/src/agents/agent-serializer.ts +38 -38
  84. package/src/agents/discover-agents.ts +121 -118
  85. package/src/config/config.ts +740 -858
  86. package/src/config/defaults.ts +96 -96
  87. package/src/config/drift-detector.ts +211 -211
  88. package/src/config/markers.ts +327 -327
  89. package/src/config/resilient-parser.ts +109 -108
  90. package/src/config/suggestions.ts +74 -74
  91. package/src/config/types.ts +199 -0
  92. package/src/extension/async-notifier.ts +123 -89
  93. package/src/extension/autonomous-policy.ts +169 -169
  94. package/src/extension/cross-extension-rpc.ts +104 -104
  95. package/src/extension/help.ts +47 -47
  96. package/src/extension/import-index.ts +69 -69
  97. package/src/extension/management.ts +395 -382
  98. package/src/extension/notification-router.ts +116 -116
  99. package/src/extension/notification-sink.ts +51 -51
  100. package/src/extension/project-init.ts +168 -168
  101. package/src/extension/register.ts +859 -668
  102. package/src/extension/registration/artifact-cleanup.ts +15 -15
  103. package/src/extension/registration/command-utils.ts +54 -54
  104. package/src/extension/registration/commands.ts +559 -452
  105. package/src/extension/registration/compaction-guard.ts +125 -125
  106. package/src/extension/registration/subagent-helpers.ts +102 -102
  107. package/src/extension/registration/subagent-tools.ts +220 -159
  108. package/src/extension/registration/team-tool.ts +159 -99
  109. package/src/extension/registration/viewers.ts +29 -0
  110. package/src/extension/result-watcher.ts +128 -128
  111. package/src/extension/run-bundle-schema.ts +89 -89
  112. package/src/extension/run-export.ts +73 -73
  113. package/src/extension/run-import.ts +84 -84
  114. package/src/extension/run-index.ts +94 -94
  115. package/src/extension/run-maintenance.ts +142 -142
  116. package/src/extension/session-summary.ts +8 -8
  117. package/src/extension/team-manager-command.ts +96 -96
  118. package/src/extension/team-recommendation.ts +188 -188
  119. package/src/extension/team-tool/api.ts +5 -2
  120. package/src/extension/team-tool/cancel.ts +224 -209
  121. package/src/extension/team-tool/config-patch.ts +36 -36
  122. package/src/extension/team-tool/context.ts +60 -60
  123. package/src/extension/team-tool/doctor.ts +242 -242
  124. package/src/extension/team-tool/handle-settings.ts +421 -195
  125. package/src/extension/team-tool/inspect.ts +41 -41
  126. package/src/extension/team-tool/lifecycle-actions.ts +139 -139
  127. package/src/extension/team-tool/parallel-dispatch.ts +156 -156
  128. package/src/extension/team-tool/plan.ts +19 -19
  129. package/src/extension/team-tool/respond.ts +112 -111
  130. package/src/extension/team-tool/run.ts +246 -229
  131. package/src/extension/team-tool/status.ts +110 -110
  132. package/src/extension/team-tool-types.ts +13 -13
  133. package/src/extension/team-tool.ts +344 -344
  134. package/src/extension/tool-result.ts +16 -16
  135. package/src/extension/validate-resources.ts +77 -77
  136. package/src/hooks/registry.ts +61 -61
  137. package/src/hooks/types.ts +40 -40
  138. package/src/i18n.ts +184 -184
  139. package/src/observability/correlation.ts +35 -35
  140. package/src/observability/event-to-metric.ts +68 -68
  141. package/src/observability/exporters/adapter.ts +30 -30
  142. package/src/observability/exporters/otlp-exporter.ts +106 -92
  143. package/src/observability/exporters/prometheus-exporter.ts +54 -54
  144. package/src/observability/metric-registry.ts +87 -87
  145. package/src/observability/metric-retention.ts +54 -54
  146. package/src/observability/metric-sink.ts +81 -56
  147. package/src/observability/metrics-primitives.ts +167 -167
  148. package/src/prompt/prompt-runtime.ts +72 -72
  149. package/src/runtime/adaptive-plan.ts +338 -0
  150. package/src/runtime/agent-control.ts +169 -169
  151. package/src/runtime/agent-memory.ts +72 -72
  152. package/src/runtime/agent-observability.ts +114 -114
  153. package/src/runtime/async-marker.ts +26 -26
  154. package/src/runtime/async-runner.ts +153 -153
  155. package/src/runtime/attention-events.ts +28 -28
  156. package/src/runtime/auto-resume.ts +100 -100
  157. package/src/runtime/background-runner.ts +122 -89
  158. package/src/runtime/cancellation.ts +61 -61
  159. package/src/runtime/capability-inventory.ts +116 -116
  160. package/src/runtime/child-pi-pool.ts +68 -0
  161. package/src/runtime/child-pi.ts +541 -461
  162. package/src/runtime/code-summary.ts +247 -247
  163. package/src/runtime/compaction-summary.ts +271 -271
  164. package/src/runtime/concurrency.ts +58 -58
  165. package/src/runtime/crash-recovery.ts +317 -301
  166. package/src/runtime/crew-agent-records.ts +379 -281
  167. package/src/runtime/crew-agent-runtime.ts +60 -60
  168. package/src/runtime/cross-extension-rpc.ts +72 -0
  169. package/src/runtime/custom-tools/irc-tool.ts +201 -201
  170. package/src/runtime/custom-tools/submit-result-tool.ts +90 -90
  171. package/src/runtime/deadletter.ts +47 -47
  172. package/src/runtime/delivery-coordinator.ts +176 -176
  173. package/src/runtime/delta-conflict.ts +360 -360
  174. package/src/runtime/diagnostic-export.ts +102 -102
  175. package/src/runtime/direct-run.ts +35 -35
  176. package/src/runtime/effectiveness.ts +82 -81
  177. package/src/runtime/errors/crew-errors.ts +166 -0
  178. package/src/runtime/event-stream-bridge.ts +92 -92
  179. package/src/runtime/foreground-control.ts +82 -82
  180. package/src/runtime/green-contract.ts +46 -46
  181. package/src/runtime/group-join.ts +234 -106
  182. package/src/runtime/heartbeat-watcher.ts +145 -124
  183. package/src/runtime/iteration-hooks.ts +267 -267
  184. package/src/runtime/live-agent-control.ts +88 -88
  185. package/src/runtime/live-agent-manager.ts +377 -179
  186. package/src/runtime/live-control-realtime.ts +36 -36
  187. package/src/runtime/live-session-runtime.ts +676 -600
  188. package/src/runtime/loop-gates.ts +129 -129
  189. package/src/runtime/manifest-cache.ts +263 -263
  190. package/src/runtime/mcp-proxy.ts +113 -113
  191. package/src/runtime/metric-parser.ts +40 -40
  192. package/src/runtime/model-fallback.ts +282 -274
  193. package/src/runtime/model-resolver.ts +118 -0
  194. package/src/runtime/output-validator.ts +187 -187
  195. package/src/runtime/overflow-recovery.ts +175 -175
  196. package/src/runtime/parallel-research.ts +44 -44
  197. package/src/runtime/parallel-utils.ts +156 -156
  198. package/src/runtime/parent-guard.ts +80 -80
  199. package/src/runtime/phase-progress.ts +217 -217
  200. package/src/runtime/pi-args.ts +165 -165
  201. package/src/runtime/pi-json-output.ts +111 -111
  202. package/src/runtime/pi-spawn.ts +167 -167
  203. package/src/runtime/policy-engine.ts +79 -79
  204. package/src/runtime/post-checks.ts +125 -125
  205. package/src/runtime/post-exit-stdio-guard.ts +86 -86
  206. package/src/runtime/process-status.ts +97 -73
  207. package/src/runtime/progress-event-coalescer.ts +43 -43
  208. package/src/runtime/recovery-recipes.ts +74 -74
  209. package/src/runtime/retry-executor.ts +81 -81
  210. package/src/runtime/role-permission.ts +39 -39
  211. package/src/runtime/run-tracker.ts +99 -0
  212. package/src/runtime/runtime-policy.ts +21 -0
  213. package/src/runtime/runtime-resolver.ts +94 -91
  214. package/src/runtime/scheduler.ts +294 -0
  215. package/src/runtime/semaphore.ts +131 -131
  216. package/src/runtime/sensitive-paths.ts +92 -92
  217. package/src/runtime/session-usage.ts +79 -79
  218. package/src/runtime/settings-store.ts +103 -0
  219. package/src/runtime/sidechain-output.ts +29 -29
  220. package/src/runtime/skill-instructions.ts +222 -222
  221. package/src/runtime/stale-reconciler.ts +198 -189
  222. package/src/runtime/streaming-output.ts +47 -0
  223. package/src/runtime/subagent-manager.ts +404 -400
  224. package/src/runtime/subprocess-tool-registry.ts +67 -67
  225. package/src/runtime/task-display.ts +38 -38
  226. package/src/runtime/task-graph-scheduler.ts +122 -122
  227. package/src/runtime/task-graph.ts +207 -207
  228. package/src/runtime/task-output-context.ts +177 -177
  229. package/src/runtime/task-packet.ts +93 -93
  230. package/src/runtime/task-quality.ts +207 -207
  231. package/src/runtime/task-runner/capabilities.ts +78 -78
  232. package/src/runtime/task-runner/live-executor.ts +131 -113
  233. package/src/runtime/task-runner/progress.ts +119 -119
  234. package/src/runtime/task-runner/prompt-builder.ts +139 -139
  235. package/src/runtime/task-runner/prompt-pipeline.ts +64 -64
  236. package/src/runtime/task-runner/result-utils.ts +14 -14
  237. package/src/runtime/task-runner/run-projection.ts +103 -103
  238. package/src/runtime/task-runner/state-helpers.ts +22 -22
  239. package/src/runtime/task-runner.ts +469 -459
  240. package/src/runtime/team-runner.ts +693 -945
  241. package/src/runtime/usage-tracker.ts +71 -0
  242. package/src/runtime/worker-heartbeat.ts +21 -21
  243. package/src/runtime/worker-startup.ts +57 -57
  244. package/src/runtime/workflow-state.ts +187 -187
  245. package/src/runtime/yield-handler.ts +190 -190
  246. package/src/schema/config-schema.ts +172 -168
  247. package/src/schema/team-tool-schema.ts +126 -126
  248. package/src/schema/validation-types.ts +151 -148
  249. package/src/skills/discover-skills.ts +67 -67
  250. package/src/skills/skill-templates.ts +374 -374
  251. package/src/state/active-run-registry.ts +227 -191
  252. package/src/state/artifact-store.ts +130 -129
  253. package/src/state/atomic-write.ts +262 -195
  254. package/src/state/blob-store.ts +116 -116
  255. package/src/state/contracts.ts +111 -111
  256. package/src/state/event-log-rotation.ts +161 -158
  257. package/src/state/event-log.ts +383 -303
  258. package/src/state/event-reconstructor.ts +217 -217
  259. package/src/state/jsonl-writer.ts +82 -82
  260. package/src/state/locks.ts +146 -146
  261. package/src/state/mailbox.ts +446 -405
  262. package/src/state/state-store.ts +364 -351
  263. package/src/state/task-claims.ts +44 -44
  264. package/src/state/types.ts +285 -285
  265. package/src/state/usage.ts +29 -29
  266. package/src/subagents/async-entry.ts +1 -1
  267. package/src/subagents/index.ts +3 -3
  268. package/src/subagents/live/control.ts +1 -1
  269. package/src/subagents/live/manager.ts +1 -1
  270. package/src/subagents/live/realtime.ts +1 -1
  271. package/src/subagents/live/session-runtime.ts +1 -1
  272. package/src/subagents/manager.ts +1 -1
  273. package/src/subagents/spawn.ts +1 -1
  274. package/src/teams/discover-teams.ts +116 -116
  275. package/src/teams/team-config.ts +27 -27
  276. package/src/teams/team-serializer.ts +38 -38
  277. package/src/types/diff.d.ts +18 -18
  278. package/src/ui/agent-management-overlay.ts +144 -144
  279. package/src/ui/crew-widget.ts +487 -370
  280. package/src/ui/dashboard-panes/agents-pane.ts +109 -28
  281. package/src/ui/dashboard-panes/cancellation-pane.ts +42 -42
  282. package/src/ui/dashboard-panes/capability-pane.ts +59 -59
  283. package/src/ui/dashboard-panes/health-pane.ts +30 -30
  284. package/src/ui/dashboard-panes/mailbox-pane.ts +35 -35
  285. package/src/ui/dashboard-panes/progress-pane.ts +30 -30
  286. package/src/ui/dashboard-panes/transcript-pane.ts +10 -10
  287. package/src/ui/heartbeat-aggregator.ts +63 -63
  288. package/src/ui/keybinding-map.ts +97 -94
  289. package/src/ui/live-conversation-overlay.ts +152 -0
  290. package/src/ui/live-run-sidebar.ts +180 -180
  291. package/src/ui/mascot.ts +442 -442
  292. package/src/ui/overlays/agent-picker-overlay.ts +57 -57
  293. package/src/ui/overlays/confirm-overlay.ts +58 -58
  294. package/src/ui/overlays/mailbox-compose-overlay.ts +144 -144
  295. package/src/ui/overlays/mailbox-compose-preview.ts +63 -63
  296. package/src/ui/overlays/mailbox-detail-overlay.ts +122 -122
  297. package/src/ui/pi-ui-compat.ts +57 -57
  298. package/src/ui/powerbar-publisher.ts +221 -197
  299. package/src/ui/render-scheduler.ts +216 -143
  300. package/src/ui/run-action-dispatcher.ts +118 -118
  301. package/src/ui/run-dashboard.ts +526 -464
  302. package/src/ui/run-event-bus.ts +208 -208
  303. package/src/ui/run-snapshot-cache.ts +826 -777
  304. package/src/ui/settings-overlay.ts +721 -0
  305. package/src/ui/snapshot-types.ts +86 -70
  306. package/src/ui/theme-adapter.ts +190 -190
  307. package/src/ui/tool-progress-formatter.ts +89 -0
  308. package/src/ui/transcript-cache.ts +94 -94
  309. package/src/ui/transcript-viewer.ts +335 -335
  310. package/src/utils/conflict-detect.ts +662 -0
  311. package/src/utils/file-coalescer.ts +86 -86
  312. package/src/utils/frontmatter.ts +68 -68
  313. package/src/utils/fs-watch.ts +88 -31
  314. package/src/utils/gh-protocol.ts +479 -0
  315. package/src/utils/ids.ts +17 -17
  316. package/src/utils/incremental-reader.ts +104 -104
  317. package/src/utils/internal-error.ts +6 -6
  318. package/src/utils/names.ts +27 -27
  319. package/src/utils/paths.ts +102 -63
  320. package/src/utils/redaction.ts +44 -44
  321. package/src/utils/safe-paths.ts +47 -47
  322. package/src/utils/scan-cache.ts +136 -136
  323. package/src/utils/sse-parser.ts +134 -134
  324. package/src/utils/task-name-generator.ts +337 -337
  325. package/src/utils/timings.ts +33 -33
  326. package/src/utils/visual.ts +243 -198
  327. package/src/workflows/discover-workflows.ts +139 -139
  328. package/src/workflows/validate-workflow.ts +40 -40
  329. package/src/workflows/workflow-config.ts +26 -26
  330. package/src/workflows/workflow-serializer.ts +32 -32
  331. package/src/worktree/branch-freshness.ts +45 -45
  332. package/src/worktree/cleanup.ts +75 -75
  333. package/src/worktree/worktree-manager.ts +188 -188
  334. package/teams/default.team.md +12 -12
  335. package/teams/fast-fix.team.md +11 -11
  336. package/teams/implementation.team.md +18 -18
  337. package/teams/parallel-research.team.md +14 -14
  338. package/teams/research.team.md +11 -11
  339. package/teams/review.team.md +12 -12
  340. package/tsconfig.json +19 -19
  341. package/workflows/default.workflow.md +30 -30
  342. package/workflows/fast-fix.workflow.md +23 -23
  343. package/workflows/implementation.workflow.md +43 -43
  344. package/workflows/parallel-research.workflow.md +46 -46
  345. package/workflows/research.workflow.md +22 -22
  346. package/workflows/review.workflow.md +30 -30
  347. package/skills/task-packet/SKILL.md +0 -28
  348. package/skills/verify-evidence/SKILL.md +0 -27
@@ -1,400 +1,404 @@
1
- import * as fs from "node:fs";
2
- import * as path from "node:path";
3
- import { loadRunManifestById } from "../state/state-store.ts";
4
- import type { PiTeamsToolResult } from "../extension/tool-result.ts";
5
- import { DEFAULT_SUBAGENT } from "../config/defaults.ts";
6
- import { projectCrewRoot } from "../utils/paths.ts";
7
- import { DEFAULT_PATHS } from "../config/defaults.ts";
8
- import { logInternalError } from "../utils/internal-error.ts";
9
- import { redactSecrets } from "../utils/redaction.ts";
10
-
11
- export type SubagentStatus = "queued" | "running" | "completed" | "failed" | "cancelled" | "error" | "blocked" | "stopped";
12
-
13
- export interface SubagentSpawnOptions {
14
- cwd: string;
15
- type: string;
16
- description: string;
17
- prompt: string;
18
- background: boolean;
19
- model?: string;
20
- skill?: string | string[] | false;
21
- maxTurns?: number;
22
- ownerSessionGeneration?: number;
23
- }
24
-
25
- export interface SubagentRecord {
26
- id: string;
27
- runId?: string;
28
- type: string;
29
- description: string;
30
- prompt: string;
31
- status: SubagentStatus;
32
- startedAt: number;
33
- completedAt?: number;
34
- result?: string;
35
- error?: string;
36
- resultConsumed?: boolean;
37
- model?: string;
38
- skill?: string | string[] | false;
39
- background: boolean;
40
- ownerSessionGeneration?: number;
41
- stuckNotified?: boolean;
42
- blockedAt?: number;
43
- promise?: Promise<void>;
44
- // Phase 1.6: Telemetry baseline fields
45
- turnCount?: number;
46
- terminated?: boolean;
47
- durationMs?: number;
48
- }
49
-
50
- type SpawnRunner = (options: SubagentSpawnOptions, signal?: AbortSignal) => Promise<PiTeamsToolResult>;
51
- type Notify = (record: SubagentRecord) => void;
52
- type NotifyEvent = (type: string, data: Record<string, unknown>) => void;
53
-
54
- interface QueuedSpawn {
55
- record: SubagentRecord;
56
- options: SubagentSpawnOptions;
57
- runner: SpawnRunner;
58
- signal?: AbortSignal;
59
- }
60
-
61
- function isValidSubagentId(id: string): boolean {
62
- return /^[a-z0-9_]+$/i.test(id) && id.length <= 128;
63
- }
64
-
65
- function persistedSubagentPath(cwd: string, id: string): string {
66
- if (!isValidSubagentId(id)) throw new Error(`Invalid subagent id: ${id}`);
67
- return path.join(projectCrewRoot(cwd), DEFAULT_PATHS.state.subagentsSubdir, `${id}.json`);
68
- }
69
-
70
- function serializableRecord(record: SubagentRecord): SubagentRecord {
71
- const { promise: _promise, ...rest } = record;
72
- return rest;
73
- }
74
-
75
- export function savePersistedSubagentRecord(cwd: string, record: SubagentRecord): void {
76
- try {
77
- const filePath = persistedSubagentPath(cwd, record.id);
78
- fs.mkdirSync(path.dirname(filePath), { recursive: true });
79
- fs.writeFileSync(filePath, `${JSON.stringify(redactSecrets(serializableRecord(record)), null, 2)}\n`, "utf-8");
80
- } catch (error) {
81
- logInternalError("subagent-manager.save", error, `id=${record.id}`);
82
- }
83
- }
84
-
85
- export function readPersistedSubagentRecord(cwd: string, id: string): SubagentRecord | undefined {
86
- try {
87
- const parsed = JSON.parse(fs.readFileSync(persistedSubagentPath(cwd, id), "utf-8"));
88
- return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed as SubagentRecord : undefined;
89
- } catch {
90
- return undefined;
91
- }
92
- }
93
-
94
- function resultText(result: PiTeamsToolResult): string {
95
- return result.content?.map((item) => item.type === "text" ? item.text : "").filter(Boolean).join("\n") ?? "";
96
- }
97
-
98
- function detailsRunId(result: PiTeamsToolResult): string | undefined {
99
- const details = result.details as { runId?: unknown } | undefined;
100
- return typeof details?.runId === "string" ? details.runId : undefined;
101
- }
102
-
103
- function totalRunTurns(cwd: string, runId: string | undefined): number | undefined {
104
- if (!runId) return undefined;
105
- const loaded = loadRunManifestById(cwd, runId);
106
- if (!loaded) return undefined;
107
- let total = 0;
108
- let hasTurns = false;
109
- for (const task of loaded.tasks) {
110
- const turns = task.usage?.turns ?? task.agentProgress?.turns;
111
- if (typeof turns === "number" && Number.isFinite(turns)) {
112
- total += turns;
113
- hasTurns = true;
114
- }
115
- }
116
- return hasTurns ? total : undefined;
117
- }
118
-
119
- export class SubagentManager {
120
- private readonly records = new Map<string, SubagentRecord>();
121
- private readonly cwdByRecord = new Map<string, string>();
122
- private readonly controllers = new Map<string, AbortController>();
123
- private readonly controllerCleanup = new Map<string, () => void>();
124
- private queue: QueuedSpawn[] = [];
125
- private runningBackground = 0;
126
- private counter = 0;
127
- private maxConcurrent: number;
128
- private readonly onComplete?: Notify;
129
- private readonly onEvent?: NotifyEvent;
130
- private readonly pollIntervalMs: number;
131
-
132
- constructor(maxConcurrent = 4, onComplete?: Notify, pollIntervalMs = 1000, onEvent?: NotifyEvent) {
133
- this.maxConcurrent = maxConcurrent;
134
- this.onComplete = onComplete;
135
- this.onEvent = onEvent;
136
- this.pollIntervalMs = pollIntervalMs;
137
- }
138
-
139
- spawn(options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): SubagentRecord {
140
- const record: SubagentRecord = {
141
- id: `agent_${Date.now().toString(36)}_${(++this.counter).toString(36)}`,
142
- type: options.type,
143
- description: options.description,
144
- prompt: options.prompt,
145
- status: options.background && this.runningBackground >= this.maxConcurrent ? "queued" : "running",
146
- startedAt: Date.now(),
147
- model: options.model,
148
- skill: options.skill,
149
- background: options.background,
150
- ownerSessionGeneration: options.ownerSessionGeneration,
151
- };
152
- this.records.set(record.id, record);
153
- this.cwdByRecord.set(record.id, options.cwd);
154
- savePersistedSubagentRecord(options.cwd, record);
155
- if (record.status === "queued") {
156
- this.queue.push({ record, options, runner, signal });
157
- return record;
158
- }
159
- this.start(record, options, runner, signal);
160
- return record;
161
- }
162
-
163
- getRecord(id: string): SubagentRecord | undefined {
164
- return this.records.get(id);
165
- }
166
-
167
- listAgents(): SubagentRecord[] {
168
- return [...this.records.values()].sort((a, b) => b.startedAt - a.startedAt);
169
- }
170
-
171
- abort(id: string): boolean {
172
- const record = this.records.get(id);
173
- if (!record) return false;
174
- if (record.status === "queued") {
175
- this.queue = this.queue.filter((entry) => entry.record.id !== id);
176
- this.markStopped(record);
177
- return true;
178
- }
179
- if (record.status !== "running" && record.status !== "blocked") return false;
180
- this.controllers.get(id)?.abort();
181
- this.markStopped(record);
182
- return true;
183
- }
184
-
185
- abortAll(): number {
186
- let count = 0;
187
- for (const entry of this.queue) {
188
- this.markStopped(entry.record);
189
- count++;
190
- }
191
- this.queue = [];
192
- for (const record of this.records.values()) {
193
- if (record.status === "running" || record.status === "blocked") {
194
- this.controllers.get(record.id)?.abort();
195
- this.markStopped(record);
196
- count++;
197
- }
198
- }
199
- return count;
200
- }
201
-
202
- async waitForAll(): Promise<void> {
203
- while (true) {
204
- this.drainQueue();
205
- const pending = this.listAgents().filter((record) => record.status === "running" || record.status === "queued").map((record) => record.promise).filter((promise): promise is Promise<void> => Boolean(promise));
206
- if (!pending.length) break;
207
- await Promise.allSettled(pending);
208
- }
209
- }
210
-
211
- async waitForRecord(id: string): Promise<SubagentRecord | undefined> {
212
- while (true) {
213
- const record = this.records.get(id);
214
- if (!record) return undefined;
215
- if (record.status !== "running" && record.status !== "queued") return record;
216
- if (record.promise) await record.promise.catch(() => { /* status already set to error */ });
217
- else await new Promise((resolve) => setTimeout(resolve, 100));
218
- }
219
- }
220
-
221
- setMaxConcurrent(value: number): void {
222
- this.maxConcurrent = Math.max(1, Math.floor(value));
223
- this.drainQueue();
224
- }
225
-
226
- private start(record: SubagentRecord, options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): void {
227
- if (options.background) this.runningBackground++;
228
- record.status = "running";
229
- record.startedAt = Date.now();
230
- record.completedAt = undefined;
231
- const runSignal = this.createRunSignal(record.id, signal);
232
- savePersistedSubagentRecord(options.cwd, record);
233
- record.promise = (async () => {
234
- try {
235
- const result = await runner(options, runSignal);
236
- if (record.status === "stopped") return;
237
- record.runId = detailsRunId(result);
238
- record.result = resultText(result);
239
- savePersistedSubagentRecord(options.cwd, record);
240
- if (result.isError) {
241
- record.status = "error";
242
- record.error = record.result;
243
- throw new Error(record.error);
244
- }
245
- if (record.runId) await this.pollRunToTerminal(options.cwd, record);
246
- else record.status = "completed";
247
- } catch (error) {
248
- if (record.status === "stopped" || runSignal.aborted) {
249
- record.status = "stopped";
250
- return;
251
- }
252
- record.status = "error";
253
- record.error = error instanceof Error ? error.message : String(error);
254
- throw error; // H4: Propagate rejection so callers awaiting record.promise see the error
255
- } finally {
256
- this.cleanupRunSignal(record.id);
257
- if (options.background) this.runningBackground = Math.max(0, this.runningBackground - 1);
258
- if (record.status !== "blocked") record.completedAt = record.completedAt ?? Date.now();
259
- savePersistedSubagentRecord(options.cwd, record);
260
- if (record.status === "completed" || record.status === "failed" || record.status === "cancelled" || record.status === "error" || record.status === "stopped") {
261
- // Phase 1.6: Populate telemetry fields
262
- record.turnCount = record.turnCount ?? totalRunTurns(options.cwd, record.runId);
263
- record.durationMs = record.completedAt ? Math.max(0, record.completedAt - record.startedAt) : undefined;
264
- savePersistedSubagentRecord(options.cwd, record);
265
- this.onComplete?.(record);
266
- }
267
- this.drainQueue();
268
- }
269
- })();
270
- }
271
-
272
- private markStopped(record: SubagentRecord): void {
273
- record.status = "stopped";
274
- record.completedAt = Date.now();
275
- const cwd = this.cwdByRecord.get(record.id);
276
- if (cwd) savePersistedSubagentRecord(cwd, record);
277
- }
278
-
279
- private createRunSignal(id: string, signal?: AbortSignal): AbortSignal {
280
- const controller = new AbortController();
281
- this.controllers.set(id, controller);
282
- if (signal?.aborted) {
283
- controller.abort();
284
- return controller.signal;
285
- }
286
- if (signal) {
287
- const abort = (): void => controller.abort();
288
- signal.addEventListener("abort", abort, { once: true });
289
- this.controllerCleanup.set(id, () => signal.removeEventListener("abort", abort));
290
- }
291
- return controller.signal;
292
- }
293
-
294
- private cleanupRunSignal(id: string): void {
295
- this.controllerCleanup.get(id)?.();
296
- this.controllerCleanup.delete(id);
297
- this.controllers.delete(id);
298
- }
299
-
300
- private drainQueue(): void {
301
- while (this.queue.length > 0 && this.runningBackground < this.maxConcurrent) {
302
- const next = this.queue.shift();
303
- if (!next || next.record.status !== "queued") continue;
304
- this.start(next.record, next.options, next.runner, next.signal);
305
- }
306
- }
307
-
308
- private async pollRunToTerminal(cwd: string, record: SubagentRecord): Promise<void> {
309
- while (record.runId && (record.status === "running" || record.status === "blocked")) {
310
- const loaded = loadRunManifestById(cwd, record.runId);
311
- if (!loaded) {
312
- await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
313
- continue;
314
- }
315
- if (loaded.manifest.status === "completed") {
316
- record.status = "completed";
317
- record.error = undefined;
318
- record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
319
- record.completedAt = Date.now();
320
- savePersistedSubagentRecord(cwd, record);
321
- return;
322
- }
323
- if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
324
- record.status = loaded.manifest.status;
325
- record.error = loaded.manifest.summary;
326
- record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
327
- record.completedAt = Date.now();
328
- savePersistedSubagentRecord(cwd, record);
329
- return;
330
- }
331
- if (loaded.manifest.status === "blocked") {
332
- record.status = "blocked";
333
- record.error = undefined;
334
- if (!record.blockedAt) {
335
- record.blockedAt = Date.now();
336
- record.stuckNotified = false;
337
- record.completedAt = undefined;
338
- this.onComplete?.(record);
339
- this.scheduleStuckBlockedNotify(cwd, record);
340
- this.scheduleBlockedTerminalPoll(cwd, record);
341
- }
342
- savePersistedSubagentRecord(cwd, record);
343
- return;
344
- }
345
- await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
346
- }
347
- }
348
-
349
- private scheduleBlockedTerminalPoll(cwd: string, record: SubagentRecord): void {
350
- const poll = (): void => {
351
- const current = this.records.get(record.id);
352
- if (!current || current.status !== "blocked" || !current.runId) return;
353
- const loaded = loadRunManifestById(cwd, current.runId);
354
- if (!loaded || loaded.manifest.status === "blocked" || loaded.manifest.status === "running" || loaded.manifest.status === "planning" || loaded.manifest.status === "queued") {
355
- const timer = setTimeout(poll, this.pollIntervalMs);
356
- timer.unref();
357
- return;
358
- }
359
- const persisted = readPersistedSubagentRecord(cwd, current.id);
360
- current.resultConsumed = current.resultConsumed || persisted?.resultConsumed;
361
- if (loaded.manifest.status === "completed") {
362
- current.status = "completed";
363
- current.error = undefined;
364
- } else if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
365
- current.status = loaded.manifest.status;
366
- current.error = loaded.manifest.summary;
367
- } else return;
368
- current.completedAt = Date.now();
369
- current.turnCount = current.turnCount ?? totalRunTurns(cwd, current.runId);
370
- current.durationMs = Math.max(0, current.completedAt - current.startedAt);
371
- savePersistedSubagentRecord(cwd, current);
372
- this.onComplete?.(current);
373
- };
374
- const timer = setTimeout(poll, this.pollIntervalMs);
375
- timer.unref();
376
- }
377
-
378
- private scheduleStuckBlockedNotify(cwd: string, record: SubagentRecord): void {
379
- const threshold = DEFAULT_SUBAGENT.stuckBlockedNotifyMs;
380
- const fire = (): void => {
381
- const current = this.records.get(record.id);
382
- if (!current || current.status !== "blocked" || !current.blockedAt || current.stuckNotified) return;
383
- current.stuckNotified = true;
384
- this.onEvent?.("subagent.stuck-blocked", {
385
- event: "subagent.stuck-blocked",
386
- id: current.id,
387
- runId: current.runId,
388
- durationMs: Math.max(0, Date.now() - current.blockedAt),
389
- ownerSessionGeneration: current.ownerSessionGeneration,
390
- });
391
- savePersistedSubagentRecord(cwd, current);
392
- };
393
- if (threshold <= 0) {
394
- fire();
395
- return;
396
- }
397
- const timer = setTimeout(fire, threshold);
398
- timer.unref();
399
- }
400
- }
1
+ import * as fs from "node:fs";
2
+ import * as path from "node:path";
3
+ import { loadRunManifestById } from "../state/state-store.ts";
4
+ import type { PiTeamsToolResult } from "../extension/tool-result.ts";
5
+ import { DEFAULT_SUBAGENT } from "../config/defaults.ts";
6
+ import { projectCrewRoot } from "../utils/paths.ts";
7
+ import { DEFAULT_PATHS } from "../config/defaults.ts";
8
+ import { logInternalError } from "../utils/internal-error.ts";
9
+ import { redactSecrets } from "../utils/redaction.ts";
10
+
11
+ export type SubagentStatus = "queued" | "running" | "completed" | "failed" | "cancelled" | "error" | "blocked" | "stopped";
12
+
13
+ export interface SubagentSpawnOptions {
14
+ cwd: string;
15
+ type: string;
16
+ description: string;
17
+ prompt: string;
18
+ background: boolean;
19
+ model?: string;
20
+ skill?: string | string[] | false;
21
+ maxTurns?: number;
22
+ ownerSessionGeneration?: number;
23
+ }
24
+
25
+ export interface SubagentRecord {
26
+ id: string;
27
+ runId?: string;
28
+ type: string;
29
+ description: string;
30
+ prompt: string;
31
+ status: SubagentStatus;
32
+ startedAt: number;
33
+ completedAt?: number;
34
+ result?: string;
35
+ error?: string;
36
+ resultConsumed?: boolean;
37
+ model?: string;
38
+ skill?: string | string[] | false;
39
+ background: boolean;
40
+ ownerSessionGeneration?: number;
41
+ stuckNotified?: boolean;
42
+ blockedAt?: number;
43
+ promise?: Promise<void>;
44
+ // Phase 1.6: Telemetry baseline fields
45
+ turnCount?: number;
46
+ terminated?: boolean;
47
+ durationMs?: number;
48
+ }
49
+
50
+ type SpawnRunner = (options: SubagentSpawnOptions, signal?: AbortSignal) => Promise<PiTeamsToolResult>;
51
+ type Notify = (record: SubagentRecord) => void;
52
+ type NotifyEvent = (type: string, data: Record<string, unknown>) => void;
53
+
54
+ interface QueuedSpawn {
55
+ record: SubagentRecord;
56
+ options: SubagentSpawnOptions;
57
+ runner: SpawnRunner;
58
+ signal?: AbortSignal;
59
+ }
60
+
61
+ function isValidSubagentId(id: string): boolean {
62
+ return /^[a-z0-9_]+$/i.test(id) && id.length <= 128;
63
+ }
64
+
65
+ function persistedSubagentPath(cwd: string, id: string): string {
66
+ if (!isValidSubagentId(id)) throw new Error(`Invalid subagent id: ${id}`);
67
+ return path.join(projectCrewRoot(cwd), DEFAULT_PATHS.state.subagentsSubdir, `${id}.json`);
68
+ }
69
+
70
+ function serializableRecord(record: SubagentRecord): SubagentRecord {
71
+ const { promise: _promise, ...rest } = record;
72
+ return rest;
73
+ }
74
+
75
+ export function savePersistedSubagentRecord(cwd: string, record: SubagentRecord): void {
76
+ try {
77
+ const filePath = persistedSubagentPath(cwd, record.id);
78
+ fs.mkdirSync(path.dirname(filePath), { recursive: true });
79
+ fs.writeFileSync(filePath, `${JSON.stringify(redactSecrets(serializableRecord(record)), null, 2)}\n`, "utf-8");
80
+ } catch (error) {
81
+ logInternalError("subagent-manager.save", error, `id=${record.id}`);
82
+ }
83
+ }
84
+
85
+ export function readPersistedSubagentRecord(cwd: string, id: string): SubagentRecord | undefined {
86
+ try {
87
+ const parsed = JSON.parse(fs.readFileSync(persistedSubagentPath(cwd, id), "utf-8"));
88
+ return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed as SubagentRecord : undefined;
89
+ } catch {
90
+ return undefined;
91
+ }
92
+ }
93
+
94
+ function resultText(result: PiTeamsToolResult): string {
95
+ return result.content?.map((item) => item.type === "text" ? item.text : "").filter(Boolean).join("\n") ?? "";
96
+ }
97
+
98
+ function detailsRunId(result: PiTeamsToolResult): string | undefined {
99
+ const details = result.details as { runId?: unknown } | undefined;
100
+ return typeof details?.runId === "string" ? details.runId : undefined;
101
+ }
102
+
103
+ function totalRunTurns(cwd: string, runId: string | undefined): number | undefined {
104
+ if (!runId) return undefined;
105
+ const loaded = loadRunManifestById(cwd, runId);
106
+ if (!loaded) return undefined;
107
+ let total = 0;
108
+ let hasTurns = false;
109
+ for (const task of loaded.tasks) {
110
+ const turns = task.usage?.turns ?? task.agentProgress?.turns;
111
+ if (typeof turns === "number" && Number.isFinite(turns)) {
112
+ total += turns;
113
+ hasTurns = true;
114
+ }
115
+ }
116
+ return hasTurns ? total : undefined;
117
+ }
118
+
119
+ export class SubagentManager {
120
+ private readonly records = new Map<string, SubagentRecord>();
121
+ private readonly cwdByRecord = new Map<string, string>();
122
+ private readonly controllers = new Map<string, AbortController>();
123
+ private readonly controllerCleanup = new Map<string, () => void>();
124
+ private queue: QueuedSpawn[] = [];
125
+ private runningBackground = 0;
126
+ private counter = 0;
127
+ private maxConcurrent: number;
128
+ private readonly onComplete?: Notify;
129
+ private readonly onEvent?: NotifyEvent;
130
+ private readonly pollIntervalMs: number;
131
+
132
+ constructor(maxConcurrent = 4, onComplete?: Notify, pollIntervalMs = 1000, onEvent?: NotifyEvent) {
133
+ this.maxConcurrent = maxConcurrent;
134
+ this.onComplete = onComplete;
135
+ this.onEvent = onEvent;
136
+ this.pollIntervalMs = pollIntervalMs;
137
+ }
138
+
139
+ spawn(options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): SubagentRecord {
140
+ const record: SubagentRecord = {
141
+ id: `agent_${Date.now().toString(36)}_${(++this.counter).toString(36)}`,
142
+ type: options.type,
143
+ description: options.description,
144
+ prompt: options.prompt,
145
+ status: options.background && this.runningBackground >= this.maxConcurrent ? "queued" : "running",
146
+ startedAt: Date.now(),
147
+ model: options.model,
148
+ skill: options.skill,
149
+ background: options.background,
150
+ ownerSessionGeneration: options.ownerSessionGeneration,
151
+ };
152
+ this.records.set(record.id, record);
153
+ this.cwdByRecord.set(record.id, options.cwd);
154
+ savePersistedSubagentRecord(options.cwd, record);
155
+ if (record.status === "queued") {
156
+ this.queue.push({ record, options, runner, signal });
157
+ return record;
158
+ }
159
+ this.start(record, options, runner, signal);
160
+ return record;
161
+ }
162
+
163
+ getRecord(id: string): SubagentRecord | undefined {
164
+ return this.records.get(id);
165
+ }
166
+
167
+ listAgents(): SubagentRecord[] {
168
+ return [...this.records.values()].sort((a, b) => b.startedAt - a.startedAt);
169
+ }
170
+
171
+ abort(id: string, reason?: string): boolean {
172
+ const record = this.records.get(id);
173
+ if (!record) return false;
174
+ if (record.status === "queued") {
175
+ this.queue = this.queue.filter((entry) => entry.record.id !== id);
176
+ this.markStopped(record, reason ?? "Aborted by caller.");
177
+ return true;
178
+ }
179
+ if (record.status !== "running" && record.status !== "blocked") return false;
180
+ this.controllers.get(id)?.abort();
181
+ this.markStopped(record, reason ?? "Aborted by caller.");
182
+ return true;
183
+ }
184
+
185
+ abortAll(reason?: string): number {
186
+ let count = 0;
187
+ const stopReason = reason ?? "Aborted (session switch or shutdown).";
188
+ for (const entry of this.queue) {
189
+ this.markStopped(entry.record, stopReason);
190
+ count++;
191
+ }
192
+ this.queue = [];
193
+ for (const record of this.records.values()) {
194
+ if (record.status === "running" || record.status === "blocked") {
195
+ this.controllers.get(record.id)?.abort();
196
+ this.markStopped(record, stopReason);
197
+ count++;
198
+ }
199
+ }
200
+ return count;
201
+ }
202
+
203
+ async waitForAll(): Promise<void> {
204
+ while (true) {
205
+ this.drainQueue();
206
+ const pending = this.listAgents().filter((record) => record.status === "running" || record.status === "queued").map((record) => record.promise).filter((promise): promise is Promise<void> => Boolean(promise));
207
+ if (!pending.length) break;
208
+ await Promise.allSettled(pending);
209
+ }
210
+ }
211
+
212
+ async waitForRecord(id: string): Promise<SubagentRecord | undefined> {
213
+ while (true) {
214
+ const record = this.records.get(id);
215
+ if (!record) return undefined;
216
+ if (record.status !== "running" && record.status !== "queued") return record;
217
+ if (record.promise) await record.promise.catch(() => { /* status already set to error */ });
218
+ else await new Promise((resolve) => setTimeout(resolve, 100));
219
+ }
220
+ }
221
+
222
+ setMaxConcurrent(value: number): void {
223
+ this.maxConcurrent = Math.max(1, Math.floor(value));
224
+ this.drainQueue();
225
+ }
226
+
227
+ private start(record: SubagentRecord, options: SubagentSpawnOptions, runner: SpawnRunner, signal?: AbortSignal): void {
228
+ if (options.background) this.runningBackground++;
229
+ record.status = "running";
230
+ record.startedAt = Date.now();
231
+ record.completedAt = undefined;
232
+ const runSignal = this.createRunSignal(record.id, signal);
233
+ savePersistedSubagentRecord(options.cwd, record);
234
+ record.promise = (async () => {
235
+ try {
236
+ const result = await runner(options, runSignal);
237
+ if (record.status === "stopped") return;
238
+ record.runId = detailsRunId(result);
239
+ record.result = resultText(result);
240
+ savePersistedSubagentRecord(options.cwd, record);
241
+ if (result.isError) {
242
+ record.status = "error";
243
+ record.error = record.result;
244
+ throw new Error(record.error);
245
+ }
246
+ if (record.runId) await this.pollRunToTerminal(options.cwd, record);
247
+ else record.status = "completed";
248
+ } catch (error) {
249
+ if (record.status === "stopped" || runSignal.aborted) {
250
+ const abortReason = runSignal.aborted ? "Signal aborted — agent cancelled by parent (session switch, user cancel, or tool timeout)." : undefined;
251
+ record.status = "stopped";
252
+ if (!record.error) record.error = abortReason ?? (error instanceof Error ? error.message : String(error));
253
+ return;
254
+ }
255
+ record.status = "error";
256
+ record.error = error instanceof Error ? error.message : String(error);
257
+ throw error; // H4: Propagate rejection so callers awaiting record.promise see the error
258
+ } finally {
259
+ this.cleanupRunSignal(record.id);
260
+ if (options.background) this.runningBackground = Math.max(0, this.runningBackground - 1);
261
+ if (record.status !== "blocked") record.completedAt = record.completedAt ?? Date.now();
262
+ savePersistedSubagentRecord(options.cwd, record);
263
+ if (record.status === "completed" || record.status === "failed" || record.status === "cancelled" || record.status === "error" || record.status === "stopped") {
264
+ // Phase 1.6: Populate telemetry fields
265
+ record.turnCount = record.turnCount ?? totalRunTurns(options.cwd, record.runId);
266
+ record.durationMs = record.completedAt ? Math.max(0, record.completedAt - record.startedAt) : undefined;
267
+ savePersistedSubagentRecord(options.cwd, record);
268
+ this.onComplete?.(record);
269
+ }
270
+ this.drainQueue();
271
+ }
272
+ })();
273
+ }
274
+
275
+ private markStopped(record: SubagentRecord, reason?: string): void {
276
+ record.status = "stopped";
277
+ record.completedAt = Date.now();
278
+ if (reason && !record.error) record.error = reason;
279
+ const cwd = this.cwdByRecord.get(record.id);
280
+ if (cwd) savePersistedSubagentRecord(cwd, record);
281
+ }
282
+
283
+ private createRunSignal(id: string, signal?: AbortSignal): AbortSignal {
284
+ const controller = new AbortController();
285
+ this.controllers.set(id, controller);
286
+ if (signal?.aborted) {
287
+ controller.abort();
288
+ return controller.signal;
289
+ }
290
+ if (signal) {
291
+ const abort = (): void => controller.abort();
292
+ signal.addEventListener("abort", abort, { once: true });
293
+ this.controllerCleanup.set(id, () => signal.removeEventListener("abort", abort));
294
+ }
295
+ return controller.signal;
296
+ }
297
+
298
+ private cleanupRunSignal(id: string): void {
299
+ this.controllerCleanup.get(id)?.();
300
+ this.controllerCleanup.delete(id);
301
+ this.controllers.delete(id);
302
+ }
303
+
304
+ private drainQueue(): void {
305
+ while (this.queue.length > 0 && this.runningBackground < this.maxConcurrent) {
306
+ const next = this.queue.shift();
307
+ if (!next || next.record.status !== "queued") continue;
308
+ this.start(next.record, next.options, next.runner, next.signal);
309
+ }
310
+ }
311
+
312
+ private async pollRunToTerminal(cwd: string, record: SubagentRecord): Promise<void> {
313
+ while (record.runId && (record.status === "running" || record.status === "blocked")) {
314
+ const loaded = loadRunManifestById(cwd, record.runId);
315
+ if (!loaded) {
316
+ await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
317
+ continue;
318
+ }
319
+ if (loaded.manifest.status === "completed") {
320
+ record.status = "completed";
321
+ record.error = undefined;
322
+ record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
323
+ record.completedAt = Date.now();
324
+ savePersistedSubagentRecord(cwd, record);
325
+ return;
326
+ }
327
+ if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
328
+ record.status = loaded.manifest.status;
329
+ record.error = loaded.manifest.summary;
330
+ record.turnCount = record.turnCount ?? totalRunTurns(cwd, record.runId);
331
+ record.completedAt = Date.now();
332
+ savePersistedSubagentRecord(cwd, record);
333
+ return;
334
+ }
335
+ if (loaded.manifest.status === "blocked") {
336
+ record.status = "blocked";
337
+ record.error = undefined;
338
+ if (!record.blockedAt) {
339
+ record.blockedAt = Date.now();
340
+ record.stuckNotified = false;
341
+ record.completedAt = undefined;
342
+ this.onComplete?.(record);
343
+ this.scheduleStuckBlockedNotify(cwd, record);
344
+ this.scheduleBlockedTerminalPoll(cwd, record);
345
+ }
346
+ savePersistedSubagentRecord(cwd, record);
347
+ return;
348
+ }
349
+ await new Promise((resolve) => setTimeout(resolve, this.pollIntervalMs));
350
+ }
351
+ }
352
+
353
+ private scheduleBlockedTerminalPoll(cwd: string, record: SubagentRecord): void {
354
+ const poll = (): void => {
355
+ const current = this.records.get(record.id);
356
+ if (!current || current.status !== "blocked" || !current.runId) return;
357
+ const loaded = loadRunManifestById(cwd, current.runId);
358
+ if (!loaded || loaded.manifest.status === "blocked" || loaded.manifest.status === "running" || loaded.manifest.status === "planning" || loaded.manifest.status === "queued") {
359
+ const timer = setTimeout(poll, this.pollIntervalMs);
360
+ timer.unref();
361
+ return;
362
+ }
363
+ const persisted = readPersistedSubagentRecord(cwd, current.id);
364
+ current.resultConsumed = current.resultConsumed || persisted?.resultConsumed;
365
+ if (loaded.manifest.status === "completed") {
366
+ current.status = "completed";
367
+ current.error = undefined;
368
+ } else if (loaded.manifest.status === "failed" || loaded.manifest.status === "cancelled") {
369
+ current.status = loaded.manifest.status;
370
+ current.error = loaded.manifest.summary;
371
+ } else return;
372
+ current.completedAt = Date.now();
373
+ current.turnCount = current.turnCount ?? totalRunTurns(cwd, current.runId);
374
+ current.durationMs = Math.max(0, current.completedAt - current.startedAt);
375
+ savePersistedSubagentRecord(cwd, current);
376
+ this.onComplete?.(current);
377
+ };
378
+ const timer = setTimeout(poll, this.pollIntervalMs);
379
+ timer.unref();
380
+ }
381
+
382
+ private scheduleStuckBlockedNotify(cwd: string, record: SubagentRecord): void {
383
+ const threshold = DEFAULT_SUBAGENT.stuckBlockedNotifyMs;
384
+ const fire = (): void => {
385
+ const current = this.records.get(record.id);
386
+ if (!current || current.status !== "blocked" || !current.blockedAt || current.stuckNotified) return;
387
+ current.stuckNotified = true;
388
+ this.onEvent?.("subagent.stuck-blocked", {
389
+ event: "subagent.stuck-blocked",
390
+ id: current.id,
391
+ runId: current.runId,
392
+ durationMs: Math.max(0, Date.now() - current.blockedAt),
393
+ ownerSessionGeneration: current.ownerSessionGeneration,
394
+ });
395
+ savePersistedSubagentRecord(cwd, current);
396
+ };
397
+ if (threshold <= 0) {
398
+ fire();
399
+ return;
400
+ }
401
+ const timer = setTimeout(fire, threshold);
402
+ timer.unref();
403
+ }
404
+ }