@ai-devkit/agent-manager 0.33.0 → 0.34.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 +11 -2
- package/dist/AgentManager.d.ts.map +1 -1
- package/dist/AgentManager.js +37 -6
- package/dist/AgentManager.js.map +1 -1
- package/dist/__tests__/AgentManager.test.js +212 -5
- package/dist/__tests__/AgentManager.test.js.map +1 -1
- package/dist/__tests__/capacity/codex.test.js +2 -1
- package/dist/__tests__/capacity/codex.test.js.map +1 -1
- package/dist/__tests__/capacity/index.test.js +6 -3
- package/dist/__tests__/capacity/index.test.js.map +1 -1
- package/dist/__tests__/capacity/zai.test.js +213 -0
- package/dist/__tests__/capacity/zai.test.js.map +1 -0
- package/dist/__tests__/{adapters → providers/copilot}/CopilotAdapter.test.js +19 -14
- package/dist/__tests__/providers/copilot/CopilotAdapter.test.js.map +1 -0
- package/dist/__tests__/{adapters → providers/gemini}/GeminiCliAdapter.test.js +37 -29
- package/dist/__tests__/providers/gemini/GeminiCliAdapter.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js +189 -0
- package/dist/__tests__/providers/opencode/OpenCodeAdapter.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js +89 -0
- package/dist/__tests__/providers/opencode/OpenCodeAgentMapper.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js +94 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionLocator.test.js.map +1 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js +147 -0
- package/dist/__tests__/providers/opencode/OpenCodeSessionParser.test.js.map +1 -0
- package/dist/__tests__/runtime/HerdrAgentDiscovery.test.js +164 -0
- package/dist/__tests__/runtime/HerdrAgentDiscovery.test.js.map +1 -0
- package/dist/__tests__/runtime/HerdrAgentRuntime.test.js +37 -1
- package/dist/__tests__/runtime/HerdrAgentRuntime.test.js.map +1 -1
- package/dist/__tests__/utils/AgentRegistry.test.js +20 -13
- package/dist/__tests__/utils/AgentRegistry.test.js.map +1 -1
- package/dist/adapters/index.d.ts +3 -3
- package/dist/adapters/index.d.ts.map +1 -1
- package/dist/adapters/index.js +3 -3
- package/dist/adapters/index.js.map +1 -1
- package/dist/capacity/codex.d.ts +1 -0
- package/dist/capacity/codex.d.ts.map +1 -1
- package/dist/capacity/codex.js +9 -4
- package/dist/capacity/codex.js.map +1 -1
- package/dist/capacity/index.d.ts +8 -0
- package/dist/capacity/index.d.ts.map +1 -1
- package/dist/capacity/index.js +10 -9
- package/dist/capacity/index.js.map +1 -1
- package/dist/capacity/types.d.ts +8 -0
- package/dist/capacity/types.d.ts.map +1 -1
- package/dist/capacity/types.js.map +1 -1
- package/dist/capacity/zai.d.ts +14 -0
- package/dist/capacity/zai.d.ts.map +1 -0
- package/dist/capacity/zai.js +203 -0
- package/dist/capacity/zai.js.map +1 -0
- package/dist/index.d.ts +8 -6
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +5 -4
- package/dist/index.js.map +1 -1
- package/dist/{adapters → providers/copilot}/CopilotAdapter.d.ts +11 -25
- package/dist/providers/copilot/CopilotAdapter.d.ts.map +1 -0
- package/dist/providers/copilot/CopilotAdapter.js +108 -0
- package/dist/providers/copilot/CopilotAdapter.js.map +1 -0
- package/dist/providers/copilot/CopilotAgentMapper.d.ts +9 -0
- package/dist/providers/copilot/CopilotAgentMapper.d.ts.map +1 -0
- package/dist/providers/copilot/CopilotAgentMapper.js +37 -0
- package/dist/providers/copilot/CopilotAgentMapper.js.map +1 -0
- package/dist/providers/copilot/CopilotSessionLocator.d.ts +19 -0
- package/dist/providers/copilot/CopilotSessionLocator.d.ts.map +1 -0
- package/dist/providers/copilot/CopilotSessionLocator.js +41 -0
- package/dist/providers/copilot/CopilotSessionLocator.js.map +1 -0
- package/dist/providers/copilot/CopilotSessionParser.d.ts +30 -0
- package/dist/providers/copilot/CopilotSessionParser.d.ts.map +1 -0
- package/dist/providers/copilot/CopilotSessionParser.js +181 -0
- package/dist/providers/copilot/CopilotSessionParser.js.map +1 -0
- package/dist/providers/gemini/GeminiAgentMapper.d.ts +9 -0
- package/dist/providers/gemini/GeminiAgentMapper.d.ts.map +1 -0
- package/dist/providers/gemini/GeminiAgentMapper.js +37 -0
- package/dist/providers/gemini/GeminiAgentMapper.js.map +1 -0
- package/dist/providers/gemini/GeminiCliAdapter.d.ts +52 -0
- package/dist/providers/gemini/GeminiCliAdapter.d.ts.map +1 -0
- package/dist/providers/gemini/GeminiCliAdapter.js +188 -0
- package/dist/providers/gemini/GeminiCliAdapter.js.map +1 -0
- package/dist/providers/gemini/GeminiSessionLocator.d.ts +30 -0
- package/dist/providers/gemini/GeminiSessionLocator.d.ts.map +1 -0
- package/dist/providers/gemini/GeminiSessionLocator.js +142 -0
- package/dist/providers/gemini/GeminiSessionLocator.js.map +1 -0
- package/dist/providers/gemini/GeminiSessionParser.d.ts +94 -0
- package/dist/providers/gemini/GeminiSessionParser.d.ts.map +1 -0
- package/dist/providers/gemini/GeminiSessionParser.js +166 -0
- package/dist/providers/gemini/GeminiSessionParser.js.map +1 -0
- package/dist/{adapters → providers/opencode}/OpenCodeAdapter.d.ts +6 -12
- package/dist/providers/opencode/OpenCodeAdapter.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeAdapter.js +86 -0
- package/dist/providers/opencode/OpenCodeAdapter.js.map +1 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.d.ts +11 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.js +50 -0
- package/dist/providers/opencode/OpenCodeAgentMapper.js.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.d.ts +21 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.js +99 -0
- package/dist/providers/opencode/OpenCodeSessionLocator.js.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionParser.d.ts +18 -0
- package/dist/providers/opencode/OpenCodeSessionParser.d.ts.map +1 -0
- package/dist/providers/opencode/OpenCodeSessionParser.js +104 -0
- package/dist/providers/opencode/OpenCodeSessionParser.js.map +1 -0
- package/dist/runtime/herdr/HerdrAgentDiscovery.d.ts +18 -0
- package/dist/runtime/herdr/HerdrAgentDiscovery.d.ts.map +1 -0
- package/dist/runtime/herdr/HerdrAgentDiscovery.js +183 -0
- package/dist/runtime/herdr/HerdrAgentDiscovery.js.map +1 -0
- package/dist/runtime/herdr/HerdrAgentRuntime.d.ts +5 -0
- package/dist/runtime/herdr/HerdrAgentRuntime.d.ts.map +1 -1
- package/dist/runtime/herdr/HerdrAgentRuntime.js +26 -8
- package/dist/runtime/herdr/HerdrAgentRuntime.js.map +1 -1
- package/dist/runtime/herdr/HerdrResponseParsers.d.ts +2 -0
- package/dist/runtime/herdr/HerdrResponseParsers.d.ts.map +1 -1
- package/dist/runtime/herdr/HerdrResponseParsers.js +12 -1
- package/dist/runtime/herdr/HerdrResponseParsers.js.map +1 -1
- package/dist/utils/AgentRegistry.d.ts.map +1 -1
- package/dist/utils/AgentRegistry.js +5 -5
- package/dist/utils/AgentRegistry.js.map +1 -1
- package/package.json +1 -1
- package/src/AgentManager.ts +67 -5
- package/src/__tests__/AgentManager.test.ts +231 -5
- package/src/__tests__/capacity/codex.test.ts +94 -26
- package/src/__tests__/capacity/fixtures/zai-auth.json +6 -0
- package/src/__tests__/capacity/fixtures/zai-quota-invalid.json +7 -0
- package/src/__tests__/capacity/fixtures/zai-quota.json +38 -0
- package/src/__tests__/capacity/index.test.ts +10 -3
- package/src/__tests__/capacity/zai.test.ts +186 -0
- package/src/__tests__/{adapters → providers/copilot}/CopilotAdapter.test.ts +13 -16
- package/src/__tests__/{adapters → providers/gemini}/GeminiCliAdapter.test.ts +39 -31
- package/src/__tests__/providers/opencode/OpenCodeAdapter.test.ts +197 -0
- package/src/__tests__/providers/opencode/OpenCodeAgentMapper.test.ts +83 -0
- package/src/__tests__/providers/opencode/OpenCodeSessionLocator.test.ts +84 -0
- package/src/__tests__/providers/opencode/OpenCodeSessionParser.test.ts +148 -0
- package/src/__tests__/runtime/HerdrAgentDiscovery.test.ts +131 -0
- package/src/__tests__/runtime/HerdrAgentRuntime.test.ts +20 -1
- package/src/__tests__/utils/AgentRegistry.test.ts +20 -10
- package/src/adapters/index.ts +3 -3
- package/src/capacity/codex.ts +96 -28
- package/src/capacity/index.ts +29 -11
- package/src/capacity/types.ts +8 -0
- package/src/capacity/zai.ts +237 -0
- package/src/index.ts +18 -6
- package/src/providers/copilot/CopilotAdapter.ts +145 -0
- package/src/providers/copilot/CopilotAgentMapper.ts +36 -0
- package/src/providers/copilot/CopilotSessionLocator.ts +57 -0
- package/src/providers/copilot/CopilotSessionParser.ts +289 -0
- package/src/providers/gemini/GeminiAgentMapper.ts +36 -0
- package/src/providers/gemini/GeminiCliAdapter.ts +266 -0
- package/src/providers/gemini/GeminiSessionLocator.ts +183 -0
- package/src/providers/gemini/GeminiSessionParser.ts +271 -0
- package/src/providers/opencode/OpenCodeAdapter.ts +112 -0
- package/src/providers/opencode/OpenCodeAgentMapper.ts +61 -0
- package/src/providers/opencode/OpenCodeSessionLocator.ts +123 -0
- package/src/providers/opencode/OpenCodeSessionParser.ts +145 -0
- package/src/runtime/herdr/HerdrAgentDiscovery.ts +178 -0
- package/src/runtime/herdr/HerdrAgentRuntime.ts +33 -7
- package/src/runtime/herdr/HerdrResponseParsers.ts +11 -1
- package/src/utils/AgentRegistry.ts +8 -6
- package/dist/__tests__/adapters/CopilotAdapter.test.js.map +0 -1
- package/dist/__tests__/adapters/GeminiCliAdapter.test.js.map +0 -1
- package/dist/__tests__/adapters/OpenCodeAdapter.test.js +0 -635
- package/dist/__tests__/adapters/OpenCodeAdapter.test.js.map +0 -1
- package/dist/adapters/CopilotAdapter.d.ts.map +0 -1
- package/dist/adapters/CopilotAdapter.js +0 -329
- package/dist/adapters/CopilotAdapter.js.map +0 -1
- package/dist/adapters/GeminiCliAdapter.d.ts +0 -98
- package/dist/adapters/GeminiCliAdapter.d.ts.map +0 -1
- package/dist/adapters/GeminiCliAdapter.js +0 -513
- package/dist/adapters/GeminiCliAdapter.js.map +0 -1
- package/dist/adapters/OpenCodeAdapter.d.ts.map +0 -1
- package/dist/adapters/OpenCodeAdapter.js +0 -286
- package/dist/adapters/OpenCodeAdapter.js.map +0 -1
- package/src/__tests__/adapters/OpenCodeAdapter.test.ts +0 -543
- package/src/adapters/CopilotAdapter.ts +0 -483
- package/src/adapters/GeminiCliAdapter.ts +0 -699
- package/src/adapters/OpenCodeAdapter.ts +0 -369
package/dist/AgentManager.d.ts
CHANGED
|
@@ -6,7 +6,8 @@
|
|
|
6
6
|
*/
|
|
7
7
|
import type { AgentAdapter, AgentInfo, SessionSummary, ListSessionsOptions, ProcessInfo } from "./adapters/AgentAdapter.js";
|
|
8
8
|
import { type AgentSortKey } from "./utils/sortAgents.js";
|
|
9
|
-
import { AgentRegistry } from "./utils/AgentRegistry.js";
|
|
9
|
+
import { AgentRegistry, type AgentRuntimeProvider } from "./utils/AgentRegistry.js";
|
|
10
|
+
import { type HerdrAgentPane } from "./runtime/herdr/HerdrAgentDiscovery.js";
|
|
10
11
|
type ProcessSnapshotCapture = (namePatterns: readonly string[]) => Promise<ProcessInfo[]>;
|
|
11
12
|
export interface ListAgentsOptions {
|
|
12
13
|
/**
|
|
@@ -15,6 +16,11 @@ export interface ListAgentsOptions {
|
|
|
15
16
|
*/
|
|
16
17
|
sortBy?: AgentSortKey;
|
|
17
18
|
}
|
|
19
|
+
export interface AgentManagerOptions {
|
|
20
|
+
runtimeProvider?: AgentRuntimeProvider | (() => AgentRuntimeProvider | Promise<AgentRuntimeProvider>);
|
|
21
|
+
fetchHerdrAgentPanes?: () => Promise<readonly HerdrAgentPane[]>;
|
|
22
|
+
onRuntimeDiscoveryError?: (error: unknown) => void;
|
|
23
|
+
}
|
|
18
24
|
export declare class AgentNotRunningError extends Error {
|
|
19
25
|
agentName: string;
|
|
20
26
|
constructor(agentName: string);
|
|
@@ -36,9 +42,10 @@ export declare class AgentNotRunningError extends Error {
|
|
|
36
42
|
*/
|
|
37
43
|
export declare class AgentManager {
|
|
38
44
|
private readonly captureSnapshot;
|
|
45
|
+
private readonly options;
|
|
39
46
|
private adapters;
|
|
40
47
|
private registry;
|
|
41
|
-
constructor(registry?: AgentRegistry, captureSnapshot?: ProcessSnapshotCapture);
|
|
48
|
+
constructor(registry?: AgentRegistry, captureSnapshot?: ProcessSnapshotCapture, options?: AgentManagerOptions);
|
|
42
49
|
/**
|
|
43
50
|
* Register an adapter for a specific agent type
|
|
44
51
|
*
|
|
@@ -96,6 +103,8 @@ export declare class AgentManager {
|
|
|
96
103
|
* ```
|
|
97
104
|
*/
|
|
98
105
|
listAgents(options?: ListAgentsOptions): Promise<AgentInfo[]>;
|
|
106
|
+
private resolveHerdrPanes;
|
|
107
|
+
private resolveRuntimeProvider;
|
|
99
108
|
private toRegistryEntry;
|
|
100
109
|
togglePin(agentName: string): boolean;
|
|
101
110
|
/**
|
|
@@ -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,EACnB,WAAW,EACZ,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAc,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,
|
|
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,EACnB,WAAW,EACZ,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAc,KAAK,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,EACL,aAAa,EAEb,KAAK,oBAAoB,EAE1B,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAIL,KAAK,cAAc,EACpB,MAAM,wCAAwC,CAAC;AAEhD,KAAK,sBAAsB,GAAG,CAAC,YAAY,EAAE,SAAS,MAAM,EAAE,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;AAE1F,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;AAED,qBAAa,oBAAqB,SAAQ,KAAK;IAC1B,SAAS,EAAE,MAAM;gBAAjB,SAAS,EAAE,MAAM;CAIrC;AAED;;;;;;;;;;;;;;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;YAiFrD,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;;;;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
|
@@ -4,8 +4,9 @@
|
|
|
4
4
|
* Orchestrates agent detection across multiple adapter types.
|
|
5
5
|
* Manages adapter registration and aggregates results from all adapters.
|
|
6
6
|
*/ import { sortAgents } from "./utils/sortAgents.js";
|
|
7
|
-
import { AgentRegistry } from "./utils/AgentRegistry.js";
|
|
7
|
+
import { AgentRegistry, parseTmuxRuntimeRef } from "./utils/AgentRegistry.js";
|
|
8
8
|
import { captureProcessSnapshot, filterByProcessNames } from "./utils/process.js";
|
|
9
|
+
import { findMatchingHerdrPane, fetchHerdrAgentPanes, herdrPaneToRuntimeRef } from "./runtime/herdr/HerdrAgentDiscovery.js";
|
|
9
10
|
export class AgentNotRunningError extends Error {
|
|
10
11
|
agentName;
|
|
11
12
|
constructor(agentName){
|
|
@@ -29,10 +30,12 @@ export class AgentNotRunningError extends Error {
|
|
|
29
30
|
* ```
|
|
30
31
|
*/ export class AgentManager {
|
|
31
32
|
captureSnapshot;
|
|
33
|
+
options;
|
|
32
34
|
adapters = new Map();
|
|
33
35
|
registry;
|
|
34
|
-
constructor(registry = AgentRegistry.default(), captureSnapshot = captureProcessSnapshot){
|
|
36
|
+
constructor(registry = AgentRegistry.default(), captureSnapshot = captureProcessSnapshot, options = {}){
|
|
35
37
|
this.captureSnapshot = captureSnapshot;
|
|
38
|
+
this.options = options;
|
|
36
39
|
this.registry = registry;
|
|
37
40
|
}
|
|
38
41
|
/**
|
|
@@ -158,7 +161,8 @@ export class AgentNotRunningError extends Error {
|
|
|
158
161
|
identityKey(entry.type, entry.pid),
|
|
159
162
|
entry
|
|
160
163
|
]));
|
|
161
|
-
const
|
|
164
|
+
const herdrPanes = allAgents.length > 0 ? await this.resolveHerdrPanes() : [];
|
|
165
|
+
const entries = allAgents.map((agent)=>this.toRegistryEntry(agent, preExistingByIdentity.get(identityKey(agent.type, agent.pid)), herdrPanes));
|
|
162
166
|
if (entries.length > 0) this.registry.registerBatch(entries);
|
|
163
167
|
this.registry.pruneIfDue();
|
|
164
168
|
for (const agent of allAgents){
|
|
@@ -172,9 +176,25 @@ export class AgentNotRunningError extends Error {
|
|
|
172
176
|
const sortKey = options?.sortBy ?? "status";
|
|
173
177
|
return sortAgents(allAgents, sortKey);
|
|
174
178
|
}
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
179
|
+
async resolveHerdrPanes() {
|
|
180
|
+
const runtimeProvider = await this.resolveRuntimeProvider();
|
|
181
|
+
if (runtimeProvider !== "herdr") return [];
|
|
182
|
+
try {
|
|
183
|
+
return await (this.options.fetchHerdrAgentPanes ?? fetchHerdrAgentPanes)();
|
|
184
|
+
} catch (error) {
|
|
185
|
+
this.options.onRuntimeDiscoveryError?.(error);
|
|
186
|
+
return [];
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
async resolveRuntimeProvider() {
|
|
190
|
+
const runtimeProvider = this.options.runtimeProvider;
|
|
191
|
+
return typeof runtimeProvider === "function" ? await runtimeProvider() : runtimeProvider;
|
|
192
|
+
}
|
|
193
|
+
toRegistryEntry(agent, existing, herdrPanes = []) {
|
|
194
|
+
const name = existing?.name ?? agent.name;
|
|
195
|
+
const existingHasAuthoritativeRuntimeRef = existing?.runtime === "herdr" || Boolean(parseTmuxRuntimeRef(existing?.runtimeRef));
|
|
196
|
+
const baseEntry = {
|
|
197
|
+
name,
|
|
178
198
|
type: agent.type,
|
|
179
199
|
pid: agent.pid,
|
|
180
200
|
runtime: existing?.runtime ?? "tmux",
|
|
@@ -185,6 +205,17 @@ export class AgentNotRunningError extends Error {
|
|
|
185
205
|
sessionFilePath: agent.sessionFilePath ?? "",
|
|
186
206
|
pinned: existing?.pinned ?? agent.pinned ?? false
|
|
187
207
|
};
|
|
208
|
+
if (!existingHasAuthoritativeRuntimeRef) {
|
|
209
|
+
const herdrPane = findMatchingHerdrPane(agent, herdrPanes);
|
|
210
|
+
if (herdrPane) {
|
|
211
|
+
return {
|
|
212
|
+
...baseEntry,
|
|
213
|
+
runtime: "herdr",
|
|
214
|
+
runtimeRef: herdrPaneToRuntimeRef(herdrPane, name)
|
|
215
|
+
};
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
return baseEntry;
|
|
188
219
|
}
|
|
189
220
|
togglePin(agentName) {
|
|
190
221
|
const entry = this.registry.lookup(agentName);
|
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 { AgentRegistry, type RegistryEntry } from \"./utils/AgentRegistry.js\";\nimport { captureProcessSnapshot, filterByProcessNames } from \"./utils/process.js\";\n\ntype ProcessSnapshotCapture = (namePatterns: readonly string[]) => Promise<ProcessInfo[]>;\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 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 ) {\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 } 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 entries = allAgents.map((agent) =>\n this.toRegistryEntry(agent, preExistingByIdentity.get(identityKey(agent.type, agent.pid))),\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 private toRegistryEntry(agent: AgentInfo, existing?: RegistryEntry): RegistryEntry {\n return {\n name: existing?.name ?? agent.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\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 /**\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","captureProcessSnapshot","filterByProcessNames","AgentNotRunningError","Error","agentName","name","AgentManager","adapters","Map","registry","default","captureSnapshot","registerAdapter","adapter","adapterKey","type","has","set","unregisterAdapter","delete","getAdapters","Array","from","values","getAdapter","get","hasAdapter","listAgents","options","allAgents","errors","processNames","Set","flatMap","processes","length","adapterPromises","map","agents","detectAgents","error","err","String","push","results","Promise","all","result","console","forEach","message","identityKey","pid","preExistingByIdentity","list","entry","entries","agent","toRegistryEntry","registerBatch","pruneIfDue","pinned","updatedAt","lastActive","Date","sortKey","sortBy","existing","runtime","runtimeRef","cwd","projectPath","startedAt","toISOString","sessionId","sessionFilePath","togglePin","lookup","isAlive","prune","listSessions","opts","targetAdapters","filter","undefined","merged","sort","a","b","getTime","getAdapterCount","size","clear","resolveAgent","input","registryEntry","registryAgent","find","lowerInput","toLowerCase","exactMatch","matches","includes"],"mappings":"AAAA;;;;;CAKC,GASD,SAASA,UAAU,QAA2B,wBAAwB;AACtE,SAASC,aAAa,QAA4B,2BAA2B;AAC7E,SAASC,sBAAsB,EAAEC,oBAAoB,QAAQ,qBAAqB;AAYlF,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,WAA0BV,cAAcW,OAAO,EAAE,EACjD,AAAiBC,kBAA0CX,sBAAsB,CACjF;aADiBW,kBAAAA;QAEjB,IAAI,CAACF,QAAQ,GAAGA;IAClB;IAEA;;;;;;;;;;GAUC,GACDG,gBAAgBC,OAAqB,EAAQ;QAC3C,MAAMC,aAAaD,QAAQE,IAAI;QAE/B,IAAI,IAAI,CAACR,QAAQ,CAACS,GAAG,CAACF,aAAa;YACjC,MAAM,IAAIX,MAAM,CAAC,kBAAkB,EAAEW,WAAW,uBAAuB,CAAC;QAC1E;QAEA,IAAI,CAACP,QAAQ,CAACU,GAAG,CAACH,YAAYD;IAChC;IAEA;;;;;GAKC,GACDK,kBAAkBH,IAAY,EAAW;QACvC,OAAO,IAAI,CAACR,QAAQ,CAACY,MAAM,CAACJ;IAC9B;IAEA;;;;GAIC,GACDK,cAA8B;QAC5B,OAAOC,MAAMC,IAAI,CAAC,IAAI,CAACf,QAAQ,CAACgB,MAAM;IACxC;IAEA;;;;;GAKC,GACDC,WAAWT,IAAY,EAA4B;QACjD,OAAO,IAAI,CAACR,QAAQ,CAACkB,GAAG,CAACV;IAC3B;IAEA;;;;;GAKC,GACDW,WAAWX,IAAY,EAAW;QAChC,OAAO,IAAI,CAACR,QAAQ,CAACS,GAAG,CAACD;IAC3B;IAEA;;;;;;;;;;;;;;;;GAgBC,GACD,MAAMY,WAAWC,OAA2B,EAAwB;QAClE,MAAMC,YAAyB,EAAE;QACjC,MAAMC,SAAgD,EAAE;QAExD,MAAMvB,WAAWc,MAAMC,IAAI,CAAC,IAAI,CAACf,QAAQ,CAACgB,MAAM;QAChD,MAAMQ,eAAeV,MAAMC,IAAI,CAC7B,IAAIU,IACFzB,SAAS0B,OAAO,CAAC,CAACpB,UAAaA,QAAQkB,YAAY,GAAG;mBAAIlB,QAAQkB,YAAY;aAAC,GAAG,EAAE;QAGxF,IAAIG,YAAoC,EAAE;QAC1C,IAAIH,aAAaI,MAAM,GAAG,GAAG;YAC3B,IAAI;gBACFD,YAAY,MAAM,IAAI,CAACvB,eAAe,CAACoB;YACzC,EAAE,OAAM;gBACNG,YAAY,EAAE;YAChB;QACF;QAEA,wFAAwF;QACxF,MAAME,kBAAkB7B,SAAS8B,GAAG,CAAC,OAAOxB;YAC1C,IAAI;gBACF,MAAMyB,SAASzB,QAAQkB,YAAY,GAC/B,MAAMlB,QAAQ0B,YAAY,CAAC;oBACzBL,WAAWjC,qBAAqBiC,WAAWrB,QAAQkB,YAAY;gBACjE,KACA,MAAMlB,QAAQ0B,YAAY;gBAC9B,OAAO;oBAAExB,MAAMF,QAAQE,IAAI;oBAAEuB;oBAAQE,OAAO;gBAAK;YACnD,EAAE,OAAOA,OAAO;gBACd,mEAAmE;gBACnE,MAAMC,MAAMD,iBAAiBrC,QAAQqC,QAAQ,IAAIrC,MAAMuC,OAAOF;gBAC9DV,OAAOa,IAAI,CAAC;oBAAE5B,MAAMF,QAAQE,IAAI;oBAAEyB,OAAOC;gBAAI;gBAC7C,OAAO;oBAAE1B,MAAMF,QAAQE,IAAI;oBAAEuB,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;gBACzBX,UAAUc,IAAI,IAAII,OAAOT,MAAM;YACjC;QACF;QAEA,mEAAmE;QACnE,IAAIR,OAAOK,MAAM,GAAG,GAAG;YACrBa,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEV,OAAOK,MAAM,CAAC,mBAAmB,CAAC;YAC5DL,OAAOmB,OAAO,CAAC,CAAC,EAAElC,IAAI,EAAEyB,KAAK,EAAE;gBAC7BQ,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAEzB,KAAK,EAAE,EAAEyB,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAMC,cAAc,CAACpC,MAAcqC,MAAwB,GAAGrC,KAAK,CAAC,EAAEqC,KAAK;QAC3E,MAAMC,wBAAwB,IAAI7C,IAChC,IAAI,CAACC,QAAQ,CAAC6C,IAAI,GAAGjB,GAAG,CAAC,CAACkB,QAAU;gBAACJ,YAAYI,MAAMxC,IAAI,EAAEwC,MAAMH,GAAG;gBAAGG;aAAM;QAEjF,MAAMC,UAAU3B,UAAUQ,GAAG,CAAC,CAACoB,QAC7B,IAAI,CAACC,eAAe,CAACD,OAAOJ,sBAAsB5B,GAAG,CAAC0B,YAAYM,MAAM1C,IAAI,EAAE0C,MAAML,GAAG;QAEzF,IAAII,QAAQrB,MAAM,GAAG,GAAG,IAAI,CAAC1B,QAAQ,CAACkD,aAAa,CAACH;QACpD,IAAI,CAAC/C,QAAQ,CAACmD,UAAU;QAExB,KAAK,MAAMH,SAAS5B,UAAW;YAC7B,MAAM0B,QAAQF,sBAAsB5B,GAAG,CAAC0B,YAAYM,MAAM1C,IAAI,EAAE0C,MAAML,GAAG;YACzE,IAAIG,OAAO;gBACTE,MAAMpD,IAAI,GAAGkD,MAAMlD,IAAI;gBACvBoD,MAAMI,MAAM,GAAGN,MAAMM,MAAM;gBAC3B,IAAIN,MAAMM,MAAM,IAAIN,MAAMO,SAAS,EAAEL,MAAMM,UAAU,GAAG,IAAIC,KAAKT,MAAMO,SAAS;YAClF;QACF;QAEA,MAAMG,UAAwBrC,SAASsC,UAAU;QACjD,OAAOpE,WAAW+B,WAAWoC;IAC/B;IAEQP,gBAAgBD,KAAgB,EAAEU,QAAwB,EAAiB;QACjF,OAAO;YACL9D,MAAM8D,UAAU9D,QAAQoD,MAAMpD,IAAI;YAClCU,MAAM0C,MAAM1C,IAAI;YAChBqC,KAAKK,MAAML,GAAG;YACdgB,SAASD,UAAUC,WAAW;YAC9BC,YAAYF,UAAUE,cAAc;YACpCC,KAAKb,MAAMc,WAAW;YACtBC,WAAWL,UAAUK,aAAa,IAAIR,OAAOS,WAAW;YACxDC,WAAWjB,MAAMiB,SAAS;YAC1BC,iBAAiBlB,MAAMkB,eAAe,IAAI;YAC1Cd,QAAQM,UAAUN,UAAUJ,MAAMI,MAAM,IAAI;QAC9C;IACF;IAEAe,UAAUxE,SAAiB,EAAW;QACpC,MAAMmD,QAAQ,IAAI,CAAC9C,QAAQ,CAACoE,MAAM,CAACzE;QACnC,IAAI,CAACmD,SAAS,CAAC,IAAI,CAAC9C,QAAQ,CAACqE,OAAO,CAACvB,QAAQ;YAC3C,IAAIA,OAAO,IAAI,CAAC9C,QAAQ,CAACsE,KAAK;YAC9B,MAAM,IAAI7E,qBAAqBE;QACjC;QACA,MAAMyD,SAAS,IAAI,CAACpD,QAAQ,CAACmE,SAAS,CAACrB,MAAMxC,IAAI,EAAEwC,MAAMH,GAAG;QAC5D,IAAIS,WAAW,MAAM,MAAM,IAAI3D,qBAAqBE;QACpD,OAAOyD;IACT;IAEA;;;;;;;;;;;GAWC,GACD,MAAMmB,aAAaC,IAA0B,EAA6B;QACxE,MAAMC,iBAAiB7D,MAAMC,IAAI,CAAC,IAAI,CAACf,QAAQ,CAACgB,MAAM,IAAI4D,MAAM,CAC9D,CAACtE,UAAYoE,MAAMlE,SAASqE,aAAavE,QAAQE,IAAI,KAAKkE,KAAKlE,IAAI;QAGrE,MAAMe,SAAgD,EAAE;QAExD,MAAMc,UAAU,MAAMC,QAAQC,GAAG,CAC/BoC,eAAe7C,GAAG,CAAC,OAAOxB;YACxB,IAAI;gBACF,OAAO,MAAMA,QAAQmE,YAAY,CAACC;YACpC,EAAE,OAAOzC,OAAO;gBACd,MAAMC,MAAMD,iBAAiBrC,QAAQqC,QAAQ,IAAIrC,MAAMuC,OAAOF;gBAC9DV,OAAOa,IAAI,CAAC;oBAAE5B,MAAMF,QAAQE,IAAI;oBAAEyB,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAIX,OAAOK,MAAM,GAAG,GAAG;YACrBa,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEV,OAAOK,MAAM,CAAC,oCAAoC,CAAC;YAC7E,KAAK,MAAM,EAAEpB,IAAI,EAAEyB,KAAK,EAAE,IAAIV,OAAQ;gBACpCkB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAEzB,KAAK,EAAE,EAAEyB,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAMmC,SAA2B,EAAE;QACnC,KAAK,MAAM/B,QAAQV,QAAS;YAC1ByC,OAAO1C,IAAI,IAAIW;QACjB;QAEA+B,OAAOC,IAAI,CAAC,CAACC,GAAGC,IAAMA,EAAEzB,UAAU,CAAC0B,OAAO,KAAKF,EAAExB,UAAU,CAAC0B,OAAO;QACnE,OAAOJ;IACT;IAEA;;;;GAIC,GACDK,kBAA0B;QACxB,OAAO,IAAI,CAACnF,QAAQ,CAACoF,IAAI;IAC3B;IAEA;;GAEC,GACDC,QAAc;QACZ,IAAI,CAACrF,QAAQ,CAACqF,KAAK;IACrB;IAEA;;;;;;GAMC,GACDC,aAAaC,KAAa,EAAExD,MAAmB,EAAkC;QAC/E,IAAI,CAACwD,SAASxD,OAAOH,MAAM,KAAK,GAAG,OAAO;QAE1C,qFAAqF;QACrF,MAAM4D,gBAAgB,IAAI,CAACtF,QAAQ,CAACoE,MAAM,CAACiB;QAC3C,IAAIC,eAAe;YACjB,MAAMC,gBAAgB1D,OAAO2D,IAAI,CAAC,CAACV,IAAMA,EAAEnC,GAAG,KAAK2C,cAAc3C,GAAG;YACpE,IAAI4C,eAAe,OAAOA;QAC5B;QAEA,MAAME,aAAaJ,MAAMK,WAAW;QAEpC,oCAAoC;QACpC,MAAMC,aAAa9D,OAAO2D,IAAI,CAAC,CAACV,IAAMA,EAAElF,IAAI,CAAC8F,WAAW,OAAOD;QAC/D,IAAIE,YAAY,OAAOA;QAEvB,wCAAwC;QACxC,MAAMC,UAAU/D,OAAO6C,MAAM,CAAC,CAACI,IAAMA,EAAElF,IAAI,CAAC8F,WAAW,GAAGG,QAAQ,CAACJ;QAEnE,IAAIG,QAAQlE,MAAM,KAAK,GAAG,OAAOkE,OAAO,CAAC,EAAE;QAC3C,IAAIA,QAAQlE,MAAM,GAAG,GAAG,OAAOkE;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 { captureProcessSnapshot, filterByProcessNames } from \"./utils/process.js\";\nimport {\n findMatchingHerdrPane,\n fetchHerdrAgentPanes,\n herdrPaneToRuntimeRef,\n type HerdrAgentPane,\n} from \"./runtime/herdr/HerdrAgentDiscovery.js\";\n\ntype ProcessSnapshotCapture = (namePatterns: readonly string[]) => Promise<ProcessInfo[]>;\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 } 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 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 /**\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","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","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","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,SAASC,sBAAsB,EAAEC,oBAAoB,QAAQ,qBAAqB;AAClF,SACEC,qBAAqB,EACrBC,oBAAoB,EACpBC,qBAAqB,QAEhB,yCAAyC;AAoBhD,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;YACzC,EAAE,OAAM;gBACNG,YAAY,EAAE;YAChB;QACF;QAEA,wFAAwF;QACxF,MAAME,kBAAkB7B,SAAS8B,GAAG,CAAC,OAAOvB;YAC1C,IAAI;gBACF,MAAMwB,SAASxB,QAAQiB,YAAY,GAC/B,MAAMjB,QAAQyB,YAAY,CAAC;oBACzBL,WAAWpC,qBAAqBoC,WAAWpB,QAAQiB,YAAY;gBACjE,KACA,MAAMjB,QAAQyB,YAAY;gBAC9B,OAAO;oBAAEvB,MAAMF,QAAQE,IAAI;oBAAEsB;oBAAQE,OAAO;gBAAK;YACnD,EAAE,OAAOA,OAAO;gBACd,mEAAmE;gBACnE,MAAMC,MAAMD,iBAAiBrC,QAAQqC,QAAQ,IAAIrC,MAAMuC,OAAOF;gBAC9DV,OAAOa,IAAI,CAAC;oBAAE3B,MAAMF,QAAQE,IAAI;oBAAEwB,OAAOC;gBAAI;gBAC7C,OAAO;oBAAEzB,MAAMF,QAAQE,IAAI;oBAAEsB,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;gBACzBX,UAAUc,IAAI,IAAII,OAAOT,MAAM;YACjC;QACF;QAEA,mEAAmE;QACnE,IAAIR,OAAOK,MAAM,GAAG,GAAG;YACrBa,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEV,OAAOK,MAAM,CAAC,mBAAmB,CAAC;YAC5DL,OAAOmB,OAAO,CAAC,CAAC,EAAEjC,IAAI,EAAEwB,KAAK,EAAE;gBAC7BQ,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAExB,KAAK,EAAE,EAAEwB,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAMC,cAAc,CAACnC,MAAcoC,MAAwB,GAAGpC,KAAK,CAAC,EAAEoC,KAAK;QAC3E,MAAMC,wBAAwB,IAAI7C,IAChC,IAAI,CAACC,QAAQ,CAAC6C,IAAI,GAAGjB,GAAG,CAAC,CAACkB,QAAU;gBAACJ,YAAYI,MAAMvC,IAAI,EAAEuC,MAAMH,GAAG;gBAAGG;aAAM;QAEjF,MAAMC,aAAa3B,UAAUM,MAAM,GAAG,IAAI,MAAM,IAAI,CAACsB,iBAAiB,KAAK,EAAE;QAC7E,MAAMC,UAAU7B,UAAUQ,GAAG,CAAC,CAACsB,QAC7B,IAAI,CAACC,eAAe,CAClBD,OACAN,sBAAsB3B,GAAG,CAACyB,YAAYQ,MAAM3C,IAAI,EAAE2C,MAAMP,GAAG,IAC3DI;QAGJ,IAAIE,QAAQvB,MAAM,GAAG,GAAG,IAAI,CAAC1B,QAAQ,CAACoD,aAAa,CAACH;QACpD,IAAI,CAACjD,QAAQ,CAACqD,UAAU;QAExB,KAAK,MAAMH,SAAS9B,UAAW;YAC7B,MAAM0B,QAAQF,sBAAsB3B,GAAG,CAACyB,YAAYQ,MAAM3C,IAAI,EAAE2C,MAAMP,GAAG;YACzE,IAAIG,OAAO;gBACTI,MAAMtD,IAAI,GAAGkD,MAAMlD,IAAI;gBACvBsD,MAAMI,MAAM,GAAGR,MAAMQ,MAAM;gBAC3B,IAAIR,MAAMQ,MAAM,IAAIR,MAAMS,SAAS,EAAEL,MAAMM,UAAU,GAAG,IAAIC,KAAKX,MAAMS,SAAS;YAClF;QACF;QAEA,MAAMG,UAAwBvD,SAASwD,UAAU;QACjD,OAAO1E,WAAWmC,WAAWsC;IAC/B;IAEA,MAAcV,oBAAwD;QACpE,MAAMY,kBAAkB,MAAM,IAAI,CAACC,sBAAsB;QACzD,IAAID,oBAAoB,SAAS,OAAO,EAAE;QAE1C,IAAI;YACF,OAAO,MAAM,AAAC,CAAA,IAAI,CAACzD,OAAO,CAACZ,oBAAoB,IAAIA,oBAAmB;QACxE,EAAE,OAAOwC,OAAO;YACd,IAAI,CAAC5B,OAAO,CAAC2D,uBAAuB,GAAG/B;YACvC,OAAO,EAAE;QACX;IACF;IAEA,MAAc8B,yBAAoE;QAChF,MAAMD,kBAAkB,IAAI,CAACzD,OAAO,CAACyD,eAAe;QACpD,OAAO,OAAOA,oBAAoB,aAAa,MAAMA,oBAAoBA;IAC3E;IAEQT,gBACND,KAAgB,EAChBa,QAAwB,EACxBhB,aAAwC,EAAE,EAC3B;QACf,MAAMnD,OAAOmE,UAAUnE,QAAQsD,MAAMtD,IAAI;QACzC,MAAMoE,qCACJD,UAAUE,YAAY,WAAWC,QAAQ/E,oBAAoB4E,UAAUI;QACzE,MAAMC,YAA2B;YAC/BxE;YACAW,MAAM2C,MAAM3C,IAAI;YAChBoC,KAAKO,MAAMP,GAAG;YACdsB,SAASF,UAAUE,WAAW;YAC9BE,YAAYJ,UAAUI,cAAc;YACpCE,KAAKnB,MAAMoB,WAAW;YACtBC,WAAWR,UAAUQ,aAAa,IAAId,OAAOe,WAAW;YACxDC,WAAWvB,MAAMuB,SAAS;YAC1BC,iBAAiBxB,MAAMwB,eAAe,IAAI;YAC1CpB,QAAQS,UAAUT,UAAUJ,MAAMI,MAAM,IAAI;QAC9C;QAEA,IAAI,CAACU,oCAAoC;YACvC,MAAMW,YAAYrF,sBAAsB4D,OAAOH;YAC/C,IAAI4B,WAAW;gBACb,OAAO;oBACL,GAAGP,SAAS;oBACZH,SAAS;oBACTE,YAAY3E,sBAAsBmF,WAAW/E;gBAC/C;YACF;QACF;QAEA,OAAOwE;IACT;IAEAQ,UAAUjF,SAAiB,EAAW;QACpC,MAAMmD,QAAQ,IAAI,CAAC9C,QAAQ,CAAC6E,MAAM,CAAClF;QACnC,IAAI,CAACmD,SAAS,CAAC,IAAI,CAAC9C,QAAQ,CAAC8E,OAAO,CAAChC,QAAQ;YAC3C,IAAIA,OAAO,IAAI,CAAC9C,QAAQ,CAAC+E,KAAK;YAC9B,MAAM,IAAItF,qBAAqBE;QACjC;QACA,MAAM2D,SAAS,IAAI,CAACtD,QAAQ,CAAC4E,SAAS,CAAC9B,MAAMvC,IAAI,EAAEuC,MAAMH,GAAG;QAC5D,IAAIW,WAAW,MAAM,MAAM,IAAI7D,qBAAqBE;QACpD,OAAO2D;IACT;IAEA;;;;;;;;;;;GAWC,GACD,MAAM0B,aAAaC,IAA0B,EAA6B;QACxE,MAAMC,iBAAiBrE,MAAMC,IAAI,CAAC,IAAI,CAAChB,QAAQ,CAACiB,MAAM,IAAIoE,MAAM,CAC9D,CAAC9E,UAAY4E,MAAM1E,SAAS6E,aAAa/E,QAAQE,IAAI,KAAK0E,KAAK1E,IAAI;QAGrE,MAAMc,SAAgD,EAAE;QAExD,MAAMc,UAAU,MAAMC,QAAQC,GAAG,CAC/B6C,eAAetD,GAAG,CAAC,OAAOvB;YACxB,IAAI;gBACF,OAAO,MAAMA,QAAQ2E,YAAY,CAACC;YACpC,EAAE,OAAOlD,OAAO;gBACd,MAAMC,MAAMD,iBAAiBrC,QAAQqC,QAAQ,IAAIrC,MAAMuC,OAAOF;gBAC9DV,OAAOa,IAAI,CAAC;oBAAE3B,MAAMF,QAAQE,IAAI;oBAAEwB,OAAOC;gBAAI;gBAC7C,OAAO,EAAE;YACX;QACF;QAGF,IAAIX,OAAOK,MAAM,GAAG,GAAG;YACrBa,QAAQR,KAAK,CAAC,CAAC,SAAS,EAAEV,OAAOK,MAAM,CAAC,oCAAoC,CAAC;YAC7E,KAAK,MAAM,EAAEnB,IAAI,EAAEwB,KAAK,EAAE,IAAIV,OAAQ;gBACpCkB,QAAQR,KAAK,CAAC,CAAC,IAAI,EAAExB,KAAK,EAAE,EAAEwB,MAAMU,OAAO,EAAE;YAC/C;QACF;QAEA,MAAM4C,SAA2B,EAAE;QACnC,KAAK,MAAMxC,QAAQV,QAAS;YAC1BkD,OAAOnD,IAAI,IAAIW;QACjB;QAEAwC,OAAOC,IAAI,CAAC,CAACC,GAAGC,IAAMA,EAAEhC,UAAU,CAACiC,OAAO,KAAKF,EAAE/B,UAAU,CAACiC,OAAO;QACnE,OAAOJ;IACT;IAEA;;;;GAIC,GACDK,kBAA0B;QACxB,OAAO,IAAI,CAAC5F,QAAQ,CAAC6F,IAAI;IAC3B;IAEA;;GAEC,GACDC,QAAc;QACZ,IAAI,CAAC9F,QAAQ,CAAC8F,KAAK;IACrB;IAEA;;;;;;GAMC,GACDC,aAAaC,KAAa,EAAEjE,MAAmB,EAAkC;QAC/E,IAAI,CAACiE,SAASjE,OAAOH,MAAM,KAAK,GAAG,OAAO;QAE1C,qFAAqF;QACrF,MAAMqE,gBAAgB,IAAI,CAAC/F,QAAQ,CAAC6E,MAAM,CAACiB;QAC3C,IAAIC,eAAe;YACjB,MAAMC,gBAAgBnE,OAAOoE,IAAI,CAAC,CAACV,IAAMA,EAAE5C,GAAG,KAAKoD,cAAcpD,GAAG;YACpE,IAAIqD,eAAe,OAAOA;QAC5B;QAEA,MAAME,aAAaJ,MAAMK,WAAW;QAEpC,oCAAoC;QACpC,MAAMC,aAAavE,OAAOoE,IAAI,CAAC,CAACV,IAAMA,EAAE3F,IAAI,CAACuG,WAAW,OAAOD;QAC/D,IAAIE,YAAY,OAAOA;QAEvB,wCAAwC;QACxC,MAAMC,UAAUxE,OAAOsD,MAAM,CAAC,CAACI,IAAMA,EAAE3F,IAAI,CAACuG,WAAW,GAAGG,QAAQ,CAACJ;QAEnE,IAAIG,QAAQ3E,MAAM,KAAK,GAAG,OAAO2E,OAAO,CAAC,EAAE;QAC3C,IAAIA,QAAQ3E,MAAM,GAAG,GAAG,OAAO2E;QAE/B,OAAO;IACT;AACF"}
|
|
@@ -398,6 +398,14 @@ describe("AgentManager", ()=>{
|
|
|
398
398
|
force: true
|
|
399
399
|
});
|
|
400
400
|
});
|
|
401
|
+
function useHerdrPanes(panes) {
|
|
402
|
+
const fetchHerdrPanes = vi.fn().mockResolvedValue(panes);
|
|
403
|
+
scopedManager = new AgentManager(registry, undefined, {
|
|
404
|
+
runtimeProvider: "herdr",
|
|
405
|
+
fetchHerdrAgentPanes: fetchHerdrPanes
|
|
406
|
+
});
|
|
407
|
+
return fetchHerdrPanes;
|
|
408
|
+
}
|
|
401
409
|
it("persists every detected agent to the registry", async ()=>{
|
|
402
410
|
scopedManager.registerAdapter(new MockAdapter("claude", [
|
|
403
411
|
createMockAgent({
|
|
@@ -423,6 +431,206 @@ describe("AgentManager", ()=>{
|
|
|
423
431
|
});
|
|
424
432
|
expect(entries[0].startedAt).toMatch(/^\d{4}-\d{2}-\d{2}T/);
|
|
425
433
|
});
|
|
434
|
+
it("persists a Herdr runtime ref for manually discovered agents matched by session id", async ()=>{
|
|
435
|
+
useHerdrPanes([
|
|
436
|
+
{
|
|
437
|
+
agent: "codex",
|
|
438
|
+
agentSessionId: "01a0a05e-751f-7850-9fd7-247e297f9ac5",
|
|
439
|
+
cwd: "/cwd/ai-devkit",
|
|
440
|
+
paneId: "w31:p1",
|
|
441
|
+
workspaceId: "w31",
|
|
442
|
+
tabId: "w31:t1"
|
|
443
|
+
}
|
|
444
|
+
]);
|
|
445
|
+
scopedManager.registerAdapter(new MockAdapter("codex", [
|
|
446
|
+
createMockAgent({
|
|
447
|
+
name: "ai-devkit-44290",
|
|
448
|
+
type: "codex",
|
|
449
|
+
pid: process.pid,
|
|
450
|
+
sessionId: "01a0a05e-751f-7850-9fd7-247e297f9ac5",
|
|
451
|
+
sessionFilePath: "/path/codex.jsonl",
|
|
452
|
+
projectPath: "/cwd/ai-devkit"
|
|
453
|
+
})
|
|
454
|
+
]));
|
|
455
|
+
await scopedManager.listAgents();
|
|
456
|
+
expect(registry.list()[0]).toMatchObject({
|
|
457
|
+
name: "ai-devkit-44290",
|
|
458
|
+
type: "codex",
|
|
459
|
+
pid: process.pid,
|
|
460
|
+
runtime: "herdr",
|
|
461
|
+
runtimeRef: {
|
|
462
|
+
session: "default",
|
|
463
|
+
paneId: "w31:p1",
|
|
464
|
+
workspaceId: "w31",
|
|
465
|
+
tabId: "w31:t1",
|
|
466
|
+
agentName: "ai-devkit-44290"
|
|
467
|
+
}
|
|
468
|
+
});
|
|
469
|
+
});
|
|
470
|
+
it("upgrades an existing tmux fallback row when Herdr later matches by session id", async ()=>{
|
|
471
|
+
useHerdrPanes([
|
|
472
|
+
{
|
|
473
|
+
agent: "codex",
|
|
474
|
+
agentSessionId: "01a0a05e-751f-7850-9fd7-247e297f9ac5",
|
|
475
|
+
cwd: "/cwd/ai-devkit",
|
|
476
|
+
paneId: "w31:p1"
|
|
477
|
+
}
|
|
478
|
+
]);
|
|
479
|
+
registry.register({
|
|
480
|
+
name: "ai-devkit-44290",
|
|
481
|
+
type: "codex",
|
|
482
|
+
pid: process.pid,
|
|
483
|
+
runtime: "tmux",
|
|
484
|
+
runtimeRef: null,
|
|
485
|
+
cwd: "/cwd/ai-devkit",
|
|
486
|
+
startedAt: "2026-09-14T00:00:00.000Z",
|
|
487
|
+
sessionId: "01a0a05e-751f-7850-9fd7-247e297f9ac5",
|
|
488
|
+
sessionFilePath: "/path/codex.jsonl"
|
|
489
|
+
});
|
|
490
|
+
scopedManager.registerAdapter(new MockAdapter("codex", [
|
|
491
|
+
createMockAgent({
|
|
492
|
+
name: "ai-devkit-44290",
|
|
493
|
+
type: "codex",
|
|
494
|
+
pid: process.pid,
|
|
495
|
+
sessionId: "01a0a05e-751f-7850-9fd7-247e297f9ac5",
|
|
496
|
+
sessionFilePath: "/path/codex.jsonl",
|
|
497
|
+
projectPath: "/cwd/ai-devkit"
|
|
498
|
+
})
|
|
499
|
+
]));
|
|
500
|
+
await scopedManager.listAgents();
|
|
501
|
+
expect(registry.lookup("ai-devkit-44290")).toMatchObject({
|
|
502
|
+
runtime: "herdr",
|
|
503
|
+
runtimeRef: {
|
|
504
|
+
session: "default",
|
|
505
|
+
paneId: "w31:p1",
|
|
506
|
+
agentName: "ai-devkit-44290"
|
|
507
|
+
}
|
|
508
|
+
});
|
|
509
|
+
});
|
|
510
|
+
it("upgrades an existing tmux fallback row when Herdr matches by foreground pid", async ()=>{
|
|
511
|
+
useHerdrPanes([
|
|
512
|
+
{
|
|
513
|
+
agent: "codex",
|
|
514
|
+
cwd: "/cwd/ai-devkit",
|
|
515
|
+
paneId: "w31:p1",
|
|
516
|
+
foregroundPids: [
|
|
517
|
+
44290
|
|
518
|
+
]
|
|
519
|
+
},
|
|
520
|
+
{
|
|
521
|
+
agent: "codex",
|
|
522
|
+
cwd: "/cwd/ai-devkit",
|
|
523
|
+
paneId: "w32:p1",
|
|
524
|
+
foregroundPids: [
|
|
525
|
+
process.pid
|
|
526
|
+
]
|
|
527
|
+
}
|
|
528
|
+
]);
|
|
529
|
+
registry.register({
|
|
530
|
+
name: "ai-devkit-64904",
|
|
531
|
+
type: "codex",
|
|
532
|
+
pid: process.pid,
|
|
533
|
+
runtime: "tmux",
|
|
534
|
+
runtimeRef: null,
|
|
535
|
+
cwd: "/cwd/ai-devkit",
|
|
536
|
+
startedAt: "2026-09-14T00:00:00.000Z",
|
|
537
|
+
sessionId: `pid-${process.pid}`,
|
|
538
|
+
sessionFilePath: ""
|
|
539
|
+
});
|
|
540
|
+
scopedManager.registerAdapter(new MockAdapter("codex", [
|
|
541
|
+
createMockAgent({
|
|
542
|
+
name: "ai-devkit-64904",
|
|
543
|
+
type: "codex",
|
|
544
|
+
pid: process.pid,
|
|
545
|
+
sessionId: `pid-${process.pid}`,
|
|
546
|
+
sessionFilePath: undefined,
|
|
547
|
+
projectPath: "/cwd/ai-devkit"
|
|
548
|
+
})
|
|
549
|
+
]));
|
|
550
|
+
await scopedManager.listAgents();
|
|
551
|
+
expect(registry.lookup("ai-devkit-64904")).toMatchObject({
|
|
552
|
+
runtime: "herdr",
|
|
553
|
+
runtimeRef: {
|
|
554
|
+
session: "default",
|
|
555
|
+
paneId: "w32:p1",
|
|
556
|
+
agentName: "ai-devkit-64904"
|
|
557
|
+
}
|
|
558
|
+
});
|
|
559
|
+
});
|
|
560
|
+
it("fetches Herdr panes automatically when the configured runtime is Herdr", async ()=>{
|
|
561
|
+
const fetchHerdrPanes = useHerdrPanes([
|
|
562
|
+
{
|
|
563
|
+
agent: "codex",
|
|
564
|
+
cwd: "/cwd/ai-devkit",
|
|
565
|
+
paneId: "w32:p1",
|
|
566
|
+
foregroundPids: [
|
|
567
|
+
process.pid
|
|
568
|
+
]
|
|
569
|
+
}
|
|
570
|
+
]);
|
|
571
|
+
scopedManager.registerAdapter(new MockAdapter("codex", [
|
|
572
|
+
createMockAgent({
|
|
573
|
+
name: "ai-devkit-64904",
|
|
574
|
+
type: "codex",
|
|
575
|
+
pid: process.pid,
|
|
576
|
+
sessionId: `pid-${process.pid}`,
|
|
577
|
+
sessionFilePath: undefined,
|
|
578
|
+
projectPath: "/cwd/ai-devkit"
|
|
579
|
+
})
|
|
580
|
+
]));
|
|
581
|
+
await scopedManager.listAgents();
|
|
582
|
+
expect(fetchHerdrPanes).toHaveBeenCalledOnce();
|
|
583
|
+
expect(registry.lookup("ai-devkit-64904")).toMatchObject({
|
|
584
|
+
runtime: "herdr",
|
|
585
|
+
runtimeRef: {
|
|
586
|
+
session: "default",
|
|
587
|
+
paneId: "w32:p1",
|
|
588
|
+
agentName: "ai-devkit-64904"
|
|
589
|
+
}
|
|
590
|
+
});
|
|
591
|
+
});
|
|
592
|
+
it("upgrades an existing tmux row with a malformed runtime ref when Herdr matches", async ()=>{
|
|
593
|
+
useHerdrPanes([
|
|
594
|
+
{
|
|
595
|
+
agent: "codex",
|
|
596
|
+
cwd: "/cwd/ai-devkit",
|
|
597
|
+
paneId: "w32:p1",
|
|
598
|
+
foregroundPids: [
|
|
599
|
+
process.pid
|
|
600
|
+
]
|
|
601
|
+
}
|
|
602
|
+
]);
|
|
603
|
+
registry.register({
|
|
604
|
+
name: "ai-devkit-64904",
|
|
605
|
+
type: "codex",
|
|
606
|
+
pid: process.pid,
|
|
607
|
+
runtime: "tmux",
|
|
608
|
+
runtimeRef: {},
|
|
609
|
+
cwd: "/cwd/ai-devkit",
|
|
610
|
+
startedAt: "2026-09-14T00:00:00.000Z",
|
|
611
|
+
sessionId: `pid-${process.pid}`,
|
|
612
|
+
sessionFilePath: ""
|
|
613
|
+
});
|
|
614
|
+
scopedManager.registerAdapter(new MockAdapter("codex", [
|
|
615
|
+
createMockAgent({
|
|
616
|
+
name: "ai-devkit-64904",
|
|
617
|
+
type: "codex",
|
|
618
|
+
pid: process.pid,
|
|
619
|
+
sessionId: `pid-${process.pid}`,
|
|
620
|
+
sessionFilePath: undefined,
|
|
621
|
+
projectPath: "/cwd/ai-devkit"
|
|
622
|
+
})
|
|
623
|
+
]));
|
|
624
|
+
await scopedManager.listAgents();
|
|
625
|
+
expect(registry.lookup("ai-devkit-64904")).toMatchObject({
|
|
626
|
+
runtime: "herdr",
|
|
627
|
+
runtimeRef: {
|
|
628
|
+
session: "default",
|
|
629
|
+
paneId: "w32:p1",
|
|
630
|
+
agentName: "ai-devkit-64904"
|
|
631
|
+
}
|
|
632
|
+
});
|
|
633
|
+
});
|
|
426
634
|
it("prunes entries for dead pids", async ()=>{
|
|
427
635
|
registry.register({
|
|
428
636
|
name: "dead",
|
|
@@ -659,7 +867,7 @@ describe("AgentManager", ()=>{
|
|
|
659
867
|
expect(agents[0].pinned).toBe(false);
|
|
660
868
|
expect(agents[0].lastActive.toISOString()).toBe("2026-01-01T00:00:00.000Z");
|
|
661
869
|
});
|
|
662
|
-
it("
|
|
870
|
+
it("does not persist passive cwd changes when an existing cwd is present", async ()=>{
|
|
663
871
|
const adapter = new MockAdapter("claude", [
|
|
664
872
|
createMockAgent({
|
|
665
873
|
name: "changing",
|
|
@@ -681,10 +889,9 @@ describe("AgentManager", ()=>{
|
|
|
681
889
|
await scopedManager.listAgents();
|
|
682
890
|
const upserts = databaseOperations.filter((sql)=>/^\s*INSERT INTO agents/i.test(sql));
|
|
683
891
|
const transactions = databaseOperations.filter((sql)=>/^\s*(BEGIN|COMMIT)/i.test(sql));
|
|
684
|
-
expect(upserts).toHaveLength(
|
|
685
|
-
expect(
|
|
686
|
-
expect(
|
|
687
|
-
expect(registry.lookup("changing")?.cwd).toBe("/cwd/after");
|
|
892
|
+
expect(upserts).toHaveLength(0);
|
|
893
|
+
expect(transactions).toHaveLength(0);
|
|
894
|
+
expect(registry.lookup("changing")?.cwd).toBe("/cwd/before");
|
|
688
895
|
});
|
|
689
896
|
it("prunes newly dead entries only when the passive cadence is due", async ()=>{
|
|
690
897
|
registry.register({
|