@zhushanwen/pi-subagent-workflow 7.3.4 → 8.0.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 (166) hide show
  1. package/README.md +39 -12
  2. package/agents/analyst.md +61 -0
  3. package/agents/coder.md +70 -0
  4. package/agents/debugger.md +67 -0
  5. package/agents/doc-reviewer.md +3 -3
  6. package/agents/explorer.md +50 -18
  7. package/agents/general-purpose.md +19 -8
  8. package/agents/orchestrator.md +37 -32
  9. package/agents/planner.md +45 -11
  10. package/agents/researcher.md +53 -11
  11. package/agents/reviewer.md +74 -0
  12. package/package.json +1 -1
  13. package/skills/workflow-script-format/SKILL.md +1 -1
  14. package/src/execution/__tests__/__fixtures__/truncline.snapshot.json +1 -0
  15. package/src/execution/__tests__/agent-registry.test.ts +13 -11
  16. package/src/execution/__tests__/ask-user-transit-e2e.test.ts +10 -4
  17. package/src/execution/__tests__/before-agent-start-injection.test.ts +132 -0
  18. package/src/execution/__tests__/bg-notify-render.test.ts +15 -15
  19. package/src/execution/__tests__/chatmode-first-round-closure-service.test.ts +365 -0
  20. package/src/execution/__tests__/chatmode-first-round-closure-spawn.test.ts +190 -0
  21. package/src/execution/__tests__/chatmode-round-notify-real-chain.test.ts +215 -0
  22. package/src/execution/__tests__/conversation-wiring.test.ts +198 -0
  23. package/src/execution/__tests__/crash-recovery.test.ts +8 -2
  24. package/src/execution/__tests__/delivery-methods.test.ts +385 -0
  25. package/src/execution/__tests__/epipe-fallback.test.ts +241 -0
  26. package/src/execution/__tests__/execute-and-await-worktree.test.ts +49 -2
  27. package/src/execution/__tests__/execute-nesting.test.ts +20 -72
  28. package/src/execution/__tests__/execution-record.test.ts +199 -0
  29. package/src/execution/__tests__/finalize-record.test.ts +170 -14
  30. package/src/execution/__tests__/format.test.ts +131 -7
  31. package/src/execution/__tests__/gc-timer.test.ts +184 -0
  32. package/src/execution/__tests__/get-record-for-action-restart.test.ts +254 -0
  33. package/src/execution/__tests__/helpers/spawn-mock.ts +25 -7
  34. package/src/execution/__tests__/index-session-start-identity.test.ts +371 -0
  35. package/src/execution/__tests__/index-session-start.test.ts +257 -5
  36. package/src/execution/__tests__/lifecycle-manager-lock.test.ts +211 -0
  37. package/src/execution/__tests__/lifecycle-manager.test.ts +337 -0
  38. package/src/execution/__tests__/lifecycle-predicates.test.ts +116 -0
  39. package/src/execution/__tests__/list-component.test.ts +59 -5
  40. package/src/execution/__tests__/list-fields.test.ts +109 -0
  41. package/src/execution/__tests__/model-resolver.test.ts +38 -1
  42. package/src/execution/__tests__/nested-visibility-env-propagation.test.ts +287 -0
  43. package/src/execution/__tests__/nested-visibility.test.ts +325 -0
  44. package/src/execution/__tests__/notifier-flush.test.ts +209 -7
  45. package/src/execution/__tests__/one-shot-upgrade.test.ts +205 -0
  46. package/src/execution/__tests__/parent-child-matrix.test.ts +336 -0
  47. package/src/execution/__tests__/record-store.test.ts +158 -52
  48. package/src/execution/__tests__/recursive-visibility-baseline.test.ts +11 -12
  49. package/src/execution/__tests__/recursive-visibility-env.test.ts +18 -20
  50. package/src/execution/__tests__/resource-policy.test.ts +109 -0
  51. package/src/execution/__tests__/run-and-finalize-chatmode.test.ts +267 -0
  52. package/src/execution/__tests__/run-spawn-chatmode-settled.test.ts +253 -0
  53. package/src/execution/__tests__/run-spawn-edges.test.ts +18 -25
  54. package/src/execution/__tests__/run-spawn-integration.test.ts +29 -25
  55. package/src/execution/__tests__/run-spawn-resume.test.ts +322 -0
  56. package/src/execution/__tests__/run-spawn-rpc-mode.test.ts +14 -11
  57. package/src/execution/__tests__/session-pending.test.ts +61 -2
  58. package/src/execution/__tests__/session-reconstructor.test.ts +4 -4
  59. package/src/execution/__tests__/session-runner-epipe.test.ts +178 -0
  60. package/src/execution/__tests__/session-runner-schema-env.test.ts +15 -21
  61. package/src/execution/__tests__/session-start-reaper.test.ts +10 -8
  62. package/src/execution/__tests__/spawn-args.test.ts +127 -49
  63. package/src/execution/__tests__/spawn-worktree-guidance.test.ts +1 -0
  64. package/src/execution/__tests__/spawned-children.test.ts +92 -0
  65. package/src/execution/__tests__/status-refactor.test.ts +345 -0
  66. package/src/execution/__tests__/stdin-writer.test.ts +97 -0
  67. package/src/execution/__tests__/subagent-service-message-close.test.ts +598 -0
  68. package/src/execution/__tests__/subagent-service-parent-guard.test.ts +180 -0
  69. package/src/execution/__tests__/subagent-service.test.ts +49 -11
  70. package/src/execution/__tests__/timeout-integration.test.ts +27 -13
  71. package/src/execution/__tests__/tool-action.test.ts +12 -10
  72. package/src/execution/__tests__/truncline-snapshot.test.ts +81 -0
  73. package/src/execution/__tests__/turn-limiter-semantics.test.ts +194 -0
  74. package/src/execution/__tests__/worktree-manager.test.ts +292 -89
  75. package/src/execution/__tests__/worktree-pid-registration.integration.test.ts +13 -12
  76. package/src/execution/argv-mirror.ts +21 -2
  77. package/src/execution/execution-record.ts +126 -9
  78. package/src/execution/finalize-record.ts +90 -13
  79. package/src/execution/host-mode.ts +1 -1
  80. package/src/execution/lifecycle-manager.ts +484 -0
  81. package/src/execution/lifecycle-predicates.ts +65 -0
  82. package/src/execution/manifest-store.ts +61 -16
  83. package/src/execution/model-resolver.ts +26 -5
  84. package/src/execution/notifier.ts +69 -12
  85. package/src/execution/pi-invocation.ts +21 -1
  86. package/src/execution/record-store.ts +554 -107
  87. package/src/execution/session-pending.ts +116 -45
  88. package/src/execution/session-reconstructor.ts +224 -7
  89. package/src/execution/session-runner.ts +290 -75
  90. package/src/execution/sessions-index.ts +304 -0
  91. package/src/execution/stdin-writer.ts +93 -7
  92. package/src/execution/stream-sink.ts +20 -3
  93. package/src/execution/subagent-service.ts +867 -138
  94. package/src/execution/turn-limiter.ts +14 -0
  95. package/src/execution/types.ts +204 -22
  96. package/src/execution/worktree-manager.ts +128 -49
  97. package/src/execution/worktree-registry.ts +13 -2
  98. package/src/index.ts +277 -19
  99. package/src/injectors/subagent-list-injector.ts +26 -8
  100. package/src/injectors/workflow-list-injector.ts +25 -8
  101. package/src/interface/__tests__/subagent-tool-prompt.test.ts +18 -5
  102. package/src/interface/__tests__/tool-render.test.ts +15 -13
  103. package/src/interface/bg-notify-render.ts +32 -8
  104. package/src/interface/command-actions.ts +26 -7
  105. package/src/interface/commands.ts +21 -22
  106. package/src/interface/format.ts +44 -17
  107. package/src/interface/gui-mappers.ts +6 -8
  108. package/src/interface/helpers.ts +170 -10
  109. package/src/interface/list-component.ts +53 -14
  110. package/src/interface/subagent-actions.ts +235 -17
  111. package/src/interface/subagent-tool.ts +82 -17
  112. package/src/interface/subagents.ts +2 -1
  113. package/src/interface/tool-render.ts +11 -24
  114. package/src/interface/tool-workflow.ts +20 -35
  115. package/src/interface/views/WorkflowsView.ts +89 -32
  116. package/src/interface/views/__tests__/WorkflowsView-signature.test.ts +264 -0
  117. package/src/interface/views/__tests__/detail-content-session-file.test.ts +1 -1
  118. package/src/interface/views/format.ts +3 -3
  119. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +325 -0
  120. package/src/orchestration/__tests__/args-validator.test.ts +1 -1
  121. package/src/orchestration/__tests__/config-loader.test.ts +38 -0
  122. package/src/orchestration/__tests__/error-recovery-handlers.test.ts +394 -4
  123. package/src/orchestration/__tests__/error-recovery-workflow-call.test.ts +4 -4
  124. package/src/orchestration/__tests__/execute-agent-call.test.ts +95 -0
  125. package/src/orchestration/__tests__/jsonl-run-store-session-file.test.ts +657 -18
  126. package/src/orchestration/__tests__/launcher-nested-workflow.test.ts +0 -2
  127. package/src/orchestration/__tests__/lifecycle.test.ts +332 -149
  128. package/src/orchestration/__tests__/skill-discovery.test.ts +157 -0
  129. package/src/orchestration/__tests__/test-mocks.ts +191 -0
  130. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +98 -0
  131. package/src/orchestration/__tests__/workflow-nesting-e2e.test.ts +0 -2
  132. package/src/orchestration/__tests__/workflow-script-lint-memo.test.ts +110 -0
  133. package/src/orchestration/__tests__/workflows-e2e.test.ts +1 -1
  134. package/src/orchestration/agent-opts-resolver.ts +4 -1
  135. package/src/orchestration/args-validator.ts +2 -2
  136. package/src/orchestration/config-loader.ts +30 -1
  137. package/src/orchestration/error-recovery.ts +133 -29
  138. package/src/orchestration/execute-agent-call.ts +31 -7
  139. package/src/orchestration/jsonl-run-store.ts +287 -40
  140. package/src/orchestration/launcher.ts +7 -1
  141. package/src/orchestration/lifecycle.ts +135 -132
  142. package/src/orchestration/models/__tests__/trace.test.ts +408 -0
  143. package/src/orchestration/models/budget.ts +1 -1
  144. package/src/orchestration/models/run-runtime.ts +15 -17
  145. package/src/orchestration/models/run-spec.ts +2 -2
  146. package/src/orchestration/models/run-state.ts +3 -3
  147. package/src/orchestration/models/trace.ts +95 -15
  148. package/src/orchestration/models/types.ts +8 -9
  149. package/src/orchestration/models/workflow-run.ts +50 -71
  150. package/src/orchestration/models/workflow-script.ts +32 -1
  151. package/src/orchestration/skill-discovery.ts +30 -0
  152. package/src/orchestration/worker-handle.ts +1 -1
  153. package/src/orchestration/worker-host.ts +1 -1
  154. package/src/orchestration/worker-script-builder.ts +29 -10
  155. package/src/shared/__tests__/agent-ref.test.ts +34 -0
  156. package/src/shared/__tests__/resource-discovery-manifest-cache.test.ts +280 -0
  157. package/src/shared/__tests__/resource-discovery.test.ts +55 -0
  158. package/src/shared/__tests__/schema-jsonify.test.ts +81 -0
  159. package/src/shared/agent-ref.ts +16 -0
  160. package/src/shared/resource-discovery.ts +147 -58
  161. package/src/shared/schema-jsonify.ts +53 -0
  162. package/workflows/README.md +4 -4
  163. package/agents/code-reviewer.md +0 -47
  164. package/agents/context-builder.md +0 -21
  165. package/agents/oracle.md +0 -34
  166. package/agents/worker.md +0 -20
