@vitos-pizza/vitos-pizza 0.1.0 → 0.2.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.
- package/AGENTS.md +158 -158
- package/CHANGELOG.md +42 -24
- package/LICENSE +21 -21
- package/README.md +237 -234
- package/assets/logo.svg +28 -28
- package/node_modules/@vitos-pizza/agent-mode/extensions/index.ts +138 -117
- package/node_modules/@vitos-pizza/agent-mode/package.json +3 -3
- package/node_modules/@vitos-pizza/agent-mode/src/apply-mode.ts +60 -60
- package/node_modules/@vitos-pizza/agent-mode/src/mode-events.ts +102 -102
- package/node_modules/@vitos-pizza/agent-mode/src/modes.ts +43 -43
- package/node_modules/@vitos-pizza/agent-mode/src/plan-instructions.ts +85 -44
- package/node_modules/@vitos-pizza/agent-mode/tests/apply-mode.test.ts +36 -36
- package/node_modules/@vitos-pizza/agent-mode/tests/mode-events.test.ts +22 -22
- package/node_modules/@vitos-pizza/agent-mode/tests/modes.test.ts +38 -38
- package/node_modules/@vitos-pizza/agent-mode/tests/plan-instructions.test.ts +68 -45
- package/node_modules/@vitos-pizza/agent-mode/tsconfig.json +13 -13
- package/node_modules/@vitos-pizza/agent-mode/vitest.config.ts +8 -8
- package/node_modules/@vitos-pizza/keybindings/extensions/index.ts +104 -104
- package/node_modules/@vitos-pizza/keybindings/package.json +1 -1
- package/node_modules/@vitos-pizza/keybindings/presets/shortcuts.json +3 -3
- package/node_modules/@vitos-pizza/keybindings/src/bind.ts +57 -57
- package/node_modules/@vitos-pizza/keybindings/src/config.ts +84 -84
- package/node_modules/@vitos-pizza/keybindings/src/registry.ts +43 -43
- package/node_modules/@vitos-pizza/keybindings/src/types.ts +54 -54
- package/node_modules/@vitos-pizza/keybindings/tests/config.test.ts +83 -83
- package/node_modules/@vitos-pizza/keybindings/tests/registry.test.ts +142 -142
- package/node_modules/@vitos-pizza/keybindings/tsconfig.json +13 -13
- package/node_modules/@vitos-pizza/keybindings/vitest.config.ts +8 -8
- package/node_modules/@vitos-pizza/permission-system/extensions/index.ts +162 -162
- package/node_modules/@vitos-pizza/permission-system/package.json +2 -2
- package/node_modules/@vitos-pizza/permission-system/src/bash/matcher.ts +54 -51
- package/node_modules/@vitos-pizza/permission-system/src/bash/parser.ts +40 -40
- package/node_modules/@vitos-pizza/permission-system/src/config-paths.ts +23 -23
- package/node_modules/@vitos-pizza/permission-system/src/detect-preset-mode.ts +61 -61
- package/node_modules/@vitos-pizza/permission-system/src/expand-home.ts +12 -12
- package/node_modules/@vitos-pizza/permission-system/src/forwarding/forwarder.ts +224 -224
- package/node_modules/@vitos-pizza/permission-system/src/forwarding/subagent-registry.ts +61 -61
- package/node_modules/@vitos-pizza/permission-system/src/gates/path-utils.ts +66 -66
- package/node_modules/@vitos-pizza/permission-system/src/gates/pipeline.ts +151 -155
- package/node_modules/@vitos-pizza/permission-system/src/handlers/before-agent-start.ts +86 -65
- package/node_modules/@vitos-pizza/permission-system/src/handlers/skill-input.ts +1 -1
- package/node_modules/@vitos-pizza/permission-system/src/handlers/tool-call-gate.ts +204 -199
- package/node_modules/@vitos-pizza/permission-system/src/mode-api.ts +44 -44
- package/node_modules/@vitos-pizza/permission-system/src/permission-evaluator.ts +233 -216
- package/node_modules/@vitos-pizza/permission-system/src/permission-session.ts +62 -62
- package/node_modules/@vitos-pizza/permission-system/src/policy-loader.ts +211 -211
- package/node_modules/@vitos-pizza/permission-system/src/sanitizers/skill-prompt.ts +99 -99
- package/node_modules/@vitos-pizza/permission-system/src/sanitizers/system-prompt.ts +221 -221
- package/node_modules/@vitos-pizza/permission-system/src/service.ts +66 -66
- package/node_modules/@vitos-pizza/permission-system/src/session-approvals.ts +4 -1
- package/node_modules/@vitos-pizza/permission-system/src/types.ts +80 -80
- package/node_modules/@vitos-pizza/permission-system/src/ui/approval-summary.ts +43 -43
- package/node_modules/@vitos-pizza/permission-system/src/ui/config-modal.ts +36 -36
- package/node_modules/@vitos-pizza/permission-system/src/ui/permission-dialog.ts +52 -52
- package/node_modules/@vitos-pizza/permission-system/src/wildcard-matcher.ts +81 -81
- package/node_modules/@vitos-pizza/permission-system/src/yaml-frontmatter.ts +84 -84
- package/node_modules/@vitos-pizza/permission-system/src/yolo/runtime-api.ts +49 -49
- package/node_modules/@vitos-pizza/permission-system/tests/bash-parser.test.ts +32 -32
- package/node_modules/@vitos-pizza/permission-system/tests/before-agent-start.test.ts +61 -0
- package/node_modules/@vitos-pizza/permission-system/tests/detect-preset-mode.test.ts +68 -68
- package/node_modules/@vitos-pizza/permission-system/tests/forwarder.test.ts +55 -55
- package/node_modules/@vitos-pizza/permission-system/tests/gates.test.ts +170 -61
- package/node_modules/@vitos-pizza/permission-system/tests/path-utils.test.ts +68 -68
- package/node_modules/@vitos-pizza/permission-system/tests/permission-evaluator.test.ts +82 -55
- package/node_modules/@vitos-pizza/permission-system/tests/policy-loader.test.ts +25 -25
- package/node_modules/@vitos-pizza/permission-system/tests/system-prompt-sanitizer.test.ts +22 -22
- package/node_modules/@vitos-pizza/permission-system/tests/wildcard-matcher.test.ts +27 -27
- package/node_modules/@vitos-pizza/permission-system/tsconfig.json +13 -13
- package/node_modules/@vitos-pizza/permission-system/vitest.config.ts +8 -8
- package/node_modules/@vitos-pizza/question/package.json +1 -1
- package/node_modules/@vitos-pizza/question/src/question-ui.ts +5 -1
- package/node_modules/@vitos-pizza/question/src/register-tool.ts +1 -4
- package/node_modules/@vitos-pizza/question/tests/question-ui-width-cache.test.ts +45 -0
- package/node_modules/@vitos-pizza/question/tests/register-tool-render.test.ts +48 -45
- package/node_modules/@vitos-pizza/session-title/extensions/index.ts +137 -137
- package/node_modules/@vitos-pizza/session-title/package.json +3 -3
- package/node_modules/@vitos-pizza/session-title/src/build-title-prompt.ts +16 -16
- package/node_modules/@vitos-pizza/session-title/src/extract-title-context.ts +55 -55
- package/node_modules/@vitos-pizza/session-title/src/fast-rules.ts +48 -48
- package/node_modules/@vitos-pizza/session-title/src/load-settings.ts +70 -70
- package/node_modules/@vitos-pizza/session-title/src/normalize-title.ts +20 -20
- package/node_modules/@vitos-pizza/session-title/src/parse-title-response.ts +6 -6
- package/node_modules/@vitos-pizza/session-title/src/resolve-title-model.ts +33 -33
- package/node_modules/@vitos-pizza/session-title/src/schedule-title-job.ts +53 -53
- package/node_modules/@vitos-pizza/session-title/src/should-schedule.ts +30 -30
- package/node_modules/@vitos-pizza/session-title/src/title-agent.ts +60 -60
- package/node_modules/@vitos-pizza/session-title/src/truncate-input.ts +10 -10
- package/node_modules/@vitos-pizza/session-title/src/types.ts +29 -29
- package/node_modules/@vitos-pizza/session-title/tests/build-title-prompt.test.ts +34 -34
- package/node_modules/@vitos-pizza/session-title/tests/extension.test.ts +198 -198
- package/node_modules/@vitos-pizza/session-title/tests/extract-title-context.test.ts +48 -48
- package/node_modules/@vitos-pizza/session-title/tests/fast-rules.test.ts +43 -43
- package/node_modules/@vitos-pizza/session-title/tests/parse-title-response.test.ts +34 -34
- package/node_modules/@vitos-pizza/session-title/tests/should-schedule.test.ts +30 -30
- package/node_modules/@vitos-pizza/session-title/tests/title-agent.test.ts +154 -154
- package/node_modules/@vitos-pizza/session-title/tests/truncate-input.test.ts +15 -15
- package/node_modules/@vitos-pizza/session-title/tsconfig.json +13 -13
- package/node_modules/@vitos-pizza/session-title/vitest.config.ts +8 -8
- package/node_modules/@vitos-pizza/subagents/agents/planner.md +46 -43
- package/node_modules/@vitos-pizza/subagents/agents/scout.md +40 -40
- package/node_modules/@vitos-pizza/subagents/agents/title.md +13 -13
- package/node_modules/@vitos-pizza/subagents/agents/worker.md +36 -36
- package/node_modules/@vitos-pizza/subagents/extensions/index.ts +164 -164
- package/node_modules/@vitos-pizza/subagents/package.json +1 -1
- package/node_modules/@vitos-pizza/subagents/skills/subagents/SKILL.md +86 -80
- package/node_modules/@vitos-pizza/subagents/src/agents.ts +166 -166
- package/node_modules/@vitos-pizza/subagents/src/async/job-tracker.ts +71 -71
- package/node_modules/@vitos-pizza/subagents/src/async/spawn-async.ts +220 -220
- package/node_modules/@vitos-pizza/subagents/src/async/storage.ts +57 -57
- package/node_modules/@vitos-pizza/subagents/src/env.ts +4 -4
- package/node_modules/@vitos-pizza/subagents/src/execute-subagent.ts +285 -285
- package/node_modules/@vitos-pizza/subagents/src/progress.ts +187 -187
- package/node_modules/@vitos-pizza/subagents/src/render-animation.ts +40 -40
- package/node_modules/@vitos-pizza/subagents/src/render.ts +428 -428
- package/node_modules/@vitos-pizza/subagents/src/rpc/channels.ts +21 -21
- package/node_modules/@vitos-pizza/subagents/src/rpc/client.ts +144 -144
- package/node_modules/@vitos-pizza/subagents/src/rpc/server.ts +11 -11
- package/node_modules/@vitos-pizza/subagents/src/run-agent.ts +313 -313
- package/node_modules/@vitos-pizza/subagents/src/runtime.ts +102 -102
- package/node_modules/@vitos-pizza/subagents/src/schema.ts +62 -62
- package/node_modules/@vitos-pizza/subagents/src/spawn.ts +104 -104
- package/node_modules/@vitos-pizza/subagents/src/types.ts +92 -92
- package/node_modules/@vitos-pizza/subagents/src/utils.ts +100 -100
- package/node_modules/@vitos-pizza/subagents/src/wait.ts +159 -159
- package/node_modules/@vitos-pizza/subagents/tests/agents.test.ts +19 -19
- package/node_modules/@vitos-pizza/subagents/tests/execute-subagent.test.ts +96 -96
- package/node_modules/@vitos-pizza/subagents/tests/extension.test.ts +42 -42
- package/node_modules/@vitos-pizza/subagents/tests/progress.test.ts +206 -206
- package/node_modules/@vitos-pizza/subagents/tests/rpc.test.ts +74 -74
- package/node_modules/@vitos-pizza/subagents/tsconfig.json +13 -13
- package/node_modules/@vitos-pizza/subagents/vitest.config.ts +8 -8
- package/node_modules/@vitos-pizza/ui-enhancements/package.json +1 -1
- package/node_modules/@vitos-pizza/ui-enhancements/src/chrome/agent-mode-badge.ts +60 -60
- package/node_modules/@vitos-pizza/ui-enhancements/src/chrome/border-status.ts +14 -1
- package/node_modules/@vitos-pizza/ui-enhancements/src/chrome/resize-recovery.ts +63 -0
- package/node_modules/@vitos-pizza/ui-enhancements/tests/agent-mode-badge.test.ts +35 -35
- package/node_modules/@vitos-pizza/ui-enhancements/tests/resize-recovery.test.ts +135 -0
- package/node_modules/@vitos-pizza/websearch/package.json +1 -1
- package/node_modules/node-addon-api/tools/conversion.js +0 -0
- package/node_modules/node-gyp-build/bin.js +0 -0
- package/node_modules/node-gyp-build/build-test.js +0 -0
- package/node_modules/node-gyp-build/optional.js +0 -0
- package/package.json +11 -9
- package/packages/agent-mode/extensions/index.ts +138 -117
- package/packages/agent-mode/package.json +3 -3
- package/packages/agent-mode/src/apply-mode.ts +60 -60
- package/packages/agent-mode/src/mode-events.ts +102 -102
- package/packages/agent-mode/src/modes.ts +43 -43
- package/packages/agent-mode/src/plan-instructions.ts +85 -44
- package/packages/agent-mode/tsconfig.json +13 -13
- package/packages/agent-mode/vitest.config.ts +8 -8
- package/packages/keybindings/extensions/index.ts +104 -104
- package/packages/keybindings/package.json +1 -1
- package/packages/keybindings/presets/shortcuts.json +3 -3
- package/packages/keybindings/src/bind.ts +57 -57
- package/packages/keybindings/src/config.ts +84 -84
- package/packages/keybindings/src/registry.ts +43 -43
- package/packages/keybindings/src/types.ts +54 -54
- package/packages/keybindings/tsconfig.json +13 -13
- package/packages/keybindings/vitest.config.ts +8 -8
- package/packages/permission-system/extensions/index.ts +162 -162
- package/packages/permission-system/node_modules/web-tree-sitter/debug/tree-sitter.wasm +0 -0
- package/packages/permission-system/node_modules/web-tree-sitter/tree-sitter.wasm +0 -0
- package/packages/permission-system/package.json +2 -2
- package/packages/permission-system/src/bash/matcher.ts +54 -51
- package/packages/permission-system/src/bash/parser.ts +40 -40
- package/packages/permission-system/src/config-paths.ts +23 -23
- package/packages/permission-system/src/detect-preset-mode.ts +61 -61
- package/packages/permission-system/src/expand-home.ts +12 -12
- package/packages/permission-system/src/forwarding/forwarder.ts +224 -224
- package/packages/permission-system/src/forwarding/subagent-registry.ts +61 -61
- package/packages/permission-system/src/gates/path-utils.ts +66 -66
- package/packages/permission-system/src/gates/pipeline.ts +151 -155
- package/packages/permission-system/src/handlers/before-agent-start.ts +86 -65
- package/packages/permission-system/src/handlers/skill-input.ts +1 -1
- package/packages/permission-system/src/handlers/tool-call-gate.ts +204 -199
- package/packages/permission-system/src/mode-api.ts +44 -44
- package/packages/permission-system/src/permission-evaluator.ts +233 -216
- package/packages/permission-system/src/permission-session.ts +62 -62
- package/packages/permission-system/src/policy-loader.ts +211 -211
- package/packages/permission-system/src/sanitizers/skill-prompt.ts +99 -99
- package/packages/permission-system/src/sanitizers/system-prompt.ts +221 -221
- package/packages/permission-system/src/service.ts +66 -66
- package/packages/permission-system/src/session-approvals.ts +4 -1
- package/packages/permission-system/src/types.ts +80 -80
- package/packages/permission-system/src/ui/approval-summary.ts +43 -43
- package/packages/permission-system/src/ui/config-modal.ts +36 -36
- package/packages/permission-system/src/ui/permission-dialog.ts +52 -52
- package/packages/permission-system/src/wildcard-matcher.ts +81 -81
- package/packages/permission-system/src/yaml-frontmatter.ts +84 -84
- package/packages/permission-system/src/yolo/runtime-api.ts +49 -49
- package/packages/permission-system/tsconfig.json +13 -13
- package/packages/permission-system/vitest.config.ts +8 -8
- package/packages/question/package.json +1 -1
- package/packages/question/src/question-ui.ts +5 -1
- package/packages/question/src/register-tool.ts +1 -4
- package/packages/session-title/extensions/index.ts +137 -137
- package/packages/session-title/package.json +3 -3
- package/packages/session-title/src/build-title-prompt.ts +16 -16
- package/packages/session-title/src/extract-title-context.ts +55 -55
- package/packages/session-title/src/fast-rules.ts +48 -48
- package/packages/session-title/src/load-settings.ts +70 -70
- package/packages/session-title/src/normalize-title.ts +20 -20
- package/packages/session-title/src/parse-title-response.ts +6 -6
- package/packages/session-title/src/resolve-title-model.ts +33 -33
- package/packages/session-title/src/schedule-title-job.ts +53 -53
- package/packages/session-title/src/should-schedule.ts +30 -30
- package/packages/session-title/src/title-agent.ts +60 -60
- package/packages/session-title/src/truncate-input.ts +10 -10
- package/packages/session-title/src/types.ts +29 -29
- package/packages/session-title/tsconfig.json +13 -13
- package/packages/session-title/vitest.config.ts +8 -8
- package/packages/subagents/agents/planner.md +46 -43
- package/packages/subagents/agents/scout.md +40 -40
- package/packages/subagents/agents/title.md +13 -13
- package/packages/subagents/agents/worker.md +36 -36
- package/packages/subagents/extensions/index.ts +164 -164
- package/packages/subagents/package.json +1 -1
- package/packages/subagents/skills/subagents/SKILL.md +86 -80
- package/packages/subagents/src/agents.ts +166 -166
- package/packages/subagents/src/async/job-tracker.ts +71 -71
- package/packages/subagents/src/async/spawn-async.ts +220 -220
- package/packages/subagents/src/async/storage.ts +57 -57
- package/packages/subagents/src/env.ts +4 -4
- package/packages/subagents/src/execute-subagent.ts +285 -285
- package/packages/subagents/src/progress.ts +187 -187
- package/packages/subagents/src/render-animation.ts +40 -40
- package/packages/subagents/src/render.ts +428 -428
- package/packages/subagents/src/rpc/channels.ts +21 -21
- package/packages/subagents/src/rpc/client.ts +144 -144
- package/packages/subagents/src/rpc/server.ts +11 -11
- package/packages/subagents/src/run-agent.ts +313 -313
- package/packages/subagents/src/runtime.ts +102 -102
- package/packages/subagents/src/schema.ts +62 -62
- package/packages/subagents/src/spawn.ts +104 -104
- package/packages/subagents/src/types.ts +92 -92
- package/packages/subagents/src/utils.ts +100 -100
- package/packages/subagents/src/wait.ts +159 -159
- package/packages/subagents/tsconfig.json +13 -13
- package/packages/subagents/vitest.config.ts +8 -8
- package/packages/todoist/extensions/index.ts +163 -0
- package/packages/todoist/package.json +34 -0
- package/packages/todoist/src/events.ts +7 -0
- package/packages/todoist/src/silent-render.ts +8 -0
- package/packages/todoist/src/state.ts +147 -0
- package/packages/todoist/src/tools.ts +175 -0
- package/packages/todoist/src/types.ts +34 -0
- package/packages/todoist/src/widget.ts +109 -0
- package/packages/todoist/tsconfig.json +13 -0
- package/packages/todoist/vitest.config.ts +8 -0
- package/packages/ui-enhancements/package.json +1 -1
- package/packages/ui-enhancements/src/chrome/agent-mode-badge.ts +60 -60
- package/packages/ui-enhancements/src/chrome/border-status.ts +14 -1
- package/packages/ui-enhancements/src/chrome/resize-recovery.ts +63 -0
- package/packages/websearch/package.json +1 -1
- package/scripts/sync-pi-manifest.mjs +1 -1
- package/packages/permission-system/node_modules/web-tree-sitter/lib/tree-sitter.cjs +0 -2988
- package/packages/permission-system/node_modules/web-tree-sitter/lib/tree-sitter.wasm +0 -0
- package/packages/permission-system/node_modules/web-tree-sitter/lib/tree-sitter.wasm.map +0 -1
|
@@ -1,159 +1,159 @@
|
|
|
1
|
-
import * as fs from "node:fs";
|
|
2
|
-
import type { AsyncJobTracker, SubagentEventBus } from "./async/job-tracker.ts";
|
|
3
|
-
import { summarizeAsyncRun } from "./async/spawn-async.ts";
|
|
4
|
-
import {
|
|
5
|
-
getAsyncRootFromSessionFile,
|
|
6
|
-
getRunDir,
|
|
7
|
-
isTerminalState,
|
|
8
|
-
} from "./async/storage.ts";
|
|
9
|
-
import { SUBAGENT_ASYNC_COMPLETE } from "./rpc/channels.ts";
|
|
10
|
-
import type { AsyncRunSummary, WaitInput, WaitResult } from "./types.ts";
|
|
11
|
-
|
|
12
|
-
const DEFAULT_TIMEOUT_MS = 30 * 60 * 1000;
|
|
13
|
-
const DEFAULT_POLL_INTERVAL_MS = 500;
|
|
14
|
-
|
|
15
|
-
export interface WaitDeps {
|
|
16
|
-
tracker: AsyncJobTracker;
|
|
17
|
-
sessionFile?: string | null;
|
|
18
|
-
events?: SubagentEventBus;
|
|
19
|
-
now?: () => number;
|
|
20
|
-
sleep?: (ms: number) => Promise<void>;
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
function listRunDirs(asyncRoot: string): string[] {
|
|
24
|
-
if (!fs.existsSync(asyncRoot)) return [];
|
|
25
|
-
try {
|
|
26
|
-
return fs
|
|
27
|
-
.readdirSync(asyncRoot, { withFileTypes: true })
|
|
28
|
-
.filter((entry) => entry.isDirectory())
|
|
29
|
-
.map((entry) => getRunDir(asyncRoot, entry.name));
|
|
30
|
-
} catch {
|
|
31
|
-
return [];
|
|
32
|
-
}
|
|
33
|
-
}
|
|
34
|
-
|
|
35
|
-
function listActiveRuns(
|
|
36
|
-
tracker: AsyncJobTracker,
|
|
37
|
-
sessionFile: string | null | undefined,
|
|
38
|
-
): AsyncRunSummary[] {
|
|
39
|
-
const asyncRoot = getAsyncRootFromSessionFile(sessionFile ?? null);
|
|
40
|
-
const summaries: AsyncRunSummary[] = [];
|
|
41
|
-
const seen = new Set<string>();
|
|
42
|
-
|
|
43
|
-
for (const status of tracker.list()) {
|
|
44
|
-
if (isTerminalState(status.state)) continue;
|
|
45
|
-
seen.add(status.id);
|
|
46
|
-
summaries.push({
|
|
47
|
-
id: status.id,
|
|
48
|
-
state: status.state,
|
|
49
|
-
agents: status.agents,
|
|
50
|
-
exitCode: status.exitCode,
|
|
51
|
-
});
|
|
52
|
-
}
|
|
53
|
-
|
|
54
|
-
for (const runDir of listRunDirs(asyncRoot)) {
|
|
55
|
-
const summary = summarizeAsyncRun(runDir);
|
|
56
|
-
if (!summary || isTerminalState(summary.state) || seen.has(summary.id)) continue;
|
|
57
|
-
summaries.push(summary);
|
|
58
|
-
}
|
|
59
|
-
|
|
60
|
-
return summaries;
|
|
61
|
-
}
|
|
62
|
-
|
|
63
|
-
function collectCompleted(
|
|
64
|
-
tracker: AsyncJobTracker,
|
|
65
|
-
sessionFile: string | null | undefined,
|
|
66
|
-
idPrefix?: string,
|
|
67
|
-
): AsyncRunSummary[] {
|
|
68
|
-
const asyncRoot = getAsyncRootFromSessionFile(sessionFile ?? null);
|
|
69
|
-
const completed: AsyncRunSummary[] = [];
|
|
70
|
-
const seen = new Set<string>();
|
|
71
|
-
|
|
72
|
-
for (const status of tracker.list()) {
|
|
73
|
-
if (!isTerminalState(status.state)) continue;
|
|
74
|
-
if (idPrefix && !status.id.startsWith(idPrefix)) continue;
|
|
75
|
-
seen.add(status.id);
|
|
76
|
-
const runDir = getRunDir(asyncRoot, status.id);
|
|
77
|
-
const summary = summarizeAsyncRun(runDir);
|
|
78
|
-
completed.push(
|
|
79
|
-
summary ?? {
|
|
80
|
-
id: status.id,
|
|
81
|
-
state: status.state,
|
|
82
|
-
agents: status.agents,
|
|
83
|
-
exitCode: status.exitCode,
|
|
84
|
-
},
|
|
85
|
-
);
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
return completed;
|
|
89
|
-
}
|
|
90
|
-
|
|
91
|
-
export async function waitForSubagents(
|
|
92
|
-
input: WaitInput,
|
|
93
|
-
deps: WaitDeps,
|
|
94
|
-
): Promise<WaitResult> {
|
|
95
|
-
const timeoutMs = input.timeoutMs ?? DEFAULT_TIMEOUT_MS;
|
|
96
|
-
const now = deps.now ?? (() => Date.now());
|
|
97
|
-
const sleep =
|
|
98
|
-
deps.sleep ?? ((ms: number) => new Promise((resolve) => setTimeout(resolve, ms)));
|
|
99
|
-
const deadline = now() + timeoutMs;
|
|
100
|
-
const waitAll = input.all === true;
|
|
101
|
-
const idPrefix = input.id;
|
|
102
|
-
|
|
103
|
-
let wake = false;
|
|
104
|
-
const unsub = deps.events?.on(SUBAGENT_ASYNC_COMPLETE, () => {
|
|
105
|
-
wake = true;
|
|
106
|
-
});
|
|
107
|
-
|
|
108
|
-
try {
|
|
109
|
-
while (now() < deadline) {
|
|
110
|
-
const active = listActiveRuns(deps.tracker, deps.sessionFile).filter((run) =>
|
|
111
|
-
idPrefix ? run.id.startsWith(idPrefix) : true,
|
|
112
|
-
);
|
|
113
|
-
|
|
114
|
-
if (active.length === 0) {
|
|
115
|
-
return {
|
|
116
|
-
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
117
|
-
timedOut: false,
|
|
118
|
-
activeRemaining: 0,
|
|
119
|
-
};
|
|
120
|
-
}
|
|
121
|
-
|
|
122
|
-
if (!waitAll) {
|
|
123
|
-
const completed = collectCompleted(deps.tracker, deps.sessionFile, idPrefix);
|
|
124
|
-
if (completed.length > 0) {
|
|
125
|
-
return {
|
|
126
|
-
completed,
|
|
127
|
-
timedOut: false,
|
|
128
|
-
activeRemaining: active.length,
|
|
129
|
-
};
|
|
130
|
-
}
|
|
131
|
-
} else {
|
|
132
|
-
const stillActive = active.filter((run) => !isTerminalState(run.state));
|
|
133
|
-
if (stillActive.length === 0) {
|
|
134
|
-
return {
|
|
135
|
-
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
136
|
-
timedOut: false,
|
|
137
|
-
activeRemaining: 0,
|
|
138
|
-
};
|
|
139
|
-
}
|
|
140
|
-
}
|
|
141
|
-
|
|
142
|
-
wake = false;
|
|
143
|
-
await sleep(DEFAULT_POLL_INTERVAL_MS);
|
|
144
|
-
if (!wake) continue;
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
const activeRemaining = listActiveRuns(deps.tracker, deps.sessionFile).filter((run) =>
|
|
148
|
-
idPrefix ? run.id.startsWith(idPrefix) : true,
|
|
149
|
-
).length;
|
|
150
|
-
|
|
151
|
-
return {
|
|
152
|
-
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
153
|
-
timedOut: true,
|
|
154
|
-
activeRemaining,
|
|
155
|
-
};
|
|
156
|
-
} finally {
|
|
157
|
-
unsub?.();
|
|
158
|
-
}
|
|
159
|
-
}
|
|
1
|
+
import * as fs from "node:fs";
|
|
2
|
+
import type { AsyncJobTracker, SubagentEventBus } from "./async/job-tracker.ts";
|
|
3
|
+
import { summarizeAsyncRun } from "./async/spawn-async.ts";
|
|
4
|
+
import {
|
|
5
|
+
getAsyncRootFromSessionFile,
|
|
6
|
+
getRunDir,
|
|
7
|
+
isTerminalState,
|
|
8
|
+
} from "./async/storage.ts";
|
|
9
|
+
import { SUBAGENT_ASYNC_COMPLETE } from "./rpc/channels.ts";
|
|
10
|
+
import type { AsyncRunSummary, WaitInput, WaitResult } from "./types.ts";
|
|
11
|
+
|
|
12
|
+
const DEFAULT_TIMEOUT_MS = 30 * 60 * 1000;
|
|
13
|
+
const DEFAULT_POLL_INTERVAL_MS = 500;
|
|
14
|
+
|
|
15
|
+
export interface WaitDeps {
|
|
16
|
+
tracker: AsyncJobTracker;
|
|
17
|
+
sessionFile?: string | null;
|
|
18
|
+
events?: SubagentEventBus;
|
|
19
|
+
now?: () => number;
|
|
20
|
+
sleep?: (ms: number) => Promise<void>;
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
function listRunDirs(asyncRoot: string): string[] {
|
|
24
|
+
if (!fs.existsSync(asyncRoot)) return [];
|
|
25
|
+
try {
|
|
26
|
+
return fs
|
|
27
|
+
.readdirSync(asyncRoot, { withFileTypes: true })
|
|
28
|
+
.filter((entry) => entry.isDirectory())
|
|
29
|
+
.map((entry) => getRunDir(asyncRoot, entry.name));
|
|
30
|
+
} catch {
|
|
31
|
+
return [];
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
function listActiveRuns(
|
|
36
|
+
tracker: AsyncJobTracker,
|
|
37
|
+
sessionFile: string | null | undefined,
|
|
38
|
+
): AsyncRunSummary[] {
|
|
39
|
+
const asyncRoot = getAsyncRootFromSessionFile(sessionFile ?? null);
|
|
40
|
+
const summaries: AsyncRunSummary[] = [];
|
|
41
|
+
const seen = new Set<string>();
|
|
42
|
+
|
|
43
|
+
for (const status of tracker.list()) {
|
|
44
|
+
if (isTerminalState(status.state)) continue;
|
|
45
|
+
seen.add(status.id);
|
|
46
|
+
summaries.push({
|
|
47
|
+
id: status.id,
|
|
48
|
+
state: status.state,
|
|
49
|
+
agents: status.agents,
|
|
50
|
+
exitCode: status.exitCode,
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
for (const runDir of listRunDirs(asyncRoot)) {
|
|
55
|
+
const summary = summarizeAsyncRun(runDir);
|
|
56
|
+
if (!summary || isTerminalState(summary.state) || seen.has(summary.id)) continue;
|
|
57
|
+
summaries.push(summary);
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
return summaries;
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
function collectCompleted(
|
|
64
|
+
tracker: AsyncJobTracker,
|
|
65
|
+
sessionFile: string | null | undefined,
|
|
66
|
+
idPrefix?: string,
|
|
67
|
+
): AsyncRunSummary[] {
|
|
68
|
+
const asyncRoot = getAsyncRootFromSessionFile(sessionFile ?? null);
|
|
69
|
+
const completed: AsyncRunSummary[] = [];
|
|
70
|
+
const seen = new Set<string>();
|
|
71
|
+
|
|
72
|
+
for (const status of tracker.list()) {
|
|
73
|
+
if (!isTerminalState(status.state)) continue;
|
|
74
|
+
if (idPrefix && !status.id.startsWith(idPrefix)) continue;
|
|
75
|
+
seen.add(status.id);
|
|
76
|
+
const runDir = getRunDir(asyncRoot, status.id);
|
|
77
|
+
const summary = summarizeAsyncRun(runDir);
|
|
78
|
+
completed.push(
|
|
79
|
+
summary ?? {
|
|
80
|
+
id: status.id,
|
|
81
|
+
state: status.state,
|
|
82
|
+
agents: status.agents,
|
|
83
|
+
exitCode: status.exitCode,
|
|
84
|
+
},
|
|
85
|
+
);
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
return completed;
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
export async function waitForSubagents(
|
|
92
|
+
input: WaitInput,
|
|
93
|
+
deps: WaitDeps,
|
|
94
|
+
): Promise<WaitResult> {
|
|
95
|
+
const timeoutMs = input.timeoutMs ?? DEFAULT_TIMEOUT_MS;
|
|
96
|
+
const now = deps.now ?? (() => Date.now());
|
|
97
|
+
const sleep =
|
|
98
|
+
deps.sleep ?? ((ms: number) => new Promise((resolve) => setTimeout(resolve, ms)));
|
|
99
|
+
const deadline = now() + timeoutMs;
|
|
100
|
+
const waitAll = input.all === true;
|
|
101
|
+
const idPrefix = input.id;
|
|
102
|
+
|
|
103
|
+
let wake = false;
|
|
104
|
+
const unsub = deps.events?.on(SUBAGENT_ASYNC_COMPLETE, () => {
|
|
105
|
+
wake = true;
|
|
106
|
+
});
|
|
107
|
+
|
|
108
|
+
try {
|
|
109
|
+
while (now() < deadline) {
|
|
110
|
+
const active = listActiveRuns(deps.tracker, deps.sessionFile).filter((run) =>
|
|
111
|
+
idPrefix ? run.id.startsWith(idPrefix) : true,
|
|
112
|
+
);
|
|
113
|
+
|
|
114
|
+
if (active.length === 0) {
|
|
115
|
+
return {
|
|
116
|
+
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
117
|
+
timedOut: false,
|
|
118
|
+
activeRemaining: 0,
|
|
119
|
+
};
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
if (!waitAll) {
|
|
123
|
+
const completed = collectCompleted(deps.tracker, deps.sessionFile, idPrefix);
|
|
124
|
+
if (completed.length > 0) {
|
|
125
|
+
return {
|
|
126
|
+
completed,
|
|
127
|
+
timedOut: false,
|
|
128
|
+
activeRemaining: active.length,
|
|
129
|
+
};
|
|
130
|
+
}
|
|
131
|
+
} else {
|
|
132
|
+
const stillActive = active.filter((run) => !isTerminalState(run.state));
|
|
133
|
+
if (stillActive.length === 0) {
|
|
134
|
+
return {
|
|
135
|
+
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
136
|
+
timedOut: false,
|
|
137
|
+
activeRemaining: 0,
|
|
138
|
+
};
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
|
|
142
|
+
wake = false;
|
|
143
|
+
await sleep(DEFAULT_POLL_INTERVAL_MS);
|
|
144
|
+
if (!wake) continue;
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
const activeRemaining = listActiveRuns(deps.tracker, deps.sessionFile).filter((run) =>
|
|
148
|
+
idPrefix ? run.id.startsWith(idPrefix) : true,
|
|
149
|
+
).length;
|
|
150
|
+
|
|
151
|
+
return {
|
|
152
|
+
completed: collectCompleted(deps.tracker, deps.sessionFile, idPrefix),
|
|
153
|
+
timedOut: true,
|
|
154
|
+
activeRemaining,
|
|
155
|
+
};
|
|
156
|
+
} finally {
|
|
157
|
+
unsub?.();
|
|
158
|
+
}
|
|
159
|
+
}
|
|
@@ -1,19 +1,19 @@
|
|
|
1
|
-
import { describe, expect, it } from "vitest";
|
|
2
|
-
import { discoverAgents, getBuiltinAgentsDir } from "../src/agents.ts";
|
|
3
|
-
|
|
4
|
-
describe("discoverAgents", () => {
|
|
5
|
-
it("loads builtin scout, planner, and worker agents", () => {
|
|
6
|
-
const { agents, builtinAgentsDir } = discoverAgents(process.cwd(), "both");
|
|
7
|
-
expect(builtinAgentsDir).toBe(getBuiltinAgentsDir());
|
|
8
|
-
const names = agents.map((agent) => agent.name).sort();
|
|
9
|
-
expect(names).toEqual(["planner", "scout", "title", "worker"]);
|
|
10
|
-
});
|
|
11
|
-
|
|
12
|
-
it("prefers builtin agents over duplicates", () => {
|
|
13
|
-
const scout = discoverAgents(process.cwd(), "both").agents.find(
|
|
14
|
-
(agent) => agent.name === "scout",
|
|
15
|
-
);
|
|
16
|
-
expect(scout?.source).toBe("builtin");
|
|
17
|
-
expect(scout?.tools).toContain("read");
|
|
18
|
-
});
|
|
19
|
-
});
|
|
1
|
+
import { describe, expect, it } from "vitest";
|
|
2
|
+
import { discoverAgents, getBuiltinAgentsDir } from "../src/agents.ts";
|
|
3
|
+
|
|
4
|
+
describe("discoverAgents", () => {
|
|
5
|
+
it("loads builtin scout, planner, and worker agents", () => {
|
|
6
|
+
const { agents, builtinAgentsDir } = discoverAgents(process.cwd(), "both");
|
|
7
|
+
expect(builtinAgentsDir).toBe(getBuiltinAgentsDir());
|
|
8
|
+
const names = agents.map((agent) => agent.name).sort();
|
|
9
|
+
expect(names).toEqual(["planner", "scout", "title", "worker"]);
|
|
10
|
+
});
|
|
11
|
+
|
|
12
|
+
it("prefers builtin agents over duplicates", () => {
|
|
13
|
+
const scout = discoverAgents(process.cwd(), "both").agents.find(
|
|
14
|
+
(agent) => agent.name === "scout",
|
|
15
|
+
);
|
|
16
|
+
expect(scout?.source).toBe("builtin");
|
|
17
|
+
expect(scout?.tools).toContain("read");
|
|
18
|
+
});
|
|
19
|
+
});
|
|
@@ -1,96 +1,96 @@
|
|
|
1
|
-
import { describe, expect, it, vi } from "vitest";
|
|
2
|
-
import { buildSubagentDetails, executeSubagent } from "../src/execute-subagent.ts";
|
|
3
|
-
import type { AgentConfig } from "../src/agents.ts";
|
|
4
|
-
|
|
5
|
-
vi.mock("../src/run-agent.ts", () => ({
|
|
6
|
-
runSingleAgent: vi.fn(async (input) => ({
|
|
7
|
-
agent: input.agentName,
|
|
8
|
-
agentSource: "builtin" as const,
|
|
9
|
-
task: input.task,
|
|
10
|
-
exitCode: 0,
|
|
11
|
-
messages: [
|
|
12
|
-
{
|
|
13
|
-
role: "assistant",
|
|
14
|
-
content: [{ type: "text", text: `output:${input.task}` }],
|
|
15
|
-
api: "test",
|
|
16
|
-
provider: "test",
|
|
17
|
-
model: "test",
|
|
18
|
-
usage: { input: 1, output: 1, cacheRead: 0, cacheWrite: 0, totalTokens: 2 },
|
|
19
|
-
stopReason: "stop",
|
|
20
|
-
timestamp: Date.now(),
|
|
21
|
-
},
|
|
22
|
-
],
|
|
23
|
-
stderr: "",
|
|
24
|
-
usage: {
|
|
25
|
-
input: 1,
|
|
26
|
-
output: 1,
|
|
27
|
-
cacheRead: 0,
|
|
28
|
-
cacheWrite: 0,
|
|
29
|
-
cost: 0,
|
|
30
|
-
contextTokens: 2,
|
|
31
|
-
turns: 1,
|
|
32
|
-
},
|
|
33
|
-
step: input.step,
|
|
34
|
-
})),
|
|
35
|
-
}));
|
|
36
|
-
|
|
37
|
-
vi.mock("../src/agents.ts", async (importOriginal) => {
|
|
38
|
-
const actual = await importOriginal<typeof import("../src/agents.ts")>();
|
|
39
|
-
return {
|
|
40
|
-
...actual,
|
|
41
|
-
discoverAgents: () => ({
|
|
42
|
-
agents: [
|
|
43
|
-
{
|
|
44
|
-
name: "scout",
|
|
45
|
-
description: "scout",
|
|
46
|
-
systemPrompt: "scout",
|
|
47
|
-
source: "builtin",
|
|
48
|
-
filePath: "/scout.md",
|
|
49
|
-
},
|
|
50
|
-
{
|
|
51
|
-
name: "planner",
|
|
52
|
-
description: "planner",
|
|
53
|
-
systemPrompt: "planner",
|
|
54
|
-
source: "builtin",
|
|
55
|
-
filePath: "/planner.md",
|
|
56
|
-
},
|
|
57
|
-
] satisfies AgentConfig[],
|
|
58
|
-
projectAgentsDir: null,
|
|
59
|
-
builtinAgentsDir: "/agents",
|
|
60
|
-
}),
|
|
61
|
-
};
|
|
62
|
-
});
|
|
63
|
-
|
|
64
|
-
describe("executeSubagent", () => {
|
|
65
|
-
it("rejects invalid mode combinations", async () => {
|
|
66
|
-
const result = await executeSubagent(
|
|
67
|
-
{ agent: "scout", task: "a", chain: [{ agent: "scout", task: "b" }] },
|
|
68
|
-
{ cwd: process.cwd() },
|
|
69
|
-
);
|
|
70
|
-
expect(result.content[0]).toMatchObject({ type: "text" });
|
|
71
|
-
expect((result.content[0] as { text: string }).text).toContain("Invalid parameters");
|
|
72
|
-
});
|
|
73
|
-
|
|
74
|
-
it("passes {previous} through chain steps", async () => {
|
|
75
|
-
const result = await executeSubagent(
|
|
76
|
-
{
|
|
77
|
-
chain: [
|
|
78
|
-
{ agent: "scout", task: "first" },
|
|
79
|
-
{ agent: "planner", task: "plan from {previous}" },
|
|
80
|
-
],
|
|
81
|
-
},
|
|
82
|
-
{ cwd: process.cwd() },
|
|
83
|
-
);
|
|
84
|
-
expect(result.details?.mode).toBe("chain");
|
|
85
|
-
expect(result.details?.results).toHaveLength(2);
|
|
86
|
-
expect((result.content[0] as { text: string }).text).toContain("output:plan from output:first");
|
|
87
|
-
});
|
|
88
|
-
});
|
|
89
|
-
|
|
90
|
-
describe("buildSubagentDetails", () => {
|
|
91
|
-
it("builds details with mode and scope", () => {
|
|
92
|
-
const details = buildSubagentDetails("single", "both", null, []);
|
|
93
|
-
expect(details.mode).toBe("single");
|
|
94
|
-
expect(details.agentScope).toBe("both");
|
|
95
|
-
});
|
|
96
|
-
});
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { buildSubagentDetails, executeSubagent } from "../src/execute-subagent.ts";
|
|
3
|
+
import type { AgentConfig } from "../src/agents.ts";
|
|
4
|
+
|
|
5
|
+
vi.mock("../src/run-agent.ts", () => ({
|
|
6
|
+
runSingleAgent: vi.fn(async (input) => ({
|
|
7
|
+
agent: input.agentName,
|
|
8
|
+
agentSource: "builtin" as const,
|
|
9
|
+
task: input.task,
|
|
10
|
+
exitCode: 0,
|
|
11
|
+
messages: [
|
|
12
|
+
{
|
|
13
|
+
role: "assistant",
|
|
14
|
+
content: [{ type: "text", text: `output:${input.task}` }],
|
|
15
|
+
api: "test",
|
|
16
|
+
provider: "test",
|
|
17
|
+
model: "test",
|
|
18
|
+
usage: { input: 1, output: 1, cacheRead: 0, cacheWrite: 0, totalTokens: 2 },
|
|
19
|
+
stopReason: "stop",
|
|
20
|
+
timestamp: Date.now(),
|
|
21
|
+
},
|
|
22
|
+
],
|
|
23
|
+
stderr: "",
|
|
24
|
+
usage: {
|
|
25
|
+
input: 1,
|
|
26
|
+
output: 1,
|
|
27
|
+
cacheRead: 0,
|
|
28
|
+
cacheWrite: 0,
|
|
29
|
+
cost: 0,
|
|
30
|
+
contextTokens: 2,
|
|
31
|
+
turns: 1,
|
|
32
|
+
},
|
|
33
|
+
step: input.step,
|
|
34
|
+
})),
|
|
35
|
+
}));
|
|
36
|
+
|
|
37
|
+
vi.mock("../src/agents.ts", async (importOriginal) => {
|
|
38
|
+
const actual = await importOriginal<typeof import("../src/agents.ts")>();
|
|
39
|
+
return {
|
|
40
|
+
...actual,
|
|
41
|
+
discoverAgents: () => ({
|
|
42
|
+
agents: [
|
|
43
|
+
{
|
|
44
|
+
name: "scout",
|
|
45
|
+
description: "scout",
|
|
46
|
+
systemPrompt: "scout",
|
|
47
|
+
source: "builtin",
|
|
48
|
+
filePath: "/scout.md",
|
|
49
|
+
},
|
|
50
|
+
{
|
|
51
|
+
name: "planner",
|
|
52
|
+
description: "planner",
|
|
53
|
+
systemPrompt: "planner",
|
|
54
|
+
source: "builtin",
|
|
55
|
+
filePath: "/planner.md",
|
|
56
|
+
},
|
|
57
|
+
] satisfies AgentConfig[],
|
|
58
|
+
projectAgentsDir: null,
|
|
59
|
+
builtinAgentsDir: "/agents",
|
|
60
|
+
}),
|
|
61
|
+
};
|
|
62
|
+
});
|
|
63
|
+
|
|
64
|
+
describe("executeSubagent", () => {
|
|
65
|
+
it("rejects invalid mode combinations", async () => {
|
|
66
|
+
const result = await executeSubagent(
|
|
67
|
+
{ agent: "scout", task: "a", chain: [{ agent: "scout", task: "b" }] },
|
|
68
|
+
{ cwd: process.cwd() },
|
|
69
|
+
);
|
|
70
|
+
expect(result.content[0]).toMatchObject({ type: "text" });
|
|
71
|
+
expect((result.content[0] as { text: string }).text).toContain("Invalid parameters");
|
|
72
|
+
});
|
|
73
|
+
|
|
74
|
+
it("passes {previous} through chain steps", async () => {
|
|
75
|
+
const result = await executeSubagent(
|
|
76
|
+
{
|
|
77
|
+
chain: [
|
|
78
|
+
{ agent: "scout", task: "first" },
|
|
79
|
+
{ agent: "planner", task: "plan from {previous}" },
|
|
80
|
+
],
|
|
81
|
+
},
|
|
82
|
+
{ cwd: process.cwd() },
|
|
83
|
+
);
|
|
84
|
+
expect(result.details?.mode).toBe("chain");
|
|
85
|
+
expect(result.details?.results).toHaveLength(2);
|
|
86
|
+
expect((result.content[0] as { text: string }).text).toContain("output:plan from output:first");
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
|
|
90
|
+
describe("buildSubagentDetails", () => {
|
|
91
|
+
it("builds details with mode and scope", () => {
|
|
92
|
+
const details = buildSubagentDetails("single", "both", null, []);
|
|
93
|
+
expect(details.mode).toBe("single");
|
|
94
|
+
expect(details.agentScope).toBe("both");
|
|
95
|
+
});
|
|
96
|
+
});
|
|
@@ -1,42 +1,42 @@
|
|
|
1
|
-
import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
|
|
2
|
-
import { describe, expect, it, vi } from "vitest";
|
|
3
|
-
import { registerSubagents } from "../extensions/index.ts";
|
|
4
|
-
import { SUBAGENT_PARENT_SESSION_ENV } from "../src/env.ts";
|
|
5
|
-
import { SUBAGENTS_READY } from "../src/rpc/channels.ts";
|
|
6
|
-
|
|
7
|
-
describe("registerSubagents", () => {
|
|
8
|
-
it("sets parent session env on session_start and emits ready", () => {
|
|
9
|
-
const handlers = new Map<string, (event: unknown, ctx: ExtensionContext) => void>();
|
|
10
|
-
const emitted: Array<{ channel: string; payload: unknown }> = [];
|
|
11
|
-
const pi = {
|
|
12
|
-
on: vi.fn((event: string, handler: (event: unknown, ctx: ExtensionContext) => void) => {
|
|
13
|
-
handlers.set(event, handler);
|
|
14
|
-
}),
|
|
15
|
-
registerTool: vi.fn(),
|
|
16
|
-
events: {
|
|
17
|
-
emit: vi.fn((channel: string, payload: unknown) => {
|
|
18
|
-
emitted.push({ channel, payload });
|
|
19
|
-
}),
|
|
20
|
-
on: vi.fn(() => () => {}),
|
|
21
|
-
},
|
|
22
|
-
};
|
|
23
|
-
|
|
24
|
-
delete process.env[SUBAGENT_PARENT_SESSION_ENV];
|
|
25
|
-
registerSubagents(pi as never);
|
|
26
|
-
|
|
27
|
-
const ctx = {
|
|
28
|
-
cwd: process.cwd(),
|
|
29
|
-
sessionManager: {
|
|
30
|
-
getSessionId: () => "parent-session-123",
|
|
31
|
-
getSessionFile: () => null,
|
|
32
|
-
},
|
|
33
|
-
} as unknown as ExtensionContext;
|
|
34
|
-
|
|
35
|
-
handlers.get("session_start")?.({}, ctx);
|
|
36
|
-
expect(process.env[SUBAGENT_PARENT_SESSION_ENV]).toBe("parent-session-123");
|
|
37
|
-
expect(emitted.some((entry) => entry.channel === SUBAGENTS_READY)).toBe(true);
|
|
38
|
-
|
|
39
|
-
handlers.get("session_shutdown")?.({}, ctx);
|
|
40
|
-
expect(process.env[SUBAGENT_PARENT_SESSION_ENV]).toBeUndefined();
|
|
41
|
-
});
|
|
42
|
-
});
|
|
1
|
+
import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
|
|
2
|
+
import { describe, expect, it, vi } from "vitest";
|
|
3
|
+
import { registerSubagents } from "../extensions/index.ts";
|
|
4
|
+
import { SUBAGENT_PARENT_SESSION_ENV } from "../src/env.ts";
|
|
5
|
+
import { SUBAGENTS_READY } from "../src/rpc/channels.ts";
|
|
6
|
+
|
|
7
|
+
describe("registerSubagents", () => {
|
|
8
|
+
it("sets parent session env on session_start and emits ready", () => {
|
|
9
|
+
const handlers = new Map<string, (event: unknown, ctx: ExtensionContext) => void>();
|
|
10
|
+
const emitted: Array<{ channel: string; payload: unknown }> = [];
|
|
11
|
+
const pi = {
|
|
12
|
+
on: vi.fn((event: string, handler: (event: unknown, ctx: ExtensionContext) => void) => {
|
|
13
|
+
handlers.set(event, handler);
|
|
14
|
+
}),
|
|
15
|
+
registerTool: vi.fn(),
|
|
16
|
+
events: {
|
|
17
|
+
emit: vi.fn((channel: string, payload: unknown) => {
|
|
18
|
+
emitted.push({ channel, payload });
|
|
19
|
+
}),
|
|
20
|
+
on: vi.fn(() => () => {}),
|
|
21
|
+
},
|
|
22
|
+
};
|
|
23
|
+
|
|
24
|
+
delete process.env[SUBAGENT_PARENT_SESSION_ENV];
|
|
25
|
+
registerSubagents(pi as never);
|
|
26
|
+
|
|
27
|
+
const ctx = {
|
|
28
|
+
cwd: process.cwd(),
|
|
29
|
+
sessionManager: {
|
|
30
|
+
getSessionId: () => "parent-session-123",
|
|
31
|
+
getSessionFile: () => null,
|
|
32
|
+
},
|
|
33
|
+
} as unknown as ExtensionContext;
|
|
34
|
+
|
|
35
|
+
handlers.get("session_start")?.({}, ctx);
|
|
36
|
+
expect(process.env[SUBAGENT_PARENT_SESSION_ENV]).toBe("parent-session-123");
|
|
37
|
+
expect(emitted.some((entry) => entry.channel === SUBAGENTS_READY)).toBe(true);
|
|
38
|
+
|
|
39
|
+
handlers.get("session_shutdown")?.({}, ctx);
|
|
40
|
+
expect(process.env[SUBAGENT_PARENT_SESSION_ENV]).toBeUndefined();
|
|
41
|
+
});
|
|
42
|
+
});
|