@zhushanwen/pi-subagent-workflow 8.6.0 → 8.8.0

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 (364) hide show
  1. package/README.md +5 -11
  2. package/package.json +13 -20
  3. package/relay/relay.mjs +1 -1
  4. package/skills/workflow-script-format/SKILL.md +32 -13
  5. package/src/host/__tests__/pi-host.test.ts +235 -0
  6. package/src/host/pi-host.ts +163 -0
  7. package/src/index.ts +141 -138
  8. package/src/injectors/__tests__/engine-awareness.test.ts +292 -0
  9. package/src/injectors/__tests__/engine-section-stability.test.ts +343 -0
  10. package/src/injectors/__tests__/model-list-injector.test.ts +18 -21
  11. package/src/injectors/__tests__/subagent-list-injector.test.ts +59 -19
  12. package/src/injectors/__tests__/workflow-list-injector.test.ts +31 -17
  13. package/src/injectors/engine-awareness.ts +119 -0
  14. package/src/injectors/model-list-injector.ts +15 -58
  15. package/src/injectors/subagent-list-injector.ts +56 -114
  16. package/src/injectors/workflow-list-injector.ts +37 -70
  17. package/src/interface/__tests__/detectors.test.ts +48 -37
  18. package/src/interface/__tests__/subagent-tool-path-guard.test.ts +2 -2
  19. package/src/interface/__tests__/subagent-tool-prompt.test.ts +6 -3
  20. package/src/interface/__tests__/tool-workflow-run-builtin-name.test.ts +216 -0
  21. package/src/interface/__tests__/tool-workflow-script-generate.test.ts +17 -6
  22. package/src/interface/__tests__/tool-workflow-throw-paths.test.ts +38 -2
  23. package/src/interface/bg-notify-render.ts +7 -17
  24. package/src/interface/command-actions.ts +5 -15
  25. package/src/interface/commands.ts +4 -4
  26. package/src/interface/format.ts +17 -6
  27. package/src/interface/gui-mappers.ts +18 -22
  28. package/src/interface/helpers.ts +9 -130
  29. package/src/interface/list-component.ts +5 -5
  30. package/src/interface/list-shared.ts +3 -3
  31. package/src/interface/list-view.ts +3 -3
  32. package/src/interface/subagent-actions.ts +66 -439
  33. package/src/interface/subagent-tool-schema.ts +21 -17
  34. package/src/interface/subagent-tool.ts +12 -7
  35. package/src/interface/subagents.ts +4 -4
  36. package/src/interface/tool-render.ts +37 -15
  37. package/src/interface/tool-workflow-script.ts +34 -76
  38. package/src/interface/tool-workflow.ts +57 -94
  39. package/src/interface/views/WorkflowsView.ts +8 -16
  40. package/src/interface/views/__tests__/WorkflowsView-signature.test.ts +3 -3
  41. package/src/interface/views/__tests__/detail-content-session-file.test.ts +2 -2
  42. package/src/interface/views/detail-content.ts +4 -4
  43. package/src/interface/views/format.ts +23 -62
  44. package/src/{orchestration/jsonl-run-store.ts → jsonl-run-store.ts} +137 -225
  45. package/agents/analyst.md +0 -61
  46. package/agents/coder.md +0 -70
  47. package/agents/debugger.md +0 -67
  48. package/agents/doc-reviewer.md +0 -50
  49. package/agents/explorer.md +0 -64
  50. package/agents/general-purpose.md +0 -32
  51. package/agents/orchestrator.md +0 -63
  52. package/agents/planner.md +0 -54
  53. package/agents/researcher.md +0 -65
  54. package/agents/reviewer.md +0 -74
  55. package/src/execution/__tests__/__fixtures__/notifier-golden-snapshots.json +0 -53
  56. package/src/execution/__tests__/__fixtures__/truncline.snapshot.json +0 -1
  57. package/src/execution/__tests__/agent-registry.test.ts +0 -363
  58. package/src/execution/__tests__/agent-result-mapper.test.ts +0 -150
  59. package/src/execution/__tests__/alive-store.test.ts +0 -147
  60. package/src/execution/__tests__/ask-user-transit-e2e.test.ts +0 -490
  61. package/src/execution/__tests__/before-agent-start-injection.test.ts +0 -132
  62. package/src/execution/__tests__/bg-notify-render.test.ts +0 -329
  63. package/src/execution/__tests__/channel-registry-handshake.test.ts +0 -243
  64. package/src/execution/__tests__/chat-engine-routing.test.ts +0 -601
  65. package/src/execution/__tests__/chatmode-first-round-closure-service.test.ts +0 -365
  66. package/src/execution/__tests__/chatmode-first-round-closure-spawn.test.ts +0 -190
  67. package/src/execution/__tests__/chatmode-round-notify-real-chain.test.ts +0 -215
  68. package/src/execution/__tests__/concurrency-pool.test.ts +0 -250
  69. package/src/execution/__tests__/config.test.ts +0 -110
  70. package/src/execution/__tests__/conversation-wiring.test.ts +0 -198
  71. package/src/execution/__tests__/crash-recovery.test.ts +0 -320
  72. package/src/execution/__tests__/delivery-methods.test.ts +0 -422
  73. package/src/execution/__tests__/dialog-queue.test.ts +0 -299
  74. package/src/execution/__tests__/epipe-fallback.test.ts +0 -241
  75. package/src/execution/__tests__/execute-and-await-worktree.test.ts +0 -243
  76. package/src/execution/__tests__/execute-nesting.test.ts +0 -323
  77. package/src/execution/__tests__/execute-options-mapper.test.ts +0 -200
  78. package/src/execution/__tests__/execution-record.test.ts +0 -1394
  79. package/src/execution/__tests__/explicit-agent-ref-guard.test.ts +0 -171
  80. package/src/execution/__tests__/finalize-record.test.ts +0 -368
  81. package/src/execution/__tests__/finalized-marker.test.ts +0 -82
  82. package/src/execution/__tests__/format-schema-instruction.test.ts +0 -169
  83. package/src/execution/__tests__/format.test.ts +0 -458
  84. package/src/execution/__tests__/gc-timer.test.ts +0 -184
  85. package/src/execution/__tests__/get-record-for-action-restart.test.ts +0 -254
  86. package/src/execution/__tests__/gui-mode-dispatch.test.ts +0 -59
  87. package/src/execution/__tests__/helpers/mock-extension-api.ts +0 -30
  88. package/src/execution/__tests__/helpers/spawn-mock.ts +0 -242
  89. package/src/execution/__tests__/host-mode.test.ts +0 -87
  90. package/src/execution/__tests__/index-session-start-identity.test.ts +0 -371
  91. package/src/execution/__tests__/index-session-start.test.ts +0 -672
  92. package/src/execution/__tests__/lifecycle-manager-lock.test.ts +0 -211
  93. package/src/execution/__tests__/lifecycle-manager.test.ts +0 -383
  94. package/src/execution/__tests__/lifecycle-predicates.test.ts +0 -116
  95. package/src/execution/__tests__/list-component.test.ts +0 -401
  96. package/src/execution/__tests__/list-fields.test.ts +0 -140
  97. package/src/execution/__tests__/manifest-parentid.test.ts +0 -117
  98. package/src/execution/__tests__/model-resolver.test.ts +0 -465
  99. package/src/execution/__tests__/nested-visibility-env-propagation.test.ts +0 -287
  100. package/src/execution/__tests__/nested-visibility.test.ts +0 -325
  101. package/src/execution/__tests__/notifier-flush.test.ts +0 -612
  102. package/src/execution/__tests__/notifier-golden-snapshot.test.ts +0 -261
  103. package/src/execution/__tests__/notify-ledger.test.ts +0 -826
  104. package/src/execution/__tests__/one-shot-upgrade.test.ts +0 -205
  105. package/src/execution/__tests__/output-collector.test.ts +0 -358
  106. package/src/execution/__tests__/parent-child-matrix.test.ts +0 -336
  107. package/src/execution/__tests__/path-encoding.test.ts +0 -104
  108. package/src/execution/__tests__/pi-invocation.test.ts +0 -134
  109. package/src/execution/__tests__/record-store.test.ts +0 -1057
  110. package/src/execution/__tests__/records-cwd-isolation.test.ts +0 -91
  111. package/src/execution/__tests__/recursive-visibility-baseline.test.ts +0 -339
  112. package/src/execution/__tests__/recursive-visibility-env.test.ts +0 -273
  113. package/src/execution/__tests__/relay-agent.test.ts +0 -448
  114. package/src/execution/__tests__/relay-env.test.ts +0 -42
  115. package/src/execution/__tests__/resource-policy.test.ts +0 -109
  116. package/src/execution/__tests__/rpc-mode.test.ts +0 -89
  117. package/src/execution/__tests__/run-and-finalize-chatmode.test.ts +0 -267
  118. package/src/execution/__tests__/run-spawn-chatmode-settled.test.ts +0 -253
  119. package/src/execution/__tests__/run-spawn-edges.test.ts +0 -687
  120. package/src/execution/__tests__/run-spawn-integration.test.ts +0 -933
  121. package/src/execution/__tests__/run-spawn-resume.test.ts +0 -322
  122. package/src/execution/__tests__/run-spawn-rpc-mode.test.ts +0 -196
  123. package/src/execution/__tests__/run-spawn-stdout-callback-throw.test.ts +0 -199
  124. package/src/execution/__tests__/sdk-contract.test.ts +0 -273
  125. package/src/execution/__tests__/session-context-resolver.test.ts +0 -167
  126. package/src/execution/__tests__/session-file-gc.test.ts +0 -293
  127. package/src/execution/__tests__/session-pending.test.ts +0 -197
  128. package/src/execution/__tests__/session-reconstructor.test.ts +0 -379
  129. package/src/execution/__tests__/session-runner-epipe.test.ts +0 -178
  130. package/src/execution/__tests__/session-runner-schema-env.test.ts +0 -350
  131. package/src/execution/__tests__/session-start-reaper.test.ts +0 -233
  132. package/src/execution/__tests__/spawn-args.test.ts +0 -445
  133. package/src/execution/__tests__/spawn-event-adapter-rpc.test.ts +0 -189
  134. package/src/execution/__tests__/spawn-event-adapter.test.ts +0 -167
  135. package/src/execution/__tests__/spawn-worktree-guidance.test.ts +0 -207
  136. package/src/execution/__tests__/spawned-children.test.ts +0 -92
  137. package/src/execution/__tests__/start-sync-model-guard.test.ts +0 -150
  138. package/src/execution/__tests__/startup-config-declaration.test.ts +0 -35
  139. package/src/execution/__tests__/status-refactor.test.ts +0 -345
  140. package/src/execution/__tests__/stdin-writer.test.ts +0 -463
  141. package/src/execution/__tests__/stream-sink-retirement.test.ts +0 -261
  142. package/src/execution/__tests__/subagent-service-abort.test.ts +0 -60
  143. package/src/execution/__tests__/subagent-service-message-close.test.ts +0 -629
  144. package/src/execution/__tests__/subagent-service-parent-guard.test.ts +0 -180
  145. package/src/execution/__tests__/subagent-service.test.ts +0 -788
  146. package/src/execution/__tests__/subprocess-agent-runner-routing.test.ts +0 -310
  147. package/src/execution/__tests__/subprocess-agent-runner.test.ts +0 -599
  148. package/src/execution/__tests__/temp-prompt.test.ts +0 -53
  149. package/src/execution/__tests__/timeout-integration.test.ts +0 -615
  150. package/src/execution/__tests__/tombstone-store.test.ts +0 -73
  151. package/src/execution/__tests__/tool-action.test.ts +0 -476
  152. package/src/execution/__tests__/truncline-snapshot.test.ts +0 -81
  153. package/src/execution/__tests__/turn-limiter-semantics.test.ts +0 -194
  154. package/src/execution/__tests__/turn-limiter.test.ts +0 -65
  155. package/src/execution/__tests__/ui-channels.test.ts +0 -187
  156. package/src/execution/__tests__/ui-interaction-model.test.ts +0 -67
  157. package/src/execution/__tests__/ui-request-handler-factory.test.ts +0 -370
  158. package/src/execution/__tests__/ui-request-handler.test.ts +0 -204
  159. package/src/execution/__tests__/ui-request-observability.test.ts +0 -114
  160. package/src/execution/__tests__/ui-request-queue.test.ts +0 -133
  161. package/src/execution/__tests__/worktree-manager.test.ts +0 -633
  162. package/src/execution/__tests__/worktree-pid-registration.integration.test.ts +0 -229
  163. package/src/execution/__tests__/worktree-reconcile.integration.test.ts +0 -181
  164. package/src/execution/__tests__/worktree-registry.test.ts +0 -199
  165. package/src/execution/agent-registry.ts +0 -199
  166. package/src/execution/agent-result-mapper.ts +0 -90
  167. package/src/execution/alive-store.ts +0 -92
  168. package/src/execution/argv-mirror.ts +0 -113
  169. package/src/execution/best-effort.ts +0 -38
  170. package/src/execution/channel-registry-access.ts +0 -144
  171. package/src/execution/concurrency-pool.ts +0 -116
  172. package/src/execution/config.ts +0 -96
  173. package/src/execution/dialog-queue.ts +0 -330
  174. package/src/execution/engine/__tests__/common/data-dir.test.ts +0 -53
  175. package/src/execution/engine/__tests__/common/errors.test.ts +0 -132
  176. package/src/execution/engine/__tests__/common/event-journal.test.ts +0 -177
  177. package/src/execution/engine/__tests__/common/kill-chain.test.ts +0 -192
  178. package/src/execution/engine/__tests__/common/nesting-guard.test.ts +0 -81
  179. package/src/execution/engine/__tests__/common/persona-router.test.ts +0 -123
  180. package/src/execution/engine/__tests__/common/pool-manager.test.ts +0 -154
  181. package/src/execution/engine/__tests__/common/schema-emulation.test.ts +0 -128
  182. package/src/execution/engine/__tests__/conformance/__fixtures__/pi-golden-events.json +0 -28
  183. package/src/execution/engine/__tests__/conformance/agent-event-invariants.ts +0 -141
  184. package/src/execution/engine/__tests__/conformance/contract.abort.test.ts +0 -109
  185. package/src/execution/engine/__tests__/conformance/contract.agent-events.test.ts +0 -101
  186. package/src/execution/engine/__tests__/conformance/contract.probe.test.ts +0 -77
  187. package/src/execution/engine/__tests__/conformance/contract.read-degradation.test.ts +0 -104
  188. package/src/execution/engine/__tests__/conformance/contract.relay.test.ts +0 -342
  189. package/src/execution/engine/__tests__/conformance/engine-conformance.live.test.ts +0 -201
  190. package/src/execution/engine/__tests__/conformance/golden-replay.pi.test.ts +0 -76
  191. package/src/execution/engine/__tests__/conformance/golden-replay.zcode.test.ts +0 -79
  192. package/src/execution/engine/__tests__/engine-discovery.test.ts +0 -87
  193. package/src/execution/engine/__tests__/engines-declaration.test.ts +0 -36
  194. package/src/execution/engine/__tests__/model-prompt.test.ts +0 -85
  195. package/src/execution/engine/__tests__/paths.test.ts +0 -39
  196. package/src/execution/engine/__tests__/registry.test.ts +0 -120
  197. package/src/execution/engine/__tests__/routing.test.ts +0 -231
  198. package/src/execution/engine/common/data-dir.ts +0 -62
  199. package/src/execution/engine/common/errors.ts +0 -183
  200. package/src/execution/engine/common/event-journal.ts +0 -254
  201. package/src/execution/engine/common/journal-replay.ts +0 -62
  202. package/src/execution/engine/common/kill-chain.ts +0 -221
  203. package/src/execution/engine/common/nesting-guard.ts +0 -50
  204. package/src/execution/engine/common/persona-router.ts +0 -108
  205. package/src/execution/engine/common/pool-manager.ts +0 -226
  206. package/src/execution/engine/common/schema-emulation.ts +0 -189
  207. package/src/execution/engine/common/session-view-projection.ts +0 -51
  208. package/src/execution/engine/engine-discovery.ts +0 -65
  209. package/src/execution/engine/engines/pi/__tests__/pi-engine.test.ts +0 -469
  210. package/src/execution/engine/engines/pi/__tests__/reader.test.ts +0 -155
  211. package/src/execution/engine/engines/pi/__tests__/task-spec-mapper.test.ts +0 -164
  212. package/src/execution/engine/engines/pi/pi-engine.ts +0 -415
  213. package/src/execution/engine/engines/pi/reader.ts +0 -48
  214. package/src/execution/engine/engines/pi/registration.ts +0 -35
  215. package/src/execution/engine/engines/pi/task-spec-mapper.ts +0 -100
  216. package/src/execution/engine/engines/zcode/__tests__/__fixtures__/zcode-golden-spawn.json +0 -39
  217. package/src/execution/engine/engines/zcode/__tests__/launcher.test.ts +0 -150
  218. package/src/execution/engine/engines/zcode/__tests__/parser.test.ts +0 -246
  219. package/src/execution/engine/engines/zcode/__tests__/preparer.test.ts +0 -228
  220. package/src/execution/engine/engines/zcode/__tests__/reader.test.ts +0 -210
  221. package/src/execution/engine/engines/zcode/__tests__/registration.test.ts +0 -64
  222. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.live.test.ts +0 -127
  223. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.test.ts +0 -580
  224. package/src/execution/engine/engines/zcode/constants.ts +0 -43
  225. package/src/execution/engine/engines/zcode/golden-sample.ts +0 -39
  226. package/src/execution/engine/engines/zcode/launcher.ts +0 -161
  227. package/src/execution/engine/engines/zcode/parser.ts +0 -436
  228. package/src/execution/engine/engines/zcode/preparer.ts +0 -363
  229. package/src/execution/engine/engines/zcode/reader.ts +0 -381
  230. package/src/execution/engine/engines/zcode/registration.ts +0 -37
  231. package/src/execution/engine/engines/zcode/zcode-engine.ts +0 -658
  232. package/src/execution/engine/host-task-spec.ts +0 -47
  233. package/src/execution/engine/model-prompt.ts +0 -59
  234. package/src/execution/engine/paths.ts +0 -42
  235. package/src/execution/engine/port.ts +0 -153
  236. package/src/execution/engine/registry.ts +0 -123
  237. package/src/execution/engine/routing.ts +0 -218
  238. package/src/execution/engine/types.ts +0 -309
  239. package/src/execution/execute-options-mapper.ts +0 -112
  240. package/src/execution/execution-record.ts +0 -965
  241. package/src/execution/finalize-record.ts +0 -251
  242. package/src/execution/finalized-marker.ts +0 -51
  243. package/src/execution/get-state-handshake.ts +0 -104
  244. package/src/execution/host-mode.ts +0 -56
  245. package/src/execution/idle-gc.ts +0 -47
  246. package/src/execution/lifecycle-manager.ts +0 -513
  247. package/src/execution/lifecycle-predicates.ts +0 -65
  248. package/src/execution/manifest-store.ts +0 -256
  249. package/src/execution/model-config-service.ts +0 -219
  250. package/src/execution/model-resolver.ts +0 -248
  251. package/src/execution/notifier.ts +0 -330
  252. package/src/execution/notify-ledger.ts +0 -580
  253. package/src/execution/output-collector.ts +0 -228
  254. package/src/execution/path-encoding.ts +0 -60
  255. package/src/execution/pi-invocation.ts +0 -120
  256. package/src/execution/record-entry.ts +0 -132
  257. package/src/execution/record-store.ts +0 -1234
  258. package/src/execution/relay-env.ts +0 -37
  259. package/src/execution/session-context-resolver.ts +0 -64
  260. package/src/execution/session-file-gc.ts +0 -120
  261. package/src/execution/session-pending.ts +0 -164
  262. package/src/execution/session-reconstructor.ts +0 -678
  263. package/src/execution/session-runner.ts +0 -1781
  264. package/src/execution/sessions-index.ts +0 -304
  265. package/src/execution/spawn-event-adapter.ts +0 -363
  266. package/src/execution/stdin-writer.ts +0 -198
  267. package/src/execution/stream-sink.ts +0 -126
  268. package/src/execution/subagent-service.ts +0 -2141
  269. package/src/execution/subprocess-agent-runner.ts +0 -317
  270. package/src/execution/temp-prompt.ts +0 -62
  271. package/src/execution/tombstone-store.ts +0 -72
  272. package/src/execution/turn-limiter.ts +0 -102
  273. package/src/execution/types.ts +0 -973
  274. package/src/execution/ui-channels.ts +0 -216
  275. package/src/execution/ui-interaction-model.ts +0 -48
  276. package/src/execution/ui-request-handler-factory.ts +0 -212
  277. package/src/execution/ui-request-observability.ts +0 -81
  278. package/src/execution/ui-request-queue.ts +0 -184
  279. package/src/execution/worktree-manager.ts +0 -688
  280. package/src/execution/worktree-registry.ts +0 -222
  281. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +0 -349
  282. package/src/orchestration/__tests__/agent-call-catch-fallback.test.ts +0 -202
  283. package/src/orchestration/__tests__/agent-call-stream.test.ts +0 -152
  284. package/src/orchestration/__tests__/args-validator.test.ts +0 -143
  285. package/src/orchestration/__tests__/config-loader.test.ts +0 -503
  286. package/src/orchestration/__tests__/error-recovery-handlers.test.ts +0 -809
  287. package/src/orchestration/__tests__/error-recovery-postmessage-defense.test.ts +0 -405
  288. package/src/orchestration/__tests__/error-recovery-serialize-failed-result.test.ts +0 -56
  289. package/src/orchestration/__tests__/error-recovery-workflow-call.test.ts +0 -166
  290. package/src/orchestration/__tests__/execute-agent-call.test.ts +0 -451
  291. package/src/orchestration/__tests__/jsonl-run-store-corrupt-entry.test.ts +0 -150
  292. package/src/orchestration/__tests__/jsonl-run-store-loadall-sources.test.ts +0 -171
  293. package/src/orchestration/__tests__/jsonl-run-store-retention.test.ts +0 -202
  294. package/src/orchestration/__tests__/jsonl-run-store-session-file.test.ts +0 -1030
  295. package/src/orchestration/__tests__/launcher-nested-workflow.test.ts +0 -600
  296. package/src/orchestration/__tests__/lifecycle-runid-injection.test.ts +0 -96
  297. package/src/orchestration/__tests__/lifecycle.test.ts +0 -659
  298. package/src/orchestration/__tests__/non-cloneable-return-e2e.test.ts +0 -95
  299. package/src/orchestration/__tests__/review-fix-loop-e2e.test.ts +0 -2025
  300. package/src/orchestration/__tests__/script-lint.test.ts +0 -831
  301. package/src/orchestration/__tests__/skill-discovery.test.ts +0 -201
  302. package/src/orchestration/__tests__/test-mocks.ts +0 -197
  303. package/src/orchestration/__tests__/worker-exit-without-result.test.ts +0 -368
  304. package/src/orchestration/__tests__/worker-host.test.ts +0 -120
  305. package/src/orchestration/__tests__/worker-returnmeta-passthrough.test.ts +0 -164
  306. package/src/orchestration/__tests__/worker-script-builder-runtime.test.ts +0 -563
  307. package/src/orchestration/__tests__/worker-script-builder.test.ts +0 -309
  308. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +0 -129
  309. package/src/orchestration/__tests__/workflow-nesting-e2e.test.ts +0 -317
  310. package/src/orchestration/__tests__/workflow-script-lint-memo.test.ts +0 -110
  311. package/src/orchestration/__tests__/workflows-e2e.test.ts +0 -537
  312. package/src/orchestration/agent-opts-resolver.ts +0 -174
  313. package/src/orchestration/args-validator.ts +0 -127
  314. package/src/orchestration/config-loader.ts +0 -315
  315. package/src/orchestration/error-recovery.ts +0 -1018
  316. package/src/orchestration/execute-agent-call.ts +0 -256
  317. package/src/orchestration/launcher.ts +0 -455
  318. package/src/orchestration/lifecycle.ts +0 -399
  319. package/src/orchestration/models/__tests__/budget.test.ts +0 -307
  320. package/src/orchestration/models/__tests__/trace.test.ts +0 -408
  321. package/src/orchestration/models/agent-call.ts +0 -83
  322. package/src/orchestration/models/budget.ts +0 -118
  323. package/src/orchestration/models/ports.ts +0 -164
  324. package/src/orchestration/models/run-runtime.ts +0 -104
  325. package/src/orchestration/models/run-spec.ts +0 -83
  326. package/src/orchestration/models/run-state.ts +0 -44
  327. package/src/orchestration/models/trace.ts +0 -183
  328. package/src/orchestration/models/types.ts +0 -301
  329. package/src/orchestration/models/workflow-run.ts +0 -254
  330. package/src/orchestration/models/workflow-script-registry.ts +0 -35
  331. package/src/orchestration/models/workflow-script.ts +0 -117
  332. package/src/orchestration/script-lint.ts +0 -766
  333. package/src/orchestration/skill-discovery.ts +0 -115
  334. package/src/orchestration/worker-handle.ts +0 -115
  335. package/src/orchestration/worker-host.ts +0 -98
  336. package/src/orchestration/worker-script-builder.ts +0 -421
  337. package/src/orchestration/workflow-files.ts +0 -85
  338. package/src/orchestration/workflow-script-registry-impl.ts +0 -133
  339. package/src/shared/__tests__/agent-ref.test.ts +0 -34
  340. package/src/shared/__tests__/meta-parser.test.ts +0 -304
  341. package/src/shared/__tests__/model-ref.test.ts +0 -306
  342. package/src/shared/__tests__/resource-discovery-manifest-cache.test.ts +0 -280
  343. package/src/shared/__tests__/resource-discovery.test.ts +0 -736
  344. package/src/shared/__tests__/resource-meta.test.ts +0 -51
  345. package/src/shared/__tests__/schema-jsonify.test.ts +0 -81
  346. package/src/shared/__tests__/timer-delay.test.ts +0 -61
  347. package/src/shared/agent-event.ts +0 -13
  348. package/src/shared/agent-ref.ts +0 -57
  349. package/src/shared/meta-parser.ts +0 -261
  350. package/src/shared/model-ref.ts +0 -286
  351. package/src/shared/resource-discovery.ts +0 -811
  352. package/src/shared/resource-meta.ts +0 -65
  353. package/src/shared/schema-env.ts +0 -44
  354. package/src/shared/schema-jsonify.ts +0 -58
  355. package/src/shared/timer-delay.ts +0 -54
  356. package/src/shared/xml-injection.ts +0 -35
  357. package/workflows/README.md +0 -81
  358. package/workflows/_shared/agent-refs.cjs +0 -40
  359. package/workflows/chain.js +0 -131
  360. package/workflows/map-reduce.js +0 -174
  361. package/workflows/parallel.js +0 -148
  362. package/workflows/review-fix-loop-utils.cjs +0 -1309
  363. package/workflows/review-fix-loop.js +0 -1404
  364. package/workflows/scatter-gather.js +0 -178