@@ -190,11 +190,48 @@ describe("resolveModel — thinkingLevel resolution", () => {
190
190
  expect(r.thinkingLevel).toBe("medium");
191
191
  });
192
192
 
193
- it("ctxModel path: thinkingLevel undefined when no override anywhere", () => {
193
+ it("ctxModel path: thinkingLevel undefined when no override and ctxModel non-reasoning", () => {
194
194
  const reg = makeRegistry([]);
195
+ // ctxModel(makeModel 默认 reasoning=false)无 override → 非 reasoning 不支持 thinking
195
196
  const r = resolveModel(undefined, reg, undefined, ctxModel);
196
197
  expect(r.thinkingLevel).toBeUndefined();
197
198
  });
199
+
200
+ // ── 默认 thinking 兜底:无显式指定时用「模型最高可用档」(不落 pi 默认 medium)──
201
+
202
+ it("ctxModel path: no override + reasoning model with map → highest mapped level", () => {
203
+ const reg = makeRegistry([]);
204
+ const m = makeModel({ id: "r-main", provider: "rm", reasoning: true, thinkingLevelMap: { low: 1, high: 2, xhigh: 3 } });
205
+ const r = resolveModel(undefined, reg, undefined, m);
206
+ expect(r.thinkingLevel).toBe("xhigh");
207
+ });
208
+
209
+ it("ctxModel path: no override + reasoning model with map containing max → max", () => {
210
+ const reg = makeRegistry([]);
211
+ const m = makeModel({ id: "r-max", provider: "rm", reasoning: true, thinkingLevelMap: { high: 2, max: 99 } });
212
+ const r = resolveModel(undefined, reg, undefined, m);
213
+ expect(r.thinkingLevel).toBe("max");
214
+ });
215
+
216
+ it("ctxModel path: no override + reasoning model without map → xhigh (pi-legal ceiling)", () => {
217
+ const reg = makeRegistry([]);
218
+ // 无 map 时不能用 "max"——pi 侧 model:"max" 后缀非法会导致 model 解析失败
219
+ const m = makeModel({ id: "r-nomap", provider: "rm", reasoning: true });
220
+ const r = resolveModel(undefined, reg, undefined, m);
221
+ expect(r.thinkingLevel).toBe("xhigh");
222
+ });
223
+
224
+ it("override path: no thinkingLevel param → defaults to model's highest mapped level", () => {
225
+ const m = makeModel({
226
+ id: "default-thinking",
227
+ provider: "dt",
228
+ reasoning: true,
229
+ thinkingLevelMap: { minimal: 1, medium: 2, high: 3 },
230
+ });
231
+ const reg = makeRegistry([m]);
232
+ const r = resolveModel(undefined, reg, { model: "dt/default-thinking" });
233
+ expect(r.thinkingLevel).toBe("high");
234
+ });
198
235
  });
