@ai-devkit/agent-manager 0.37.0 → 0.38.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/dist/AgentManager.d.ts +0 -4
- package/dist/AgentManager.d.ts.map +1 -1
- package/dist/AgentManager.js +2 -2
- package/dist/AgentManager.js.map +1 -1
- package/dist/__tests__/capacity/anthropic.test.js +115 -0
- package/dist/__tests__/capacity/anthropic.test.js.map +1 -0
- package/dist/__tests__/capacity/boundaries.test.js +45 -0
- package/dist/__tests__/capacity/boundaries.test.js.map +1 -0
- package/dist/__tests__/capacity/claude.test.js +3 -3
- package/dist/__tests__/capacity/claude.test.js.map +1 -1
- package/dist/__tests__/capacity/codex.test.js +3 -1
- package/dist/__tests__/capacity/codex.test.js.map +1 -1
- package/dist/__tests__/capacity/openai.test.js +21 -12
- package/dist/__tests__/capacity/openai.test.js.map +1 -1
- package/dist/__tests__/capacity/wham.test.js +1 -1
- package/dist/__tests__/capacity/wham.test.js.map +1 -1
- package/dist/__tests__/capacity/zai.test.js +7 -8
- package/dist/__tests__/capacity/zai.test.js.map +1 -1
- package/dist/__tests__/harnesses/claude/durable/ClaudePrintAgent.integration.test.js +2 -1
- package/dist/__tests__/harnesses/claude/durable/ClaudePrintAgent.integration.test.js.map +1 -1
- package/dist/__tests__/harnesses/claude/durable/ClaudePrintRunner.test.js.map +1 -1
- package/dist/__tests__/harnesses/codex/durable/CodexCliProbe.test.js +7 -15
- package/dist/__tests__/harnesses/codex/durable/CodexCliProbe.test.js.map +1 -1
- package/dist/__tests__/harnesses/codex/durable/CodexPrintAgent.integration.test.js +3 -1
- package/dist/__tests__/harnesses/codex/durable/CodexPrintAgent.integration.test.js.map +1 -1
- package/dist/__tests__/harnesses/codex/durable/CodexPrintAgentService.test.js +11 -26
- package/dist/__tests__/harnesses/codex/durable/CodexPrintAgentService.test.js.map +1 -1
- package/dist/__tests__/harnesses/codex/durable/CodexPrintRunner.test.js +3 -7
- package/dist/__tests__/harnesses/codex/durable/CodexPrintRunner.test.js.map +1 -1
- package/dist/__tests__/harnesses/pi/durable/PiCliProbe.test.js +5 -11
- package/dist/__tests__/harnesses/pi/durable/PiCliProbe.test.js.map +1 -1
- package/dist/__tests__/harnesses/pi/durable/PiPrintAgentService.test.js +9 -19
- package/dist/__tests__/harnesses/pi/durable/PiPrintAgentService.test.js.map +1 -1
- package/dist/__tests__/harnesses/pi/durable/PiPrintRunner.test.js +3 -7
- package/dist/__tests__/harnesses/pi/durable/PiPrintRunner.test.js.map +1 -1
- package/dist/__tests__/{readiness → harnesses/readiness}/AgentReadiness.test.js +31 -10
- package/dist/__tests__/harnesses/readiness/AgentReadiness.test.js.map +1 -0
- package/dist/__tests__/harnesses/runtimeProfiles.test.js +89 -0
- package/dist/__tests__/harnesses/runtimeProfiles.test.js.map +1 -0
- package/dist/__tests__/print/DurableAgent.test.js +4 -6
- package/dist/__tests__/print/DurableAgent.test.js.map +1 -1
- package/dist/__tests__/print/DurableAgentRepository.test.js +1 -12
- package/dist/__tests__/print/DurableAgentRepository.test.js.map +1 -1
- package/dist/__tests__/runtime/ManagedAgentRuntime.test.js.map +1 -1
- package/dist/__tests__/terminal/TmuxManager.test.js +1 -1
- package/dist/__tests__/terminal/TmuxManager.test.js.map +1 -1
- package/dist/capacity/index.d.ts +9 -19
- package/dist/capacity/index.d.ts.map +1 -1
- package/dist/capacity/index.js +11 -4
- package/dist/capacity/index.js.map +1 -1
- package/dist/capacity/providers/anthropic.d.ts +9 -0
- package/dist/capacity/providers/anthropic.d.ts.map +1 -0
- package/dist/capacity/{claude.js → providers/anthropic.js} +10 -128
- package/dist/capacity/providers/anthropic.js.map +1 -0
- package/dist/capacity/providers/openai-platform.d.ts +14 -0
- package/dist/capacity/providers/openai-platform.d.ts.map +1 -0
- package/dist/capacity/providers/openai-platform.js +120 -0
- package/dist/capacity/providers/openai-platform.js.map +1 -0
- package/dist/capacity/providers/openai-subscription.d.ts +20 -0
- package/dist/capacity/providers/openai-subscription.d.ts.map +1 -0
- package/dist/capacity/providers/openai-subscription.js +129 -0
- package/dist/capacity/providers/openai-subscription.js.map +1 -0
- package/dist/capacity/providers/zai.d.ts +8 -0
- package/dist/capacity/providers/zai.d.ts.map +1 -0
- package/dist/capacity/{zai.js → providers/zai.js} +17 -62
- package/dist/capacity/providers/zai.js.map +1 -0
- package/dist/capacity/sources/claude.d.ts +9 -0
- package/dist/capacity/sources/claude.d.ts.map +1 -0
- package/dist/capacity/sources/claude.js +17 -0
- package/dist/capacity/sources/claude.js.map +1 -0
- package/dist/capacity/{codex.d.ts → sources/codex.d.ts} +5 -9
- package/dist/capacity/sources/codex.d.ts.map +1 -0
- package/dist/capacity/{codex.js → sources/codex.js} +31 -84
- package/dist/capacity/sources/codex.js.map +1 -0
- package/dist/capacity/sources/pi.d.ts +30 -0
- package/dist/capacity/sources/pi.d.ts.map +1 -0
- package/dist/capacity/sources/pi.js +81 -0
- package/dist/capacity/sources/pi.js.map +1 -0
- package/dist/capacity/types.d.ts +2 -0
- package/dist/capacity/types.d.ts.map +1 -1
- package/dist/capacity/types.js +1 -1
- package/dist/capacity/types.js.map +1 -1
- package/dist/harnesses/claude/credentials.d.ts +10 -0
- package/dist/harnesses/claude/credentials.d.ts.map +1 -0
- package/dist/harnesses/claude/credentials.js +112 -0
- package/dist/harnesses/claude/credentials.js.map +1 -0
- package/dist/harnesses/claude/readiness.d.ts +3 -0
- package/dist/harnesses/claude/readiness.d.ts.map +1 -0
- package/dist/harnesses/claude/readiness.js +63 -0
- package/dist/harnesses/claude/readiness.js.map +1 -0
- package/dist/harnesses/codex/credentials.d.ts +15 -0
- package/dist/harnesses/codex/credentials.d.ts.map +1 -0
- package/dist/harnesses/codex/credentials.js +58 -0
- package/dist/harnesses/codex/credentials.js.map +1 -0
- package/dist/harnesses/codex/readiness.d.ts +3 -0
- package/dist/harnesses/codex/readiness.d.ts.map +1 -0
- package/dist/harnesses/codex/readiness.js +60 -0
- package/dist/harnesses/codex/readiness.js.map +1 -0
- package/dist/harnesses/copilot/readiness.d.ts +3 -0
- package/dist/harnesses/copilot/readiness.d.ts.map +1 -0
- package/dist/harnesses/copilot/readiness.js +39 -0
- package/dist/harnesses/copilot/readiness.js.map +1 -0
- package/dist/harnesses/opencode/readiness.d.ts +3 -0
- package/dist/harnesses/opencode/readiness.d.ts.map +1 -0
- package/dist/harnesses/opencode/readiness.js +50 -0
- package/dist/harnesses/opencode/readiness.js.map +1 -0
- package/dist/harnesses/pi/credentials.d.ts +22 -0
- package/dist/harnesses/pi/credentials.d.ts.map +1 -0
- package/dist/harnesses/pi/credentials.js +98 -0
- package/dist/harnesses/pi/credentials.js.map +1 -0
- package/dist/harnesses/pi/readiness.d.ts +3 -0
- package/dist/harnesses/pi/readiness.d.ts.map +1 -0
- package/dist/harnesses/pi/readiness.js +96 -0
- package/dist/harnesses/pi/readiness.js.map +1 -0
- package/dist/harnesses/readiness/AgentReadiness.d.ts +6 -0
- package/dist/harnesses/readiness/AgentReadiness.d.ts.map +1 -0
- package/dist/harnesses/readiness/AgentReadiness.js +68 -0
- package/dist/harnesses/readiness/AgentReadiness.js.map +1 -0
- package/dist/harnesses/readiness/checks.d.ts +31 -0
- package/dist/harnesses/readiness/checks.d.ts.map +1 -0
- package/dist/harnesses/readiness/checks.js +270 -0
- package/dist/harnesses/readiness/checks.js.map +1 -0
- package/dist/{readiness/AgentReadiness.d.ts → harnesses/readiness/types.d.ts} +19 -10
- package/dist/harnesses/readiness/types.d.ts.map +1 -0
- package/dist/harnesses/readiness/types.js +4 -0
- package/dist/harnesses/readiness/types.js.map +1 -0
- package/dist/harnesses/runtimeProfiles.d.ts +11 -0
- package/dist/harnesses/runtimeProfiles.d.ts.map +1 -0
- package/dist/harnesses/runtimeProfiles.js +99 -0
- package/dist/harnesses/runtimeProfiles.js.map +1 -0
- package/dist/index.d.ts +16 -42
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +9 -23
- package/dist/index.js.map +1 -1
- package/dist/runtime/ManagedAgentRuntime.d.ts +3 -3
- package/dist/runtime/ManagedAgentRuntime.d.ts.map +1 -1
- package/dist/runtime/ManagedAgentRuntime.js +6 -20
- package/dist/runtime/ManagedAgentRuntime.js.map +1 -1
- package/dist/runtime/herdr/HerdrAgentDiscovery.d.ts.map +1 -1
- package/dist/runtime/herdr/HerdrAgentDiscovery.js +2 -43
- package/dist/runtime/herdr/HerdrAgentDiscovery.js.map +1 -1
- package/dist/runtime/tmux/PidPolling.d.ts +1 -1
- package/dist/runtime/tmux/PidPolling.d.ts.map +1 -1
- package/dist/runtime/tmux/PidPolling.js.map +1 -1
- package/dist/runtime/tmux/TmuxAgentRuntime.d.ts +1 -1
- package/dist/runtime/tmux/TmuxAgentRuntime.d.ts.map +1 -1
- package/dist/runtime/tmux/TmuxAgentRuntime.js +1 -1
- package/dist/runtime/tmux/TmuxAgentRuntime.js.map +1 -1
- package/dist/runtime/tmux/TmuxManager.d.ts +12 -0
- package/dist/runtime/tmux/TmuxManager.d.ts.map +1 -0
- package/dist/{terminal → runtime/tmux}/TmuxManager.js +9 -28
- package/dist/runtime/tmux/TmuxManager.js.map +1 -0
- package/dist/runtime/types.d.ts +1 -1
- package/dist/runtime/types.d.ts.map +1 -1
- package/dist/runtime/types.js.map +1 -1
- package/dist/terminal/TerminalFocusManager.d.ts +4 -32
- package/dist/terminal/TerminalFocusManager.d.ts.map +1 -1
- package/dist/terminal/TerminalFocusManager.js +17 -267
- package/dist/terminal/TerminalFocusManager.js.map +1 -1
- package/dist/terminal/TtyWriter.d.ts +1 -44
- package/dist/terminal/TtyWriter.d.ts.map +1 -1
- package/dist/terminal/TtyWriter.js +11 -347
- package/dist/terminal/TtyWriter.js.map +1 -1
- package/dist/terminal/backends/index.d.ts +5 -0
- package/dist/terminal/backends/index.d.ts.map +1 -0
- package/dist/terminal/backends/index.js +19 -0
- package/dist/terminal/backends/index.js.map +1 -0
- package/dist/terminal/backends/iterm2.d.ts +3 -0
- package/dist/terminal/backends/iterm2.d.ts.map +1 -0
- package/dist/terminal/backends/iterm2.js +135 -0
- package/dist/terminal/backends/iterm2.js.map +1 -0
- package/dist/terminal/backends/shared.d.ts +7 -0
- package/dist/terminal/backends/shared.d.ts.map +1 -0
- package/dist/terminal/backends/shared.js +35 -0
- package/dist/terminal/backends/shared.js.map +1 -0
- package/dist/terminal/backends/terminal-app.d.ts +3 -0
- package/dist/terminal/backends/terminal-app.d.ts.map +1 -0
- package/dist/terminal/backends/terminal-app.js +129 -0
- package/dist/terminal/backends/terminal-app.js.map +1 -0
- package/dist/terminal/backends/tmux.d.ts +3 -0
- package/dist/terminal/backends/tmux.d.ts.map +1 -0
- package/dist/terminal/backends/tmux.js +76 -0
- package/dist/terminal/backends/tmux.js.map +1 -0
- package/dist/terminal/backends/wezterm.d.ts +3 -0
- package/dist/terminal/backends/wezterm.d.ts.map +1 -0
- package/dist/terminal/backends/wezterm.js +71 -0
- package/dist/terminal/backends/wezterm.js.map +1 -0
- package/dist/terminal/types.d.ts +21 -0
- package/dist/terminal/types.d.ts.map +1 -0
- package/dist/terminal/types.js +10 -0
- package/dist/terminal/types.js.map +1 -0
- package/package.json +1 -1
- package/src/AgentManager.ts +1 -1
- package/src/__tests__/capacity/anthropic.test.ts +105 -0
- package/src/__tests__/capacity/boundaries.test.ts +36 -0
- package/src/__tests__/capacity/claude.test.ts +3 -8
- package/src/__tests__/capacity/codex.test.ts +12 -35
- package/src/__tests__/capacity/fixtures/openai-codex-auth.json +1 -1
- package/src/__tests__/capacity/openai.test.ts +15 -15
- package/src/__tests__/capacity/wham.test.ts +1 -1
- package/src/__tests__/capacity/zai.test.ts +27 -62
- package/src/__tests__/harnesses/claude/durable/ClaudePrintAgent.integration.test.ts +2 -2
- package/src/__tests__/harnesses/claude/durable/ClaudePrintRunner.test.ts +1 -1
- package/src/__tests__/harnesses/codex/durable/CodexCliProbe.test.ts +14 -19
- package/src/__tests__/harnesses/codex/durable/CodexPrintAgent.integration.test.ts +3 -3
- package/src/__tests__/harnesses/codex/durable/CodexPrintAgentService.test.ts +24 -30
- package/src/__tests__/harnesses/codex/durable/CodexPrintRunner.test.ts +4 -8
- package/src/__tests__/harnesses/pi/durable/PiCliProbe.test.ts +7 -11
- package/src/__tests__/harnesses/pi/durable/PiPrintAgentService.test.ts +19 -22
- package/src/__tests__/harnesses/pi/durable/PiPrintRunner.test.ts +7 -8
- package/src/__tests__/{readiness → harnesses/readiness}/AgentReadiness.test.ts +36 -9
- package/src/__tests__/harnesses/runtimeProfiles.test.ts +110 -0
- package/src/__tests__/print/DurableAgent.test.ts +4 -13
- package/src/__tests__/print/DurableAgentRepository.test.ts +1 -13
- package/src/__tests__/runtime/ManagedAgentRuntime.test.ts +1 -1
- package/src/__tests__/terminal/TmuxManager.test.ts +1 -1
- package/src/capacity/index.ts +29 -23
- package/src/capacity/{claude.ts → providers/anthropic.ts} +16 -131
- package/src/capacity/providers/openai-platform.ts +150 -0
- package/src/capacity/{wham.ts → providers/openai-subscription.ts} +86 -8
- package/src/capacity/{zai.ts → providers/zai.ts} +29 -111
- package/src/capacity/sources/claude.ts +30 -0
- package/src/capacity/{codex.ts → sources/codex.ts} +54 -124
- package/src/capacity/sources/pi.ts +131 -0
- package/src/capacity/types.ts +3 -0
- package/src/harnesses/claude/credentials.ts +122 -0
- package/src/harnesses/claude/readiness.ts +69 -0
- package/src/harnesses/codex/credentials.ts +81 -0
- package/src/harnesses/codex/readiness.ts +72 -0
- package/src/harnesses/copilot/readiness.ts +33 -0
- package/src/harnesses/opencode/readiness.ts +48 -0
- package/src/harnesses/pi/credentials.ts +129 -0
- package/src/harnesses/pi/readiness.ts +104 -0
- package/src/harnesses/readiness/AgentReadiness.ts +87 -0
- package/src/harnesses/readiness/checks.ts +336 -0
- package/src/harnesses/readiness/types.ts +90 -0
- package/src/{utils/agents.ts → harnesses/runtimeProfiles.ts} +34 -38
- package/src/index.ts +15 -128
- package/src/runtime/ManagedAgentRuntime.ts +7 -22
- package/src/runtime/herdr/HerdrAgentDiscovery.ts +3 -19
- package/src/runtime/tmux/PidPolling.ts +1 -1
- package/src/runtime/tmux/TmuxAgentRuntime.ts +1 -1
- package/src/{terminal → runtime/tmux}/TmuxManager.ts +9 -30
- package/src/runtime/types.ts +1 -1
- package/src/terminal/TerminalFocusManager.ts +19 -292
- package/src/terminal/TtyWriter.ts +16 -336
- package/src/terminal/backends/index.ts +21 -0
- package/src/terminal/backends/iterm2.ts +126 -0
- package/src/terminal/backends/shared.ts +42 -0
- package/src/terminal/backends/terminal-app.ts +122 -0
- package/src/terminal/backends/tmux.ts +62 -0
- package/src/terminal/backends/wezterm.ts +65 -0
- package/src/terminal/types.ts +22 -0
- package/dist/__tests__/readiness/AgentReadiness.test.js.map +0 -1
- package/dist/__tests__/utils/agents.test.js +0 -93
- package/dist/__tests__/utils/agents.test.js.map +0 -1
- package/dist/capacity/claude.d.ts +0 -17
- package/dist/capacity/claude.d.ts.map +0 -1
- package/dist/capacity/claude.js.map +0 -1
- package/dist/capacity/codex.d.ts.map +0 -1
- package/dist/capacity/codex.js.map +0 -1
- package/dist/capacity/openai.d.ts +0 -47
- package/dist/capacity/openai.d.ts.map +0 -1
- package/dist/capacity/openai.js +0 -307
- package/dist/capacity/openai.js.map +0 -1
- package/dist/capacity/wham.d.ts +0 -16
- package/dist/capacity/wham.d.ts.map +0 -1
- package/dist/capacity/wham.js +0 -66
- package/dist/capacity/wham.js.map +0 -1
- package/dist/capacity/zai.d.ts +0 -14
- package/dist/capacity/zai.d.ts.map +0 -1
- package/dist/capacity/zai.js.map +0 -1
- package/dist/readiness/AgentReadiness.d.ts.map +0 -1
- package/dist/readiness/AgentReadiness.js +0 -614
- package/dist/readiness/AgentReadiness.js.map +0 -1
- package/dist/terminal/TmuxManager.d.ts +0 -28
- package/dist/terminal/TmuxManager.d.ts.map +0 -1
- package/dist/terminal/TmuxManager.js.map +0 -1
- package/dist/terminal/index.d.ts +0 -5
- package/dist/terminal/index.d.ts.map +0 -1
- package/dist/terminal/index.js +0 -5
- package/dist/terminal/index.js.map +0 -1
- package/dist/utils/agents.d.ts +0 -16
- package/dist/utils/agents.d.ts.map +0 -1
- package/dist/utils/agents.js +0 -117
- package/dist/utils/agents.js.map +0 -1
- package/dist/utils/index.d.ts +0 -8
- package/dist/utils/index.d.ts.map +0 -1
- package/dist/utils/index.js +0 -6
- package/dist/utils/index.js.map +0 -1
- package/src/__tests__/utils/agents.test.ts +0 -105
- package/src/capacity/openai.ts +0 -355
- package/src/readiness/AgentReadiness.ts +0 -730
- package/src/terminal/index.ts +0 -4
- package/src/utils/index.ts +0 -14
package/dist/AgentManager.d.ts
CHANGED
|
@@ -21,10 +21,6 @@ export interface AgentManagerOptions {
|
|
|
21
21
|
fetchHerdrAgentPanes?: () => Promise<readonly HerdrAgentPane[]>;
|
|
22
22
|
onRuntimeDiscoveryError?: (error: unknown) => void;
|
|
23
23
|
}
|
|
24
|
-
export declare class AgentNotRunningError extends Error {
|
|
25
|
-
agentName: string;
|
|
26
|
-
constructor(agentName: string);
|
|
27
|
-
}
|
|
28
24
|
/**
|
|
29
25
|
* Agent Manager Class
|
|
30
26
|
*
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AgentManager.d.ts","sourceRoot":"","sources":["../src/AgentManager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EACV,YAAY,EACZ,SAAS,EACT,cAAc,EACd,mBAAmB,EAEpB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAc,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,EACL,aAAa,EAEb,KAAK,oBAAoB,EAE1B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAGL,KAAK,sBAAsB,EAC5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAIL,KAAK,cAAc,EACpB,MAAM,wCAAwC,CAAC;AAEhD,MAAM,WAAW,iBAAiB;IAChC;;;OAGG;IACH,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB;AAED,MAAM,WAAW,mBAAmB;IAClC,eAAe,CAAC,EACZ,oBAAoB,GACpB,CAAC,MAAM,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC;IACjE,oBAAoB,CAAC,EAAE,MAAM,OAAO,CAAC,SAAS,cAAc,EAAE,CAAC,CAAC;IAChE,uBAAuB,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD;
|
|
1
|
+
{"version":3,"file":"AgentManager.d.ts","sourceRoot":"","sources":["../src/AgentManager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EACV,YAAY,EACZ,SAAS,EACT,cAAc,EACd,mBAAmB,EAEpB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAc,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,EACL,aAAa,EAEb,KAAK,oBAAoB,EAE1B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAGL,KAAK,sBAAsB,EAC5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAIL,KAAK,cAAc,EACpB,MAAM,wCAAwC,CAAC;AAEhD,MAAM,WAAW,iBAAiB;IAChC;;;OAGG;IACH,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB;AAED,MAAM,WAAW,mBAAmB;IAClC,eAAe,CAAC,EACZ,oBAAoB,GACpB,CAAC,MAAM,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC;IACjE,oBAAoB,CAAC,EAAE,MAAM,OAAO,CAAC,SAAS,cAAc,EAAE,CAAC,CAAC;IAChE,uBAAuB,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD;AASD;;;;;;;;;;;;;;GAcG;AACH,qBAAa,YAAY;IAMrB,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN1B,OAAO,CAAC,QAAQ,CAAwC;IACxD,OAAO,CAAC,QAAQ,CAAgB;gBAG9B,QAAQ,GAAE,aAAuC,EAChC,eAAe,GAAE,sBAA+C,EAChE,OAAO,GAAE,mBAAwB;IAKpD;;;;;;;;;;OAUG;IACH,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAU5C;;;;;OAKG;IACH,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIxC;;;;OAIG;IACH,WAAW,IAAI,YAAY,EAAE;IAI7B;;;;;OAKG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAIlD;;;;;OAKG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIjC;;;;;;;;;;;;;;;;OAgBG;IACG,UAAU,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAmFnE;;;;OAIG;IACH,OAAO,CAAC,kBAAkB;YAYZ,iBAAiB;YAYjB,sBAAsB;IAKpC,OAAO,CAAC,eAAe;IAmCvB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAWrC;;;;;;;;;;;OAWG;IACG,YAAY,CAAC,IAAI,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAmCzE,4EAA4E;IACtE,gBAAgB,CACpB,SAAS,EAAE,MAAM,EACjB,IAAI,CAAC,EAAE,IAAI,CAAC,mBAAmB,EAAE,MAAM,CAAC,GACvC,OAAO,CAAC,cAAc,EAAE,CAAC;IA8B5B;;;;OAIG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;;;;;OAMG;IACH,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,SAAS,GAAG,SAAS,EAAE,GAAG,IAAI;CAwBjF"}
|
package/dist/AgentManager.js
CHANGED
|
@@ -7,13 +7,13 @@
|
|
|
7
7
|
import { AgentRegistry, parseTmuxRuntimeRef } from "./utils/AgentRegistry.js";
|
|
8
8
|
import { captureProcessSnapshot, filterByProcessNames } from "./utils/process.js";
|
|
9
9
|
import { findMatchingHerdrPane, fetchHerdrAgentPanes, herdrPaneToRuntimeRef } from "./runtime/herdr/HerdrAgentDiscovery.js";
|
|
10
|
-
|
|
10
|
+
let AgentNotRunningError = class AgentNotRunningError extends Error {
|
|
11
11
|
agentName;
|
|
12
12
|
constructor(agentName){
|
|
13
13
|
super(`Agent "${agentName}" is no longer running.`), this.agentName = agentName;
|
|
14
14
|
this.name = "AgentNotRunningError";
|
|
15
15
|
}
|
|
16
|
-
}
|
|
16
|
+
};
|
|
17
17
|
/**
|
|
18
18
|
* Agent Manager Class
|
|
19
19
|
*
|
package/dist/AgentManager.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/AgentManager.ts"],"sourcesContent":["/**\n * Agent Manager\n *\n * Orchestrates agent detection across multiple adapter types.\n * Manages adapter registration and aggregates results from all adapters.\n */\n\nimport type {\n AgentAdapter,\n AgentInfo,\n SessionSummary,\n ListSessionsOptions,\n ProcessInfo,\n} from \"./adapters/AgentAdapter.js\";\nimport { sortAgents, type AgentSortKey } from \"./utils/sortAgents.js\";\nimport {\n AgentRegistry,\n parseTmuxRuntimeRef,\n type AgentRuntimeProvider,\n type RegistryEntry,\n} from \"./utils/AgentRegistry.js\";\nimport {\n captureProcessSnapshot,\n filterByProcessNames,\n type ProcessSnapshotCapture,\n} from \"./utils/process.js\";\nimport {\n findMatchingHerdrPane,\n fetchHerdrAgentPanes,\n herdrPaneToRuntimeRef,\n type HerdrAgentPane,\n} from \"./runtime/herdr/HerdrAgentDiscovery.js\";\n\nexport interface ListAgentsOptions {\n /**\n * Sort key for the returned list. Defaults to `status`, which orders\n * waiting → running → idle → unknown, then by name for stability.\n */\n sortBy?: AgentSortKey;\n}\n\nexport interface AgentManagerOptions {\n runtimeProvider?:\n | AgentRuntimeProvider\n | (() => AgentRuntimeProvider | Promise<AgentRuntimeProvider>);\n fetchHerdrAgentPanes?: () => Promise<readonly HerdrAgentPane[]>;\n onRuntimeDiscoveryError?: (error: unknown) => void;\n}\n\nexport class AgentNotRunningError extends Error {\n constructor(public agentName: string) {\n super(`Agent \"${agentName}\" is no longer running.`);\n this.name = \"AgentNotRunningError\";\n }\n}\n\n/**\n * Agent Manager Class\n *\n * Central manager for detecting AI agents across different types.\n * Supports multiple adapters (Claude Code, Gemini CLI, etc.)\n *\n * @example\n * ```typescript\n * const manager = new AgentManager();\n * manager.registerAdapter(new ClaudeCodeAdapter());\n *\n * const agents = await manager.listAgents();\n * console.log(`Found ${agents.length} agents`);\n * ```\n */\nexport class AgentManager {\n private adapters: Map<string, AgentAdapter> = new Map();\n private registry: AgentRegistry;\n\n constructor(\n registry: AgentRegistry = AgentRegistry.default(),\n private readonly captureSnapshot: ProcessSnapshotCapture = captureProcessSnapshot,\n private readonly options: AgentManagerOptions = {},\n ) {\n this.registry = registry;\n }\n\n /**\n * Register an adapter for a specific agent type\n *\n * @param adapter Agent adapter to register\n * @throws Error if an adapter for this type is already registered\n *\n * @example\n * ```typescript\n * manager.registerAdapter(new ClaudeCodeAdapter());\n * ```\n */\n registerAdapter(adapter: AgentAdapter): void {\n const adapterKey = adapter.type;\n\n if (this.adapters.has(adapterKey)) {\n throw new Error(`Adapter for type \"${adapterKey}\" is already registered`);\n }\n\n this.adapters.set(adapterKey, adapter);\n }\n\n /**\n * Unregister an adapter by type\n *\n * @param type Agent type to unregister\n * @returns True if adapter was removed, false if not found\n */\n unregisterAdapter(type: string): boolean {\n return this.adapters.delete(type);\n }\n\n /**\n * Get all registered adapters\n *\n * @returns Array of registered adapters\n */\n getAdapters(): AgentAdapter[] {\n return Array.from(this.adapters.values());\n }\n\n /**\n * Get a registered adapter by type\n *\n * @param type Agent type to look up\n * @returns The adapter, or undefined if not registered\n */\n getAdapter(type: string): AgentAdapter | undefined {\n return this.adapters.get(type);\n }\n\n /**\n * Check if an adapter is registered for a specific type\n *\n * @param type Agent type to check\n * @returns True if adapter is registered\n */\n hasAdapter(type: string): boolean {\n return this.adapters.has(type);\n }\n\n /**\n * List all running AI agents detected by registered adapters\n *\n * Queries all registered adapters and aggregates results.\n * Handles errors gracefully - if one adapter fails, others still run.\n *\n * @returns Array of detected agents from all adapters\n *\n * @example\n * ```typescript\n * const agents = await manager.listAgents();\n *\n * agents.forEach(agent => {\n * console.log(`${agent.name}: ${agent.status}`);\n * });\n * ```\n */\n async listAgents(options?: ListAgentsOptions): Promise<AgentInfo[]> {\n const allAgents: AgentInfo[] = [];\n const errors: Array<{ type: string; error: Error }> = [];\n\n const adapters = Array.from(this.adapters.values());\n const processNames = Array.from(\n new Set(\n adapters.flatMap((adapter) => (adapter.processNames ? [...adapter.processNames] : [])),\n ),\n );\n let processes: readonly ProcessInfo[] = [];\n if (processNames.length > 0) {\n try {\n processes = await this.captureSnapshot(processNames, {\n isCandidate: (process) => this.isCandidateProcess(adapters, process),\n });\n } catch {\n processes = [];\n }\n }\n\n // Query all adapters in parallel using executable-scoped slices of the shared snapshot.\n const adapterPromises = adapters.map(async (adapter) => {\n try {\n const agents = adapter.processNames\n ? await adapter.detectAgents({\n processes: filterByProcessNames(processes, adapter.processNames),\n })\n : await adapter.detectAgents();\n return { type: adapter.type, agents, error: null };\n } catch (error) {\n // Capture error but don't throw - allow other adapters to continue\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return { type: adapter.type, agents: [], error: err };\n }\n });\n\n const results = await Promise.all(adapterPromises);\n\n // Aggregate all successful results\n for (const result of results) {\n if (result.error === null) {\n allAgents.push(...result.agents);\n }\n }\n\n // Log errors if any (but don't throw - partial results are useful)\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed:`);\n errors.forEach(({ type, error }) => {\n console.error(` - ${type}: ${error.message}`);\n });\n }\n\n const identityKey = (type: string, pid: number): string => `${type}:${pid}`;\n const preExistingByIdentity = new Map(\n this.registry.list().map((entry) => [identityKey(entry.type, entry.pid), entry]),\n );\n const herdrPanes = allAgents.length > 0 ? await this.resolveHerdrPanes() : [];\n const entries = allAgents.map((agent) =>\n this.toRegistryEntry(\n agent,\n preExistingByIdentity.get(identityKey(agent.type, agent.pid)),\n herdrPanes,\n ),\n );\n if (entries.length > 0) this.registry.registerBatch(entries);\n this.registry.pruneIfDue();\n\n for (const agent of allAgents) {\n const entry = preExistingByIdentity.get(identityKey(agent.type, agent.pid));\n if (entry) {\n agent.name = entry.name;\n agent.pinned = entry.pinned;\n if (entry.pinned && entry.updatedAt) agent.lastActive = new Date(entry.updatedAt);\n }\n }\n\n const sortKey: AgentSortKey = options?.sortBy ?? \"status\";\n return sortAgents(allAgents, sortKey);\n }\n\n /**\n * Only processes some adapter would actually handle are worth enriching\n * with cwd/start time. Broad matchers such as \"node\" otherwise pull in\n * every Node process on the machine.\n */\n private isCandidateProcess(adapters: readonly AgentAdapter[], process: ProcessInfo): boolean {\n return adapters.some((adapter) => {\n if (!adapter.processNames) return false;\n if (filterByProcessNames([process], adapter.processNames).length === 0) return false;\n try {\n return adapter.canHandle(process);\n } catch {\n return false;\n }\n });\n }\n\n private async resolveHerdrPanes(): Promise<readonly HerdrAgentPane[]> {\n const runtimeProvider = await this.resolveRuntimeProvider();\n if (runtimeProvider !== \"herdr\") return [];\n\n try {\n return await (this.options.fetchHerdrAgentPanes ?? fetchHerdrAgentPanes)();\n } catch (error) {\n this.options.onRuntimeDiscoveryError?.(error);\n return [];\n }\n }\n\n private async resolveRuntimeProvider(): Promise<AgentRuntimeProvider | undefined> {\n const runtimeProvider = this.options.runtimeProvider;\n return typeof runtimeProvider === \"function\" ? await runtimeProvider() : runtimeProvider;\n }\n\n private toRegistryEntry(\n agent: AgentInfo,\n existing?: RegistryEntry,\n herdrPanes: readonly HerdrAgentPane[] = [],\n ): RegistryEntry {\n const name = existing?.name ?? agent.name;\n const existingHasAuthoritativeRuntimeRef =\n existing?.runtime === \"herdr\" || Boolean(parseTmuxRuntimeRef(existing?.runtimeRef));\n const baseEntry: RegistryEntry = {\n name,\n type: agent.type,\n pid: agent.pid,\n runtime: existing?.runtime ?? \"tmux\",\n runtimeRef: existing?.runtimeRef ?? null,\n cwd: agent.projectPath,\n startedAt: existing?.startedAt ?? new Date().toISOString(),\n sessionId: agent.sessionId,\n sessionFilePath: agent.sessionFilePath ?? \"\",\n pinned: existing?.pinned ?? agent.pinned ?? false,\n };\n\n if (!existingHasAuthoritativeRuntimeRef) {\n const herdrPane = findMatchingHerdrPane(agent, herdrPanes);\n if (herdrPane) {\n return {\n ...baseEntry,\n runtime: \"herdr\",\n runtimeRef: herdrPaneToRuntimeRef(herdrPane, name),\n };\n }\n }\n\n return baseEntry;\n }\n\n togglePin(agentName: string): boolean {\n const entry = this.registry.lookup(agentName);\n if (!entry || !this.registry.isAlive(entry)) {\n if (entry) this.registry.prune();\n throw new AgentNotRunningError(agentName);\n }\n const pinned = this.registry.togglePin(entry.type, entry.pid);\n if (pinned === null) throw new AgentNotRunningError(agentName);\n return pinned;\n }\n\n /**\n * List historical sessions across every registered adapter.\n *\n * When `opts.type` is set, adapters whose `type` doesn't match are\n * skipped without being called. The remaining adapters' results are\n * merged and sorted by `lastActive` descending. Adapter failures are\n * caught (one-line stderr warning) so one broken adapter doesn't hide\n * the others.\n *\n * @param opts Filter options computed by the CLI; the manager passes\n * them through to each adapter unchanged.\n */\n async listSessions(opts?: ListSessionsOptions): Promise<SessionSummary[]> {\n const targetAdapters = Array.from(this.adapters.values()).filter(\n (adapter) => opts?.type === undefined || adapter.type === opts.type,\n );\n\n const errors: Array<{ type: string; error: Error }> = [];\n\n const results = await Promise.all(\n targetAdapters.map(async (adapter) => {\n try {\n return await adapter.listSessions(opts);\n } catch (error) {\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return [];\n }\n }),\n );\n\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed to list sessions:`);\n for (const { type, error } of errors) {\n console.error(` - ${type}: ${error.message}`);\n }\n }\n\n const merged: SessionSummary[] = [];\n for (const list of results) {\n merged.push(...list);\n }\n\n merged.sort((a, b) => b.lastActive.getTime() - a.lastActive.getTime());\n return merged;\n }\n\n /** Resolve an exact historical session ID across the selected harnesses. */\n async findSessionsById(\n sessionId: string,\n opts?: Pick<ListSessionsOptions, \"type\">,\n ): Promise<SessionSummary[]> {\n const targetAdapters = Array.from(this.adapters.values()).filter(\n (adapter) => opts?.type === undefined || adapter.type === opts.type,\n );\n const errors: Array<{ type: string; error: Error }> = [];\n\n const results = await Promise.all(\n targetAdapters.map(async (adapter) => {\n try {\n if (adapter.findSessionsById) return await adapter.findSessionsById(sessionId);\n const sessions = await adapter.listSessions(opts);\n return sessions.filter((session) => session.sessionId === sessionId);\n } catch (error) {\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return [];\n }\n }),\n );\n\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed to find session by ID:`);\n for (const { type, error } of errors) {\n console.error(` - ${type}: ${error.message}`);\n }\n }\n\n return results.flat();\n }\n\n /**\n * Get count of registered adapters\n *\n * @returns Number of registered adapters\n */\n getAdapterCount(): number {\n return this.adapters.size;\n }\n\n /**\n * Clear all registered adapters\n */\n clear(): void {\n this.adapters.clear();\n }\n\n /**\n * Resolve an agent by name (exact or partial match)\n *\n * @param input Name to search for\n * @param agents List of agents to search within\n * @returns Matched agent (unique), array of agents (ambiguous), or null (none)\n */\n resolveAgent(input: string, agents: AgentInfo[]): AgentInfo | AgentInfo[] | null {\n if (!input || agents.length === 0) return null;\n\n // Registry-first: if name is in registry and its PID is in the agent list, return it\n const registryEntry = this.registry.lookup(input);\n if (registryEntry) {\n const registryAgent = agents.find((a) => a.pid === registryEntry.pid);\n if (registryAgent) return registryAgent;\n }\n\n const lowerInput = input.toLowerCase();\n\n // 1. Exact match (case-insensitive)\n const exactMatch = agents.find((a) => a.name.toLowerCase() === lowerInput);\n if (exactMatch) return exactMatch;\n\n // 2. Partial match (prefix or contains)\n const matches = agents.filter((a) => a.name.toLowerCase().includes(lowerInput));\n\n if (matches.length === 1) return matches[0];\n if (matches.length > 1) return matches;\n\n return null;\n }\n}\n"],"names":["sortAgents","AgentRegistry","parseTmuxRuntimeRef","captureProcessSnapshot","filterByProcessNames","findMatchingHerdrPane","fetchHerdrAgentPanes","herdrPaneToRuntimeRef","AgentNotRunningError","Error","agentName","name","AgentManager","adapters","Map","registry","default","captureSnapshot","options","registerAdapter","adapter","adapterKey","type","has","set","unregisterAdapter","delete","getAdapters","Array","from","values","getAdapter","get","hasAdapter","listAgents","allAgents","errors","processNames","Set","flatMap","processes","length","isCandidate","process","isCandidateProcess","adapterPromises","map","agents","detectAgents","error","err","String","push","results","Promise","all","result","console","forEach","message","identityKey","pid","preExistingByIdentity","list","entry","herdrPanes","resolveHerdrPanes","entries","agent","toRegistryEntry","registerBatch","pruneIfDue","pinned","updatedAt","lastActive","Date","sortKey","sortBy","some","canHandle","runtimeProvider","resolveRuntimeProvider","onRuntimeDiscoveryError","existing","existingHasAuthoritativeRuntimeRef","runtime","Boolean","runtimeRef","baseEntry","cwd","projectPath","startedAt","toISOString","sessionId","sessionFilePath","herdrPane","togglePin","lookup","isAlive","prune","listSessions","opts","targetAdapters","filter","undefined","merged","sort","a","b","getTime","findSessionsById","sessions","session","flat","getAdapterCount","size","clear","resolveAgent","input","registryEntry","registryAgent","find","lowerInput","toLowerCase","exactMatch","matches","includes"],"mappings":"AAAA;;;;;CAKC,GASD,SAASA,UAAU,QAA2B,wBAAwB;AACtE,SACEC,aAAa,EACbC,mBAAmB,QAGd,2BAA2B;AAClC,SACEC,sBAAsB,EACtBC,oBAAoB,QAEf,qBAAqB;AAC5B,SACEC,qBAAqB,EACrBC,oBAAoB,EACpBC,qBAAqB,QAEhB,yCAAyC;AAkBhD,OAAO,MAAMC,6BAA6BC;;IACxC,YAAY,AAAOC,SAAiB,CAAE;QACpC,KAAK,CAAC,CAAC,OAAO,EAAEA,UAAU,uBAAuB,CAAC,QADjCA,YAAAA;QAEjB,IAAI,CAACC,IAAI,GAAG;IACd;AACF;AAEA;;;;;;;;;;;;;;CAcC,GACD,OAAO,MAAMC;;;IACHC,WAAsC,IAAIC,MAAM;IAChDC,SAAwB;IAEhC,YACEA,WAA0Bd,cAAce,OAAO,EAAE,EACjD,AAAiBC,kBAA0Cd,sBAAsB,EACjF,AAAiBe,UAA+B,CAAC,CAAC,CAClD;aAFiBD,kBAAAA;aACAC,UAAAA;QAEjB,IAAI,CAACH,QAAQ,GAAGA;IAClB;IAEA;;;;;;;;;;GAUC,GACDI,gBAAgBC,OAAqB,EAAQ;QAC3C,MAAMC,aAAaD,QAAQE,IAAI;QAE/B,IAAI,IAAI,CAACT,QAAQ,CAACU,GAAG,CAACF,aAAa;YACjC,MAAM,IAAIZ,MAAM,CAAC,kBAAkB,EAAEY,WAAW,uBAAuB,CAAC;QAC1E;QAEA,IAAI,CAACR,QAAQ,CAACW,GAAG,CAACH,YAAYD;IAChC;IAEA;;;;;GAKC,GACDK,kBAAkBH,IAAY,EAAW;QACvC,OAAO,IAAI,CAACT,QAAQ,CAACa,MAAM,CAACJ;IAC9B;IAEA;;;;GAIC,GACDK,cAA8B;QAC5B,OAAOC,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM;IACxC;IAEA;;;;;GAKC,GACDC,WAAWT,IAAY,EAA4B;QACjD,OAAO,IAAI,CAACT,QAAQ,CAACmB,GAAG,CAACV;IAC3B;IAEA;;;;;GAKC,GACDW,WAAWX,IAAY,EAAW;QAChC,OAAO,IAAI,CAACT,QAAQ,CAACU,GAAG,CAACD;IAC3B;IAEA;;;;;;;;;;;;;;;;GAgBC,GACD,MAAMY,WAAWhB,OAA2B,EAAwB;QAClE,MAAMiB,YAAyB,EAAE;QACjC,MAAMC,SAAgD,EAAE;QAExD,MAAMvB,WAAWe,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM;QAChD,MAAMO,eAAeT,MAAMC,IAAI,CAC7B,IAAIS,IACFzB,SAAS0B,OAAO,CAAC,CAACnB,UAAaA,QAAQiB,YAAY,GAAG;mBAAIjB,QAAQiB,YAAY;aAAC,GAAG,EAAE;QAGxF,IAAIG,YAAoC,EAAE;QAC1C,IAAIH,aAAaI,MAAM,GAAG,GAAG;YAC3B,IAAI;gBACFD,YAAY,MAAM,IAAI,CAACvB,eAAe,CAACoB,cAAc;oBACnDK,aAAa,CAACC,UAAY,IAAI,CAACC,kBAAkB,CAAC/B,UAAU8B;gBAC9D;YACF,EAAE,OAAM;gBACNH,YAAY,EAAE;YAChB;QACF;QAEA,wFAAwF;QACxF,MAAMK,kBAAkBhC,SAASiC,GAAG,CAAC,OAAO1B;YAC1C,IAAI;gBACF,MAAM2B,SAAS3B,QAAQiB,YAAY,GAC/B,MAAMjB,QAAQ4B,YAAY,CAAC;oBACzBR,WAAWpC,qBAAqBoC,WAAWpB,QAAQiB,YAAY;gBACjE,KACA,MAAMjB,QAAQ4B,YAAY;gBAC9B,OAAO;oBAAE1B,MAAMF,QAAQE,IAAI;oBAAEyB;oBAAQE,OAAO;gBAAK;YACnD,EAAE,OAAOA,OAAO;gBACd,mEAAmE;gBACnE,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO;oBAAE5B,MAAMF,QAAQE,IAAI;oBAAEyB,QAAQ,EAAE;oBAAEE,OAAOC;gBAAI;YACtD;QACF;QAEA,MAAMG,UAAU,MAAMC,QAAQC,GAAG,CAACV;QAElC,mCAAmC;QACnC,KAAK,MAAMW,UAAUH,QAAS;YAC5B,IAAIG,OAAOP,KAAK,KAAK,MAAM;gBACzBd,UAAUiB,IAAI,IAAII,OAAOT,MAAM;YACjC;QACF;QAEA,mEAAmE;QACnE,IAAIX,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,mBAAmB,CAAC;YAC5DL,OAAOsB,OAAO,CAAC,CAAC,EAAEpC,IAAI,EAAE2B,KAAK,EAAE;gBAC7BQ,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAMC,cAAc,CAACtC,MAAcuC,MAAwB,GAAGvC,KAAK,CAAC,EAAEuC,KAAK;QAC3E,MAAMC,wBAAwB,IAAIhD,IAChC,IAAI,CAACC,QAAQ,CAACgD,IAAI,GAAGjB,GAAG,CAAC,CAACkB,QAAU;gBAACJ,YAAYI,MAAM1C,IAAI,EAAE0C,MAAMH,GAAG;gBAAGG;aAAM;QAEjF,MAAMC,aAAa9B,UAAUM,MAAM,GAAG,IAAI,MAAM,IAAI,CAACyB,iBAAiB,KAAK,EAAE;QAC7E,MAAMC,UAAUhC,UAAUW,GAAG,CAAC,CAACsB,QAC7B,IAAI,CAACC,eAAe,CAClBD,OACAN,sBAAsB9B,GAAG,CAAC4B,YAAYQ,MAAM9C,IAAI,EAAE8C,MAAMP,GAAG,IAC3DI;QAGJ,IAAIE,QAAQ1B,MAAM,GAAG,GAAG,IAAI,CAAC1B,QAAQ,CAACuD,aAAa,CAACH;QACpD,IAAI,CAACpD,QAAQ,CAACwD,UAAU;QAExB,KAAK,MAAMH,SAASjC,UAAW;YAC7B,MAAM6B,QAAQF,sBAAsB9B,GAAG,CAAC4B,YAAYQ,MAAM9C,IAAI,EAAE8C,MAAMP,GAAG;YACzE,IAAIG,OAAO;gBACTI,MAAMzD,IAAI,GAAGqD,MAAMrD,IAAI;gBACvByD,MAAMI,MAAM,GAAGR,MAAMQ,MAAM;gBAC3B,IAAIR,MAAMQ,MAAM,IAAIR,MAAMS,SAAS,EAAEL,MAAMM,UAAU,GAAG,IAAIC,KAAKX,MAAMS,SAAS;YAClF;QACF;QAEA,MAAMG,UAAwB1D,SAAS2D,UAAU;QACjD,OAAO7E,WAAWmC,WAAWyC;IAC/B;IAEA;;;;GAIC,GACD,AAAQhC,mBAAmB/B,QAAiC,EAAE8B,OAAoB,EAAW;QAC3F,OAAO9B,SAASiE,IAAI,CAAC,CAAC1D;YACpB,IAAI,CAACA,QAAQiB,YAAY,EAAE,OAAO;YAClC,IAAIjC,qBAAqB;gBAACuC;aAAQ,EAAEvB,QAAQiB,YAAY,EAAEI,MAAM,KAAK,GAAG,OAAO;YAC/E,IAAI;gBACF,OAAOrB,QAAQ2D,SAAS,CAACpC;YAC3B,EAAE,OAAM;gBACN,OAAO;YACT;QACF;IACF;IAEA,MAAcuB,oBAAwD;QACpE,MAAMc,kBAAkB,MAAM,IAAI,CAACC,sBAAsB;QACzD,IAAID,oBAAoB,SAAS,OAAO,EAAE;QAE1C,IAAI;YACF,OAAO,MAAM,AAAC,CAAA,IAAI,CAAC9D,OAAO,CAACZ,oBAAoB,IAAIA,oBAAmB;QACxE,EAAE,OAAO2C,OAAO;YACd,IAAI,CAAC/B,OAAO,CAACgE,uBAAuB,GAAGjC;YACvC,OAAO,EAAE;QACX;IACF;IAEA,MAAcgC,yBAAoE;QAChF,MAAMD,kBAAkB,IAAI,CAAC9D,OAAO,CAAC8D,eAAe;QACpD,OAAO,OAAOA,oBAAoB,aAAa,MAAMA,oBAAoBA;IAC3E;IAEQX,gBACND,KAAgB,EAChBe,QAAwB,EACxBlB,aAAwC,EAAE,EAC3B;QACf,MAAMtD,OAAOwE,UAAUxE,QAAQyD,MAAMzD,IAAI;QACzC,MAAMyE,qCACJD,UAAUE,YAAY,WAAWC,QAAQpF,oBAAoBiF,UAAUI;QACzE,MAAMC,YAA2B;YAC/B7E;YACAW,MAAM8C,MAAM9C,IAAI;YAChBuC,KAAKO,MAAMP,GAAG;YACdwB,SAASF,UAAUE,WAAW;YAC9BE,YAAYJ,UAAUI,cAAc;YACpCE,KAAKrB,MAAMsB,WAAW;YACtBC,WAAWR,UAAUQ,aAAa,IAAIhB,OAAOiB,WAAW;YACxDC,WAAWzB,MAAMyB,SAAS;YAC1BC,iBAAiB1B,MAAM0B,eAAe,IAAI;YAC1CtB,QAAQW,UAAUX,UAAUJ,MAAMI,MAAM,IAAI;QAC9C;QAEA,IAAI,CAACY,oCAAoC;YACvC,MAAMW,YAAY1F,sBAAsB+D,OAAOH;YAC/C,IAAI8B,WAAW;gBACb,OAAO;oBACL,GAAGP,SAAS;oBACZH,SAAS;oBACTE,YAAYhF,sBAAsBwF,WAAWpF;gBAC/C;YACF;QACF;QAEA,OAAO6E;IACT;IAEAQ,UAAUtF,SAAiB,EAAW;QACpC,MAAMsD,QAAQ,IAAI,CAACjD,QAAQ,CAACkF,MAAM,CAACvF;QACnC,IAAI,CAACsD,SAAS,CAAC,IAAI,CAACjD,QAAQ,CAACmF,OAAO,CAAClC,QAAQ;YAC3C,IAAIA,OAAO,IAAI,CAACjD,QAAQ,CAACoF,KAAK;YAC9B,MAAM,IAAI3F,qBAAqBE;QACjC;QACA,MAAM8D,SAAS,IAAI,CAACzD,QAAQ,CAACiF,SAAS,CAAChC,MAAM1C,IAAI,EAAE0C,MAAMH,GAAG;QAC5D,IAAIW,WAAW,MAAM,MAAM,IAAIhE,qBAAqBE;QACpD,OAAO8D;IACT;IAEA;;;;;;;;;;;GAWC,GACD,MAAM4B,aAAaC,IAA0B,EAA6B;QACxE,MAAMC,iBAAiB1E,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM,IAAIyE,MAAM,CAC9D,CAACnF,UAAYiF,MAAM/E,SAASkF,aAAapF,QAAQE,IAAI,KAAK+E,KAAK/E,IAAI;QAGrE,MAAMc,SAAgD,EAAE;QAExD,MAAMiB,UAAU,MAAMC,QAAQC,GAAG,CAC/B+C,eAAexD,GAAG,CAAC,OAAO1B;YACxB,IAAI;gBACF,OAAO,MAAMA,QAAQgF,YAAY,CAACC;YACpC,EAAE,OAAOpD,OAAO;gBACd,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAId,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,oCAAoC,CAAC;YAC7E,KAAK,MAAM,EAAEnB,IAAI,EAAE2B,KAAK,EAAE,IAAIb,OAAQ;gBACpCqB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAM8C,SAA2B,EAAE;QACnC,KAAK,MAAM1C,QAAQV,QAAS;YAC1BoD,OAAOrD,IAAI,IAAIW;QACjB;QAEA0C,OAAOC,IAAI,CAAC,CAACC,GAAGC,IAAMA,EAAElC,UAAU,CAACmC,OAAO,KAAKF,EAAEjC,UAAU,CAACmC,OAAO;QACnE,OAAOJ;IACT;IAEA,0EAA0E,GAC1E,MAAMK,iBACJjB,SAAiB,EACjBQ,IAAwC,EACb;QAC3B,MAAMC,iBAAiB1E,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM,IAAIyE,MAAM,CAC9D,CAACnF,UAAYiF,MAAM/E,SAASkF,aAAapF,QAAQE,IAAI,KAAK+E,KAAK/E,IAAI;QAErE,MAAMc,SAAgD,EAAE;QAExD,MAAMiB,UAAU,MAAMC,QAAQC,GAAG,CAC/B+C,eAAexD,GAAG,CAAC,OAAO1B;YACxB,IAAI;gBACF,IAAIA,QAAQ0F,gBAAgB,EAAE,OAAO,MAAM1F,QAAQ0F,gBAAgB,CAACjB;gBACpE,MAAMkB,WAAW,MAAM3F,QAAQgF,YAAY,CAACC;gBAC5C,OAAOU,SAASR,MAAM,CAAC,CAACS,UAAYA,QAAQnB,SAAS,KAAKA;YAC5D,EAAE,OAAO5C,OAAO;gBACd,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAId,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,yCAAyC,CAAC;YAClF,KAAK,MAAM,EAAEnB,IAAI,EAAE2B,KAAK,EAAE,IAAIb,OAAQ;gBACpCqB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,OAAON,QAAQ4D,IAAI;IACrB;IAEA;;;;GAIC,GACDC,kBAA0B;QACxB,OAAO,IAAI,CAACrG,QAAQ,CAACsG,IAAI;IAC3B;IAEA;;GAEC,GACDC,QAAc;QACZ,IAAI,CAACvG,QAAQ,CAACuG,KAAK;IACrB;IAEA;;;;;;GAMC,GACDC,aAAaC,KAAa,EAAEvE,MAAmB,EAAkC;QAC/E,IAAI,CAACuE,SAASvE,OAAON,MAAM,KAAK,GAAG,OAAO;QAE1C,qFAAqF;QACrF,MAAM8E,gBAAgB,IAAI,CAACxG,QAAQ,CAACkF,MAAM,CAACqB;QAC3C,IAAIC,eAAe;YACjB,MAAMC,gBAAgBzE,OAAO0E,IAAI,CAAC,CAACd,IAAMA,EAAE9C,GAAG,KAAK0D,cAAc1D,GAAG;YACpE,IAAI2D,eAAe,OAAOA;QAC5B;QAEA,MAAME,aAAaJ,MAAMK,WAAW;QAEpC,oCAAoC;QACpC,MAAMC,aAAa7E,OAAO0E,IAAI,CAAC,CAACd,IAAMA,EAAEhG,IAAI,CAACgH,WAAW,OAAOD;QAC/D,IAAIE,YAAY,OAAOA;QAEvB,wCAAwC;QACxC,MAAMC,UAAU9E,OAAOwD,MAAM,CAAC,CAACI,IAAMA,EAAEhG,IAAI,CAACgH,WAAW,GAAGG,QAAQ,CAACJ;QAEnE,IAAIG,QAAQpF,MAAM,KAAK,GAAG,OAAOoF,OAAO,CAAC,EAAE;QAC3C,IAAIA,QAAQpF,MAAM,GAAG,GAAG,OAAOoF;QAE/B,OAAO;IACT;AACF"}
|
|
1
|
+
{"version":3,"sources":["../src/AgentManager.ts"],"sourcesContent":["/**\n * Agent Manager\n *\n * Orchestrates agent detection across multiple adapter types.\n * Manages adapter registration and aggregates results from all adapters.\n */\n\nimport type {\n AgentAdapter,\n AgentInfo,\n SessionSummary,\n ListSessionsOptions,\n ProcessInfo,\n} from \"./adapters/AgentAdapter.js\";\nimport { sortAgents, type AgentSortKey } from \"./utils/sortAgents.js\";\nimport {\n AgentRegistry,\n parseTmuxRuntimeRef,\n type AgentRuntimeProvider,\n type RegistryEntry,\n} from \"./utils/AgentRegistry.js\";\nimport {\n captureProcessSnapshot,\n filterByProcessNames,\n type ProcessSnapshotCapture,\n} from \"./utils/process.js\";\nimport {\n findMatchingHerdrPane,\n fetchHerdrAgentPanes,\n herdrPaneToRuntimeRef,\n type HerdrAgentPane,\n} from \"./runtime/herdr/HerdrAgentDiscovery.js\";\n\nexport interface ListAgentsOptions {\n /**\n * Sort key for the returned list. Defaults to `status`, which orders\n * waiting → running → idle → unknown, then by name for stability.\n */\n sortBy?: AgentSortKey;\n}\n\nexport interface AgentManagerOptions {\n runtimeProvider?:\n | AgentRuntimeProvider\n | (() => AgentRuntimeProvider | Promise<AgentRuntimeProvider>);\n fetchHerdrAgentPanes?: () => Promise<readonly HerdrAgentPane[]>;\n onRuntimeDiscoveryError?: (error: unknown) => void;\n}\n\nclass AgentNotRunningError extends Error {\n constructor(public agentName: string) {\n super(`Agent \"${agentName}\" is no longer running.`);\n this.name = \"AgentNotRunningError\";\n }\n}\n\n/**\n * Agent Manager Class\n *\n * Central manager for detecting AI agents across different types.\n * Supports multiple adapters (Claude Code, Gemini CLI, etc.)\n *\n * @example\n * ```typescript\n * const manager = new AgentManager();\n * manager.registerAdapter(new ClaudeCodeAdapter());\n *\n * const agents = await manager.listAgents();\n * console.log(`Found ${agents.length} agents`);\n * ```\n */\nexport class AgentManager {\n private adapters: Map<string, AgentAdapter> = new Map();\n private registry: AgentRegistry;\n\n constructor(\n registry: AgentRegistry = AgentRegistry.default(),\n private readonly captureSnapshot: ProcessSnapshotCapture = captureProcessSnapshot,\n private readonly options: AgentManagerOptions = {},\n ) {\n this.registry = registry;\n }\n\n /**\n * Register an adapter for a specific agent type\n *\n * @param adapter Agent adapter to register\n * @throws Error if an adapter for this type is already registered\n *\n * @example\n * ```typescript\n * manager.registerAdapter(new ClaudeCodeAdapter());\n * ```\n */\n registerAdapter(adapter: AgentAdapter): void {\n const adapterKey = adapter.type;\n\n if (this.adapters.has(adapterKey)) {\n throw new Error(`Adapter for type \"${adapterKey}\" is already registered`);\n }\n\n this.adapters.set(adapterKey, adapter);\n }\n\n /**\n * Unregister an adapter by type\n *\n * @param type Agent type to unregister\n * @returns True if adapter was removed, false if not found\n */\n unregisterAdapter(type: string): boolean {\n return this.adapters.delete(type);\n }\n\n /**\n * Get all registered adapters\n *\n * @returns Array of registered adapters\n */\n getAdapters(): AgentAdapter[] {\n return Array.from(this.adapters.values());\n }\n\n /**\n * Get a registered adapter by type\n *\n * @param type Agent type to look up\n * @returns The adapter, or undefined if not registered\n */\n getAdapter(type: string): AgentAdapter | undefined {\n return this.adapters.get(type);\n }\n\n /**\n * Check if an adapter is registered for a specific type\n *\n * @param type Agent type to check\n * @returns True if adapter is registered\n */\n hasAdapter(type: string): boolean {\n return this.adapters.has(type);\n }\n\n /**\n * List all running AI agents detected by registered adapters\n *\n * Queries all registered adapters and aggregates results.\n * Handles errors gracefully - if one adapter fails, others still run.\n *\n * @returns Array of detected agents from all adapters\n *\n * @example\n * ```typescript\n * const agents = await manager.listAgents();\n *\n * agents.forEach(agent => {\n * console.log(`${agent.name}: ${agent.status}`);\n * });\n * ```\n */\n async listAgents(options?: ListAgentsOptions): Promise<AgentInfo[]> {\n const allAgents: AgentInfo[] = [];\n const errors: Array<{ type: string; error: Error }> = [];\n\n const adapters = Array.from(this.adapters.values());\n const processNames = Array.from(\n new Set(\n adapters.flatMap((adapter) => (adapter.processNames ? [...adapter.processNames] : [])),\n ),\n );\n let processes: readonly ProcessInfo[] = [];\n if (processNames.length > 0) {\n try {\n processes = await this.captureSnapshot(processNames, {\n isCandidate: (process) => this.isCandidateProcess(adapters, process),\n });\n } catch {\n processes = [];\n }\n }\n\n // Query all adapters in parallel using executable-scoped slices of the shared snapshot.\n const adapterPromises = adapters.map(async (adapter) => {\n try {\n const agents = adapter.processNames\n ? await adapter.detectAgents({\n processes: filterByProcessNames(processes, adapter.processNames),\n })\n : await adapter.detectAgents();\n return { type: adapter.type, agents, error: null };\n } catch (error) {\n // Capture error but don't throw - allow other adapters to continue\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return { type: adapter.type, agents: [], error: err };\n }\n });\n\n const results = await Promise.all(adapterPromises);\n\n // Aggregate all successful results\n for (const result of results) {\n if (result.error === null) {\n allAgents.push(...result.agents);\n }\n }\n\n // Log errors if any (but don't throw - partial results are useful)\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed:`);\n errors.forEach(({ type, error }) => {\n console.error(` - ${type}: ${error.message}`);\n });\n }\n\n const identityKey = (type: string, pid: number): string => `${type}:${pid}`;\n const preExistingByIdentity = new Map(\n this.registry.list().map((entry) => [identityKey(entry.type, entry.pid), entry]),\n );\n const herdrPanes = allAgents.length > 0 ? await this.resolveHerdrPanes() : [];\n const entries = allAgents.map((agent) =>\n this.toRegistryEntry(\n agent,\n preExistingByIdentity.get(identityKey(agent.type, agent.pid)),\n herdrPanes,\n ),\n );\n if (entries.length > 0) this.registry.registerBatch(entries);\n this.registry.pruneIfDue();\n\n for (const agent of allAgents) {\n const entry = preExistingByIdentity.get(identityKey(agent.type, agent.pid));\n if (entry) {\n agent.name = entry.name;\n agent.pinned = entry.pinned;\n if (entry.pinned && entry.updatedAt) agent.lastActive = new Date(entry.updatedAt);\n }\n }\n\n const sortKey: AgentSortKey = options?.sortBy ?? \"status\";\n return sortAgents(allAgents, sortKey);\n }\n\n /**\n * Only processes some adapter would actually handle are worth enriching\n * with cwd/start time. Broad matchers such as \"node\" otherwise pull in\n * every Node process on the machine.\n */\n private isCandidateProcess(adapters: readonly AgentAdapter[], process: ProcessInfo): boolean {\n return adapters.some((adapter) => {\n if (!adapter.processNames) return false;\n if (filterByProcessNames([process], adapter.processNames).length === 0) return false;\n try {\n return adapter.canHandle(process);\n } catch {\n return false;\n }\n });\n }\n\n private async resolveHerdrPanes(): Promise<readonly HerdrAgentPane[]> {\n const runtimeProvider = await this.resolveRuntimeProvider();\n if (runtimeProvider !== \"herdr\") return [];\n\n try {\n return await (this.options.fetchHerdrAgentPanes ?? fetchHerdrAgentPanes)();\n } catch (error) {\n this.options.onRuntimeDiscoveryError?.(error);\n return [];\n }\n }\n\n private async resolveRuntimeProvider(): Promise<AgentRuntimeProvider | undefined> {\n const runtimeProvider = this.options.runtimeProvider;\n return typeof runtimeProvider === \"function\" ? await runtimeProvider() : runtimeProvider;\n }\n\n private toRegistryEntry(\n agent: AgentInfo,\n existing?: RegistryEntry,\n herdrPanes: readonly HerdrAgentPane[] = [],\n ): RegistryEntry {\n const name = existing?.name ?? agent.name;\n const existingHasAuthoritativeRuntimeRef =\n existing?.runtime === \"herdr\" || Boolean(parseTmuxRuntimeRef(existing?.runtimeRef));\n const baseEntry: RegistryEntry = {\n name,\n type: agent.type,\n pid: agent.pid,\n runtime: existing?.runtime ?? \"tmux\",\n runtimeRef: existing?.runtimeRef ?? null,\n cwd: agent.projectPath,\n startedAt: existing?.startedAt ?? new Date().toISOString(),\n sessionId: agent.sessionId,\n sessionFilePath: agent.sessionFilePath ?? \"\",\n pinned: existing?.pinned ?? agent.pinned ?? false,\n };\n\n if (!existingHasAuthoritativeRuntimeRef) {\n const herdrPane = findMatchingHerdrPane(agent, herdrPanes);\n if (herdrPane) {\n return {\n ...baseEntry,\n runtime: \"herdr\",\n runtimeRef: herdrPaneToRuntimeRef(herdrPane, name),\n };\n }\n }\n\n return baseEntry;\n }\n\n togglePin(agentName: string): boolean {\n const entry = this.registry.lookup(agentName);\n if (!entry || !this.registry.isAlive(entry)) {\n if (entry) this.registry.prune();\n throw new AgentNotRunningError(agentName);\n }\n const pinned = this.registry.togglePin(entry.type, entry.pid);\n if (pinned === null) throw new AgentNotRunningError(agentName);\n return pinned;\n }\n\n /**\n * List historical sessions across every registered adapter.\n *\n * When `opts.type` is set, adapters whose `type` doesn't match are\n * skipped without being called. The remaining adapters' results are\n * merged and sorted by `lastActive` descending. Adapter failures are\n * caught (one-line stderr warning) so one broken adapter doesn't hide\n * the others.\n *\n * @param opts Filter options computed by the CLI; the manager passes\n * them through to each adapter unchanged.\n */\n async listSessions(opts?: ListSessionsOptions): Promise<SessionSummary[]> {\n const targetAdapters = Array.from(this.adapters.values()).filter(\n (adapter) => opts?.type === undefined || adapter.type === opts.type,\n );\n\n const errors: Array<{ type: string; error: Error }> = [];\n\n const results = await Promise.all(\n targetAdapters.map(async (adapter) => {\n try {\n return await adapter.listSessions(opts);\n } catch (error) {\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return [];\n }\n }),\n );\n\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed to list sessions:`);\n for (const { type, error } of errors) {\n console.error(` - ${type}: ${error.message}`);\n }\n }\n\n const merged: SessionSummary[] = [];\n for (const list of results) {\n merged.push(...list);\n }\n\n merged.sort((a, b) => b.lastActive.getTime() - a.lastActive.getTime());\n return merged;\n }\n\n /** Resolve an exact historical session ID across the selected harnesses. */\n async findSessionsById(\n sessionId: string,\n opts?: Pick<ListSessionsOptions, \"type\">,\n ): Promise<SessionSummary[]> {\n const targetAdapters = Array.from(this.adapters.values()).filter(\n (adapter) => opts?.type === undefined || adapter.type === opts.type,\n );\n const errors: Array<{ type: string; error: Error }> = [];\n\n const results = await Promise.all(\n targetAdapters.map(async (adapter) => {\n try {\n if (adapter.findSessionsById) return await adapter.findSessionsById(sessionId);\n const sessions = await adapter.listSessions(opts);\n return sessions.filter((session) => session.sessionId === sessionId);\n } catch (error) {\n const err = error instanceof Error ? error : new Error(String(error));\n errors.push({ type: adapter.type, error: err });\n return [];\n }\n }),\n );\n\n if (errors.length > 0) {\n console.error(`Warning: ${errors.length} adapter(s) failed to find session by ID:`);\n for (const { type, error } of errors) {\n console.error(` - ${type}: ${error.message}`);\n }\n }\n\n return results.flat();\n }\n\n /**\n * Get count of registered adapters\n *\n * @returns Number of registered adapters\n */\n getAdapterCount(): number {\n return this.adapters.size;\n }\n\n /**\n * Clear all registered adapters\n */\n clear(): void {\n this.adapters.clear();\n }\n\n /**\n * Resolve an agent by name (exact or partial match)\n *\n * @param input Name to search for\n * @param agents List of agents to search within\n * @returns Matched agent (unique), array of agents (ambiguous), or null (none)\n */\n resolveAgent(input: string, agents: AgentInfo[]): AgentInfo | AgentInfo[] | null {\n if (!input || agents.length === 0) return null;\n\n // Registry-first: if name is in registry and its PID is in the agent list, return it\n const registryEntry = this.registry.lookup(input);\n if (registryEntry) {\n const registryAgent = agents.find((a) => a.pid === registryEntry.pid);\n if (registryAgent) return registryAgent;\n }\n\n const lowerInput = input.toLowerCase();\n\n // 1. Exact match (case-insensitive)\n const exactMatch = agents.find((a) => a.name.toLowerCase() === lowerInput);\n if (exactMatch) return exactMatch;\n\n // 2. Partial match (prefix or contains)\n const matches = agents.filter((a) => a.name.toLowerCase().includes(lowerInput));\n\n if (matches.length === 1) return matches[0];\n if (matches.length > 1) return matches;\n\n return null;\n }\n}\n"],"names":["sortAgents","AgentRegistry","parseTmuxRuntimeRef","captureProcessSnapshot","filterByProcessNames","findMatchingHerdrPane","fetchHerdrAgentPanes","herdrPaneToRuntimeRef","AgentNotRunningError","Error","agentName","name","AgentManager","adapters","Map","registry","default","captureSnapshot","options","registerAdapter","adapter","adapterKey","type","has","set","unregisterAdapter","delete","getAdapters","Array","from","values","getAdapter","get","hasAdapter","listAgents","allAgents","errors","processNames","Set","flatMap","processes","length","isCandidate","process","isCandidateProcess","adapterPromises","map","agents","detectAgents","error","err","String","push","results","Promise","all","result","console","forEach","message","identityKey","pid","preExistingByIdentity","list","entry","herdrPanes","resolveHerdrPanes","entries","agent","toRegistryEntry","registerBatch","pruneIfDue","pinned","updatedAt","lastActive","Date","sortKey","sortBy","some","canHandle","runtimeProvider","resolveRuntimeProvider","onRuntimeDiscoveryError","existing","existingHasAuthoritativeRuntimeRef","runtime","Boolean","runtimeRef","baseEntry","cwd","projectPath","startedAt","toISOString","sessionId","sessionFilePath","herdrPane","togglePin","lookup","isAlive","prune","listSessions","opts","targetAdapters","filter","undefined","merged","sort","a","b","getTime","findSessionsById","sessions","session","flat","getAdapterCount","size","clear","resolveAgent","input","registryEntry","registryAgent","find","lowerInput","toLowerCase","exactMatch","matches","includes"],"mappings":"AAAA;;;;;CAKC,GASD,SAASA,UAAU,QAA2B,wBAAwB;AACtE,SACEC,aAAa,EACbC,mBAAmB,QAGd,2BAA2B;AAClC,SACEC,sBAAsB,EACtBC,oBAAoB,QAEf,qBAAqB;AAC5B,SACEC,qBAAqB,EACrBC,oBAAoB,EACpBC,qBAAqB,QAEhB,yCAAyC;AAkBhD,IAAA,AAAMC,uBAAN,MAAMA,6BAA6BC;;IACjC,YAAY,AAAOC,SAAiB,CAAE;QACpC,KAAK,CAAC,CAAC,OAAO,EAAEA,UAAU,uBAAuB,CAAC,QADjCA,YAAAA;QAEjB,IAAI,CAACC,IAAI,GAAG;IACd;AACF;AAEA;;;;;;;;;;;;;;CAcC,GACD,OAAO,MAAMC;;;IACHC,WAAsC,IAAIC,MAAM;IAChDC,SAAwB;IAEhC,YACEA,WAA0Bd,cAAce,OAAO,EAAE,EACjD,AAAiBC,kBAA0Cd,sBAAsB,EACjF,AAAiBe,UAA+B,CAAC,CAAC,CAClD;aAFiBD,kBAAAA;aACAC,UAAAA;QAEjB,IAAI,CAACH,QAAQ,GAAGA;IAClB;IAEA;;;;;;;;;;GAUC,GACDI,gBAAgBC,OAAqB,EAAQ;QAC3C,MAAMC,aAAaD,QAAQE,IAAI;QAE/B,IAAI,IAAI,CAACT,QAAQ,CAACU,GAAG,CAACF,aAAa;YACjC,MAAM,IAAIZ,MAAM,CAAC,kBAAkB,EAAEY,WAAW,uBAAuB,CAAC;QAC1E;QAEA,IAAI,CAACR,QAAQ,CAACW,GAAG,CAACH,YAAYD;IAChC;IAEA;;;;;GAKC,GACDK,kBAAkBH,IAAY,EAAW;QACvC,OAAO,IAAI,CAACT,QAAQ,CAACa,MAAM,CAACJ;IAC9B;IAEA;;;;GAIC,GACDK,cAA8B;QAC5B,OAAOC,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM;IACxC;IAEA;;;;;GAKC,GACDC,WAAWT,IAAY,EAA4B;QACjD,OAAO,IAAI,CAACT,QAAQ,CAACmB,GAAG,CAACV;IAC3B;IAEA;;;;;GAKC,GACDW,WAAWX,IAAY,EAAW;QAChC,OAAO,IAAI,CAACT,QAAQ,CAACU,GAAG,CAACD;IAC3B;IAEA;;;;;;;;;;;;;;;;GAgBC,GACD,MAAMY,WAAWhB,OAA2B,EAAwB;QAClE,MAAMiB,YAAyB,EAAE;QACjC,MAAMC,SAAgD,EAAE;QAExD,MAAMvB,WAAWe,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM;QAChD,MAAMO,eAAeT,MAAMC,IAAI,CAC7B,IAAIS,IACFzB,SAAS0B,OAAO,CAAC,CAACnB,UAAaA,QAAQiB,YAAY,GAAG;mBAAIjB,QAAQiB,YAAY;aAAC,GAAG,EAAE;QAGxF,IAAIG,YAAoC,EAAE;QAC1C,IAAIH,aAAaI,MAAM,GAAG,GAAG;YAC3B,IAAI;gBACFD,YAAY,MAAM,IAAI,CAACvB,eAAe,CAACoB,cAAc;oBACnDK,aAAa,CAACC,UAAY,IAAI,CAACC,kBAAkB,CAAC/B,UAAU8B;gBAC9D;YACF,EAAE,OAAM;gBACNH,YAAY,EAAE;YAChB;QACF;QAEA,wFAAwF;QACxF,MAAMK,kBAAkBhC,SAASiC,GAAG,CAAC,OAAO1B;YAC1C,IAAI;gBACF,MAAM2B,SAAS3B,QAAQiB,YAAY,GAC/B,MAAMjB,QAAQ4B,YAAY,CAAC;oBACzBR,WAAWpC,qBAAqBoC,WAAWpB,QAAQiB,YAAY;gBACjE,KACA,MAAMjB,QAAQ4B,YAAY;gBAC9B,OAAO;oBAAE1B,MAAMF,QAAQE,IAAI;oBAAEyB;oBAAQE,OAAO;gBAAK;YACnD,EAAE,OAAOA,OAAO;gBACd,mEAAmE;gBACnE,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO;oBAAE5B,MAAMF,QAAQE,IAAI;oBAAEyB,QAAQ,EAAE;oBAAEE,OAAOC;gBAAI;YACtD;QACF;QAEA,MAAMG,UAAU,MAAMC,QAAQC,GAAG,CAACV;QAElC,mCAAmC;QACnC,KAAK,MAAMW,UAAUH,QAAS;YAC5B,IAAIG,OAAOP,KAAK,KAAK,MAAM;gBACzBd,UAAUiB,IAAI,IAAII,OAAOT,MAAM;YACjC;QACF;QAEA,mEAAmE;QACnE,IAAIX,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,mBAAmB,CAAC;YAC5DL,OAAOsB,OAAO,CAAC,CAAC,EAAEpC,IAAI,EAAE2B,KAAK,EAAE;gBAC7BQ,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAMC,cAAc,CAACtC,MAAcuC,MAAwB,GAAGvC,KAAK,CAAC,EAAEuC,KAAK;QAC3E,MAAMC,wBAAwB,IAAIhD,IAChC,IAAI,CAACC,QAAQ,CAACgD,IAAI,GAAGjB,GAAG,CAAC,CAACkB,QAAU;gBAACJ,YAAYI,MAAM1C,IAAI,EAAE0C,MAAMH,GAAG;gBAAGG;aAAM;QAEjF,MAAMC,aAAa9B,UAAUM,MAAM,GAAG,IAAI,MAAM,IAAI,CAACyB,iBAAiB,KAAK,EAAE;QAC7E,MAAMC,UAAUhC,UAAUW,GAAG,CAAC,CAACsB,QAC7B,IAAI,CAACC,eAAe,CAClBD,OACAN,sBAAsB9B,GAAG,CAAC4B,YAAYQ,MAAM9C,IAAI,EAAE8C,MAAMP,GAAG,IAC3DI;QAGJ,IAAIE,QAAQ1B,MAAM,GAAG,GAAG,IAAI,CAAC1B,QAAQ,CAACuD,aAAa,CAACH;QACpD,IAAI,CAACpD,QAAQ,CAACwD,UAAU;QAExB,KAAK,MAAMH,SAASjC,UAAW;YAC7B,MAAM6B,QAAQF,sBAAsB9B,GAAG,CAAC4B,YAAYQ,MAAM9C,IAAI,EAAE8C,MAAMP,GAAG;YACzE,IAAIG,OAAO;gBACTI,MAAMzD,IAAI,GAAGqD,MAAMrD,IAAI;gBACvByD,MAAMI,MAAM,GAAGR,MAAMQ,MAAM;gBAC3B,IAAIR,MAAMQ,MAAM,IAAIR,MAAMS,SAAS,EAAEL,MAAMM,UAAU,GAAG,IAAIC,KAAKX,MAAMS,SAAS;YAClF;QACF;QAEA,MAAMG,UAAwB1D,SAAS2D,UAAU;QACjD,OAAO7E,WAAWmC,WAAWyC;IAC/B;IAEA;;;;GAIC,GACD,AAAQhC,mBAAmB/B,QAAiC,EAAE8B,OAAoB,EAAW;QAC3F,OAAO9B,SAASiE,IAAI,CAAC,CAAC1D;YACpB,IAAI,CAACA,QAAQiB,YAAY,EAAE,OAAO;YAClC,IAAIjC,qBAAqB;gBAACuC;aAAQ,EAAEvB,QAAQiB,YAAY,EAAEI,MAAM,KAAK,GAAG,OAAO;YAC/E,IAAI;gBACF,OAAOrB,QAAQ2D,SAAS,CAACpC;YAC3B,EAAE,OAAM;gBACN,OAAO;YACT;QACF;IACF;IAEA,MAAcuB,oBAAwD;QACpE,MAAMc,kBAAkB,MAAM,IAAI,CAACC,sBAAsB;QACzD,IAAID,oBAAoB,SAAS,OAAO,EAAE;QAE1C,IAAI;YACF,OAAO,MAAM,AAAC,CAAA,IAAI,CAAC9D,OAAO,CAACZ,oBAAoB,IAAIA,oBAAmB;QACxE,EAAE,OAAO2C,OAAO;YACd,IAAI,CAAC/B,OAAO,CAACgE,uBAAuB,GAAGjC;YACvC,OAAO,EAAE;QACX;IACF;IAEA,MAAcgC,yBAAoE;QAChF,MAAMD,kBAAkB,IAAI,CAAC9D,OAAO,CAAC8D,eAAe;QACpD,OAAO,OAAOA,oBAAoB,aAAa,MAAMA,oBAAoBA;IAC3E;IAEQX,gBACND,KAAgB,EAChBe,QAAwB,EACxBlB,aAAwC,EAAE,EAC3B;QACf,MAAMtD,OAAOwE,UAAUxE,QAAQyD,MAAMzD,IAAI;QACzC,MAAMyE,qCACJD,UAAUE,YAAY,WAAWC,QAAQpF,oBAAoBiF,UAAUI;QACzE,MAAMC,YAA2B;YAC/B7E;YACAW,MAAM8C,MAAM9C,IAAI;YAChBuC,KAAKO,MAAMP,GAAG;YACdwB,SAASF,UAAUE,WAAW;YAC9BE,YAAYJ,UAAUI,cAAc;YACpCE,KAAKrB,MAAMsB,WAAW;YACtBC,WAAWR,UAAUQ,aAAa,IAAIhB,OAAOiB,WAAW;YACxDC,WAAWzB,MAAMyB,SAAS;YAC1BC,iBAAiB1B,MAAM0B,eAAe,IAAI;YAC1CtB,QAAQW,UAAUX,UAAUJ,MAAMI,MAAM,IAAI;QAC9C;QAEA,IAAI,CAACY,oCAAoC;YACvC,MAAMW,YAAY1F,sBAAsB+D,OAAOH;YAC/C,IAAI8B,WAAW;gBACb,OAAO;oBACL,GAAGP,SAAS;oBACZH,SAAS;oBACTE,YAAYhF,sBAAsBwF,WAAWpF;gBAC/C;YACF;QACF;QAEA,OAAO6E;IACT;IAEAQ,UAAUtF,SAAiB,EAAW;QACpC,MAAMsD,QAAQ,IAAI,CAACjD,QAAQ,CAACkF,MAAM,CAACvF;QACnC,IAAI,CAACsD,SAAS,CAAC,IAAI,CAACjD,QAAQ,CAACmF,OAAO,CAAClC,QAAQ;YAC3C,IAAIA,OAAO,IAAI,CAACjD,QAAQ,CAACoF,KAAK;YAC9B,MAAM,IAAI3F,qBAAqBE;QACjC;QACA,MAAM8D,SAAS,IAAI,CAACzD,QAAQ,CAACiF,SAAS,CAAChC,MAAM1C,IAAI,EAAE0C,MAAMH,GAAG;QAC5D,IAAIW,WAAW,MAAM,MAAM,IAAIhE,qBAAqBE;QACpD,OAAO8D;IACT;IAEA;;;;;;;;;;;GAWC,GACD,MAAM4B,aAAaC,IAA0B,EAA6B;QACxE,MAAMC,iBAAiB1E,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM,IAAIyE,MAAM,CAC9D,CAACnF,UAAYiF,MAAM/E,SAASkF,aAAapF,QAAQE,IAAI,KAAK+E,KAAK/E,IAAI;QAGrE,MAAMc,SAAgD,EAAE;QAExD,MAAMiB,UAAU,MAAMC,QAAQC,GAAG,CAC/B+C,eAAexD,GAAG,CAAC,OAAO1B;YACxB,IAAI;gBACF,OAAO,MAAMA,QAAQgF,YAAY,CAACC;YACpC,EAAE,OAAOpD,OAAO;gBACd,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAId,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,oCAAoC,CAAC;YAC7E,KAAK,MAAM,EAAEnB,IAAI,EAAE2B,KAAK,EAAE,IAAIb,OAAQ;gBACpCqB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAM8C,SAA2B,EAAE;QACnC,KAAK,MAAM1C,QAAQV,QAAS;YAC1BoD,OAAOrD,IAAI,IAAIW;QACjB;QAEA0C,OAAOC,IAAI,CAAC,CAACC,GAAGC,IAAMA,EAAElC,UAAU,CAACmC,OAAO,KAAKF,EAAEjC,UAAU,CAACmC,OAAO;QACnE,OAAOJ;IACT;IAEA,0EAA0E,GAC1E,MAAMK,iBACJjB,SAAiB,EACjBQ,IAAwC,EACb;QAC3B,MAAMC,iBAAiB1E,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM,IAAIyE,MAAM,CAC9D,CAACnF,UAAYiF,MAAM/E,SAASkF,aAAapF,QAAQE,IAAI,KAAK+E,KAAK/E,IAAI;QAErE,MAAMc,SAAgD,EAAE;QAExD,MAAMiB,UAAU,MAAMC,QAAQC,GAAG,CAC/B+C,eAAexD,GAAG,CAAC,OAAO1B;YACxB,IAAI;gBACF,IAAIA,QAAQ0F,gBAAgB,EAAE,OAAO,MAAM1F,QAAQ0F,gBAAgB,CAACjB;gBACpE,MAAMkB,WAAW,MAAM3F,QAAQgF,YAAY,CAACC;gBAC5C,OAAOU,SAASR,MAAM,CAAC,CAACS,UAAYA,QAAQnB,SAAS,KAAKA;YAC5D,EAAE,OAAO5C,OAAO;gBACd,MAAMC,MAAMD,iBAAiBxC,QAAQwC,QAAQ,IAAIxC,MAAM0C,OAAOF;gBAC9Db,OAAOgB,IAAI,CAAC;oBAAE9B,MAAMF,QAAQE,IAAI;oBAAE2B,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAId,OAAOK,MAAM,GAAG,GAAG;YACrBgB,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEb,OAAOK,MAAM,CAAC,yCAAyC,CAAC;YAClF,KAAK,MAAM,EAAEnB,IAAI,EAAE2B,KAAK,EAAE,IAAIb,OAAQ;gBACpCqB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAE3B,KAAK,EAAE,EAAE2B,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,OAAON,QAAQ4D,IAAI;IACrB;IAEA;;;;GAIC,GACDC,kBAA0B;QACxB,OAAO,IAAI,CAACrG,QAAQ,CAACsG,IAAI;IAC3B;IAEA;;GAEC,GACDC,QAAc;QACZ,IAAI,CAACvG,QAAQ,CAACuG,KAAK;IACrB;IAEA;;;;;;GAMC,GACDC,aAAaC,KAAa,EAAEvE,MAAmB,EAAkC;QAC/E,IAAI,CAACuE,SAASvE,OAAON,MAAM,KAAK,GAAG,OAAO;QAE1C,qFAAqF;QACrF,MAAM8E,gBAAgB,IAAI,CAACxG,QAAQ,CAACkF,MAAM,CAACqB;QAC3C,IAAIC,eAAe;YACjB,MAAMC,gBAAgBzE,OAAO0E,IAAI,CAAC,CAACd,IAAMA,EAAE9C,GAAG,KAAK0D,cAAc1D,GAAG;YACpE,IAAI2D,eAAe,OAAOA;QAC5B;QAEA,MAAME,aAAaJ,MAAMK,WAAW;QAEpC,oCAAoC;QACpC,MAAMC,aAAa7E,OAAO0E,IAAI,CAAC,CAACd,IAAMA,EAAEhG,IAAI,CAACgH,WAAW,OAAOD;QAC/D,IAAIE,YAAY,OAAOA;QAEvB,wCAAwC;QACxC,MAAMC,UAAU9E,OAAOwD,MAAM,CAAC,CAACI,IAAMA,EAAEhG,IAAI,CAACgH,WAAW,GAAGG,QAAQ,CAACJ;QAEnE,IAAIG,QAAQpF,MAAM,KAAK,GAAG,OAAOoF,OAAO,CAAC,EAAE;QAC3C,IAAIA,QAAQpF,MAAM,GAAG,GAAG,OAAOoF;QAE/B,OAAO;IACT;AACF"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import * as capacity from "../../capacity/index.js";
|
|
3
|
+
import * as agentManager from "../../index.js";
|
|
4
|
+
const checkedAt = "2026-09-30T12:00:00.000Z";
|
|
5
|
+
const freshAuth = JSON.stringify({
|
|
6
|
+
anthropic: {
|
|
7
|
+
type: "oauth",
|
|
8
|
+
access: "fake-pi-anthropic-token",
|
|
9
|
+
refresh: "fake-pi-anthropic-refresh",
|
|
10
|
+
expires: Date.parse("2026-10-01T12:00:00.000Z")
|
|
11
|
+
}
|
|
12
|
+
});
|
|
13
|
+
describe("Pi Anthropic capacity", ()=>{
|
|
14
|
+
it("exports a Pi Anthropic capacity report reader", ()=>{
|
|
15
|
+
expect(capacity).toHaveProperty("getAnthropicCapacityReport");
|
|
16
|
+
expect(agentManager).toHaveProperty("getAnthropicCapacityReport");
|
|
17
|
+
});
|
|
18
|
+
it("reads Pi's Anthropic OAuth credential and reports the Pi harness", async ()=>{
|
|
19
|
+
const readFile = vi.fn(async ()=>freshAuth);
|
|
20
|
+
const fetch = vi.fn(async ()=>new Response(JSON.stringify({
|
|
21
|
+
five_hour: {
|
|
22
|
+
utilization: 20
|
|
23
|
+
}
|
|
24
|
+
}), {
|
|
25
|
+
status: 200
|
|
26
|
+
}));
|
|
27
|
+
const report = await capacity.getAnthropicCapacityReport({
|
|
28
|
+
now: ()=>new Date(checkedAt),
|
|
29
|
+
env: {
|
|
30
|
+
HOME: "/users/test"
|
|
31
|
+
},
|
|
32
|
+
readFile,
|
|
33
|
+
fetch
|
|
34
|
+
});
|
|
35
|
+
expect(readFile).toHaveBeenCalledWith("/users/test/.pi/agent/auth.json", "utf8");
|
|
36
|
+
expect(fetch).toHaveBeenCalledWith("https://api.anthropic.com/api/oauth/usage", expect.objectContaining({
|
|
37
|
+
headers: expect.objectContaining({
|
|
38
|
+
Authorization: "Bearer fake-pi-anthropic-token"
|
|
39
|
+
})
|
|
40
|
+
}));
|
|
41
|
+
expect(report).toMatchObject({
|
|
42
|
+
harness: "pi",
|
|
43
|
+
provider: "anthropic",
|
|
44
|
+
generatedAt: checkedAt,
|
|
45
|
+
authenticated: true,
|
|
46
|
+
available: "yes"
|
|
47
|
+
});
|
|
48
|
+
expect(JSON.stringify(report)).not.toContain("fake-pi-anthropic-token");
|
|
49
|
+
expect(JSON.stringify(report)).not.toContain("fake-pi-anthropic-refresh");
|
|
50
|
+
});
|
|
51
|
+
it("does not fetch for an expired Pi login and returns a stale report", async ()=>{
|
|
52
|
+
const fetch = vi.fn();
|
|
53
|
+
const report = await capacity.getAnthropicCapacityReport({
|
|
54
|
+
now: ()=>new Date(checkedAt),
|
|
55
|
+
env: {
|
|
56
|
+
HOME: "/users/test"
|
|
57
|
+
},
|
|
58
|
+
readFile: async ()=>JSON.stringify({
|
|
59
|
+
anthropic: {
|
|
60
|
+
type: "oauth",
|
|
61
|
+
access: "fake-expired-token",
|
|
62
|
+
expires: 1
|
|
63
|
+
}
|
|
64
|
+
}),
|
|
65
|
+
fetch
|
|
66
|
+
});
|
|
67
|
+
expect(fetch).not.toHaveBeenCalled();
|
|
68
|
+
expect(report).toEqual({
|
|
69
|
+
harness: "pi",
|
|
70
|
+
provider: "anthropic",
|
|
71
|
+
generatedAt: checkedAt,
|
|
72
|
+
authenticated: true,
|
|
73
|
+
available: "unknown",
|
|
74
|
+
windows: [],
|
|
75
|
+
creditsRemaining: null
|
|
76
|
+
});
|
|
77
|
+
});
|
|
78
|
+
it.each([
|
|
79
|
+
[
|
|
80
|
+
"missing file",
|
|
81
|
+
async ()=>Promise.reject(Object.assign(new Error("missing"), {
|
|
82
|
+
code: "ENOENT"
|
|
83
|
+
}))
|
|
84
|
+
],
|
|
85
|
+
[
|
|
86
|
+
"malformed file",
|
|
87
|
+
async ()=>"{"
|
|
88
|
+
],
|
|
89
|
+
[
|
|
90
|
+
"missing credential",
|
|
91
|
+
async ()=>"{}"
|
|
92
|
+
],
|
|
93
|
+
[
|
|
94
|
+
"malformed credential",
|
|
95
|
+
async ()=>JSON.stringify({
|
|
96
|
+
anthropic: {
|
|
97
|
+
type: "oauth"
|
|
98
|
+
}
|
|
99
|
+
})
|
|
100
|
+
]
|
|
101
|
+
])("rejects a %s without fetching", async (_name, readFile)=>{
|
|
102
|
+
const fetch = vi.fn();
|
|
103
|
+
await expect(capacity.getAnthropicCapacityReport({
|
|
104
|
+
now: ()=>new Date(checkedAt),
|
|
105
|
+
env: {
|
|
106
|
+
HOME: "/users/test"
|
|
107
|
+
},
|
|
108
|
+
readFile,
|
|
109
|
+
fetch
|
|
110
|
+
})).rejects.toThrow(/Pi Anthropic/);
|
|
111
|
+
expect(fetch).not.toHaveBeenCalled();
|
|
112
|
+
});
|
|
113
|
+
});
|
|
114
|
+
|
|
115
|
+
//# sourceMappingURL=anthropic.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../../src/__tests__/capacity/anthropic.test.ts"],"sourcesContent":["import { describe, expect, it, vi } from \"vitest\";\nimport * as capacity from \"../../capacity/index.js\";\nimport * as agentManager from \"../../index.js\";\n\nconst checkedAt = \"2026-09-30T12:00:00.000Z\";\nconst freshAuth = JSON.stringify({\n anthropic: {\n type: \"oauth\",\n access: \"fake-pi-anthropic-token\",\n refresh: \"fake-pi-anthropic-refresh\",\n expires: Date.parse(\"2026-10-01T12:00:00.000Z\"),\n },\n});\n\ndescribe(\"Pi Anthropic capacity\", () => {\n it(\"exports a Pi Anthropic capacity report reader\", () => {\n expect(capacity).toHaveProperty(\"getAnthropicCapacityReport\");\n expect(agentManager).toHaveProperty(\"getAnthropicCapacityReport\");\n });\n\n it(\"reads Pi's Anthropic OAuth credential and reports the Pi harness\", async () => {\n const readFile = vi.fn(async () => freshAuth);\n const fetch = vi.fn(\n async () =>\n new Response(JSON.stringify({ five_hour: { utilization: 20 } }), {\n status: 200,\n }),\n );\n\n const report = await capacity.getAnthropicCapacityReport({\n now: () => new Date(checkedAt),\n env: { HOME: \"/users/test\" },\n readFile,\n fetch,\n });\n\n expect(readFile).toHaveBeenCalledWith(\"/users/test/.pi/agent/auth.json\", \"utf8\");\n expect(fetch).toHaveBeenCalledWith(\n \"https://api.anthropic.com/api/oauth/usage\",\n expect.objectContaining({\n headers: expect.objectContaining({\n Authorization: \"Bearer fake-pi-anthropic-token\",\n }),\n }),\n );\n expect(report).toMatchObject({\n harness: \"pi\",\n provider: \"anthropic\",\n generatedAt: checkedAt,\n authenticated: true,\n available: \"yes\",\n });\n expect(JSON.stringify(report)).not.toContain(\"fake-pi-anthropic-token\");\n expect(JSON.stringify(report)).not.toContain(\"fake-pi-anthropic-refresh\");\n });\n\n it(\"does not fetch for an expired Pi login and returns a stale report\", async () => {\n const fetch = vi.fn();\n const report = await capacity.getAnthropicCapacityReport({\n now: () => new Date(checkedAt),\n env: { HOME: \"/users/test\" },\n readFile: async () =>\n JSON.stringify({\n anthropic: {\n type: \"oauth\",\n access: \"fake-expired-token\",\n expires: 1,\n },\n }),\n fetch,\n });\n\n expect(fetch).not.toHaveBeenCalled();\n expect(report).toEqual({\n harness: \"pi\",\n provider: \"anthropic\",\n generatedAt: checkedAt,\n authenticated: true,\n available: \"unknown\",\n windows: [],\n creditsRemaining: null,\n });\n });\n\n it.each([\n [\n \"missing file\",\n async () => Promise.reject(Object.assign(new Error(\"missing\"), { code: \"ENOENT\" })),\n ],\n [\"malformed file\", async () => \"{\"],\n [\"missing credential\", async () => \"{}\"],\n [\"malformed credential\", async () => JSON.stringify({ anthropic: { type: \"oauth\" } })],\n ])(\"rejects a %s without fetching\", async (_name, readFile) => {\n const fetch = vi.fn();\n await expect(\n capacity.getAnthropicCapacityReport({\n now: () => new Date(checkedAt),\n env: { HOME: \"/users/test\" },\n readFile,\n fetch,\n }),\n ).rejects.toThrow(/Pi Anthropic/);\n expect(fetch).not.toHaveBeenCalled();\n });\n});\n"],"names":["describe","expect","it","vi","capacity","agentManager","checkedAt","freshAuth","JSON","stringify","anthropic","type","access","refresh","expires","Date","parse","toHaveProperty","readFile","fn","fetch","Response","five_hour","utilization","status","report","getAnthropicCapacityReport","now","env","HOME","toHaveBeenCalledWith","objectContaining","headers","Authorization","toMatchObject","harness","provider","generatedAt","authenticated","available","not","toContain","toHaveBeenCalled","toEqual","windows","creditsRemaining","each","Promise","reject","Object","assign","Error","code","_name","rejects","toThrow"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,YAAYC,cAAc,0BAA0B;AACpD,YAAYC,kBAAkB,iBAAiB;AAE/C,MAAMC,YAAY;AAClB,MAAMC,YAAYC,KAAKC,SAAS,CAAC;IAC/BC,WAAW;QACTC,MAAM;QACNC,QAAQ;QACRC,SAAS;QACTC,SAASC,KAAKC,KAAK,CAAC;IACtB;AACF;AAEAhB,SAAS,yBAAyB;IAChCE,GAAG,iDAAiD;QAClDD,OAAOG,UAAUa,cAAc,CAAC;QAChChB,OAAOI,cAAcY,cAAc,CAAC;IACtC;IAEAf,GAAG,oEAAoE;QACrE,MAAMgB,WAAWf,GAAGgB,EAAE,CAAC,UAAYZ;QACnC,MAAMa,QAAQjB,GAAGgB,EAAE,CACjB,UACE,IAAIE,SAASb,KAAKC,SAAS,CAAC;gBAAEa,WAAW;oBAAEC,aAAa;gBAAG;YAAE,IAAI;gBAC/DC,QAAQ;YACV;QAGJ,MAAMC,SAAS,MAAMrB,SAASsB,0BAA0B,CAAC;YACvDC,KAAK,IAAM,IAAIZ,KAAKT;YACpBsB,KAAK;gBAAEC,MAAM;YAAc;YAC3BX;YACAE;QACF;QAEAnB,OAAOiB,UAAUY,oBAAoB,CAAC,mCAAmC;QACzE7B,OAAOmB,OAAOU,oBAAoB,CAChC,6CACA7B,OAAO8B,gBAAgB,CAAC;YACtBC,SAAS/B,OAAO8B,gBAAgB,CAAC;gBAC/BE,eAAe;YACjB;QACF;QAEFhC,OAAOwB,QAAQS,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVC,aAAa/B;YACbgC,eAAe;YACfC,WAAW;QACb;QACAtC,OAAOO,KAAKC,SAAS,CAACgB,SAASe,GAAG,CAACC,SAAS,CAAC;QAC7CxC,OAAOO,KAAKC,SAAS,CAACgB,SAASe,GAAG,CAACC,SAAS,CAAC;IAC/C;IAEAvC,GAAG,qEAAqE;QACtE,MAAMkB,QAAQjB,GAAGgB,EAAE;QACnB,MAAMM,SAAS,MAAMrB,SAASsB,0BAA0B,CAAC;YACvDC,KAAK,IAAM,IAAIZ,KAAKT;YACpBsB,KAAK;gBAAEC,MAAM;YAAc;YAC3BX,UAAU,UACRV,KAAKC,SAAS,CAAC;oBACbC,WAAW;wBACTC,MAAM;wBACNC,QAAQ;wBACRE,SAAS;oBACX;gBACF;YACFM;QACF;QAEAnB,OAAOmB,OAAOoB,GAAG,CAACE,gBAAgB;QAClCzC,OAAOwB,QAAQkB,OAAO,CAAC;YACrBR,SAAS;YACTC,UAAU;YACVC,aAAa/B;YACbgC,eAAe;YACfC,WAAW;YACXK,SAAS,EAAE;YACXC,kBAAkB;QACpB;IACF;IAEA3C,GAAG4C,IAAI,CAAC;QACN;YACE;YACA,UAAYC,QAAQC,MAAM,CAACC,OAAOC,MAAM,CAAC,IAAIC,MAAM,YAAY;oBAAEC,MAAM;gBAAS;SACjF;QACD;YAAC;YAAkB,UAAY;SAAI;QACnC;YAAC;YAAsB,UAAY;SAAK;QACxC;YAAC;YAAwB,UAAY5C,KAAKC,SAAS,CAAC;oBAAEC,WAAW;wBAAEC,MAAM;oBAAQ;gBAAE;SAAG;KACvF,EAAE,iCAAiC,OAAO0C,OAAOnC;QAChD,MAAME,QAAQjB,GAAGgB,EAAE;QACnB,MAAMlB,OACJG,SAASsB,0BAA0B,CAAC;YAClCC,KAAK,IAAM,IAAIZ,KAAKT;YACpBsB,KAAK;gBAAEC,MAAM;YAAc;YAC3BX;YACAE;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAClBtD,OAAOmB,OAAOoB,GAAG,CAACE,gBAAgB;IACpC;AACF"}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
import { describe, expect, it } from "vitest";
|
|
2
|
+
import { parseAnthropicCapacity } from "../../capacity/providers/anthropic.js";
|
|
3
|
+
import { resolvePiAnthropicCredential } from "../../harnesses/pi/credentials.js";
|
|
4
|
+
describe("capacity boundaries", ()=>{
|
|
5
|
+
it("keeps Pi credential extraction separate from Anthropic quota parsing", async ()=>{
|
|
6
|
+
const credential = await resolvePiAnthropicCredential({
|
|
7
|
+
env: {
|
|
8
|
+
HOME: "/users/test"
|
|
9
|
+
},
|
|
10
|
+
readFile: async ()=>JSON.stringify({
|
|
11
|
+
anthropic: {
|
|
12
|
+
type: "oauth",
|
|
13
|
+
access: "fake-boundary-token",
|
|
14
|
+
expires: Date.parse("2026-10-01T00:00:00.000Z")
|
|
15
|
+
}
|
|
16
|
+
})
|
|
17
|
+
});
|
|
18
|
+
const snapshot = parseAnthropicCapacity({
|
|
19
|
+
five_hour: {
|
|
20
|
+
utilization: 25
|
|
21
|
+
}
|
|
22
|
+
});
|
|
23
|
+
expect(credential).toEqual({
|
|
24
|
+
kind: "oauth",
|
|
25
|
+
access: "fake-boundary-token",
|
|
26
|
+
expiresMs: Date.parse("2026-10-01T00:00:00.000Z")
|
|
27
|
+
});
|
|
28
|
+
expect(snapshot).toMatchObject({
|
|
29
|
+
authenticated: true,
|
|
30
|
+
available: "yes",
|
|
31
|
+
windows: [
|
|
32
|
+
{
|
|
33
|
+
id: "session",
|
|
34
|
+
usedPercent: 25
|
|
35
|
+
}
|
|
36
|
+
],
|
|
37
|
+
creditsRemaining: null
|
|
38
|
+
});
|
|
39
|
+
expect(snapshot).not.toHaveProperty("harness");
|
|
40
|
+
expect(snapshot).not.toHaveProperty("provider");
|
|
41
|
+
expect(snapshot).not.toHaveProperty("generatedAt");
|
|
42
|
+
});
|
|
43
|
+
});
|
|
44
|
+
|
|
45
|
+
//# sourceMappingURL=boundaries.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../../src/__tests__/capacity/boundaries.test.ts"],"sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { parseAnthropicCapacity } from \"../../capacity/providers/anthropic.js\";\nimport { resolvePiAnthropicCredential } from \"../../harnesses/pi/credentials.js\";\n\ndescribe(\"capacity boundaries\", () => {\n it(\"keeps Pi credential extraction separate from Anthropic quota parsing\", async () => {\n const credential = await resolvePiAnthropicCredential({\n env: { HOME: \"/users/test\" },\n readFile: async () =>\n JSON.stringify({\n anthropic: {\n type: \"oauth\",\n access: \"fake-boundary-token\",\n expires: Date.parse(\"2026-10-01T00:00:00.000Z\"),\n },\n }),\n });\n\n const snapshot = parseAnthropicCapacity({ five_hour: { utilization: 25 } });\n\n expect(credential).toEqual({\n kind: \"oauth\",\n access: \"fake-boundary-token\",\n expiresMs: Date.parse(\"2026-10-01T00:00:00.000Z\"),\n });\n expect(snapshot).toMatchObject({\n authenticated: true,\n available: \"yes\",\n windows: [{ id: \"session\", usedPercent: 25 }],\n creditsRemaining: null,\n });\n expect(snapshot).not.toHaveProperty(\"harness\");\n expect(snapshot).not.toHaveProperty(\"provider\");\n expect(snapshot).not.toHaveProperty(\"generatedAt\");\n });\n});\n"],"names":["describe","expect","it","parseAnthropicCapacity","resolvePiAnthropicCredential","credential","env","HOME","readFile","JSON","stringify","anthropic","type","access","expires","Date","parse","snapshot","five_hour","utilization","toEqual","kind","expiresMs","toMatchObject","authenticated","available","windows","id","usedPercent","creditsRemaining","not","toHaveProperty"],"mappings":"AAAA,SAASA,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,SAAS;AAC9C,SAASC,sBAAsB,QAAQ,wCAAwC;AAC/E,SAASC,4BAA4B,QAAQ,oCAAoC;AAEjFJ,SAAS,uBAAuB;IAC9BE,GAAG,wEAAwE;QACzE,MAAMG,aAAa,MAAMD,6BAA6B;YACpDE,KAAK;gBAAEC,MAAM;YAAc;YAC3BC,UAAU,UACRC,KAAKC,SAAS,CAAC;oBACbC,WAAW;wBACTC,MAAM;wBACNC,QAAQ;wBACRC,SAASC,KAAKC,KAAK,CAAC;oBACtB;gBACF;QACJ;QAEA,MAAMC,WAAWd,uBAAuB;YAAEe,WAAW;gBAAEC,aAAa;YAAG;QAAE;QAEzElB,OAAOI,YAAYe,OAAO,CAAC;YACzBC,MAAM;YACNR,QAAQ;YACRS,WAAWP,KAAKC,KAAK,CAAC;QACxB;QACAf,OAAOgB,UAAUM,aAAa,CAAC;YAC7BC,eAAe;YACfC,WAAW;YACXC,SAAS;gBAAC;oBAAEC,IAAI;oBAAWC,aAAa;gBAAG;aAAE;YAC7CC,kBAAkB;QACpB;QACA5B,OAAOgB,UAAUa,GAAG,CAACC,cAAc,CAAC;QACpC9B,OAAOgB,UAAUa,GAAG,CAACC,cAAc,CAAC;QACpC9B,OAAOgB,UAAUa,GAAG,CAACC,cAAc,CAAC;IACtC;AACF"}
|
|
@@ -2,7 +2,7 @@ import { readFile } from "node:fs/promises";
|
|
|
2
2
|
import { fileURLToPath } from "node:url";
|
|
3
3
|
import { describe, expect, it, vi } from "vitest";
|
|
4
4
|
import * as capacity from "../../capacity/index.js";
|
|
5
|
-
import {
|
|
5
|
+
import { parseAnthropicCapacity } from "../../capacity/providers/anthropic.js";
|
|
6
6
|
import * as agentManager from "../../index.js";
|
|
7
7
|
const fixture = (name)=>readFile(fileURLToPath(new URL(`./fixtures/${name}`, import.meta.url)), "utf8");
|
|
8
8
|
describe("Claude capacity", ()=>{
|
|
@@ -227,7 +227,7 @@ describe("Claude capacity", ()=>{
|
|
|
227
227
|
})).rejects.toThrow("Claude OAuth usage request failed");
|
|
228
228
|
});
|
|
229
229
|
it("preserves valid windows in a partial payload", async ()=>{
|
|
230
|
-
const report =
|
|
230
|
+
const report = parseAnthropicCapacity(JSON.parse(await fixture("claude-usage-partial.json")));
|
|
231
231
|
expect(report.available).toBe("yes");
|
|
232
232
|
expect(report.windows).toEqual([
|
|
233
233
|
{
|
|
@@ -248,7 +248,7 @@ describe("Claude capacity", ()=>{
|
|
|
248
248
|
});
|
|
249
249
|
it("rejects a non-object usage payload", async ()=>{
|
|
250
250
|
const raw = JSON.parse(await fixture("claude-usage-malformed.json"));
|
|
251
|
-
expect(()=>
|
|
251
|
+
expect(()=>parseAnthropicCapacity(raw)).toThrow("Claude usage response is malformed");
|
|
252
252
|
});
|
|
253
253
|
it.each([
|
|
254
254
|
[
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../../src/__tests__/capacity/claude.test.ts"],"sourcesContent":["import { readFile } from \"node:fs/promises\";\nimport { fileURLToPath } from \"node:url\";\nimport { describe, expect, it, vi } from \"vitest\";\nimport * as capacity from \"../../capacity/index.js\";\nimport { parseClaudeUsage } from \"../../capacity/claude.js\";\nimport * as agentManager from \"../../index.js\";\n\nconst fixture = (name: string) =>\n readFile(fileURLToPath(new URL(`./fixtures/${name}`, import.meta.url)), \"utf8\");\n\ndescribe(\"Claude capacity\", () => {\n it(\"exposes a Claude capacity report reader\", () => {\n expect(capacity).toHaveProperty(\"getClaudeCapacityReport\");\n });\n\n it(\"exports the Claude report reader from agent-manager\", () => {\n expect(agentManager).toHaveProperty(\"getClaudeCapacityReport\");\n });\n\n it(\"uses an environment OAuth token for the exact usage request\", async () => {\n const fetch = vi.fn(\n async () =>\n new Response(JSON.stringify({ five_hour: { utilization: 20 } }), {\n status: 200,\n headers: { \"Content-Type\": \"application/json\" },\n }),\n );\n\n const report = await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-environment-token\" },\n readFile: vi.fn(),\n fetch,\n });\n\n expect(fetch).toHaveBeenCalledOnce();\n expect(fetch).toHaveBeenCalledWith(\n \"https://api.anthropic.com/api/oauth/usage\",\n expect.objectContaining({\n method: \"GET\",\n headers: {\n Accept: \"application/json\",\n Authorization: \"Bearer fake-environment-token\",\n \"Content-Type\": \"application/json\",\n \"User-Agent\": \"claude-code/2.1.0\",\n \"anthropic-beta\": \"oauth-2025-04-20\",\n },\n }),\n );\n expect(report).toMatchObject({\n harness: \"claude\",\n provider: \"anthropic\",\n authenticated: true,\n });\n });\n\n it(\"reads the OAuth token from the active Claude profile\", async () => {\n const readFile = vi.fn(async () => fixture(\"claude-credentials.json\"));\n const fetch = vi.fn(async () => new Response(JSON.stringify({}), { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\", CLAUDE_CONFIG_DIR: \"profiles/work\" },\n cwd: \"/workspace/project\",\n readFile,\n fetch,\n });\n\n expect(readFile).toHaveBeenCalledWith(\n \"/workspace/project/profiles/work/.credentials.json\",\n \"utf8\",\n );\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-claude-oauth-token-for-tests\",\n });\n });\n\n it(\"rejects an expired profile credential before fetching usage\", async () => {\n const fetch = vi.fn();\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"linux\",\n readFile: async () =>\n JSON.stringify({\n claudeAiOauth: {\n accessToken: \"fake-expired-token\",\n expiresAt: 1,\n },\n }),\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials expired\");\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"maps subscription, model, scoped, and extra-usage windows\", async () => {\n const report = await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-mapping-token\" },\n fetch: vi.fn(async () => new Response(await fixture(\"claude-usage.json\"), { status: 200 })),\n });\n\n expect(report).toMatchObject({\n harness: \"claude\",\n provider: \"anthropic\",\n generatedAt: \"2026-09-25T18:00:00.000Z\",\n authenticated: true,\n available: \"yes\",\n creditsRemaining: null,\n });\n expect(report.windows).toEqual([\n {\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 25,\n resetsAt: \"2026-09-25T20:00:00.000Z\",\n },\n {\n id: \"weekly\",\n label: \"Weekly\",\n durationMinutes: 10080,\n usedPercent: 55.5,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:sonnet:weekly\",\n label: \"Sonnet weekly\",\n durationMinutes: 10080,\n usedPercent: 61,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:opus:weekly\",\n label: \"Opus weekly\",\n durationMinutes: 10080,\n usedPercent: 72,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:weekly:claude-fable\",\n label: \"Fable weekly\",\n durationMinutes: 10080,\n usedPercent: 33,\n resetsAt: \"2026-09-29T08:00:00.000Z\",\n },\n {\n id: \"claude:extra-usage\",\n label: \"Extra usage · USD\",\n limitType: \"CREDIT_LIMIT\",\n durationMinutes: null,\n usedPercent: 25,\n resetsAt: null,\n total: 50,\n current: 12.5,\n remaining: 37.5,\n },\n ]);\n });\n\n it(\"reports a rejected OAuth token as unauthorized without leaking it\", async () => {\n const token = \"fake-rejected-token-never-expose\";\n let error: unknown;\n try {\n await capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: token },\n fetch: vi.fn(async () => new Response('{\"private\":\"response-body\"}', { status: 401 })),\n });\n } catch (caught) {\n error = caught;\n }\n expect(error).toBeInstanceOf(Error);\n expect((error as Error).message).toBe(\"Claude OAuth request unauthorized (HTTP 401)\");\n expect((error as Error).message).not.toContain(token);\n expect((error as Error).message).not.toContain(\"response-body\");\n });\n\n it(\"reports forbidden subscription usage without returning the response body\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-forbidden-token\" },\n fetch: vi.fn(async () => new Response(\"private forbidden details\", { status: 403 })),\n }),\n ).rejects.toThrow(\"Claude OAuth request forbidden (HTTP 403)\");\n });\n\n it.each([\n [\"120\", \"2026-09-25T18:02:00.000Z\"],\n [\"Fri, 25 Sep 2026 19:00:00 GMT\", \"2026-09-25T19:00:00.000Z\"],\n ])(\"honors Retry-After %s\", async (retryAfter, expected) => {\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-rate-limited-token\" },\n fetch: vi.fn(\n async () =>\n new Response(\"private rate-limit body\", {\n status: 429,\n headers: { \"Retry-After\": retryAfter },\n }),\n ),\n }),\n ).rejects.toThrow(`Claude OAuth usage rate limited until ${expected}`);\n });\n\n it(\"sanitizes malformed usage JSON\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-malformed-token\" },\n fetch: vi.fn(async () => new Response(\"private non-json response\", { status: 200 })),\n }),\n ).rejects.toThrow(\"Claude usage response is malformed\");\n });\n\n it(\"sanitizes network failures\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-network-token\" },\n fetch: vi.fn(async () => {\n throw new Error(\"private network details fake-network-token\");\n }),\n }),\n ).rejects.toThrow(\"Claude OAuth usage request failed\");\n });\n\n it(\"preserves valid windows in a partial payload\", async () => {\n const report = parseClaudeUsage(\n JSON.parse(await fixture(\"claude-usage-partial.json\")),\n \"2026-09-25T18:00:00.000Z\",\n );\n expect(report.available).toBe(\"yes\");\n expect(report.windows).toEqual([\n {\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 10,\n resetsAt: null,\n },\n {\n id: \"weekly\",\n label: \"Weekly\",\n durationMinutes: 10080,\n usedPercent: null,\n resetsAt: null,\n },\n ]);\n });\n\n it(\"rejects a non-object usage payload\", async () => {\n const raw = JSON.parse(await fixture(\"claude-usage-malformed.json\"));\n expect(() => parseClaudeUsage(raw, \"2026-09-25T18:00:00.000Z\")).toThrow(\n \"Claude usage response is malformed\",\n );\n });\n\n it.each([\n [\"missing file\", async () => Promise.reject(new Error(\"missing\"))],\n [\"malformed file\", async () => \"{\"],\n [\"missing token\", async () => JSON.stringify({ claudeAiOauth: {} })],\n ])(\"rejects %s credentials without fetching\", async (_name, readFile) => {\n const fetch = vi.fn();\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"linux\",\n readFile,\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"uses the default Claude profile under HOME\", async () => {\n const readFile = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-home-token\" } }),\n );\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n readFile,\n fetch: vi.fn(async () => new Response(\"{}\", { status: 200 })),\n });\n expect(readFile).toHaveBeenCalledWith(\"/users/test/.claude/.credentials.json\", \"utf8\");\n });\n\n it(\"falls back to the Claude Code Keychain credential on macOS\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({\n claudeAiOauth: {\n accessToken: \"fake-keychain-token\",\n expiresAt: 1_900_000_000_000,\n },\n }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledWith(\"Claude Code-credentials\");\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-keychain-token\",\n });\n });\n\n it(\"falls back to Keychain when the default profile file has no OAuth token\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-keychain-fallback-token\" } }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => JSON.stringify({}),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-keychain-fallback-token\",\n });\n });\n\n it(\"keeps environment and profile-file credentials ahead of Keychain\", async () => {\n const keychainRead = vi.fn();\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-preferred-environment-token\" },\n platform: \"darwin\",\n keychainRead,\n fetch,\n });\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-preferred-file-token\" } }),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).not.toHaveBeenCalled();\n });\n\n it(\"rejects expired Keychain credentials before fetching usage\", async () => {\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead: async () =>\n JSON.stringify({\n claudeAiOauth: { accessToken: \"fake-expired-keychain-token\", expiresAt: 1 },\n }),\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials expired\");\n\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"uses a fresh Keychain credential when the default profile file is expired\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-fresh-keychain-token\" } }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () =>\n JSON.stringify({\n claudeAiOauth: { accessToken: \"fake-expired-file-token\", expiresAt: 1 },\n }),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-fresh-keychain-token\",\n });\n });\n\n it(\"sanitizes unavailable or malformed Keychain credentials\", async () => {\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"private file error\")),\n keychainRead: async () => \"private malformed keychain payload\",\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"does not mix the default Keychain credential into a custom Claude profile\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-wrong-profile-token\" } }),\n );\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\", CLAUDE_CONFIG_DIR: \"/profiles/work\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead,\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n\n expect(keychainRead).not.toHaveBeenCalled();\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"reports other HTTP failures by status only\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-status-token\" },\n fetch: vi.fn(async () => new Response(\"private service details\", { status: 503 })),\n }),\n ).rejects.toThrow(\"Claude OAuth usage request failed (HTTP 503)\");\n });\n\n it(\"aborts a usage request at the configured timeout\", async () => {\n vi.useFakeTimers();\n try {\n const request = capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-timeout-token\" },\n timeoutMs: 10,\n fetch: vi.fn(\n async (_url, init) =>\n new Promise<Response>((_resolve, reject) => {\n init?.signal?.addEventListener(\"abort\", () =>\n reject(new Error(\"private abort details\")),\n );\n }),\n ),\n });\n const rejection = expect(request).rejects.toThrow(\"Claude OAuth usage request failed\");\n await vi.advanceTimersByTimeAsync(10);\n await rejection;\n } finally {\n vi.useRealTimers();\n }\n });\n});\n"],"names":["readFile","fileURLToPath","describe","expect","it","vi","capacity","parseClaudeUsage","agentManager","fixture","name","URL","url","toHaveProperty","fetch","fn","Response","JSON","stringify","five_hour","utilization","status","headers","report","getClaudeCapacityReport","now","Date","env","CLAUDE_CODE_OAUTH_TOKEN","toHaveBeenCalledOnce","toHaveBeenCalledWith","objectContaining","method","Accept","Authorization","toMatchObject","harness","provider","authenticated","HOME","CLAUDE_CONFIG_DIR","cwd","mock","calls","platform","claudeAiOauth","accessToken","expiresAt","rejects","toThrow","not","toHaveBeenCalled","generatedAt","available","creditsRemaining","windows","toEqual","id","label","durationMinutes","usedPercent","resetsAt","limitType","total","current","remaining","token","error","caught","toBeInstanceOf","Error","message","toBe","toContain","each","retryAfter","expected","parse","raw","Promise","reject","_name","keychainRead","useFakeTimers","request","timeoutMs","_url","init","_resolve","signal","addEventListener","rejection","advanceTimersByTimeAsync","useRealTimers"],"mappings":"AAAA,SAASA,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,aAAa,QAAQ,WAAW;AACzC,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,YAAYC,cAAc,0BAA0B;AACpD,SAASC,gBAAgB,QAAQ,2BAA2B;AAC5D,YAAYC,kBAAkB,iBAAiB;AAE/C,MAAMC,UAAU,CAACC,OACfV,SAASC,cAAc,IAAIU,IAAI,CAAC,WAAW,EAAED,MAAM,EAAE,YAAYE,GAAG,IAAI;AAE1EV,SAAS,mBAAmB;IAC1BE,GAAG,2CAA2C;QAC5CD,OAAOG,UAAUO,cAAc,CAAC;IAClC;IAEAT,GAAG,uDAAuD;QACxDD,OAAOK,cAAcK,cAAc,CAAC;IACtC;IAEAT,GAAG,+DAA+D;QAChE,MAAMU,QAAQT,GAAGU,EAAE,CACjB,UACE,IAAIC,SAASC,KAAKC,SAAS,CAAC;gBAAEC,WAAW;oBAAEC,aAAa;gBAAG;YAAE,IAAI;gBAC/DC,QAAQ;gBACRC,SAAS;oBAAE,gBAAgB;gBAAmB;YAChD;QAGJ,MAAMC,SAAS,MAAMjB,SAASkB,uBAAuB,CAAC;YACpDC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAAyB;YACzD5B,UAAUK,GAAGU,EAAE;YACfD;QACF;QAEAX,OAAOW,OAAOe,oBAAoB;QAClC1B,OAAOW,OAAOgB,oBAAoB,CAChC,6CACA3B,OAAO4B,gBAAgB,CAAC;YACtBC,QAAQ;YACRV,SAAS;gBACPW,QAAQ;gBACRC,eAAe;gBACf,gBAAgB;gBAChB,cAAc;gBACd,kBAAkB;YACpB;QACF;QAEF/B,OAAOoB,QAAQY,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVC,eAAe;QACjB;IACF;IAEAlC,GAAG,wDAAwD;QACzD,MAAMJ,WAAWK,GAAGU,EAAE,CAAC,UAAYN,QAAQ;QAC3C,MAAMK,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAASC,KAAKC,SAAS,CAAC,CAAC,IAAI;gBAAEG,QAAQ;YAAI;QAE/E,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;gBAAeC,mBAAmB;YAAgB;YAC/DC,KAAK;YACLzC;YACAc;QACF;QAEAX,OAAOH,UAAU8B,oBAAoB,CACnC,sDACA;QAEF3B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,+DAA+D;QAChE,MAAMU,QAAQT,GAAGU,EAAE;QACnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBACbC,aAAa;wBACbC,WAAW;oBACb;gBACF;YACFjC;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAClB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6DAA6D;QAC9D,MAAMmB,SAAS,MAAMjB,SAASkB,uBAAuB,CAAC;YACpDC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAAqB;YACrDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAMP,QAAQ,sBAAsB;oBAAEY,QAAQ;gBAAI;QAC1F;QAEAlB,OAAOoB,QAAQY,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVe,aAAa;YACbd,eAAe;YACfe,WAAW;YACXC,kBAAkB;QACpB;QACAnD,OAAOoB,OAAOgC,OAAO,EAAEC,OAAO,CAAC;YAC7B;gBACEC,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPI,WAAW;gBACXH,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;gBACVE,OAAO;gBACPC,SAAS;gBACTC,WAAW;YACb;SACD;IACH;IAEA7D,GAAG,qEAAqE;QACtE,MAAM8D,QAAQ;QACd,IAAIC;QACJ,IAAI;YACF,MAAM7D,SAASkB,uBAAuB,CAAC;gBACrCG,KAAK;oBAAEC,yBAAyBsC;gBAAM;gBACtCpD,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,+BAA+B;wBAAEK,QAAQ;oBAAI;YACrF;QACF,EAAE,OAAO+C,QAAQ;YACfD,QAAQC;QACV;QACAjE,OAAOgE,OAAOE,cAAc,CAACC;QAC7BnE,OAAO,AAACgE,MAAgBI,OAAO,EAAEC,IAAI,CAAC;QACtCrE,OAAO,AAACgE,MAAgBI,OAAO,EAAErB,GAAG,CAACuB,SAAS,CAACP;QAC/C/D,OAAO,AAACgE,MAAgBI,OAAO,EAAErB,GAAG,CAACuB,SAAS,CAAC;IACjD;IAEArE,GAAG,4EAA4E;QAC7E,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAuB;YACvDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,6BAA6B;oBAAEK,QAAQ;gBAAI;QACnF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAGsE,IAAI,CAAC;QACN;YAAC;YAAO;SAA2B;QACnC;YAAC;YAAiC;SAA2B;KAC9D,EAAE,yBAAyB,OAAOC,YAAYC;QAC7C,MAAMzE,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAA0B;YAC1Dd,OAAOT,GAAGU,EAAE,CACV,UACE,IAAIC,SAAS,2BAA2B;oBACtCK,QAAQ;oBACRC,SAAS;wBAAE,eAAeqD;oBAAW;gBACvC;QAEN,IACA3B,OAAO,CAACC,OAAO,CAAC,CAAC,sCAAsC,EAAE2B,UAAU;IACvE;IAEAxE,GAAG,kCAAkC;QACnC,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAuB;YACvDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,6BAA6B;oBAAEK,QAAQ;gBAAI;QACnF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,8BAA8B;QAC/B,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAqB;YACrDd,OAAOT,GAAGU,EAAE,CAAC;gBACX,MAAM,IAAIuD,MAAM;YAClB;QACF,IACAtB,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,gDAAgD;QACjD,MAAMmB,SAAShB,iBACbU,KAAK4D,KAAK,CAAC,MAAMpE,QAAQ,+BACzB;QAEFN,OAAOoB,OAAO8B,SAAS,EAAEmB,IAAI,CAAC;QAC9BrE,OAAOoB,OAAOgC,OAAO,EAAEC,OAAO,CAAC;YAC7B;gBACEC,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;SACD;IACH;IAEAzD,GAAG,sCAAsC;QACvC,MAAM0E,MAAM7D,KAAK4D,KAAK,CAAC,MAAMpE,QAAQ;QACrCN,OAAO,IAAMI,iBAAiBuE,KAAK,6BAA6B7B,OAAO,CACrE;IAEJ;IAEA7C,GAAGsE,IAAI,CAAC;QACN;YAAC;YAAgB,UAAYK,QAAQC,MAAM,CAAC,IAAIV,MAAM;SAAY;QAClE;YAAC;YAAkB,UAAY;SAAI;QACnC;YAAC;YAAiB,UAAYrD,KAAKC,SAAS,CAAC;oBAAE2B,eAAe,CAAC;gBAAE;SAAG;KACrE,EAAE,2CAA2C,OAAOoC,OAAOjF;QAC1D,MAAMc,QAAQT,GAAGU,EAAE;QACnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C;YACAc;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAClB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,8CAA8C;QAC/C,MAAMJ,WAAWK,GAAGU,EAAE,CAAC,UACrBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAAkB;YAAE;QAErE,MAAMxC,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BvC;YACAc,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;oBAAEK,QAAQ;gBAAI;QAC5D;QACAlB,OAAOH,UAAU8B,oBAAoB,CAAC,yCAAyC;IACjF;IAEA1B,GAAG,8DAA8D;QAC/D,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBACb2B,eAAe;oBACbC,aAAa;oBACbC,WAAW;gBACb;YACF;QAEF,MAAMjC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY;YACApE;QACF;QAEAX,OAAO+E,cAAcpD,oBAAoB,CAAC;QAC1C3B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,2EAA2E;QAC5E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA+B;YAAE;QAElF,MAAMhC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAYiB,KAAKC,SAAS,CAAC,CAAC;YACtCgE;YACApE;QACF;QAEAX,OAAO+E,cAAcrD,oBAAoB;QACzC1B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,oEAAoE;QACrE,MAAM8E,eAAe7E,GAAGU,EAAE;QAC1B,MAAMD,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEC,yBAAyB;YAAmC;YACnEgB,UAAU;YACVsC;YACApE;QACF;QACA,MAAMR,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBAAE2B,eAAe;wBAAEC,aAAa;oBAA4B;gBAAE;YAC/EoC;YACApE;QACF;QAEAX,OAAO+E,cAAchC,GAAG,CAACC,gBAAgB;IAC3C;IAEA/C,GAAG,8DAA8D;QAC/D,MAAMU,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY,cAAc,UACZjE,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBAAEC,aAAa;wBAA+BC,WAAW;oBAAE;gBAC5E;YACFjC;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6EAA6E;QAC9E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA4B;YAAE;QAE/E,MAAMhC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBAAEC,aAAa;wBAA2BC,WAAW;oBAAE;gBACxE;YACFmC;YACApE;QACF;QAEAX,OAAO+E,cAAcrD,oBAAoB;QACzC1B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,2DAA2D;QAC5D,MAAMU,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY,cAAc,UAAY;YAC1BpE;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6EAA6E;QAC9E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA2B;YAAE;QAE9E,MAAMhC,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;gBAAeC,mBAAmB;YAAiB;YAChEI,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY;YACApE;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAO+E,cAAchC,GAAG,CAACC,gBAAgB;QACzChD,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,8CAA8C;QAC/C,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAoB;YACpDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,2BAA2B;oBAAEK,QAAQ;gBAAI;QACjF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,oDAAoD;QACrDC,GAAG8E,aAAa;QAChB,IAAI;YACF,MAAMC,UAAU9E,SAASkB,uBAAuB,CAAC;gBAC/CG,KAAK;oBAAEC,yBAAyB;gBAAqB;gBACrDyD,WAAW;gBACXvE,OAAOT,GAAGU,EAAE,CACV,OAAOuE,MAAMC,OACX,IAAIR,QAAkB,CAACS,UAAUR;wBAC/BO,MAAME,QAAQC,iBAAiB,SAAS,IACtCV,OAAO,IAAIV,MAAM;oBAErB;YAEN;YACA,MAAMqB,YAAYxF,OAAOiF,SAASpC,OAAO,CAACC,OAAO,CAAC;YAClD,MAAM5C,GAAGuF,wBAAwB,CAAC;YAClC,MAAMD;QACR,SAAU;YACRtF,GAAGwF,aAAa;QAClB;IACF;AACF"}
|
|
1
|
+
{"version":3,"sources":["../../../src/__tests__/capacity/claude.test.ts"],"sourcesContent":["import { readFile } from \"node:fs/promises\";\nimport { fileURLToPath } from \"node:url\";\nimport { describe, expect, it, vi } from \"vitest\";\nimport * as capacity from \"../../capacity/index.js\";\nimport { parseAnthropicCapacity } from \"../../capacity/providers/anthropic.js\";\nimport * as agentManager from \"../../index.js\";\n\nconst fixture = (name: string) =>\n readFile(fileURLToPath(new URL(`./fixtures/${name}`, import.meta.url)), \"utf8\");\n\ndescribe(\"Claude capacity\", () => {\n it(\"exposes a Claude capacity report reader\", () => {\n expect(capacity).toHaveProperty(\"getClaudeCapacityReport\");\n });\n\n it(\"exports the Claude report reader from agent-manager\", () => {\n expect(agentManager).toHaveProperty(\"getClaudeCapacityReport\");\n });\n\n it(\"uses an environment OAuth token for the exact usage request\", async () => {\n const fetch = vi.fn(\n async () =>\n new Response(JSON.stringify({ five_hour: { utilization: 20 } }), {\n status: 200,\n headers: { \"Content-Type\": \"application/json\" },\n }),\n );\n\n const report = await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-environment-token\" },\n readFile: vi.fn(),\n fetch,\n });\n\n expect(fetch).toHaveBeenCalledOnce();\n expect(fetch).toHaveBeenCalledWith(\n \"https://api.anthropic.com/api/oauth/usage\",\n expect.objectContaining({\n method: \"GET\",\n headers: {\n Accept: \"application/json\",\n Authorization: \"Bearer fake-environment-token\",\n \"Content-Type\": \"application/json\",\n \"User-Agent\": \"claude-code/2.1.0\",\n \"anthropic-beta\": \"oauth-2025-04-20\",\n },\n }),\n );\n expect(report).toMatchObject({\n harness: \"claude\",\n provider: \"anthropic\",\n authenticated: true,\n });\n });\n\n it(\"reads the OAuth token from the active Claude profile\", async () => {\n const readFile = vi.fn(async () => fixture(\"claude-credentials.json\"));\n const fetch = vi.fn(async () => new Response(JSON.stringify({}), { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\", CLAUDE_CONFIG_DIR: \"profiles/work\" },\n cwd: \"/workspace/project\",\n readFile,\n fetch,\n });\n\n expect(readFile).toHaveBeenCalledWith(\n \"/workspace/project/profiles/work/.credentials.json\",\n \"utf8\",\n );\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-claude-oauth-token-for-tests\",\n });\n });\n\n it(\"rejects an expired profile credential before fetching usage\", async () => {\n const fetch = vi.fn();\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"linux\",\n readFile: async () =>\n JSON.stringify({\n claudeAiOauth: {\n accessToken: \"fake-expired-token\",\n expiresAt: 1,\n },\n }),\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials expired\");\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"maps subscription, model, scoped, and extra-usage windows\", async () => {\n const report = await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-mapping-token\" },\n fetch: vi.fn(async () => new Response(await fixture(\"claude-usage.json\"), { status: 200 })),\n });\n\n expect(report).toMatchObject({\n harness: \"claude\",\n provider: \"anthropic\",\n generatedAt: \"2026-09-25T18:00:00.000Z\",\n authenticated: true,\n available: \"yes\",\n creditsRemaining: null,\n });\n expect(report.windows).toEqual([\n {\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 25,\n resetsAt: \"2026-09-25T20:00:00.000Z\",\n },\n {\n id: \"weekly\",\n label: \"Weekly\",\n durationMinutes: 10080,\n usedPercent: 55.5,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:sonnet:weekly\",\n label: \"Sonnet weekly\",\n durationMinutes: 10080,\n usedPercent: 61,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:opus:weekly\",\n label: \"Opus weekly\",\n durationMinutes: 10080,\n usedPercent: 72,\n resetsAt: \"2026-09-30T12:00:00.000Z\",\n },\n {\n id: \"claude:weekly:claude-fable\",\n label: \"Fable weekly\",\n durationMinutes: 10080,\n usedPercent: 33,\n resetsAt: \"2026-09-29T08:00:00.000Z\",\n },\n {\n id: \"claude:extra-usage\",\n label: \"Extra usage · USD\",\n limitType: \"CREDIT_LIMIT\",\n durationMinutes: null,\n usedPercent: 25,\n resetsAt: null,\n total: 50,\n current: 12.5,\n remaining: 37.5,\n },\n ]);\n });\n\n it(\"reports a rejected OAuth token as unauthorized without leaking it\", async () => {\n const token = \"fake-rejected-token-never-expose\";\n let error: unknown;\n try {\n await capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: token },\n fetch: vi.fn(async () => new Response('{\"private\":\"response-body\"}', { status: 401 })),\n });\n } catch (caught) {\n error = caught;\n }\n expect(error).toBeInstanceOf(Error);\n expect((error as Error).message).toBe(\"Claude OAuth request unauthorized (HTTP 401)\");\n expect((error as Error).message).not.toContain(token);\n expect((error as Error).message).not.toContain(\"response-body\");\n });\n\n it(\"reports forbidden subscription usage without returning the response body\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-forbidden-token\" },\n fetch: vi.fn(async () => new Response(\"private forbidden details\", { status: 403 })),\n }),\n ).rejects.toThrow(\"Claude OAuth request forbidden (HTTP 403)\");\n });\n\n it.each([\n [\"120\", \"2026-09-25T18:02:00.000Z\"],\n [\"Fri, 25 Sep 2026 19:00:00 GMT\", \"2026-09-25T19:00:00.000Z\"],\n ])(\"honors Retry-After %s\", async (retryAfter, expected) => {\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-rate-limited-token\" },\n fetch: vi.fn(\n async () =>\n new Response(\"private rate-limit body\", {\n status: 429,\n headers: { \"Retry-After\": retryAfter },\n }),\n ),\n }),\n ).rejects.toThrow(`Claude OAuth usage rate limited until ${expected}`);\n });\n\n it(\"sanitizes malformed usage JSON\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-malformed-token\" },\n fetch: vi.fn(async () => new Response(\"private non-json response\", { status: 200 })),\n }),\n ).rejects.toThrow(\"Claude usage response is malformed\");\n });\n\n it(\"sanitizes network failures\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-network-token\" },\n fetch: vi.fn(async () => {\n throw new Error(\"private network details fake-network-token\");\n }),\n }),\n ).rejects.toThrow(\"Claude OAuth usage request failed\");\n });\n\n it(\"preserves valid windows in a partial payload\", async () => {\n const report = parseAnthropicCapacity(JSON.parse(await fixture(\"claude-usage-partial.json\")));\n expect(report.available).toBe(\"yes\");\n expect(report.windows).toEqual([\n {\n id: \"session\",\n label: \"Session\",\n durationMinutes: 300,\n usedPercent: 10,\n resetsAt: null,\n },\n {\n id: \"weekly\",\n label: \"Weekly\",\n durationMinutes: 10080,\n usedPercent: null,\n resetsAt: null,\n },\n ]);\n });\n\n it(\"rejects a non-object usage payload\", async () => {\n const raw = JSON.parse(await fixture(\"claude-usage-malformed.json\"));\n expect(() => parseAnthropicCapacity(raw)).toThrow(\"Claude usage response is malformed\");\n });\n\n it.each([\n [\"missing file\", async () => Promise.reject(new Error(\"missing\"))],\n [\"malformed file\", async () => \"{\"],\n [\"missing token\", async () => JSON.stringify({ claudeAiOauth: {} })],\n ])(\"rejects %s credentials without fetching\", async (_name, readFile) => {\n const fetch = vi.fn();\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"linux\",\n readFile,\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"uses the default Claude profile under HOME\", async () => {\n const readFile = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-home-token\" } }),\n );\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n readFile,\n fetch: vi.fn(async () => new Response(\"{}\", { status: 200 })),\n });\n expect(readFile).toHaveBeenCalledWith(\"/users/test/.claude/.credentials.json\", \"utf8\");\n });\n\n it(\"falls back to the Claude Code Keychain credential on macOS\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({\n claudeAiOauth: {\n accessToken: \"fake-keychain-token\",\n expiresAt: 1_900_000_000_000,\n },\n }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledWith(\"Claude Code-credentials\");\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-keychain-token\",\n });\n });\n\n it(\"falls back to Keychain when the default profile file has no OAuth token\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-keychain-fallback-token\" } }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => JSON.stringify({}),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-keychain-fallback-token\",\n });\n });\n\n it(\"keeps environment and profile-file credentials ahead of Keychain\", async () => {\n const keychainRead = vi.fn();\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-preferred-environment-token\" },\n platform: \"darwin\",\n keychainRead,\n fetch,\n });\n await capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-preferred-file-token\" } }),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).not.toHaveBeenCalled();\n });\n\n it(\"rejects expired Keychain credentials before fetching usage\", async () => {\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead: async () =>\n JSON.stringify({\n claudeAiOauth: { accessToken: \"fake-expired-keychain-token\", expiresAt: 1 },\n }),\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials expired\");\n\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"uses a fresh Keychain credential when the default profile file is expired\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-fresh-keychain-token\" } }),\n );\n const fetch = vi.fn(async () => new Response(\"{}\", { status: 200 }));\n\n await capacity.getClaudeCapacityReport({\n now: () => new Date(\"2026-09-25T18:00:00.000Z\"),\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () =>\n JSON.stringify({\n claudeAiOauth: { accessToken: \"fake-expired-file-token\", expiresAt: 1 },\n }),\n keychainRead,\n fetch,\n });\n\n expect(keychainRead).toHaveBeenCalledOnce();\n expect(fetch.mock.calls[0][1]?.headers).toMatchObject({\n Authorization: \"Bearer fake-fresh-keychain-token\",\n });\n });\n\n it(\"sanitizes unavailable or malformed Keychain credentials\", async () => {\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"private file error\")),\n keychainRead: async () => \"private malformed keychain payload\",\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"does not mix the default Keychain credential into a custom Claude profile\", async () => {\n const keychainRead = vi.fn(async () =>\n JSON.stringify({ claudeAiOauth: { accessToken: \"fake-wrong-profile-token\" } }),\n );\n const fetch = vi.fn();\n\n await expect(\n capacity.getClaudeCapacityReport({\n env: { HOME: \"/users/test\", CLAUDE_CONFIG_DIR: \"/profiles/work\" },\n platform: \"darwin\",\n readFile: async () => Promise.reject(new Error(\"missing\")),\n keychainRead,\n fetch,\n }),\n ).rejects.toThrow(\"Claude OAuth credentials not found or malformed\");\n\n expect(keychainRead).not.toHaveBeenCalled();\n expect(fetch).not.toHaveBeenCalled();\n });\n\n it(\"reports other HTTP failures by status only\", async () => {\n await expect(\n capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-status-token\" },\n fetch: vi.fn(async () => new Response(\"private service details\", { status: 503 })),\n }),\n ).rejects.toThrow(\"Claude OAuth usage request failed (HTTP 503)\");\n });\n\n it(\"aborts a usage request at the configured timeout\", async () => {\n vi.useFakeTimers();\n try {\n const request = capacity.getClaudeCapacityReport({\n env: { CLAUDE_CODE_OAUTH_TOKEN: \"fake-timeout-token\" },\n timeoutMs: 10,\n fetch: vi.fn(\n async (_url, init) =>\n new Promise<Response>((_resolve, reject) => {\n init?.signal?.addEventListener(\"abort\", () =>\n reject(new Error(\"private abort details\")),\n );\n }),\n ),\n });\n const rejection = expect(request).rejects.toThrow(\"Claude OAuth usage request failed\");\n await vi.advanceTimersByTimeAsync(10);\n await rejection;\n } finally {\n vi.useRealTimers();\n }\n });\n});\n"],"names":["readFile","fileURLToPath","describe","expect","it","vi","capacity","parseAnthropicCapacity","agentManager","fixture","name","URL","url","toHaveProperty","fetch","fn","Response","JSON","stringify","five_hour","utilization","status","headers","report","getClaudeCapacityReport","now","Date","env","CLAUDE_CODE_OAUTH_TOKEN","toHaveBeenCalledOnce","toHaveBeenCalledWith","objectContaining","method","Accept","Authorization","toMatchObject","harness","provider","authenticated","HOME","CLAUDE_CONFIG_DIR","cwd","mock","calls","platform","claudeAiOauth","accessToken","expiresAt","rejects","toThrow","not","toHaveBeenCalled","generatedAt","available","creditsRemaining","windows","toEqual","id","label","durationMinutes","usedPercent","resetsAt","limitType","total","current","remaining","token","error","caught","toBeInstanceOf","Error","message","toBe","toContain","each","retryAfter","expected","parse","raw","Promise","reject","_name","keychainRead","useFakeTimers","request","timeoutMs","_url","init","_resolve","signal","addEventListener","rejection","advanceTimersByTimeAsync","useRealTimers"],"mappings":"AAAA,SAASA,QAAQ,QAAQ,mBAAmB;AAC5C,SAASC,aAAa,QAAQ,WAAW;AACzC,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,EAAEC,EAAE,QAAQ,SAAS;AAClD,YAAYC,cAAc,0BAA0B;AACpD,SAASC,sBAAsB,QAAQ,wCAAwC;AAC/E,YAAYC,kBAAkB,iBAAiB;AAE/C,MAAMC,UAAU,CAACC,OACfV,SAASC,cAAc,IAAIU,IAAI,CAAC,WAAW,EAAED,MAAM,EAAE,YAAYE,GAAG,IAAI;AAE1EV,SAAS,mBAAmB;IAC1BE,GAAG,2CAA2C;QAC5CD,OAAOG,UAAUO,cAAc,CAAC;IAClC;IAEAT,GAAG,uDAAuD;QACxDD,OAAOK,cAAcK,cAAc,CAAC;IACtC;IAEAT,GAAG,+DAA+D;QAChE,MAAMU,QAAQT,GAAGU,EAAE,CACjB,UACE,IAAIC,SAASC,KAAKC,SAAS,CAAC;gBAAEC,WAAW;oBAAEC,aAAa;gBAAG;YAAE,IAAI;gBAC/DC,QAAQ;gBACRC,SAAS;oBAAE,gBAAgB;gBAAmB;YAChD;QAGJ,MAAMC,SAAS,MAAMjB,SAASkB,uBAAuB,CAAC;YACpDC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAAyB;YACzD5B,UAAUK,GAAGU,EAAE;YACfD;QACF;QAEAX,OAAOW,OAAOe,oBAAoB;QAClC1B,OAAOW,OAAOgB,oBAAoB,CAChC,6CACA3B,OAAO4B,gBAAgB,CAAC;YACtBC,QAAQ;YACRV,SAAS;gBACPW,QAAQ;gBACRC,eAAe;gBACf,gBAAgB;gBAChB,cAAc;gBACd,kBAAkB;YACpB;QACF;QAEF/B,OAAOoB,QAAQY,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVC,eAAe;QACjB;IACF;IAEAlC,GAAG,wDAAwD;QACzD,MAAMJ,WAAWK,GAAGU,EAAE,CAAC,UAAYN,QAAQ;QAC3C,MAAMK,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAASC,KAAKC,SAAS,CAAC,CAAC,IAAI;gBAAEG,QAAQ;YAAI;QAE/E,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;gBAAeC,mBAAmB;YAAgB;YAC/DC,KAAK;YACLzC;YACAc;QACF;QAEAX,OAAOH,UAAU8B,oBAAoB,CACnC,sDACA;QAEF3B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,+DAA+D;QAChE,MAAMU,QAAQT,GAAGU,EAAE;QACnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBACbC,aAAa;wBACbC,WAAW;oBACb;gBACF;YACFjC;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAClB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6DAA6D;QAC9D,MAAMmB,SAAS,MAAMjB,SAASkB,uBAAuB,CAAC;YACpDC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAAqB;YACrDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAMP,QAAQ,sBAAsB;oBAAEY,QAAQ;gBAAI;QAC1F;QAEAlB,OAAOoB,QAAQY,aAAa,CAAC;YAC3BC,SAAS;YACTC,UAAU;YACVe,aAAa;YACbd,eAAe;YACfe,WAAW;YACXC,kBAAkB;QACpB;QACAnD,OAAOoB,OAAOgC,OAAO,EAAEC,OAAO,CAAC;YAC7B;gBACEC,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPI,WAAW;gBACXH,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;gBACVE,OAAO;gBACPC,SAAS;gBACTC,WAAW;YACb;SACD;IACH;IAEA7D,GAAG,qEAAqE;QACtE,MAAM8D,QAAQ;QACd,IAAIC;QACJ,IAAI;YACF,MAAM7D,SAASkB,uBAAuB,CAAC;gBACrCG,KAAK;oBAAEC,yBAAyBsC;gBAAM;gBACtCpD,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,+BAA+B;wBAAEK,QAAQ;oBAAI;YACrF;QACF,EAAE,OAAO+C,QAAQ;YACfD,QAAQC;QACV;QACAjE,OAAOgE,OAAOE,cAAc,CAACC;QAC7BnE,OAAO,AAACgE,MAAgBI,OAAO,EAAEC,IAAI,CAAC;QACtCrE,OAAO,AAACgE,MAAgBI,OAAO,EAAErB,GAAG,CAACuB,SAAS,CAACP;QAC/C/D,OAAO,AAACgE,MAAgBI,OAAO,EAAErB,GAAG,CAACuB,SAAS,CAAC;IACjD;IAEArE,GAAG,4EAA4E;QAC7E,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAuB;YACvDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,6BAA6B;oBAAEK,QAAQ;gBAAI;QACnF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAGsE,IAAI,CAAC;QACN;YAAC;YAAO;SAA2B;QACnC;YAAC;YAAiC;SAA2B;KAC9D,EAAE,yBAAyB,OAAOC,YAAYC;QAC7C,MAAMzE,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEC,yBAAyB;YAA0B;YAC1Dd,OAAOT,GAAGU,EAAE,CACV,UACE,IAAIC,SAAS,2BAA2B;oBACtCK,QAAQ;oBACRC,SAAS;wBAAE,eAAeqD;oBAAW;gBACvC;QAEN,IACA3B,OAAO,CAACC,OAAO,CAAC,CAAC,sCAAsC,EAAE2B,UAAU;IACvE;IAEAxE,GAAG,kCAAkC;QACnC,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAuB;YACvDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,6BAA6B;oBAAEK,QAAQ;gBAAI;QACnF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,8BAA8B;QAC/B,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAqB;YACrDd,OAAOT,GAAGU,EAAE,CAAC;gBACX,MAAM,IAAIuD,MAAM;YAClB;QACF,IACAtB,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,gDAAgD;QACjD,MAAMmB,SAAShB,uBAAuBU,KAAK4D,KAAK,CAAC,MAAMpE,QAAQ;QAC/DN,OAAOoB,OAAO8B,SAAS,EAAEmB,IAAI,CAAC;QAC9BrE,OAAOoB,OAAOgC,OAAO,EAAEC,OAAO,CAAC;YAC7B;gBACEC,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;YACA;gBACEJ,IAAI;gBACJC,OAAO;gBACPC,iBAAiB;gBACjBC,aAAa;gBACbC,UAAU;YACZ;SACD;IACH;IAEAzD,GAAG,sCAAsC;QACvC,MAAM0E,MAAM7D,KAAK4D,KAAK,CAAC,MAAMpE,QAAQ;QACrCN,OAAO,IAAMI,uBAAuBuE,MAAM7B,OAAO,CAAC;IACpD;IAEA7C,GAAGsE,IAAI,CAAC;QACN;YAAC;YAAgB,UAAYK,QAAQC,MAAM,CAAC,IAAIV,MAAM;SAAY;QAClE;YAAC;YAAkB,UAAY;SAAI;QACnC;YAAC;YAAiB,UAAYrD,KAAKC,SAAS,CAAC;oBAAE2B,eAAe,CAAC;gBAAE;SAAG;KACrE,EAAE,2CAA2C,OAAOoC,OAAOjF;QAC1D,MAAMc,QAAQT,GAAGU,EAAE;QACnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C;YACAc;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAClB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,8CAA8C;QAC/C,MAAMJ,WAAWK,GAAGU,EAAE,CAAC,UACrBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAAkB;YAAE;QAErE,MAAMxC,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BvC;YACAc,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;oBAAEK,QAAQ;gBAAI;QAC5D;QACAlB,OAAOH,UAAU8B,oBAAoB,CAAC,yCAAyC;IACjF;IAEA1B,GAAG,8DAA8D;QAC/D,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBACb2B,eAAe;oBACbC,aAAa;oBACbC,WAAW;gBACb;YACF;QAEF,MAAMjC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY;YACApE;QACF;QAEAX,OAAO+E,cAAcpD,oBAAoB,CAAC;QAC1C3B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,2EAA2E;QAC5E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA+B;YAAE;QAElF,MAAMhC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAYiB,KAAKC,SAAS,CAAC,CAAC;YACtCgE;YACApE;QACF;QAEAX,OAAO+E,cAAcrD,oBAAoB;QACzC1B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,oEAAoE;QACrE,MAAM8E,eAAe7E,GAAGU,EAAE;QAC1B,MAAMD,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEC,yBAAyB;YAAmC;YACnEgB,UAAU;YACVsC;YACApE;QACF;QACA,MAAMR,SAASkB,uBAAuB,CAAC;YACrCG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBAAE2B,eAAe;wBAAEC,aAAa;oBAA4B;gBAAE;YAC/EoC;YACApE;QACF;QAEAX,OAAO+E,cAAchC,GAAG,CAACC,gBAAgB;IAC3C;IAEA/C,GAAG,8DAA8D;QAC/D,MAAMU,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY,cAAc,UACZjE,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBAAEC,aAAa;wBAA+BC,WAAW;oBAAE;gBAC5E;YACFjC;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6EAA6E;QAC9E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA4B;YAAE;QAE/E,MAAMhC,QAAQT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,MAAM;gBAAEK,QAAQ;YAAI;QAEjE,MAAMf,SAASkB,uBAAuB,CAAC;YACrCC,KAAK,IAAM,IAAIC,KAAK;YACpBC,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UACRiB,KAAKC,SAAS,CAAC;oBACb2B,eAAe;wBAAEC,aAAa;wBAA2BC,WAAW;oBAAE;gBACxE;YACFmC;YACApE;QACF;QAEAX,OAAO+E,cAAcrD,oBAAoB;QACzC1B,OAAOW,MAAM4B,IAAI,CAACC,KAAK,CAAC,EAAE,CAAC,EAAE,EAAErB,SAASa,aAAa,CAAC;YACpDD,eAAe;QACjB;IACF;IAEA9B,GAAG,2DAA2D;QAC5D,MAAMU,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;YAAc;YAC3BK,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY,cAAc,UAAY;YAC1BpE;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,6EAA6E;QAC9E,MAAM8E,eAAe7E,GAAGU,EAAE,CAAC,UACzBE,KAAKC,SAAS,CAAC;gBAAE2B,eAAe;oBAAEC,aAAa;gBAA2B;YAAE;QAE9E,MAAMhC,QAAQT,GAAGU,EAAE;QAEnB,MAAMZ,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEY,MAAM;gBAAeC,mBAAmB;YAAiB;YAChEI,UAAU;YACV5C,UAAU,UAAY+E,QAAQC,MAAM,CAAC,IAAIV,MAAM;YAC/CY;YACApE;QACF,IACAkC,OAAO,CAACC,OAAO,CAAC;QAElB9C,OAAO+E,cAAchC,GAAG,CAACC,gBAAgB;QACzChD,OAAOW,OAAOoC,GAAG,CAACC,gBAAgB;IACpC;IAEA/C,GAAG,8CAA8C;QAC/C,MAAMD,OACJG,SAASkB,uBAAuB,CAAC;YAC/BG,KAAK;gBAAEC,yBAAyB;YAAoB;YACpDd,OAAOT,GAAGU,EAAE,CAAC,UAAY,IAAIC,SAAS,2BAA2B;oBAAEK,QAAQ;gBAAI;QACjF,IACA2B,OAAO,CAACC,OAAO,CAAC;IACpB;IAEA7C,GAAG,oDAAoD;QACrDC,GAAG8E,aAAa;QAChB,IAAI;YACF,MAAMC,UAAU9E,SAASkB,uBAAuB,CAAC;gBAC/CG,KAAK;oBAAEC,yBAAyB;gBAAqB;gBACrDyD,WAAW;gBACXvE,OAAOT,GAAGU,EAAE,CACV,OAAOuE,MAAMC,OACX,IAAIR,QAAkB,CAACS,UAAUR;wBAC/BO,MAAME,QAAQC,iBAAiB,SAAS,IACtCV,OAAO,IAAIV,MAAM;oBAErB;YAEN;YACA,MAAMqB,YAAYxF,OAAOiF,SAASpC,OAAO,CAACC,OAAO,CAAC;YAClD,MAAM5C,GAAGuF,wBAAwB,CAAC;YAClC,MAAMD;QACR,SAAU;YACRtF,GAAGwF,aAAa;QAClB;IACF;AACF"}
|
|
@@ -1,5 +1,7 @@
|
|
|
1
1
|
import { describe, expect, it, vi } from "vitest";
|
|
2
|
-
import { CODEX_APP_SERVER_ARGS, parseUsage, probeCodexCapacity
|
|
2
|
+
import { CODEX_APP_SERVER_ARGS, parseUsage, probeCodexCapacity } from "../../capacity/sources/codex.js";
|
|
3
|
+
import { resolveCodexAuthPath } from "../../harnesses/codex/credentials.js";
|
|
4
|
+
import { toRateWindow } from "../../capacity/providers/openai-subscription.js";
|
|
3
5
|
const checkedAt = "2026-08-20T10:00:00.000Z";
|
|
4
6
|
const context = {
|
|
5
7
|
installed: true,
|