@selesai/code 0.9.9 → 0.9.11
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/CHANGELOG.md +11 -0
- package/dist/cli/args.js +13 -2
- package/dist/cli/args.test.js +8 -0
- package/dist/cli/credential-print.js +4 -4
- package/dist/cli.js +0 -0
- package/dist/core/agent-session-auto-handoff.test.js +24 -1
- package/dist/core/agent-session.d.ts +11 -4
- package/dist/core/agent-session.js +99 -39
- package/dist/core/compaction/branch-summarization.js +30 -30
- package/dist/core/compaction/compaction.js +81 -81
- package/dist/core/compaction/utils.js +2 -2
- package/dist/core/defaults.d.ts +1 -0
- package/dist/core/defaults.js +9 -0
- package/dist/core/export-html/template.css +1066 -1066
- package/dist/core/export-html/template.html +55 -55
- package/dist/core/export-html/template.js +1864 -1864
- package/dist/core/export-html/vendor/highlight.min.js +1212 -1212
- package/dist/core/export-html/vendor/marked.min.js +78 -78
- package/dist/core/extensions/loader.js +108 -35
- package/dist/core/extensions/loader.test.d.ts +1 -0
- package/dist/core/extensions/loader.test.js +21 -0
- package/dist/core/extensions/types.d.ts +12 -2
- package/dist/core/keybindings.d.ts +2 -2
- package/dist/core/messages.js +7 -7
- package/dist/core/model-resolver.d.ts +1 -0
- package/dist/core/model-resolver.js +10 -4
- package/dist/core/package-manager.js +4 -4
- package/dist/core/sdk.d.ts +5 -5
- package/dist/core/sdk.js +16 -4
- package/dist/core/settings-manager.d.ts +6 -0
- package/dist/core/settings-manager.js +22 -0
- package/dist/core/tools/bash.d.ts +18 -1
- package/dist/core/tools/bash.js +41 -23
- package/dist/core/tools/index.d.ts +4 -1
- package/dist/core/tools/index.js +9 -1
- package/dist/core/tools/powershell.d.ts +15 -0
- package/dist/core/tools/powershell.js +38 -0
- package/dist/core/tools/powershell.test.d.ts +1 -0
- package/dist/core/tools/powershell.test.js +20 -0
- package/dist/extensions/context-compaction-reminder.test.ts +82 -82
- package/dist/extensions/context-compaction-reminder.ts +28 -28
- package/dist/extensions/copy-turn.test.ts +24 -1
- package/dist/extensions/copy-turn.ts +6 -1
- package/dist/extensions/handoff-new.test.ts +15 -2
- package/dist/extensions/handoff-new.ts +44 -30
- package/dist/extensions/pi-intercom/LICENSE +21 -21
- package/dist/extensions/pi-intercom/broker/client.test.ts +83 -83
- package/dist/extensions/pi-intercom/broker/extension.test.ts +387 -387
- package/dist/extensions/pi-intercom/broker/framing.test.ts +114 -114
- package/dist/extensions/pi-intercom/broker/paths.test.ts +153 -153
- package/dist/extensions/pi-intercom/broker/paths.ts +134 -134
- package/dist/extensions/pi-intercom/broker/runtime-claim.test.ts +34 -34
- package/dist/extensions/pi-intercom/broker/runtime-claim.ts +21 -21
- package/dist/extensions/pi-intercom/cwd.test.ts +40 -40
- package/dist/extensions/pi-intercom/cwd.ts +31 -31
- package/dist/extensions/pi-intercom/extension-api.ts +44 -44
- package/dist/extensions/pi-intercom/format-context.test.ts +31 -31
- package/dist/extensions/pi-intercom/format-context.ts +32 -32
- package/dist/extensions/pi-intercom/test/overlay-width.test.ts +66 -66
- package/dist/extensions/pi-intercom/ui/compose.ts +143 -143
- package/dist/extensions/pi-intercom/ui/session-list.ts +166 -166
- package/dist/extensions/pi-powerline-footer/bash-mode/shell-session.ts +286 -286
- package/dist/extensions/pi-powerline-footer/bash-mode/transcript.ts +108 -108
- package/dist/extensions/pi-powerline-footer/separators.ts +57 -57
- package/dist/extensions/pi-powerline-footer/tests/session-usage.test.ts +47 -47
- package/dist/extensions/pi-powerline-footer/tests/tps.test.ts +39 -39
- package/dist/extensions/pi-powerline-footer/theme.example.json +24 -24
- package/dist/extensions/pi-powerline-footer/theme.json +12 -12
- package/dist/extensions/pi-powerline-footer/tps.ts +345 -345
- package/dist/extensions/pi-rewind-hook/README.md +245 -245
- package/dist/extensions/pi-rewind-hook/index.ts +1445 -1445
- package/dist/extensions/pi-rewind-hook/package.json +29 -29
- package/dist/extensions/pi-subagents/install.mjs +0 -0
- package/dist/extensions/pi-web-agent/package.json +31 -31
- package/dist/extensions/pi-web-agent/src/backends/config.ts +205 -205
- package/dist/extensions/pi-web-agent/src/backends/doctor.ts +136 -136
- package/dist/extensions/pi-web-agent/src/backends/factory.ts +152 -152
- package/dist/extensions/pi-web-agent/src/backends/settings-reader.ts +25 -25
- package/dist/extensions/pi-web-agent/src/cache/ttl-cache.ts +28 -28
- package/dist/extensions/pi-web-agent/src/changelog-notice.ts +136 -136
- package/dist/extensions/pi-web-agent/src/commands/web-agent-config.ts +946 -946
- package/dist/extensions/pi-web-agent/src/extension.ts +126 -126
- package/dist/extensions/pi-web-agent/src/extract/readability.ts +118 -118
- package/dist/extensions/pi-web-agent/src/fetch/browser-resolution.ts +199 -199
- package/dist/extensions/pi-web-agent/src/fetch/firecrawl-fetch.ts +100 -100
- package/dist/extensions/pi-web-agent/src/fetch/headless-fetch.ts +117 -117
- package/dist/extensions/pi-web-agent/src/fetch/http-fetch.ts +67 -67
- package/dist/extensions/pi-web-agent/src/orchestration/answer-synthesizer.ts +60 -60
- package/dist/extensions/pi-web-agent/src/orchestration/candidate-selector.ts +50 -50
- package/dist/extensions/pi-web-agent/src/orchestration/direct-url.ts +52 -52
- package/dist/extensions/pi-web-agent/src/orchestration/evidence-quality.ts +105 -105
- package/dist/extensions/pi-web-agent/src/orchestration/evidence-ranker.ts +45 -45
- package/dist/extensions/pi-web-agent/src/orchestration/index.ts +28 -28
- package/dist/extensions/pi-web-agent/src/orchestration/query-planner.ts +47 -47
- package/dist/extensions/pi-web-agent/src/orchestration/research-orchestrator.ts +376 -376
- package/dist/extensions/pi-web-agent/src/orchestration/research-types.ts +64 -64
- package/dist/extensions/pi-web-agent/src/orchestration/research-worker.ts +181 -181
- package/dist/extensions/pi-web-agent/src/orchestration/source-profile.ts +101 -101
- package/dist/extensions/pi-web-agent/src/orchestration/stop-decider.ts +81 -81
- package/dist/extensions/pi-web-agent/src/presentation/config-store.ts +210 -210
- package/dist/extensions/pi-web-agent/src/presentation/config.ts +75 -75
- package/dist/extensions/pi-web-agent/src/presentation/explore-presentation.ts +61 -61
- package/dist/extensions/pi-web-agent/src/presentation/fetch-presentation.ts +54 -54
- package/dist/extensions/pi-web-agent/src/presentation/search-presentation.ts +41 -41
- package/dist/extensions/pi-web-agent/src/presentation/select-view.ts +20 -20
- package/dist/extensions/pi-web-agent/src/presentation/types.ts +63 -63
- package/dist/extensions/pi-web-agent/src/search/brave.ts +114 -114
- package/dist/extensions/pi-web-agent/src/search/duckduckgo.ts +72 -72
- package/dist/extensions/pi-web-agent/src/search/searxng.ts +96 -96
- package/dist/extensions/pi-web-agent/src/tools/web-explore.ts +71 -71
- package/dist/extensions/pi-web-agent/src/tools/web-fetch-headless.ts +31 -31
- package/dist/extensions/pi-web-agent/src/tools/web-fetch.ts +31 -31
- package/dist/extensions/pi-web-agent/src/tools/web-search.ts +157 -157
- package/dist/extensions/pi-web-agent/src/types.ts +88 -88
- package/dist/extensions/ponytail/package.json +8 -8
- package/dist/extensions/question/batch.ts +103 -103
- package/dist/extensions/question/constants.ts +30 -30
- package/dist/extensions/question/helpers.ts +58 -58
- package/dist/extensions/question/navigation.ts +14 -14
- package/dist/extensions/question/package.json +19 -19
- package/dist/extensions/question/schemas.ts +43 -43
- package/dist/extensions/question/selection-mode.ts +46 -46
- package/dist/extensions/question/shortcuts.ts +44 -44
- package/dist/extensions/question/types.ts +132 -132
- package/dist/extensions/test-resolve-hook-impl.mjs +6 -6
- package/dist/extensions/test-resolve-hook.mjs +6 -6
- package/dist/extensions/web-agent-onboarding.ts +222 -222
- package/dist/extensions/workflow/package.json +17 -17
- package/dist/package-manager-cli.js +71 -71
- package/dist/rpc-entry.js +0 -0
- package/dist/skills/agent-browser/SKILL.md +52 -52
- package/dist/skills/batch-grill-me/SKILL.md +19 -19
- package/dist/skills/grill-me/SKILL.md +10 -10
- package/dist/skills/handoff/SKILL.md +16 -16
- package/dist/skills/handoff-text/SKILL.md +14 -14
- package/dist/skills/implanger/SKILL.md +69 -69
- package/dist/skills/improve-codebase/REFERENCE.md +78 -78
- package/dist/skills/improve-codebase/SKILL.md +178 -178
- package/dist/skills/planger/SKILL.md +166 -166
- package/dist/skills/ponytail/SKILL.md +116 -116
- package/dist/skills/ponytail-audit/SKILL.md +42 -42
- package/dist/skills/ponytail-debt/SKILL.md +45 -45
- package/dist/skills/ponytail-gain/SKILL.md +51 -51
- package/dist/skills/ponytail-help/SKILL.md +70 -70
- package/dist/skills/ponytail-review/SKILL.md +58 -58
- package/dist/skills/selesai-handoff/SKILL.md +20 -20
- package/dist/skills/workflow-creation/SKILL.md +73 -73
- package/dist/themes/powerline-footer/theme.json +33 -33
- package/dist/utils/shell.d.ts +3 -0
- package/dist/utils/shell.js +17 -5
- package/docs/compaction.md +396 -396
- package/docs/containerization.md +111 -111
- package/docs/development.md +71 -71
- package/docs/docs.json +164 -164
- package/docs/environment-variables.md +86 -86
- package/docs/index.md +83 -83
- package/docs/json.md +82 -82
- package/docs/models.md +502 -502
- package/docs/packages.md +227 -227
- package/docs/plans/subagent-delegation/phase-0-correctness.md +264 -264
- package/docs/plans/subagent-delegation/phase-1-behavioral-contract.md +485 -485
- package/docs/plans/subagent-delegation/phase-2-context-controls.md +281 -281
- package/docs/plans/subagent-delegation/phase-3-advisory-routing.md +361 -361
- package/docs/plans/subagent-delegation/phase-4-optional-enforcement.md +380 -380
- package/docs/prompt-templates.md +95 -95
- package/docs/providers.md +293 -293
- package/docs/sdk.md +1144 -1143
- package/docs/security.md +59 -59
- package/docs/session-format.md +414 -414
- package/docs/sessions.md +145 -145
- package/docs/shared-host-extensions.md +109 -109
- package/docs/shell-aliases.md +13 -13
- package/docs/skills.md +231 -231
- package/docs/terminal-setup.md +142 -142
- package/docs/termux.md +127 -127
- package/docs/themes.md +295 -295
- package/docs/tmux.md +63 -63
- package/docs/tui.md +927 -927
- package/docs/windows.md +17 -17
- package/examples/README.md +25 -25
- package/examples/extensions/README.md +211 -211
- package/examples/extensions/auto-commit-on-exit.ts +49 -49
- package/examples/extensions/bash-spawn-hook.ts +30 -30
- package/examples/extensions/bookmark.ts +50 -50
- package/examples/extensions/border-status-editor.ts +150 -150
- package/examples/extensions/built-in-tool-renderer.ts +249 -249
- package/examples/extensions/claude-rules.ts +86 -86
- package/examples/extensions/commands.ts +72 -72
- package/examples/extensions/confirm-destructive.ts +59 -59
- package/examples/extensions/custom-compaction.ts +130 -130
- package/examples/extensions/custom-footer.ts +64 -64
- package/examples/extensions/custom-header.ts +73 -73
- package/examples/extensions/custom-provider-anthropic/index.ts +610 -610
- package/examples/extensions/custom-provider-anthropic/package-lock.json +24 -24
- package/examples/extensions/custom-provider-anthropic/package.json +19 -19
- package/examples/extensions/custom-provider-gitlab-duo/index.ts +404 -404
- package/examples/extensions/custom-provider-gitlab-duo/package.json +16 -16
- package/examples/extensions/custom-provider-gitlab-duo/test.ts +82 -82
- package/examples/extensions/dirty-repo-guard.ts +56 -56
- package/examples/extensions/doom-overlay/README.md +46 -46
- package/examples/extensions/doom-overlay/doom/build/doom.js +21 -21
- package/examples/extensions/doom-overlay/doom/build/doom.wasm +0 -0
- package/examples/extensions/doom-overlay/doom/build.sh +152 -152
- package/examples/extensions/doom-overlay/doom/doomgeneric_pi.c +72 -72
- package/examples/extensions/doom-overlay/doom-component.ts +132 -132
- package/examples/extensions/doom-overlay/doom-engine.ts +173 -173
- package/examples/extensions/doom-overlay/doom-keys.ts +104 -104
- package/examples/extensions/doom-overlay/index.ts +74 -74
- package/examples/extensions/doom-overlay/wad-finder.ts +51 -51
- package/examples/extensions/dynamic-resources/SKILL.md +8 -8
- package/examples/extensions/dynamic-resources/dynamic.json +79 -79
- package/examples/extensions/dynamic-resources/dynamic.md +5 -5
- package/examples/extensions/dynamic-resources/index.ts +15 -15
- package/examples/extensions/dynamic-tools.ts +74 -74
- package/examples/extensions/event-bus.ts +43 -43
- package/examples/extensions/file-trigger.ts +41 -41
- package/examples/extensions/git-checkpoint.ts +53 -53
- package/examples/extensions/git-merge-and-resolve.ts +115 -115
- package/examples/extensions/github-issue-autocomplete.ts +185 -185
- package/examples/extensions/gondolin/index.ts +531 -531
- package/examples/extensions/gondolin/package-lock.json +185 -185
- package/examples/extensions/gondolin/package.json +19 -19
- package/examples/extensions/handoff.ts +199 -199
- package/examples/extensions/hello.ts +26 -26
- package/examples/extensions/hidden-thinking-label.ts +53 -53
- package/examples/extensions/inline-bash.ts +94 -94
- package/examples/extensions/input-transform-streaming.ts +39 -39
- package/examples/extensions/input-transform.ts +43 -43
- package/examples/extensions/interactive-shell.ts +196 -196
- package/examples/extensions/mac-system-theme.ts +47 -47
- package/examples/extensions/message-renderer.ts +59 -59
- package/examples/extensions/minimal-mode.ts +426 -426
- package/examples/extensions/modal-editor.ts +85 -85
- package/examples/extensions/model-status.ts +31 -31
- package/examples/extensions/notify.ts +55 -55
- package/examples/extensions/overlay-qa-tests.ts +1450 -1450
- package/examples/extensions/overlay-test.ts +153 -153
- package/examples/extensions/permission-gate.ts +34 -34
- package/examples/extensions/pirate.ts +47 -47
- package/examples/extensions/plan-mode/README.md +66 -66
- package/examples/extensions/plan-mode/index.ts +390 -390
- package/examples/extensions/plan-mode/utils.ts +168 -168
- package/examples/extensions/preset.ts +436 -436
- package/examples/extensions/project-trust.ts +64 -64
- package/examples/extensions/prompt-customizer.ts +97 -97
- package/examples/extensions/protected-paths.ts +30 -30
- package/examples/extensions/provider-payload.ts +18 -18
- package/examples/extensions/qna.ts +122 -122
- package/examples/extensions/question.ts +285 -285
- package/examples/extensions/questionnaire.ts +448 -448
- package/examples/extensions/rainbow-editor.ts +88 -88
- package/examples/extensions/reload-runtime.ts +37 -37
- package/examples/extensions/rpc-demo.ts +118 -118
- package/examples/extensions/sandbox/index.ts +321 -321
- package/examples/extensions/sandbox/package-lock.json +92 -92
- package/examples/extensions/sandbox/package.json +19 -19
- package/examples/extensions/send-user-message.ts +97 -97
- package/examples/extensions/session-name.ts +27 -27
- package/examples/extensions/shutdown-command.ts +63 -63
- package/examples/extensions/snake.ts +343 -343
- package/examples/extensions/space-invaders.ts +560 -560
- package/examples/extensions/ssh.ts +220 -220
- package/examples/extensions/status-line.ts +32 -32
- package/examples/extensions/structured-output.ts +65 -65
- package/examples/extensions/subagent/README.md +175 -175
- package/examples/extensions/subagent/agents/planner.md +37 -37
- package/examples/extensions/subagent/agents/reviewer.md +35 -35
- package/examples/extensions/subagent/agents/scout.md +50 -50
- package/examples/extensions/subagent/agents/worker.md +24 -24
- package/examples/extensions/subagent/agents.ts +126 -126
- package/examples/extensions/subagent/index.ts +1015 -1015
- package/examples/extensions/subagent/prompts/implement-and-review.md +10 -10
- package/examples/extensions/subagent/prompts/implement.md +10 -10
- package/examples/extensions/subagent/prompts/scout-and-plan.md +9 -9
- package/examples/extensions/summarize.ts +209 -209
- package/examples/extensions/system-prompt-header.ts +17 -17
- package/examples/extensions/tic-tac-toe.ts +1008 -1008
- package/examples/extensions/timed-confirm.ts +70 -70
- package/examples/extensions/titlebar-spinner.ts +58 -58
- package/examples/extensions/todo.ts +297 -297
- package/examples/extensions/tool-override.ts +144 -144
- package/examples/extensions/tools.ts +146 -146
- package/examples/extensions/trigger-compact.ts +50 -50
- package/examples/extensions/truncated-tool.ts +195 -195
- package/examples/extensions/widget-placement.ts +9 -9
- package/examples/extensions/with-deps/index.ts +32 -32
- package/examples/extensions/with-deps/package-lock.json +31 -31
- package/examples/extensions/with-deps/package.json +22 -22
- package/examples/extensions/working-indicator.ts +123 -123
- package/examples/extensions/working-message-test.ts +25 -25
- package/examples/rpc-extension-ui.ts +632 -632
- package/examples/sdk/01-minimal.ts +26 -26
- package/examples/sdk/02-custom-model.ts +53 -53
- package/examples/sdk/03-custom-prompt.ts +75 -75
- package/examples/sdk/04-skills.ts +55 -55
- package/examples/sdk/05-tools.ts +48 -48
- package/examples/sdk/06-extensions.ts +99 -99
- package/examples/sdk/07-context-files.ts +47 -47
- package/examples/sdk/08-prompt-templates.ts +51 -51
- package/examples/sdk/09-api-keys-and-oauth.ts +52 -52
- package/examples/sdk/10-settings.ts +53 -53
- package/examples/sdk/11-sessions.ts +52 -52
- package/examples/sdk/12-full-control.ts +79 -79
- package/examples/sdk/13-session-runtime.ts +67 -67
- package/examples/sdk/README.md +144 -144
- package/package.json +4 -4
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { describe, expect, test } from "vitest";
|
|
2
|
+
import { createPowerShellTool } from "./powershell.js";
|
|
3
|
+
import { getPowerShellConfig, POWERSHELL_ARGS } from "../../utils/shell.js";
|
|
4
|
+
describe("powershell tool", () => {
|
|
5
|
+
test("uses a process-local execution policy bypass", () => {
|
|
6
|
+
expect(POWERSHELL_ARGS).toEqual(["-NoProfile", "-NonInteractive", "-ExecutionPolicy", "Bypass", "-Command"]);
|
|
7
|
+
});
|
|
8
|
+
test.skipIf(process.platform !== "win32")("executes commands with UTF-8 output", async () => {
|
|
9
|
+
expect(getPowerShellConfig().args).toEqual(POWERSHELL_ARGS);
|
|
10
|
+
const result = await createPowerShellTool(process.cwd()).execute("test", {
|
|
11
|
+
command: "Write-Output 'héllo €'; Get-ExecutionPolicy -Scope Process",
|
|
12
|
+
});
|
|
13
|
+
const output = result.content
|
|
14
|
+
.filter((item) => item.type === "text")
|
|
15
|
+
.map((item) => item.text ?? "")
|
|
16
|
+
.join("\n");
|
|
17
|
+
expect(output).toContain("héllo €");
|
|
18
|
+
expect(output).toContain("Bypass");
|
|
19
|
+
});
|
|
20
|
+
});
|
|
@@ -1,82 +1,82 @@
|
|
|
1
|
-
import { describe, expect, it, vi } from "vitest";
|
|
2
|
-
import type { ExtensionAPI, ExtensionContext } from "@selesai/code";
|
|
3
|
-
import contextCompactionReminderExtension, {
|
|
4
|
-
CONTEXT_COMPACTION_REMINDER_MESSAGE,
|
|
5
|
-
} from "./context-compaction-reminder.ts";
|
|
6
|
-
|
|
7
|
-
type Handler = (event: unknown, ctx: ExtensionContext) => void | Promise<void>;
|
|
8
|
-
|
|
9
|
-
function createHarness(tokens: number | null | undefined = undefined) {
|
|
10
|
-
const handlers = new Map<string, Handler>();
|
|
11
|
-
const notify = vi.fn();
|
|
12
|
-
const pi = {
|
|
13
|
-
on: vi.fn((event: string, handler: Handler) => handlers.set(event, handler)),
|
|
14
|
-
} as unknown as ExtensionAPI;
|
|
15
|
-
const ctx = {
|
|
16
|
-
getContextUsage: () =>
|
|
17
|
-
tokens === undefined ? undefined : { tokens, contextWindow: 200_000, percent: tokens === null ? null : tokens / 2_000 },
|
|
18
|
-
ui: { notify },
|
|
19
|
-
} as unknown as ExtensionContext;
|
|
20
|
-
|
|
21
|
-
contextCompactionReminderExtension(pi);
|
|
22
|
-
|
|
23
|
-
return {
|
|
24
|
-
notify,
|
|
25
|
-
setTokens: (value: number | null | undefined) => {
|
|
26
|
-
tokens = value;
|
|
27
|
-
},
|
|
28
|
-
settle: () => handlers.get("agent_settled")!({ type: "agent_settled" }, ctx),
|
|
29
|
-
compact: () => handlers.get("session_compact")!({ type: "session_compact" }, ctx),
|
|
30
|
-
};
|
|
31
|
-
}
|
|
32
|
-
|
|
33
|
-
describe("context-compaction-reminder", () => {
|
|
34
|
-
it("warns once when known context reaches 128k", async () => {
|
|
35
|
-
const { notify, setTokens, settle } = createHarness(127_999);
|
|
36
|
-
|
|
37
|
-
await settle();
|
|
38
|
-
setTokens(128_000);
|
|
39
|
-
await settle();
|
|
40
|
-
setTokens(130_000);
|
|
41
|
-
await settle();
|
|
42
|
-
|
|
43
|
-
expect(notify).toHaveBeenCalledTimes(1);
|
|
44
|
-
expect(notify).toHaveBeenCalledWith(CONTEXT_COMPACTION_REMINDER_MESSAGE, "warning");
|
|
45
|
-
});
|
|
46
|
-
|
|
47
|
-
it("does not reset the reminder for unavailable or post-compaction usage", async () => {
|
|
48
|
-
const { notify, setTokens, settle } = createHarness(128_000);
|
|
49
|
-
|
|
50
|
-
await settle();
|
|
51
|
-
setTokens(undefined);
|
|
52
|
-
await settle();
|
|
53
|
-
setTokens(null);
|
|
54
|
-
await settle();
|
|
55
|
-
setTokens(130_000);
|
|
56
|
-
await settle();
|
|
57
|
-
|
|
58
|
-
expect(notify).toHaveBeenCalledTimes(1);
|
|
59
|
-
});
|
|
60
|
-
|
|
61
|
-
it("allows another warning after successful compaction", async () => {
|
|
62
|
-
const { notify, settle, compact } = createHarness(128_000);
|
|
63
|
-
|
|
64
|
-
await settle();
|
|
65
|
-
await compact();
|
|
66
|
-
await settle();
|
|
67
|
-
|
|
68
|
-
expect(notify).toHaveBeenCalledTimes(2);
|
|
69
|
-
});
|
|
70
|
-
|
|
71
|
-
it("allows another warning after known usage drops below 128k", async () => {
|
|
72
|
-
const { notify, setTokens, settle } = createHarness(128_000);
|
|
73
|
-
|
|
74
|
-
await settle();
|
|
75
|
-
setTokens(127_999);
|
|
76
|
-
await settle();
|
|
77
|
-
setTokens(128_000);
|
|
78
|
-
await settle();
|
|
79
|
-
|
|
80
|
-
expect(notify).toHaveBeenCalledTimes(2);
|
|
81
|
-
});
|
|
82
|
-
});
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import type { ExtensionAPI, ExtensionContext } from "@selesai/code";
|
|
3
|
+
import contextCompactionReminderExtension, {
|
|
4
|
+
CONTEXT_COMPACTION_REMINDER_MESSAGE,
|
|
5
|
+
} from "./context-compaction-reminder.ts";
|
|
6
|
+
|
|
7
|
+
type Handler = (event: unknown, ctx: ExtensionContext) => void | Promise<void>;
|
|
8
|
+
|
|
9
|
+
function createHarness(tokens: number | null | undefined = undefined) {
|
|
10
|
+
const handlers = new Map<string, Handler>();
|
|
11
|
+
const notify = vi.fn();
|
|
12
|
+
const pi = {
|
|
13
|
+
on: vi.fn((event: string, handler: Handler) => handlers.set(event, handler)),
|
|
14
|
+
} as unknown as ExtensionAPI;
|
|
15
|
+
const ctx = {
|
|
16
|
+
getContextUsage: () =>
|
|
17
|
+
tokens === undefined ? undefined : { tokens, contextWindow: 200_000, percent: tokens === null ? null : tokens / 2_000 },
|
|
18
|
+
ui: { notify },
|
|
19
|
+
} as unknown as ExtensionContext;
|
|
20
|
+
|
|
21
|
+
contextCompactionReminderExtension(pi);
|
|
22
|
+
|
|
23
|
+
return {
|
|
24
|
+
notify,
|
|
25
|
+
setTokens: (value: number | null | undefined) => {
|
|
26
|
+
tokens = value;
|
|
27
|
+
},
|
|
28
|
+
settle: () => handlers.get("agent_settled")!({ type: "agent_settled" }, ctx),
|
|
29
|
+
compact: () => handlers.get("session_compact")!({ type: "session_compact" }, ctx),
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
describe("context-compaction-reminder", () => {
|
|
34
|
+
it("warns once when known context reaches 128k", async () => {
|
|
35
|
+
const { notify, setTokens, settle } = createHarness(127_999);
|
|
36
|
+
|
|
37
|
+
await settle();
|
|
38
|
+
setTokens(128_000);
|
|
39
|
+
await settle();
|
|
40
|
+
setTokens(130_000);
|
|
41
|
+
await settle();
|
|
42
|
+
|
|
43
|
+
expect(notify).toHaveBeenCalledTimes(1);
|
|
44
|
+
expect(notify).toHaveBeenCalledWith(CONTEXT_COMPACTION_REMINDER_MESSAGE, "warning");
|
|
45
|
+
});
|
|
46
|
+
|
|
47
|
+
it("does not reset the reminder for unavailable or post-compaction usage", async () => {
|
|
48
|
+
const { notify, setTokens, settle } = createHarness(128_000);
|
|
49
|
+
|
|
50
|
+
await settle();
|
|
51
|
+
setTokens(undefined);
|
|
52
|
+
await settle();
|
|
53
|
+
setTokens(null);
|
|
54
|
+
await settle();
|
|
55
|
+
setTokens(130_000);
|
|
56
|
+
await settle();
|
|
57
|
+
|
|
58
|
+
expect(notify).toHaveBeenCalledTimes(1);
|
|
59
|
+
});
|
|
60
|
+
|
|
61
|
+
it("allows another warning after successful compaction", async () => {
|
|
62
|
+
const { notify, settle, compact } = createHarness(128_000);
|
|
63
|
+
|
|
64
|
+
await settle();
|
|
65
|
+
await compact();
|
|
66
|
+
await settle();
|
|
67
|
+
|
|
68
|
+
expect(notify).toHaveBeenCalledTimes(2);
|
|
69
|
+
});
|
|
70
|
+
|
|
71
|
+
it("allows another warning after known usage drops below 128k", async () => {
|
|
72
|
+
const { notify, setTokens, settle } = createHarness(128_000);
|
|
73
|
+
|
|
74
|
+
await settle();
|
|
75
|
+
setTokens(127_999);
|
|
76
|
+
await settle();
|
|
77
|
+
setTokens(128_000);
|
|
78
|
+
await settle();
|
|
79
|
+
|
|
80
|
+
expect(notify).toHaveBeenCalledTimes(2);
|
|
81
|
+
});
|
|
82
|
+
});
|
|
@@ -1,28 +1,28 @@
|
|
|
1
|
-
import type { ExtensionAPI } from "@selesai/code";
|
|
2
|
-
|
|
3
|
-
export const CONTEXT_COMPACTION_REMINDER_THRESHOLD = 128_000;
|
|
4
|
-
export const CONTEXT_COMPACTION_REMINDER_MESSAGE =
|
|
5
|
-
`Conversation context has reached ${CONTEXT_COMPACTION_REMINDER_THRESHOLD.toLocaleString()} tokens. Run /handoff-new to summarize into a new chat, or enable Auto Handoff in Settings to do this automatically. Bigger context windows eat up tokens quickly and degrade AI quality.`;
|
|
6
|
-
|
|
7
|
-
export default function contextCompactionReminderExtension(pi: ExtensionAPI): void {
|
|
8
|
-
let reminderShown = false;
|
|
9
|
-
|
|
10
|
-
pi.on("session_compact", () => {
|
|
11
|
-
reminderShown = false;
|
|
12
|
-
});
|
|
13
|
-
|
|
14
|
-
pi.on("agent_settled", (_event, ctx) => {
|
|
15
|
-
const tokens = ctx.getContextUsage()?.tokens;
|
|
16
|
-
if (tokens === undefined || tokens === null) return;
|
|
17
|
-
|
|
18
|
-
if (tokens < CONTEXT_COMPACTION_REMINDER_THRESHOLD) {
|
|
19
|
-
reminderShown = false;
|
|
20
|
-
return;
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
if (!reminderShown) {
|
|
24
|
-
ctx.ui.notify(CONTEXT_COMPACTION_REMINDER_MESSAGE, "warning");
|
|
25
|
-
reminderShown = true;
|
|
26
|
-
}
|
|
27
|
-
});
|
|
28
|
-
}
|
|
1
|
+
import type { ExtensionAPI } from "@selesai/code";
|
|
2
|
+
|
|
3
|
+
export const CONTEXT_COMPACTION_REMINDER_THRESHOLD = 128_000;
|
|
4
|
+
export const CONTEXT_COMPACTION_REMINDER_MESSAGE =
|
|
5
|
+
`Conversation context has reached ${CONTEXT_COMPACTION_REMINDER_THRESHOLD.toLocaleString()} tokens. Run /handoff-new to summarize into a new chat, or enable Auto Handoff in Settings to do this automatically. Bigger context windows eat up tokens quickly and degrade AI quality.`;
|
|
6
|
+
|
|
7
|
+
export default function contextCompactionReminderExtension(pi: ExtensionAPI): void {
|
|
8
|
+
let reminderShown = false;
|
|
9
|
+
|
|
10
|
+
pi.on("session_compact", () => {
|
|
11
|
+
reminderShown = false;
|
|
12
|
+
});
|
|
13
|
+
|
|
14
|
+
pi.on("agent_settled", (_event, ctx) => {
|
|
15
|
+
const tokens = ctx.getContextUsage()?.tokens;
|
|
16
|
+
if (tokens === undefined || tokens === null) return;
|
|
17
|
+
|
|
18
|
+
if (tokens < CONTEXT_COMPACTION_REMINDER_THRESHOLD) {
|
|
19
|
+
reminderShown = false;
|
|
20
|
+
return;
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
if (!reminderShown) {
|
|
24
|
+
ctx.ui.notify(CONTEXT_COMPACTION_REMINDER_MESSAGE, "warning");
|
|
25
|
+
reminderShown = true;
|
|
26
|
+
}
|
|
27
|
+
});
|
|
28
|
+
}
|
|
@@ -31,7 +31,7 @@ function createHarness() {
|
|
|
31
31
|
const sent: AnyMessage[] = [];
|
|
32
32
|
const pi = {
|
|
33
33
|
on: vi.fn((event: string, handler: Handler) => {
|
|
34
|
-
handlers.set(event, handler);
|
|
34
|
+
handlers.set(event, (e: any, c: any) => handler(e, { mode: "tui", ...(c ?? {}) }));
|
|
35
35
|
}),
|
|
36
36
|
registerCommand: vi.fn((name: string, opts: Command) => {
|
|
37
37
|
commands.set(name, opts);
|
|
@@ -155,6 +155,29 @@ describe("copy-turn hash stability", () => {
|
|
|
155
155
|
expect(sent).toHaveLength(0);
|
|
156
156
|
});
|
|
157
157
|
|
|
158
|
+
it("non-tui modes skip copy markers and copy rows", async () => {
|
|
159
|
+
const { handlers, sent } = createHarness();
|
|
160
|
+
const messageEnd = handlers.get("message_end")!;
|
|
161
|
+
|
|
162
|
+
const rpcCtx = { mode: "rpc" };
|
|
163
|
+
const assistant = await messageEnd(
|
|
164
|
+
{ type: "message_end", message: { role: "assistant", content: [{ type: "text", text: "result text" }] } },
|
|
165
|
+
rpcCtx,
|
|
166
|
+
);
|
|
167
|
+
expect(assistant).toBeUndefined(); // no ⧉ copy marker appended
|
|
168
|
+
expect(sent).toHaveLength(0); // no user copy row either
|
|
169
|
+
|
|
170
|
+
await messageEnd({ type: "message_end", message: { role: "user", content: "ask me" } }, rpcCtx);
|
|
171
|
+
expect(sent).toHaveLength(0);
|
|
172
|
+
|
|
173
|
+
// TUI mode still annotates as before.
|
|
174
|
+
const tuiResult = await messageEnd(
|
|
175
|
+
{ type: "message_end", message: { role: "assistant", content: "tui text" } },
|
|
176
|
+
{ mode: "tui" },
|
|
177
|
+
);
|
|
178
|
+
expect(tuiResult.message.content).toMatch(/⧉ copy assistant: \/cp [0-9a-f]{6}/);
|
|
179
|
+
});
|
|
180
|
+
|
|
158
181
|
it("marks string-content assistant messages and bashExecution results", async () => {
|
|
159
182
|
const { handlers } = createHarness();
|
|
160
183
|
const messageEnd = handlers.get("message_end")!;
|
|
@@ -90,7 +90,12 @@ export default function (pi: ExtensionAPI) {
|
|
|
90
90
|
return new Text(theme.fg("dim", `⧉ copy ${role}: /cp ${h}`), 0, 0);
|
|
91
91
|
});
|
|
92
92
|
|
|
93
|
-
pi.on("message_end", async (event: any) => {
|
|
93
|
+
pi.on("message_end", async (event: any, ctx: any) => {
|
|
94
|
+
// The ⧉ copy markers are a terminal affordance: the TUI has no copy
|
|
95
|
+
// buttons, so each message is annotated with a /cp hash. Hosts like the
|
|
96
|
+
// VS Code extension run in non-TUI modes and ship their own copy UI,
|
|
97
|
+
// where these markers are just noise.
|
|
98
|
+
if (ctx?.mode !== "tui") return;
|
|
94
99
|
const m = event.message;
|
|
95
100
|
if (!m) return;
|
|
96
101
|
|
|
@@ -191,12 +191,25 @@ describe("handoff-new extension", () => {
|
|
|
191
191
|
expect(calls.userMessage).toBe("HANDOFF PROMPT");
|
|
192
192
|
});
|
|
193
193
|
|
|
194
|
-
it("non-tui mode
|
|
194
|
+
it("non-tui mode generates directly and opens new session", async () => {
|
|
195
|
+
(complete as any).mockResolvedValueOnce({
|
|
196
|
+
stopReason: "done",
|
|
197
|
+
content: [{ type: "text", text: "HANDOFF PROMPT" }],
|
|
198
|
+
});
|
|
199
|
+
const { commands } = createPiHarness();
|
|
200
|
+
const { ctx, calls } = createCtx({ mode: "rpc" });
|
|
201
|
+
await commands.get("handoff-new")!.handler("goal", ctx);
|
|
202
|
+
expect(calls.userMessage).toBe("HANDOFF PROMPT");
|
|
203
|
+
expect(calls.newSession).not.toBeNull();
|
|
204
|
+
});
|
|
205
|
+
|
|
206
|
+
it("non-tui mode generation error notifies and opens no session", async () => {
|
|
207
|
+
(complete as any).mockRejectedValueOnce(new Error("api down"));
|
|
195
208
|
const { commands } = createPiHarness();
|
|
196
209
|
const { ctx, calls } = createCtx({ mode: "rpc" });
|
|
197
210
|
await commands.get("handoff-new")!.handler("goal", ctx);
|
|
198
211
|
expect(calls.newSession).toBeNull();
|
|
199
|
-
expect(calls.notify.some((n) => /
|
|
212
|
+
expect(calls.notify.some((n) => /api down/.test(n.msg))).toBe(true);
|
|
200
213
|
});
|
|
201
214
|
|
|
202
215
|
it("no model fails cleanly", async () => {
|
|
@@ -33,11 +33,28 @@ export default function (pi: ExtensionAPI) {
|
|
|
33
33
|
});
|
|
34
34
|
}
|
|
35
35
|
|
|
36
|
-
async function
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
36
|
+
async function generateHandoffText(
|
|
37
|
+
ctx: ExtensionCommandContext,
|
|
38
|
+
aiContext: ReturnType<typeof buildAiContext>,
|
|
39
|
+
signal?: AbortSignal,
|
|
40
|
+
): Promise<string | null> {
|
|
41
|
+
const auth = await ctx.modelRegistry.getApiKeyAndHeaders(ctx.model!);
|
|
42
|
+
if (!auth.ok || !auth.apiKey) {
|
|
43
|
+
throw new Error(auth.ok ? `No API key for ${ctx.model!.provider}` : auth.error);
|
|
40
44
|
}
|
|
45
|
+
const response = await complete(ctx.model!, aiContext, {
|
|
46
|
+
apiKey: auth.apiKey,
|
|
47
|
+
headers: auth.headers,
|
|
48
|
+
signal,
|
|
49
|
+
});
|
|
50
|
+
if (response.stopReason === "aborted") return null;
|
|
51
|
+
return response.content
|
|
52
|
+
.filter((c): c is { type: "text"; text: string } => c.type === "text")
|
|
53
|
+
.map((c) => c.text)
|
|
54
|
+
.join("\n");
|
|
55
|
+
}
|
|
56
|
+
|
|
57
|
+
async function handoffNew(args: string, ctx: ExtensionCommandContext) {
|
|
41
58
|
if (!ctx.model) {
|
|
42
59
|
ctx.ui.notify("No model selected", "error");
|
|
43
60
|
return;
|
|
@@ -56,34 +73,31 @@ async function handoffNew(args: string, ctx: ExtensionCommandContext) {
|
|
|
56
73
|
const currentSessionFile = ctx.sessionManager.getSessionFile();
|
|
57
74
|
const aiContext = buildAiContext(conversationText, goal);
|
|
58
75
|
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
76
|
+
let result: string | null;
|
|
77
|
+
if (ctx.mode === "tui") {
|
|
78
|
+
result = await ctx.ui.custom<string | null>((tui, theme, _kb, done) => {
|
|
79
|
+
const loader = new BorderedLoader(tui, theme, `Generating handoff prompt...`);
|
|
80
|
+
loader.onAbort = () => done(null);
|
|
62
81
|
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
apiKey: auth.apiKey,
|
|
70
|
-
headers: auth.headers,
|
|
71
|
-
signal: loader.signal,
|
|
72
|
-
});
|
|
73
|
-
if (response.stopReason === "aborted") return null;
|
|
74
|
-
return response.content
|
|
75
|
-
.filter((c): c is { type: "text"; text: string } => c.type === "text")
|
|
76
|
-
.map((c) => c.text)
|
|
77
|
-
.join("\n");
|
|
78
|
-
})()
|
|
79
|
-
.then(done)
|
|
80
|
-
.catch((err) => {
|
|
81
|
-
console.error("handoff-new generation failed:", err);
|
|
82
|
-
done(null);
|
|
83
|
-
});
|
|
82
|
+
generateHandoffText(ctx, aiContext, loader.signal)
|
|
83
|
+
.then(done)
|
|
84
|
+
.catch((err) => {
|
|
85
|
+
console.error("handoff-new generation failed:", err);
|
|
86
|
+
done(null);
|
|
87
|
+
});
|
|
84
88
|
|
|
85
|
-
|
|
86
|
-
|
|
89
|
+
return loader;
|
|
90
|
+
});
|
|
91
|
+
} else {
|
|
92
|
+
// Non-interactive modes (RPC for the VS Code extension, print/json):
|
|
93
|
+
// no TUI loader is available, so generate directly.
|
|
94
|
+
try {
|
|
95
|
+
result = await generateHandoffText(ctx, aiContext, ctx.signal);
|
|
96
|
+
} catch (err) {
|
|
97
|
+
ctx.ui.notify(err instanceof Error ? err.message : String(err), "error");
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
87
101
|
|
|
88
102
|
if (result === null) {
|
|
89
103
|
ctx.ui.notify("Cancelled", "info");
|
|
@@ -1,21 +1,21 @@
|
|
|
1
|
-
MIT License
|
|
2
|
-
|
|
3
|
-
Copyright (c) 2026 Nico Bailon
|
|
4
|
-
|
|
5
|
-
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
-
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
-
in the Software without restriction, including without limitation the rights
|
|
8
|
-
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
-
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
-
furnished to do so, subject to the following conditions:
|
|
11
|
-
|
|
12
|
-
The above copyright notice and this permission notice shall be included in all
|
|
13
|
-
copies or substantial portions of the Software.
|
|
14
|
-
|
|
15
|
-
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
-
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
-
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
-
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
-
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
-
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
-
SOFTWARE.
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 Nico Bailon
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
|
@@ -1,83 +1,83 @@
|
|
|
1
|
-
import test from "node:test";
|
|
2
|
-
import assert from "node:assert/strict";
|
|
3
|
-
import { IntercomClient } from "./client.ts";
|
|
4
|
-
|
|
5
|
-
test("validated session lifecycle messages reach broker-message subscribers", () => {
|
|
6
|
-
const client = new IntercomClient();
|
|
7
|
-
(client as any)._sessionId = "session-1";
|
|
8
|
-
const received: unknown[] = [];
|
|
9
|
-
client.onBrokerMessage((message) => received.push(message));
|
|
10
|
-
const session = {
|
|
11
|
-
id: "session-2",
|
|
12
|
-
cwd: "/test",
|
|
13
|
-
model: "test",
|
|
14
|
-
pid: 2,
|
|
15
|
-
startedAt: 1,
|
|
16
|
-
lastActivity: 1,
|
|
17
|
-
};
|
|
18
|
-
|
|
19
|
-
(client as any).handleBrokerMessage({ type: "session_joined", session });
|
|
20
|
-
(client as any).handleBrokerMessage({ type: "presence_update", session });
|
|
21
|
-
(client as any).handleBrokerMessage({ type: "session_left", sessionId: "session-2" });
|
|
22
|
-
|
|
23
|
-
assert.deepEqual(received, [
|
|
24
|
-
{ type: "session_joined", session },
|
|
25
|
-
{ type: "presence_update", session },
|
|
26
|
-
{ type: "session_left", sessionId: "session-2" },
|
|
27
|
-
]);
|
|
28
|
-
});
|
|
29
|
-
|
|
30
|
-
test("registered feature negotiation rejects non-string feature entries", () => {
|
|
31
|
-
const client = new IntercomClient();
|
|
32
|
-
assert.throws(
|
|
33
|
-
() => (client as any).handleBrokerMessage({ type: "registered", sessionId: "session-1", features: ["valid", 123] }),
|
|
34
|
-
/Invalid registered features/,
|
|
35
|
-
);
|
|
36
|
-
});
|
|
37
|
-
|
|
38
|
-
test("malformed extension broker messages are rejected", () => {
|
|
39
|
-
const client = new IntercomClient();
|
|
40
|
-
(client as any)._sessionId = "session-1";
|
|
41
|
-
|
|
42
|
-
assert.throws(
|
|
43
|
-
() => (client as any).handleBrokerMessage({ type: "extension_owner", namespace: "test/v1", ownerId: "owner" }),
|
|
44
|
-
/Invalid extension_owner/,
|
|
45
|
-
);
|
|
46
|
-
assert.throws(
|
|
47
|
-
() => (client as any).handleBrokerMessage({ type: "extension_owner", namespace: "test/v1", ownerEpoch: "epoch" }),
|
|
48
|
-
/Invalid extension_owner/,
|
|
49
|
-
);
|
|
50
|
-
assert.throws(
|
|
51
|
-
() => (client as any).handleBrokerMessage({ type: "extension_message", namespace: "test/v1" }),
|
|
52
|
-
/Invalid extension_message/,
|
|
53
|
-
);
|
|
54
|
-
assert.throws(
|
|
55
|
-
() => (client as any).handleBrokerMessage({ type: "extension_state", namespace: "test/v1", revision: -1 }),
|
|
56
|
-
/Invalid extension_state/,
|
|
57
|
-
);
|
|
58
|
-
assert.throws(
|
|
59
|
-
() => (client as any).handleBrokerMessage({ type: "extension_state_result", namespace: "test/v1", committed: "yes", revision: 1 }),
|
|
60
|
-
/Invalid extension_state_result/,
|
|
61
|
-
);
|
|
62
|
-
assert.doesNotThrow(() => (client as any).handleBrokerMessage({
|
|
63
|
-
type: "extension_message",
|
|
64
|
-
namespace: "test/v1",
|
|
65
|
-
fromSessionId: "session-2",
|
|
66
|
-
payload: { peerOnly: true },
|
|
67
|
-
}));
|
|
68
|
-
});
|
|
69
|
-
|
|
70
|
-
test("cancelAsk ignores synchronous socket write failures", () => {
|
|
71
|
-
const client = new IntercomClient();
|
|
72
|
-
(client as any)._sessionId = "session-1";
|
|
73
|
-
(client as any).socket = {
|
|
74
|
-
destroyed: false,
|
|
75
|
-
writableEnded: false,
|
|
76
|
-
writable: true,
|
|
77
|
-
write() {
|
|
78
|
-
throw new Error("write failed");
|
|
79
|
-
},
|
|
80
|
-
};
|
|
81
|
-
|
|
82
|
-
assert.doesNotThrow(() => client.cancelAsk("ask-1"));
|
|
83
|
-
});
|
|
1
|
+
import test from "node:test";
|
|
2
|
+
import assert from "node:assert/strict";
|
|
3
|
+
import { IntercomClient } from "./client.ts";
|
|
4
|
+
|
|
5
|
+
test("validated session lifecycle messages reach broker-message subscribers", () => {
|
|
6
|
+
const client = new IntercomClient();
|
|
7
|
+
(client as any)._sessionId = "session-1";
|
|
8
|
+
const received: unknown[] = [];
|
|
9
|
+
client.onBrokerMessage((message) => received.push(message));
|
|
10
|
+
const session = {
|
|
11
|
+
id: "session-2",
|
|
12
|
+
cwd: "/test",
|
|
13
|
+
model: "test",
|
|
14
|
+
pid: 2,
|
|
15
|
+
startedAt: 1,
|
|
16
|
+
lastActivity: 1,
|
|
17
|
+
};
|
|
18
|
+
|
|
19
|
+
(client as any).handleBrokerMessage({ type: "session_joined", session });
|
|
20
|
+
(client as any).handleBrokerMessage({ type: "presence_update", session });
|
|
21
|
+
(client as any).handleBrokerMessage({ type: "session_left", sessionId: "session-2" });
|
|
22
|
+
|
|
23
|
+
assert.deepEqual(received, [
|
|
24
|
+
{ type: "session_joined", session },
|
|
25
|
+
{ type: "presence_update", session },
|
|
26
|
+
{ type: "session_left", sessionId: "session-2" },
|
|
27
|
+
]);
|
|
28
|
+
});
|
|
29
|
+
|
|
30
|
+
test("registered feature negotiation rejects non-string feature entries", () => {
|
|
31
|
+
const client = new IntercomClient();
|
|
32
|
+
assert.throws(
|
|
33
|
+
() => (client as any).handleBrokerMessage({ type: "registered", sessionId: "session-1", features: ["valid", 123] }),
|
|
34
|
+
/Invalid registered features/,
|
|
35
|
+
);
|
|
36
|
+
});
|
|
37
|
+
|
|
38
|
+
test("malformed extension broker messages are rejected", () => {
|
|
39
|
+
const client = new IntercomClient();
|
|
40
|
+
(client as any)._sessionId = "session-1";
|
|
41
|
+
|
|
42
|
+
assert.throws(
|
|
43
|
+
() => (client as any).handleBrokerMessage({ type: "extension_owner", namespace: "test/v1", ownerId: "owner" }),
|
|
44
|
+
/Invalid extension_owner/,
|
|
45
|
+
);
|
|
46
|
+
assert.throws(
|
|
47
|
+
() => (client as any).handleBrokerMessage({ type: "extension_owner", namespace: "test/v1", ownerEpoch: "epoch" }),
|
|
48
|
+
/Invalid extension_owner/,
|
|
49
|
+
);
|
|
50
|
+
assert.throws(
|
|
51
|
+
() => (client as any).handleBrokerMessage({ type: "extension_message", namespace: "test/v1" }),
|
|
52
|
+
/Invalid extension_message/,
|
|
53
|
+
);
|
|
54
|
+
assert.throws(
|
|
55
|
+
() => (client as any).handleBrokerMessage({ type: "extension_state", namespace: "test/v1", revision: -1 }),
|
|
56
|
+
/Invalid extension_state/,
|
|
57
|
+
);
|
|
58
|
+
assert.throws(
|
|
59
|
+
() => (client as any).handleBrokerMessage({ type: "extension_state_result", namespace: "test/v1", committed: "yes", revision: 1 }),
|
|
60
|
+
/Invalid extension_state_result/,
|
|
61
|
+
);
|
|
62
|
+
assert.doesNotThrow(() => (client as any).handleBrokerMessage({
|
|
63
|
+
type: "extension_message",
|
|
64
|
+
namespace: "test/v1",
|
|
65
|
+
fromSessionId: "session-2",
|
|
66
|
+
payload: { peerOnly: true },
|
|
67
|
+
}));
|
|
68
|
+
});
|
|
69
|
+
|
|
70
|
+
test("cancelAsk ignores synchronous socket write failures", () => {
|
|
71
|
+
const client = new IntercomClient();
|
|
72
|
+
(client as any)._sessionId = "session-1";
|
|
73
|
+
(client as any).socket = {
|
|
74
|
+
destroyed: false,
|
|
75
|
+
writableEnded: false,
|
|
76
|
+
writable: true,
|
|
77
|
+
write() {
|
|
78
|
+
throw new Error("write failed");
|
|
79
|
+
},
|
|
80
|
+
};
|
|
81
|
+
|
|
82
|
+
assert.doesNotThrow(() => client.cancelAsk("ask-1"));
|
|
83
|
+
});
|