199
236
 
200
237
  // ============================================================
@@ -0,0 +1,287 @@
1
+ // src/execution/__tests__/nested-visibility-env-propagation.test.ts
2
+ //
3
+ // SP-8 嵌套可见性修复验证(TC-3:PI_SUBAGENT_ROOT_SESSION_ID env 贯穿)。
4
+ //
5
+ // 验证 env 身份贯穿机制:
6
+ // TC-3a: 根进程 initSession:无 env → sessionRootId = sessionId(自己是 ROOT)
7
+ // TC-3b: 子进程 initSession:有 env → sessionRootId = env 值(贯穿真 ROOT)
8
+ // TC-3c: 孙进程 initSession:env 链全程贯穿
9
+ // TC-3d: 子进程 execute 创建的 record.rootSessionId = env 贯穿的真 ROOT
10
+ // TC-3e: 子进程 collectRecords 用 sessionRootId 过滤
11
+ //
12
+ // mock 策略与 recursive-visibility-baseline.test.ts 一致。
13
+
14
+ import type { PassThrough } from "node:stream";
15
+
16
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
17
+
18
+ // ── mock modules ──
19
+
20
+ vi.mock("node:child_process", async () => {
21
+ const { EventEmitter } = await import("node:events");
22
+ const { PassThrough } = await import("node:stream");
23
+
24
+ class FakeChild extends EventEmitter {
25
+ pid = 12345;
26
+ stdout = new PassThrough();
27
+ stderr = new PassThrough();
28
+ killed = false;
29
+ killSignal: string | undefined;
30
+ kill(sig?: string): boolean {
31
+ this.killed = true;
32
+ this.killSignal = sig;
33
+ return true;
34
+ }
35
+ }
36
+
37
+ return {
38
+ spawn: vi.fn(() => new FakeChild()),
39
+ // buildEnvBlock 用 execFile 异步取 git branch:默认 err-first 兜底(catch → branch="")
40
+ execFile: vi.fn(
41
+ (
42
+ _cmd: string,
43
+ _args: readonly string[],
44
+ _opts: unknown,
45
+ cb: (err: Error | null, stdout?: string, stderr?: string) => void,
46
+ ) => cb(new Error("execFile not configured in this test")),
47
+ ),
48
+ };
49
+ });
50
+
51
+ vi.mock("node:fs", async () => {
52
+ const actual = await import("node:fs");
53
+ return {
54
+ default: {
55
+ ...actual,
56
+ mkdirSync: vi.fn(),
57
+ existsSync: vi.fn(() => false),
58
+ appendFileSync: vi.fn(),
59
+ writeFileSync: vi.fn(),
60
+ readdirSync: vi.fn(() => []),
61
+ },
62
+ mkdirSync: vi.fn(),
63
+ existsSync: vi.fn(() => false),
64
+ appendFileSync: vi.fn(),
65
+ writeFileSync: vi.fn(),
66
+ readdirSync: vi.fn(() => []),
67
+ promises: actual.promises,
68
+ };
69
+ });
70
+
71
+ vi.mock("../alive-store.ts", async (importOriginal) => {
72
+ const actual = await importOriginal<typeof import("../alive-store.ts")>();
73
+ return {
74
+ ...actual,
75
+ writeAliveMarker: vi.fn(),
76
+ removeAliveMarker: vi.fn(),
77
+ };
78
+ });
79
+
80
+ vi.mock("../finalized-marker.ts", () => ({
81
+ writeFinalized: vi.fn(),
82
+ readFinalized: vi.fn(() => false),
83
+ }));
84
+
85
+ vi.mock("../manifest-store.ts", () => {
86
+ class FakeManifestStore {
87
+ writeManifest = vi.fn(async () => {});
88
+ readManifest = vi.fn(async () => null);
89
+ listAllSync = vi.fn(() => []);
90
+ recoverTmpFiles = vi.fn(async () => []);
91
+ }
92
+ return { ManifestStore: vi.fn(function (_recordsDir: string) { return new FakeManifestStore(); }) };
93
+ });
94
+
95
+ vi.mock("../temp-prompt.ts", () => ({
96
+ writePromptToTempFile: vi.fn(async (agent: string) => {
97
+ const safeName = agent.replace(/[^\w.-]+/g, "_");
98
+ return { dir: `/tmp/fake-${safeName}`, filePath: `/tmp/fake-${safeName}/prompt-${safeName}.md` };
99
+ }),
100
+ cleanupTempPrompt: vi.fn(async () => {}),
101
+ }));
102
+
103
+ import { spawn } from "node:child_process";
104
+
105
+ import { ModelConfigService } from "../model-config-service.ts";
106
+ import type { ModelInfo, ModelRegistryLike } from "../model-resolver.ts";
107
+ import type { RecordStore } from "../record-store.ts";
108
+ import { createRecord } from "../execution-record.ts";
109
+ import { SubagentService } from "../subagent-service.ts";
110
+
111
+ const mockSpawn = vi.mocked(spawn);
112
+
113
+ // ── env 名常量(与 subagent-service.ts 一致)──
114
+ const ENV_ROOT_SESSION_ID = "PI_SUBAGENT_ROOT_SESSION_ID";
115
+ const ENV_SELF_RECORD_ID = "PI_SUBAGENT_SELF_RECORD_ID";
116
+ const ENV_DEPTH = "PI_SUBAGENT_DEPTH";
117
+ const ENV_FORK_DEPTH = "PI_SUBAGENT_FORK_DEPTH";
118
+ const ENV_ROOT_CWD = "PI_SUBAGENT_ROOT_CWD";
119
+
120
+ // ── helpers ──
121
+
122
+ interface FakeChild {
123
+ pid: number;
124
+ stdout: PassThrough;
125
+ stderr: PassThrough;
126
+ killed: boolean;
127
+ killSignal: string | undefined;
128
+ kill(sig?: string): boolean;
129
+ emit(event: string, ...args: unknown[]): boolean;
130
+ }
131
+
132
+ function lastSpawnedChild(): FakeChild {
133
+ const result = mockSpawn.mock.results.at(-1);
134
+ if (!result) throw new Error("spawn was not called yet");
135
+ return result.value as FakeChild;
136
+ }
137
+
138
+ function getLastSpawnEnv(): Record<string, string | undefined> {
139
+ return (mockSpawn.mock.calls.at(-1)?.[2]?.env as Record<string, string | undefined>) ?? {};
140
+ }
141
+
142
+ function sessionHeader(id = "env-prop-session"): Record<string, unknown> {
143
+ return { type: "session", id, timestamp: "2026-08-11T00-00-00-000Z", cwd: "/tmp/test" };
144
+ }
145
+
146
+ function emitStdoutLine(child: FakeChild, obj: Record<string, unknown>): void {
147
+ child.stdout.write(`${JSON.stringify(obj)}\n`);
148
+ }
149
+
150
+ async function driveChildToCompletion(child: FakeChild, events: Record<string, unknown>[] = []): Promise<void> {
151
+ emitStdoutLine(child, sessionHeader());
152
+ for (const e of events) emitStdoutLine(child, e);
153
+ child.stdout.end();
154
+ child.stderr.end();
155
+ child.emit("close", 0);
156
+ }
157
+
158
+ function makeEmptyRegistry(): ModelRegistryLike {
159
+ return { getAvailable: () => [], find: () => undefined, hasConfiguredAuth: () => true };
160
+ }
161
+
162
+ function makePi() {
163
+ return { sendMessage: vi.fn(), appendEntry: vi.fn(), events: { emit: vi.fn() } };
164
+ }
165
+
166
+ const ctxModel: ModelInfo = { id: "m", name: "M", provider: "p", reasoning: false };
167
+
168
+ function setupService(sessionId: string, env?: Record<string, string>): SubagentService {
169
+ const agentDir = "/tmp/env-prop-it";
170
+ const modelService = new ModelConfigService({ agentDir });
171
+ modelService.initModel({
172
+ modelRegistry: makeEmptyRegistry(),
173
+ sessionId,
174
+ ctxModel,
175
+ });
176
+ const service = new SubagentService({
177
+ cwd: agentDir,
178
+ modelService,
179
+ getMainSessionFile: () => "/mock/main-session.jsonl",
180
+ });
181
+ service.initSession({ pi: makePi(), sessionId });
182
+ return service;
183
+ }
184
+
185
+ // ── 用例 ──
186
+
187
+ describe("TC-3: PI_SUBAGENT_ROOT_SESSION_ID env 贯穿正确", () => {
188
+ beforeEach(() => {
189
+ vi.clearAllMocks();
190
+ for (const k of [ENV_ROOT_SESSION_ID, ENV_SELF_RECORD_ID, ENV_DEPTH, ENV_FORK_DEPTH, ENV_ROOT_CWD]) {
191
+ delete process.env[k];
192
+ }
193
+ });
194
+
195
+ afterEach(() => {
196
+ for (const k of [ENV_ROOT_SESSION_ID, ENV_SELF_RECORD_ID, ENV_DEPTH, ENV_FORK_DEPTH, ENV_ROOT_CWD]) {
197
+ delete process.env[k];
198
+ }
199
+ vi.restoreAllMocks();
200
+ });
201
+
202
+ it("TC-3a: 根进程 initSession:无 env → sessionRootId = sessionId(自己是 ROOT)", () => {
203
+ const service = setupService("root-it");
204
+ const sessionRootId = Reflect.get(service, "sessionRootId") as string;
205
+ expect(sessionRootId).toBe("root-it");
206
+ });
207
+
208
+ it("TC-3b: 子进程 initSession:有 env PI_SUBAGENT_ROOT_SESSION_ID → sessionRootId = env 值(贯穿真 ROOT)", () => {
209
+ process.env[ENV_ROOT_SESSION_ID] = "real-root-session";
210
+ const service = setupService("child-session");
211
+
212
+ const sessionRootId = Reflect.get(service, "sessionRootId") as string;
213
+ expect(sessionRootId).toBe("real-root-session"); // env 贯穿,不是 child-session
214
+ });
215
+
216
+ it("TC-3c: 孙进程 initSession:env 链全程贯穿,B 的 sessionRootId 仍为 ROOT", () => {
217
+ // 模拟 B 子进程收到的 env(A 在 spawn B 时注入)
218
+ process.env[ENV_ROOT_SESSION_ID] = "root-main-session";
219
+ process.env[ENV_SELF_RECORD_ID] = "sa-a-record";
220
+ process.env[ENV_DEPTH] = "1";
221
+
222
+ const service = setupService("b-session");
223
+
224
+ // B 的 sessionRootId 仍为 ROOT(env 贯穿,不被 B 自己的 sessionId 覆盖)
225
+ const sessionRootId = Reflect.get(service, "sessionRootId") as string;
226
+ expect(sessionRootId).toBe("root-main-session");
227
+
228
+ // B 的 execCtxBaseline 记录了 B 自己的身份(recordId + depth)
229
+ const baseline = Reflect.get(service, "execCtxBaseline") as { recordId: string; depth: number };
230
+ expect(baseline.recordId).toBe("sa-a-record");
231
+ expect(baseline.depth).toBe(1);
232
+ });
233
+
234
+ it("TC-3d: 子进程 execute 创建的 record.rootSessionId = env 贯穿的真 ROOT", async () => {
235
+ process.env[ENV_ROOT_SESSION_ID] = "root-main-session";
236
+ process.env[ENV_SELF_RECORD_ID] = "sa-parent";
237
+ process.env[ENV_DEPTH] = "1";
238
+
239
+ const service = setupService("child-session");
240
+
241
+ // execute 创建 record(模拟 B 内创建 C)
242
+ const handle = await service.execute({
243
+ task: "nested task",
244
+ ctxModel,
245
+ });
246
+ const store = Reflect.get(service, "store") as RecordStore;
247
+
248
+ // collectRecords 用 sessionRootId 过滤(子进程应看到 ROOT 整棵树)
249
+ const records = store.collectRecords(100, "all", "root-main-session");
250
+ const found = records.find((r) => r.id === handle.subagentId);
251
+
252
+ expect(found).toBeDefined();
253
+ expect(found!.rootSessionId).toBe("root-main-session"); // 指向 ROOT,不是 child-session
254
+ expect(found!.parentRecordId).toBe("sa-parent"); // 父链来自基线
255
+ expect(found!.depth).toBe(2); // 基线 depth=1 + 1 = 2
256
+ });
257
+
258
+ it("TC-3e: 子进程 collectRecords 用 sessionRootId 过滤(能看到 ROOT 整棵树)", () => {
259
+ process.env[ENV_ROOT_SESSION_ID] = "root-main-session";
260
+ process.env[ENV_SELF_RECORD_ID] = "sa-parent";
261
+ process.env[ENV_DEPTH] = "1";
262
+
263
+ const service = setupService("child-session");
264
+
265
+ // 手动注入一条 running record(避免需要真正 spawn)
266
+ const store = Reflect.get(service, "store") as RecordStore;
267
+ const recordC = createRecord("sa-c", {
268
+ agent: "worker",
269
+ model: "test/model",
270
+ mode: "background",
271
+ task: "deep nested",
272
+ startedAt: Date.now(),
273
+ rootSessionId: "root-main-session",
274
+ parentRecordId: "sa-parent",
275
+ depth: 2,
276
+ });
277
+ store.register(recordC);
278
+
279
+ // collectRecords(不带 filter)用 this.sessionRootId 过滤
280
+ const viaService = service.collectRecords(100);
281
+ const ids = viaService.map((r) => r.id);
282
+
283
+ // C 的 record 归 ROOT(rootSessionId=root-main-session),能被子进程查到
284
+ expect(ids).toContain("sa-c");
285
+ expect(viaService.find((r) => r.id === "sa-c")!.rootSessionId).toBe("root-main-session");
286
+ });
287
+ });
@@ -0,0 +1,325 @@
1
+ // src/execution/__tests__/nested-visibility.test.ts
2
+ //
3
+ // SP-8 嵌套可见性修复验证(TC-1 + TC-2:RecordStore 磁盘重建)。
4
+ //
5
+ // 验证全树可见性深度 ≥2 的 subagent 在 /subagents list 中可见:
6
+ // TC-1: 嵌套 A→B→C,B 和 C 的 rootSessionId 都指向 ROOT
7
+ // TC-2: collectRecords 返回全树(含深层)
8
+ //
9
+ // 使用真实文件系统(tmpdir + 真实 .jsonl)测试 RecordStore.collectRecords 磁盘重建。
10
+ // TC-3(env 贯穿)见 nested-visibility-env-propagation.test.ts。
11
+
12
+ import * as fs from "node:fs";
13
+ import * as os from "node:os";
14
+ import * as path from "node:path";
15
+
16
+ import { afterEach, beforeEach, describe, expect, it } from "vitest";
17
+
18
+ import { createRecord } from "../execution-record.ts";
19
+ import { RecordStore } from "../record-store.ts";
20
+
21
+ // ── helpers ──
22
+
23
+ /**
24
+ * 写一个最小合法的 session.jsonl(含 identity custom entry + 1 个 assistant message)。
25
+ * 用于 collectRecords 磁盘源重建。
26
+ */
27
+ function writeSessionJsonl(
28
+ filePath: string,
29
+ identity: {
30
+ id: string;
31
+ agent: string;
32
+ mode: "background";
33
+ task: string;
34
+ startedAt: number;
35
+ rootSessionId?: string;
36
+ parentRecordId?: string;
37
+ depth?: number;
38
+ lastTs?: number;
39
+ },
40
+ assistantText = "result text",
41
+ ): void {
42
+ const lastTs = identity.lastTs ?? identity.startedAt + 1000;
43
+ const header = JSON.stringify({
44
+ type: "session", version: 3, id: "sess-uuid", timestamp: new Date(identity.startedAt).toISOString(), cwd: "/tmp",
45
+ });
46
+ const identityData: Record<string, unknown> = {
47
+ id: identity.id,
48
+ agent: identity.agent,
49
+ mode: identity.mode,
50
+ task: identity.task,
51
+ startedAt: identity.startedAt,
52
+ };
53
+ if (identity.rootSessionId !== undefined) identityData.rootSessionId = identity.rootSessionId;
54
+ if (identity.parentRecordId !== undefined) identityData.parentRecordId = identity.parentRecordId;
55
+ if (identity.depth !== undefined) identityData.depth = identity.depth;
56
+ const identityEntry = JSON.stringify({
57
+ type: "custom",
58
+ id: `id-${identity.id}`,
59
+ parentId: null,
60
+ timestamp: new Date(identity.startedAt).toISOString(),
61
+ customType: "subagent-identity",
62
+ data: identityData,
63
+ });
64
+ const assistantMsg = JSON.stringify({
65
+ type: "message",
66
+ id: `msg-${identity.id}`,
67
+ parentId: `id-${identity.id}`,
68
+ timestamp: new Date(lastTs).toISOString(),
69
+ message: {
70
+ role: "assistant",
71
+ content: [{ type: "text", text: assistantText }],
72
+ usage: { input: 10, output: 20, cacheRead: 0, cacheWrite: 0, totalTokens: 30, cost: { total: 0 } },
73
+ stopReason: "stop",
74
+ timestamp: lastTs,
75
+ },
76
+ });
77
+ fs.writeFileSync(filePath, `${header}\n${identityEntry}\n${assistantMsg}\n`, "utf-8");
78
+ }
79
+
80
+ // ============================================================
81
+ // TC-1: 嵌套 A→B→C,B 和 C 的 rootSessionId 都指向 ROOT
82
+ // ============================================================
83
+
84
+ describe("TC-1: 嵌套 A→B→C 的 rootSessionId 全部指向 ROOT", () => {
85
+ let rootDir: string;
86
+ let tmpDir: string;
87
+
88
+ beforeEach(() => {
89
+ // [DS3] 两层布局:tmpDir 深一层(对齐生产 <enc> 段结构),sessions-index.json 落
90
+ // dirname(tmpDir)=rootDir 内随 afterEach 清理,不写 os.tmpdir() 本身。
91
+ rootDir = fs.mkdtempSync(path.join(os.tmpdir(), "nested-vis-tc1-"));
92
+ tmpDir = path.join(rootDir, "sessions");
93
+ fs.mkdirSync(tmpDir);
94
+ });
95
+
96
+ afterEach(() => {
97
+ fs.rmSync(rootDir, { recursive: true, force: true, maxRetries: 5, retryDelay: 20 });
98
+ });
99
+
100
+ it("三层嵌套 record 的 rootSessionId 全部指向同一 ROOT(磁盘重建 + collectRecords 过滤)", () => {
101
+ const ROOT_SESSION = "root-session-001";
102
+ const store = new RecordStore(tmpDir);
103
+
104
+ // A: depth=0, parent=undefined(顶层)
105
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
106
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
107
+ rootSessionId: ROOT_SESSION, depth: 0,
108
+ });
109
+ // B: depth=1, parent=A(A 的子)
110
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
111
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
112
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-a", depth: 1,
113
+ });
114
+ // C: depth=2, parent=B(B 的子,最深层)
115
+ writeSessionJsonl(path.join(tmpDir, "c.jsonl"), {
116
+ id: "sa-c", agent: "worker", mode: "background", task: "task C", startedAt: 3000,
117
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-b", depth: 2,
118
+ });
119
+
120
+ const records = store.collectRecords(100, "all", ROOT_SESSION);
121
+ const ids = records.map((r) => r.id);
122
+
123
+ // 全部 3 条 record 可见
124
+ expect(ids).toContain("sa-a");
125
+ expect(ids).toContain("sa-b");
126
+ expect(ids).toContain("sa-c");
127
+ expect(records).toHaveLength(3);
128
+
129
+ // rootSessionId 全指向 ROOT
130
+ for (const r of records) {
131
+ expect(r.rootSessionId).toBe(ROOT_SESSION);
132
+ }
133
+ });
134
+
135
+ it("rootSessionId 不匹配时深层 record 被过滤(隔离防护)", () => {
136
+ const ROOT_SESSION = "root-session-001";
137
+ const OTHER_SESSION = "other-session-999";
138
+ const store = new RecordStore(tmpDir);
139
+
140
+ // A 归 ROOT
141
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
142
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
143
+ rootSessionId: ROOT_SESSION, depth: 0,
144
+ });
145
+ // B 归 OTHER(不同 session 的 subagent)
146
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
147
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
148
+ rootSessionId: OTHER_SESSION, depth: 1,
149
+ });
150
+
151
+ const records = store.collectRecords(100, "all", ROOT_SESSION);
152
+ const ids = records.map((r) => r.id);
153
+
154
+ // 只有 A 可见,B 被过滤
155
+ expect(ids).toContain("sa-a");
156
+ expect(ids).not.toContain("sa-b");
157
+ });
158
+ });
159
+
160
+ // ============================================================
161
+ // TC-2: collectRecords 返回全树(含深层)
162
+ // ============================================================
163
+
164
+ describe("TC-2: collectRecords 返回全树(含深层)", () => {
165
+ let rootDir: string;
166
+ let tmpDir: string;
167
+
168
+ beforeEach(() => {
169
+ // [DS3] 两层布局:同 TC-1(sessions-index.json 落 rootDir 内随清理)。
170
+ rootDir = fs.mkdtempSync(path.join(os.tmpdir(), "nested-vis-tc2-"));
171
+ tmpDir = path.join(rootDir, "sessions");
172
+ fs.mkdirSync(tmpDir);
173
+ });
174
+
175
+ afterEach(() => {
176
+ fs.rmSync(rootDir, { recursive: true, force: true, maxRetries: 5, retryDelay: 20 });
177
+ });
178
+
179
+ it("磁盘重建 + 内存 running 合并:全树 A→B→C 全部可见(深度 ≥2)", () => {
180
+ const ROOT_SESSION = "root-merge-001";
181
+ const store = new RecordStore(tmpDir);
182
+
183
+ // A: 磁盘终态(idle,无 sidecar → SP-2 兜底 idle)
184
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
185
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
186
+ rootSessionId: ROOT_SESSION, depth: 0,
187
+ });
188
+ // B: 磁盘终态(idle,无 sidecar)
189
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
190
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
191
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-a", depth: 1,
192
+ });
193
+ // C: 内存 running(当前正在执行)
194
+ const recordC = createRecord("sa-c", {
195
+ agent: "worker",
196
+ model: "test/model",
197
+ mode: "background",
198
+ task: "task C",
199
+ startedAt: 3000,
200
+ rootSessionId: ROOT_SESSION,
201
+ parentRecordId: "sa-b",
202
+ depth: 2,
203
+ });
204
+ store.register(recordC);
205
+
206
+ const records = store.collectRecords(100, "all", ROOT_SESSION);
207
+ const ids = records.map((r) => r.id);
208
+
209
+ // 全部 3 条可见(磁盘 A/B + 内存 C)
210
+ expect(ids).toContain("sa-a");
211
+ expect(ids).toContain("sa-b");
212
+ expect(ids).toContain("sa-c");
213
+ expect(records.length).toBeGreaterThanOrEqual(3);
214
+ });
215
+
216
+ it("depth 标签正确:A=0, B=1, C=2", () => {
217
+ const ROOT_SESSION = "root-depth-001";
218
+ const store = new RecordStore(tmpDir);
219
+
220
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
221
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
222
+ rootSessionId: ROOT_SESSION, depth: 0,
223
+ });
224
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
225
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
226
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-a", depth: 1,
227
+ });
228
+ writeSessionJsonl(path.join(tmpDir, "c.jsonl"), {
229
+ id: "sa-c", agent: "worker", mode: "background", task: "task C", startedAt: 3000,
230
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-b", depth: 2,
231
+ });
232
+
233
+ const records = store.collectRecords(100, "all", ROOT_SESSION);
234
+ const byId = new Map(records.map((r) => [r.id, r]));
235
+
236
+ expect(byId.get("sa-a")!.depth).toBe(0);
237
+ expect(byId.get("sa-b")!.depth).toBe(1);
238
+ expect(byId.get("sa-c")!.depth).toBe(2);
239
+ });
240
+
241
+ it("parentRecordId 链完整:A→(root), B→A, C→B", () => {
242
+ const ROOT_SESSION = "root-parent-001";
243
+ const store = new RecordStore(tmpDir);
244
+
245
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
246
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
247
+ rootSessionId: ROOT_SESSION, depth: 0,
248
+ });
249
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
250
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
251
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-a", depth: 1,
252
+ });
253
+ writeSessionJsonl(path.join(tmpDir, "c.jsonl"), {
254
+ id: "sa-c", agent: "worker", mode: "background", task: "task C", startedAt: 3000,
255
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-b", depth: 2,
256
+ });
257
+
258
+ const records = store.collectRecords(100, "all", ROOT_SESSION);
259
+ const byId = new Map(records.map((r) => [r.id, r]));
260
+
261
+ expect(byId.get("sa-a")!.parentRecordId).toBeUndefined(); // 顶层无父
262
+ expect(byId.get("sa-b")!.parentRecordId).toBe("sa-a");
263
+ expect(byId.get("sa-c")!.parentRecordId).toBe("sa-b");
264
+ });
265
+
266
+ it("statusFilter='running' 返回全树(v4 B-1:深层磁盘 idle 重建为 running)", () => {
267
+ const ROOT_SESSION = "root-filter-001";
268
+ const store = new RecordStore(tmpDir);
269
+
270
+ // 3 条磁盘 idle record(SP-2: 无 sidecar → 旧 idle,v4 B-1 重建为 running)
271
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
272
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
273
+ rootSessionId: ROOT_SESSION, depth: 0,
274
+ });
275
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
276
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
277
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-a", depth: 1,
278
+ });
279
+ writeSessionJsonl(path.join(tmpDir, "c.jsonl"), {
280
+ id: "sa-c", agent: "worker", mode: "background", task: "task C", startedAt: 3000,
281
+ rootSessionId: ROOT_SESSION, parentRecordId: "sa-b", depth: 2,
282
+ });
283
+
284
+ // C 是内存 running
285
+ const recordC = createRecord("sa-c", {
286
+ agent: "worker",
287
+ model: "test/model",
288
+ mode: "background",
289
+ task: "task C",
290
+ startedAt: 3000,
291
+ rootSessionId: ROOT_SESSION,
292
+ parentRecordId: "sa-b",
293
+ depth: 2,
294
+ });
295
+ store.register(recordC);
296
+
297
+ const runningRecords = store.collectRecords(100, "running", ROOT_SESSION);
298
+ const ids = runningRecords.map((r) => r.id);
299
+
300
+ // v4 B-1:磁盘 A/B 旧 idle 现重建为 running(idle 折入 running),与内存 C 一起返回
301
+ expect(ids).toHaveLength(3);
302
+ expect(ids).toContain("sa-a");
303
+ expect(ids).toContain("sa-b");
304
+ expect(ids).toContain("sa-c");
305
+ });
306
+
307
+ it("无 rootSessionFilter 时不过滤(向后兼容)", () => {
308
+ const store = new RecordStore(tmpDir);
309
+
310
+ writeSessionJsonl(path.join(tmpDir, "a.jsonl"), {
311
+ id: "sa-a", agent: "worker", mode: "background", task: "task A", startedAt: 1000,
312
+ rootSessionId: "root-1", depth: 0,
313
+ });
314
+ writeSessionJsonl(path.join(tmpDir, "b.jsonl"), {
315
+ id: "sa-b", agent: "worker", mode: "background", task: "task B", startedAt: 2000,
316
+ rootSessionId: "root-2", depth: 0,
317
+ });
318
+
319
+ // 无 filter → 全部返回
320
+ const records = store.collectRecords(100, "all", undefined);
321
+ const ids = records.map((r) => r.id);
322
+ expect(ids).toContain("sa-a");
323
+ expect(ids).toContain("sa-b");
324
+ });
325
+ });