@@ -1,350 +0,0 @@
1
- // src/__tests__/session-runner-schema-env.test.ts
2
- //
3
- // Wave 2 (issue #3): schemaEnv bridge 测试。
4
- //
5
- // 覆盖 test-matrix 用例:
6
- // - T3.9 (boundary): schemaEnv 透传——传入时 childEnv 含 PI_WORKFLOW_SCHEMA
7
- // - T3.11 (state): schemaEnv 不传 → childEnv 无 PI_WORKFLOW_SCHEMA(BC-6 tool 层不变)
8
- // - T3.16 (NFR-compatibility): schemaEnv 不传时 BC-6 childEnv 等价——不传时与合并前
9
- // 行为一致,不注入 PI_WORKFLOW_SCHEMA → structured-output tool 不注册
10
- //
11
- // 测试策略:
12
- // - applySchemaEnvToChildEnv 纯函数单测(不依赖 runSpawn/spawn mock)
13
- // - runSpawn 集成测试:通过 mock spawn 拦截 childEnv,验证 schemaEnv 实际注入
14
- //
15
- // D-A6: schemaEnv 经 RunOptions 透传到 runSpawn childEnv。
16
- // BC-6: tool 层 execute 不传 schemaEnv → childEnv 不设 PI_WORKFLOW_SCHEMA → 行为不变。
17
-
18
- import type { PassThrough } from "node:stream";
19
-
20
- import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
21
-
22
- // ── mock modules (与 run-spawn-integration.test.ts 同模式) ──
23
-
24
- vi.mock("node:child_process", async () => {
25
- const { EventEmitter } = await import("node:events");
26
- const { PassThrough } = await import("node:stream");
27
-
28
- class FakeChild extends EventEmitter {
29
- pid = 12345;
30
- stdout = new PassThrough();
31
- stdin = new PassThrough(); // runSpawn 注册 stdin EPIPE handler(session-runner),FakeChild 必须提供
32
- stderr = new PassThrough();
33
- killed = false;
34
- killSignal: string | undefined;
35
- kill(sig?: string): boolean {
36
- this.killed = true;
37
- this.killSignal = sig;
38
- return true;
39
- }
40
- }
41
-
42
- return {
43
- spawn: vi.fn(() => new FakeChild()),
44
- // buildEnvBlock 用 execFile 异步取 git branch:默认 err-first 兜底(catch → branch="")
45
- execFile: vi.fn(
46
- (
47
- _cmd: string,
48
- _args: readonly string[],
49
- _opts: unknown,
50
- cb: (err: Error | null, stdout?: string, stderr?: string) => void,
51
- ) => cb(new Error("execFile not configured in this test")),
52
- ),
53
- };
54
- });
55
-
56
- vi.mock("node:fs", async () => {
57
- const actual = await import("node:fs");
58
- return {
59
- default: {
60
- ...actual,
61
- mkdirSync: vi.fn(),
62
- existsSync: vi.fn(() => false),
63
- appendFileSync: vi.fn(),
64
- writeFileSync: vi.fn(),
65
- readdirSync: vi.fn(() => []),
66
- },
67
- mkdirSync: vi.fn(),
68
- existsSync: vi.fn(() => false),
69
- appendFileSync: vi.fn(),
70
- writeFileSync: vi.fn(),
71
- readdirSync: vi.fn(() => []),
72
- promises: actual.promises,
73
- };
74
- });
75
-
76
- vi.mock("../alive-store.ts", () => ({
77
- writeAliveMarker: vi.fn(),
78
- }));
79
-
80
- vi.mock("../temp-prompt.ts", () => ({
81
- writePromptToTempFile: vi.fn(async (agent: string) => {
82
- const safeName = agent.replace(/[^\w.-]+/g, "_");
83
- return { dir: `/tmp/fake-${safeName}`, filePath: `/tmp/fake-${safeName}/prompt-${safeName}.md` };
84
- }),
85
- cleanupTempPrompt: vi.fn(async () => {}),
86
- }));
87
-
88
- import { spawn } from "node:child_process";
89
- import * as fs from "node:fs";
90
-
91
- import { waitForSpawn } from "./helpers/spawn-mock.ts";
92
- import { createRecord } from "../execution-record.ts";
93
- import {
94
- applySchemaEnvToChildEnv,
95
- type RunOptions,
96
- runSpawn,
97
- SCHEMA_ENV_MAX_BYTES,
98
- type SessionRunnerContext,
99
- } from "../session-runner.ts";
100
- import { schemaEnvByteLength } from "../../shared/schema-env.ts";
101
-
102
- const mockSpawn = vi.mocked(spawn);
103
- const mockExistsSync = vi.mocked(fs.existsSync);
104
-
105
- interface FakeChild {
106
- pid: number;
107
- stdout: PassThrough;
108
- stderr: PassThrough;
109
- killed: boolean;
110
- killSignal: string | undefined;
111
- kill(sig?: string): boolean;
112
- emit(event: string, ...args: unknown[]): boolean;
113
- }
114
-
115
- function getLastSpawnedChild(): FakeChild {
116
- const result = mockSpawn.mock.results.at(-1);
117
- if (!result) throw new Error("spawn was not called yet");
118
- return result.value as FakeChild;
119
- }
120
-
121
- function getLastSpawnEnv(): Record<string, string | undefined> {
122
- return mockSpawn.mock.calls.at(-1)?.[2]?.env as Record<string, string | undefined> ?? {};
123
- }
124
-
125
- // ── 公共 fixture ──
126
-
127
- function makeRecord() {
128
- return createRecord("test-1", {
129
- agent: "general-purpose",
130
- model: "test/model",
131
- mode: "sync",
132
- task: "test task",
133
- startedAt: Date.now(),
134
- rootSessionId: "s1",
135
- parentRecordId: undefined,
136
- depth: 0,
137
- });
138
- }
139
-
140
- function makeRunOpts(overrides: Partial<RunOptions> = {}): RunOptions {
141
- return {
142
- resolved: { model: { provider: "test", id: "model" }, thinkingLevel: undefined },
143
- agentConfig: undefined,
144
- appendSystemPrompt: undefined,
145
- skillPath: undefined,
146
- schema: undefined,
147
- maxTurns: undefined,
148
- graceTurns: undefined,
149
- signal: undefined,
150
- onEvent: undefined,
151
- ...overrides,
152
- };
153
- }
154
-
155
- function makeCtx(overrides: Partial<SessionRunnerContext> = {}): SessionRunnerContext {
156
- return {
157
- cwd: "/fake/cwd",
158
- agentDir: "/fake/agent",
159
- skillDirs: [],
160
- mainCwd: "/fake/cwd",
161
- sessionRootId: "root-session-test",
162
- rootCwd: "/fake/cwd",
163
- ...overrides,
164
- };
165
- }
166
-
167
- // ── applySchemaEnvToChildEnv 纯函数单测 ──
168
-
169
- describe("applySchemaEnvToChildEnv (T3.9/T3.11/T3.16)", () => {
170
- // T3.11: schemaEnv 不传 → childEnv 无 PI_WORKFLOW_SCHEMA
171
- it("T3.11: schemaEnv 不传时 childEnv 不含 PI_WORKFLOW_SCHEMA(BC-6)", () => {
172
- const childEnv: Record<string, string | undefined> = { PATH: "/usr/bin" };
173
- applySchemaEnvToChildEnv(childEnv, undefined);
174
- expect(childEnv).not.toHaveProperty("PI_WORKFLOW_SCHEMA");
175
- expect(childEnv.PATH).toBe("/usr/bin"); // 其他 key 不受影响
176
- });
177
-
178
- // T3.16: schemaEnv 不传时 BC-6 childEnv 等价——合并前后行为一致
179
- it("T3.16: schemaEnv 不传时 childEnv 等价于合并前(BC-6,不注入 PI_WORKFLOW_SCHEMA)", () => {
180
- const childEnv: Record<string, string | undefined> = {};
181
- applySchemaEnvToChildEnv(childEnv, undefined);
182
- // 不传 schemaEnv 时,childEnv 应与调用前完全一致(不含 PI_WORKFLOW_SCHEMA)
183
- expect(Object.keys(childEnv)).toHaveLength(0);
184
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBeUndefined();
185
- });
186
-
187
- // T3.16 补充: schemaEnv 为空串也不注入(空串不是有效 schema)
188
- it("T3.16 补充: schemaEnv 为空串时不注入(false-ish 语义)", () => {
189
- const childEnv: Record<string, string | undefined> = {};
190
- applySchemaEnvToChildEnv(childEnv, "");
191
- expect(childEnv).not.toHaveProperty("PI_WORKFLOW_SCHEMA");
192
- });
193
-
194
- // T3.9: schemaEnv 传入 → childEnv 含 PI_WORKFLOW_SCHEMA
195
- it("T3.9: schemaEnv 传入时 childEnv 含 PI_WORKFLOW_SCHEMA", () => {
196
- const childEnv: Record<string, string | undefined> = {};
197
- const schemaJson = '{"type":"object","properties":{"x":{"type":"number"}}}';
198
- applySchemaEnvToChildEnv(childEnv, schemaJson);
199
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(schemaJson);
200
- });
201
-
202
- // T3.9 补充: schemaEnv 值为复杂 JSON 字符串时正确透传
203
- it("T3.9 补充: schemaEnv 值为复杂 JSON 字符串时完整透传", () => {
204
- const childEnv: Record<string, string | undefined> = {};
205
- const schemaJson = JSON.stringify({
206
- type: "object",
207
- properties: {
208
- name: { type: "string" },
209
- age: { type: "number", minimum: 0 },
210
- },
211
- required: ["name"],
212
- });
213
- applySchemaEnvToChildEnv(childEnv, schemaJson);
214
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(schemaJson);
215
- // 验证可以 parse 回原始结构
216
- expect(() => JSON.parse(childEnv.PI_WORKFLOW_SCHEMA!)).not.toThrow();
217
- });
218
-
219
- // T3.9 补充: schemaEnv 与已有 key 不冲突
220
- it("T3.9 补充: schemaEnv 注入不覆盖 childEnv 已有 key", () => {
221
- const childEnv: Record<string, string | undefined> = {
222
- PATH: "/usr/bin",
223
- HOME: "/home/user",
224
- };
225
- applySchemaEnvToChildEnv(childEnv, '{"x":1}');
226
- expect(childEnv.PATH).toBe("/usr/bin");
227
- expect(childEnv.HOME).toBe("/home/user");
228
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe('{"x":1}');
229
- });
230
-
231
- // [SO-DATA-4] 边界内通过:255KB(< 256KiB 上限)正常注入,行为不变
232
- it("[SO-DATA-4] 255KB schema 正常注入(上限内不误拒)", () => {
233
- const childEnv: Record<string, string | undefined> = {};
234
- // ASCII 串 byteLength === length,构造 255KB 纯 JSON body(外层再包一层合法 JSON)
235
- const padding = "x".repeat(255 * 1024);
236
- const schemaJson = JSON.stringify({ type: "object", properties: { pad: { const: padding } } });
237
- expect(schemaEnvByteLength(schemaJson)).toBeLessThanOrEqual(SCHEMA_ENV_MAX_BYTES);
238
- applySchemaEnvToChildEnv(childEnv, schemaJson);
239
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(schemaJson);
240
- });
241
-
242
- // [SO-DATA-4] 超限 fail-fast:257KB 拒绝,错误消息含实际大小 + 精简/拆分指引 + E2BIG 归因
243
- it("[SO-DATA-4] 257KB schema fail-fast 拒绝(错误含实际大小与恢复指引)", () => {
244
- const childEnv: Record<string, string | undefined> = {};
245
- const padding = "x".repeat(257 * 1024);
246
- const schemaJson = JSON.stringify({ type: "object", properties: { pad: { const: padding } } });
247
- expect(schemaEnvByteLength(schemaJson)).toBeGreaterThan(SCHEMA_ENV_MAX_BYTES);
248
- expect(() => applySchemaEnvToChildEnv(childEnv, schemaJson)).toThrow(
249
- expect.objectContaining({
250
- message: expect.stringContaining("bytes exceeds"),
251
- }),
252
- );
253
- // 错误消息含实际大小、上限、精简/拆分指引与 E2BIG 归因(可操作性验收)
254
- try {
255
- applySchemaEnvToChildEnv(childEnv, schemaJson);
256
- throw new Error("expected applySchemaEnvToChildEnv to throw");
257
- } catch (err) {
258
- const msg = err instanceof Error ? err.message : String(err);
259
- expect(msg).toContain(String(schemaEnvByteLength(schemaJson))); // 实际大小
260
- expect(msg).toContain(String(SCHEMA_ENV_MAX_BYTES)); // 上限值
261
- expect(msg).toContain("simplify the schema"); // 精简指引
262
- expect(msg).toContain("E2BIG"); // ARG_MAX/E2BIG 约束说明
263
- }
264
- // fail-fast:不写入半截值
265
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBeUndefined();
266
- });
267
- });
268
-
269
- // ── runSpawn 集成测试:schemaEnv 经 RunOptions → childEnv ──
270
-
271
- describe("runSpawn schemaEnv childEnv 注入 (T3.9/T3.11)", () => {
272
- beforeEach(() => {
273
- vi.clearAllMocks();
274
- mockExistsSync.mockReturnValue(false);
275
- });
276
-
277
- afterEach(() => {
278
- vi.restoreAllMocks();
279
- });
280
-
281
- // T3.11: schemaEnv 不传 → childEnv 的 PI_WORKFLOW_SCHEMA 保持 process.env 原值
282
- // BC-6: applySchemaEnvToChildEnv 不注入新值,但 process.env 可能已有此 key(子进程继承父环境)。
283
- // 验证点: 不传 schemaEnv 时我们的代码不修改 PI_WORKFLOW_SCHEMA。
284
- it("T3.11 (integration): RunOptions 无 schemaEnv → childEnv 继承 process.env 原值(BC-6)", async () => {
285
- const record = makeRecord();
286
- const opts = makeRunOpts({ schemaEnv: undefined });
287
- const ctx = makeCtx();
288
-
289
- const resultPromise = runSpawn(record, "test task", opts, ctx);
290
- await waitForSpawn(mockSpawn);
291
- const childEnv = getLastSpawnEnv();
292
- // BC-6: schemaEnv 未传入 → PI_WORKFLOW_SCHEMA 应为 process.env 原值(我们的代码不注入)
293
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(process.env.PI_WORKFLOW_SCHEMA);
294
-
295
- const child = getLastSpawnedChild();
296
- child.emit("close", 0);
297
- await resultPromise;
298
- });
299
-
300
- // T3.9: schemaEnv 传入 → childEnv 含 PI_WORKFLOW_SCHEMA
301
- it("T3.9 (integration): RunOptions 有 schemaEnv → childEnv 含 PI_WORKFLOW_SCHEMA", async () => {
302
- const record = makeRecord();
303
- const schemaJson = '{"type":"object","properties":{"result":{"type":"string"}}}';
304
- const opts = makeRunOpts({ schemaEnv: schemaJson });
305
- const ctx = makeCtx();
306
-
307
- const resultPromise = runSpawn(record, "test task", opts, ctx);
308
- await waitForSpawn(mockSpawn);
309
- const childEnv = getLastSpawnEnv();
310
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(schemaJson);
311
-
312
- // 关闭子进程
313
- const child = getLastSpawnedChild();
314
- child.emit("close", 0);
315
- await resultPromise;
316
- });
317
-
318
- // T3.9 + fork: schemaEnv 与 fork env 共存时不冲突
319
- it("T3.9 + fork: schemaEnv 与 fork depth env 共存不冲突", async () => {
320
- const record = createRecord("test-fork-1", {
321
- agent: "general-purpose",
322
- model: "test/model",
323
- mode: "sync",
324
- task: "test task",
325
- startedAt: Date.now(),
326
- rootSessionId: "s1",
327
- parentRecordId: undefined,
328
- depth: 1,
329
- });
330
- const schemaJson = '{"type":"object"}';
331
- const opts = makeRunOpts({
332
- schemaEnv: schemaJson,
333
- fork: true,
334
- parentForkDepth: 0,
335
- });
336
- const ctx = makeCtx();
337
-
338
- const resultPromise = runSpawn(record, "test task", opts, ctx);
339
- await waitForSpawn(mockSpawn);
340
- const childEnv = getLastSpawnEnv();
341
- // fork depth env 应存在
342
- expect(childEnv.PI_SUBAGENT_FORK_DEPTH).toBe("1");
343
- // schemaEnv 也应存在
344
- expect(childEnv.PI_WORKFLOW_SCHEMA).toBe(schemaJson);
345
-
346
- const child = getLastSpawnedChild();
347
- child.emit("close", 0);
348
- await resultPromise;
349
- });
350
- });
@@ -1,233 +0,0 @@
1
- // src/__tests__/session-start-reaper.test.ts
2
- //
3
- // 验证 session_start 的两个新行为:
4
- // 1. WTM.scan reaper 被调用(best-effort)
5
- // 2. mainSessionFile 被缓存
6
- // 3. scan 抛错不阻断启动
7
-
8
- import { beforeEach,describe, expect, it, vi } from "vitest";
9
-
10
- // ── mock modules(在 import 前声明)──
11
-
12
- vi.mock("@earendil-works/pi-coding-agent", () => ({
13
- getAgentDir: () => "/home/user/.pi/agent",
14
- }));
15
- vi.mock("@earendil-works/pi-coding-agent", () => ({
16
- getAgentDir: () => "/home/user/.pi/agent",
17
- }));
18
- vi.mock("@earendil-works/pi-ai", () => ({
19
- StringEnum: (values: string[]) => ({ type: "string", enum: values }),
20
- }));
21
- vi.mock("@earendil-works/pi-ai", () => ({
22
- StringEnum: (values: string[]) => ({ type: "string", enum: values }),
23
- }));
24
- vi.mock("typebox", () => ({
25
- Type: {
26
- Object: (props: Record<string, unknown>) => ({ type: "object", properties: props }),
27
- Optional: (schema: unknown) => ({ ...schema as object, optional: true }),
28
- String: () => ({ type: "string" }),
29
- Boolean: () => ({ type: "boolean" }),
30
- Number: () => ({ type: "number" }),
31
- Array: (items: unknown) => ({ type: "array", items }),
32
- Record: (key: unknown, value: unknown) => ({ type: "object", additionalProperties: value, key }),
33
- Unknown: () => ({ type: "unknown" }),
34
- Union: (members: unknown[]) => ({ type: "union", members }),
35
- Literal: (value: unknown) => ({ type: "literal", value }),
36
- },
37
- }));
38
-
39
- // hoisted mock 实例
40
- const { mockScan, mockCleanup } = vi.hoisted(() => ({
41
- mockScan: vi.fn(),
42
- mockCleanup: vi.fn(),
43
- }));
44
-
45
- vi.mock("../worktree-manager.ts", () => ({
46
- WorktreeManager: class {
47
- constructor(_agentDir: string) { /* mock */ }
48
- scan = mockScan;
49
- cleanup = mockCleanup;
50
- create = vi.fn();
51
- collectPatch = vi.fn();
52
- registerPid = vi.fn();
53
- },
54
- }));
55
-
56
- vi.mock("../session-file-gc.ts", () => ({
57
- maybeCleanupExpiredSessionFiles: vi.fn(),
58
- }));
59
-
60
- // mock subagent-service:避免真正构造 SubagentService(它依赖 ModelConfigService 等)
61
- const { mockInitModel, mockInitSession, mockSetUiRequestHandler, mockSetModelConfigService, mockSetSubagentService, capturedConstructorArg } =
62
- vi.hoisted(() => ({
63
- mockInitModel: vi.fn(),
64
- mockInitSession: vi.fn(),
65
- // W3: index.ts session_start 注入 UI handler 时调用
66
- mockSetUiRequestHandler: vi.fn(),
67
- mockSetModelConfigService: vi.fn(),
68
- mockSetSubagentService: vi.fn(),
69
- capturedConstructorArg: { current: undefined as unknown },
70
- }));
71
-
72
- vi.mock("../model-config-service.ts", () => ({
73
- ModelConfigService: class {
74
- initModel = mockInitModel;
75
- },
76
- getModelConfigService: () => null,
77
- setModelConfigService: mockSetModelConfigService,
78
- }));
79
-
80
- vi.mock("../subagent-service.ts", () => ({
81
- SubagentService: class {
82
- initSession = mockInitSession;
83
- // W3: index.ts session_start 注入 UI handler 时调用
84
- setUiRequestHandler = mockSetUiRequestHandler;
85
- startGcTimer = vi.fn();
86
- recoverManifestTmpFiles = vi.fn(async () => ({ deleted: 0, recovered: 0 }));
87
- constructor(init: unknown) {
88
- capturedConstructorArg.current = init;
89
- }
90
- },
91
- getSubagentService: () => null,
92
- setSubagentService: mockSetSubagentService,
93
- }));
94
-
95
- // mock commands/tools(避免触发真实注册)
96
- vi.mock("../commands/subagents.ts", () => ({
97
- registerSubagentsCommand: vi.fn(),
98
- }));
99
- vi.mock("../tools/subagent-tool.ts", () => ({
100
- registerSubagentTool: vi.fn(),
101
- }));
102
- vi.mock("../tui/bg-notify-render.ts", () => ({
103
- renderBgNotifyMessage: vi.fn(),
104
- }));
105
-
106
- // ── import 被测工厂 ──
107
- import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
108
-
109
- import subagentsExtension from "../../index.ts";
110
-
111
- // ── helpers ──
112
-
113
- /** 创建最小 mock ExtensionAPI,捕获 session_start handler。 */
114
- function createMockPi(overrides: Record<string, unknown> = {}): {
115
- pi: ExtensionAPI;
116
- getSessionStartHandler: () => ((event: unknown, ctx: unknown) => void) | undefined;
117
- } {
118
- let sessionStartHandler: ((event: unknown, ctx: unknown) => void) | undefined;
119
- const noop = (): void => { /* mock */ };
120
- const pi = new Proxy<ExtensionAPI>(overrides as ExtensionAPI, {
121
- get(target, prop: string | symbol): unknown {
122
- if (prop === "on") {
123
- return (event: string, handler: (...args: unknown[]) => unknown) => {
124
- if (event === "session_start") {
125
- sessionStartHandler = handler as (event: unknown, ctx: unknown) => void;
126
- }
127
- };
128
- }
129
- if (prop in target) return target[prop as keyof ExtensionAPI];
130
- return noop;
131
- },
132
- });
133
- return {
134
- pi,
135
- getSessionStartHandler: () => sessionStartHandler,
136
- };
137
- }
138
-
139
- /** 最小 ExtensionContext mock。 */
140
- function createMockCtx(overrides: Record<string, unknown> = {}): Record<string, unknown> {
141
- return {
142
- cwd: "/home/user/project",
143
- // [Wave1 #21] mode 必填(与 SDK ExtensionContext 契约一致);默认 tui。
144
- mode: "tui",
145
- modelRegistry: { getAvailable: () => [], find: () => undefined, hasConfiguredAuth: () => false },
146
- model: undefined,
147
- sessionManager: {
148
- getSessionId: () => "session-123",
149
- getSessionFile: () => "/home/user/.pi/agent/sessions/session-123.jsonl",
150
- getSessionDir: () => "/home/user/.pi/agent/sessions",
151
- getCwd: () => "/home/user/project",
152
- getEntries: () => [],
153
- getBranch: () => [],
154
- getLeafId: () => null,
155
- getLeafEntry: () => undefined,
156
- getEntry: () => undefined,
157
- getHeader: () => null,
158
- getTree: () => [],
159
- getSessionName: () => undefined,
160
- },
161
- ...overrides,
162
- };
163
- }
164
-
165
- // ── tests ──
166
-
167
- describe("session_start worktree reaper", () => {
168
- beforeEach(() => {
169
- vi.clearAllMocks();
170
- });
171
-
172
- it("session_start 触发 WTM.scan 调用", async () => {
173
- const { pi, getSessionStartHandler } = createMockPi();
174
- subagentsExtension(pi);
175
-
176
- const handler = getSessionStartHandler();
177
- expect(handler).toBeDefined();
178
-
179
- await handler!(
180
- { type: "session_start", reason: "startup" },
181
- createMockCtx(),
182
- );
183
-
184
- expect(mockScan).toHaveBeenCalledTimes(1);
185
- // scan 无参(全局注册表,不依赖 cwd)
186
- expect(mockScan).toHaveBeenCalledWith();
187
- });
188
-
189
- it("scan 抛错不阻断 session_start", async () => {
190
- mockScan.mockImplementation(() => {
191
- throw new Error("git not found");
192
- });
193
-
194
- const { pi, getSessionStartHandler } = createMockPi();
195
- subagentsExtension(pi);
196
-
197
- const handler = getSessionStartHandler();
198
- expect(handler).toBeDefined();
199
-
200
- // 不应抛错
201
- await expect(
202
- handler!(
203
- { type: "session_start", reason: "startup" },
204
- createMockCtx(),
205
- )
206
- ).resolves.toBeUndefined();
207
-
208
- // service 仍然被注册(启动未被阻断)
209
- expect(mockSetSubagentService).toHaveBeenCalled();
210
- });
211
-
212
- it("mainSessionFile 被缓存并传给 SubagentService", async () => {
213
- const { pi, getSessionStartHandler } = createMockPi();
214
- subagentsExtension(pi);
215
-
216
- const handler = getSessionStartHandler();
217
- await handler!(
218
- { type: "session_start", reason: "startup" },
219
- createMockCtx(),
220
- );
221
-
222
- // SubagentService 构造参数含 getMainSessionFile getter
223
- const init = capturedConstructorArg.current as {
224
- getMainSessionFile?: () => string | undefined;
225
- } | undefined;
226
- expect(init).toBeDefined();
227
- expect(init?.getMainSessionFile).toBeDefined();
228
- // 返回 session_start 时缓存的 sessionFile
229
- expect(init?.getMainSessionFile?.()).toBe(
230
- "/home/user/.pi/agent/sessions/session-123.jsonl",
231
- );
232
- });
233
- });