@snoglobe/helios 0.1.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/LICENSE +202 -0
- package/README.md +222 -0
- package/dist/app.d.ts +9 -0
- package/dist/app.d.ts.map +1 -0
- package/dist/app.js +289 -0
- package/dist/app.js.map +1 -0
- package/dist/core/agent-loop.d.ts +16 -0
- package/dist/core/agent-loop.d.ts.map +1 -0
- package/dist/core/agent-loop.js +64 -0
- package/dist/core/agent-loop.js.map +1 -0
- package/dist/core/monitor.d.ts +15 -0
- package/dist/core/monitor.d.ts.map +1 -0
- package/dist/core/monitor.js +40 -0
- package/dist/core/monitor.js.map +1 -0
- package/dist/core/orchestrator.d.ts +51 -0
- package/dist/core/orchestrator.d.ts.map +1 -0
- package/dist/core/orchestrator.js +235 -0
- package/dist/core/orchestrator.js.map +1 -0
- package/dist/core/state-machine.d.ts +19 -0
- package/dist/core/state-machine.d.ts.map +1 -0
- package/dist/core/state-machine.js +42 -0
- package/dist/core/state-machine.js.map +1 -0
- package/dist/core/stickies.d.ts +20 -0
- package/dist/core/stickies.d.ts.map +1 -0
- package/dist/core/stickies.js +41 -0
- package/dist/core/stickies.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +68 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/context-gate.d.ts +49 -0
- package/dist/memory/context-gate.d.ts.map +1 -0
- package/dist/memory/context-gate.js +109 -0
- package/dist/memory/context-gate.js.map +1 -0
- package/dist/memory/experiment-tracker.d.ts +21 -0
- package/dist/memory/experiment-tracker.d.ts.map +1 -0
- package/dist/memory/experiment-tracker.js +117 -0
- package/dist/memory/experiment-tracker.js.map +1 -0
- package/dist/memory/memory-store.d.ts +37 -0
- package/dist/memory/memory-store.d.ts.map +1 -0
- package/dist/memory/memory-store.js +150 -0
- package/dist/memory/memory-store.js.map +1 -0
- package/dist/memory/token-estimator.d.ts +7 -0
- package/dist/memory/token-estimator.d.ts.map +1 -0
- package/dist/memory/token-estimator.js +30 -0
- package/dist/memory/token-estimator.js.map +1 -0
- package/dist/metrics/analyzer.d.ts +15 -0
- package/dist/metrics/analyzer.d.ts.map +1 -0
- package/dist/metrics/analyzer.js +67 -0
- package/dist/metrics/analyzer.js.map +1 -0
- package/dist/metrics/collector.d.ts +23 -0
- package/dist/metrics/collector.d.ts.map +1 -0
- package/dist/metrics/collector.js +60 -0
- package/dist/metrics/collector.js.map +1 -0
- package/dist/metrics/parser.d.ts +21 -0
- package/dist/metrics/parser.d.ts.map +1 -0
- package/dist/metrics/parser.js +43 -0
- package/dist/metrics/parser.js.map +1 -0
- package/dist/metrics/parsers/csv.d.ts +8 -0
- package/dist/metrics/parsers/csv.d.ts.map +1 -0
- package/dist/metrics/parsers/csv.js +37 -0
- package/dist/metrics/parsers/csv.js.map +1 -0
- package/dist/metrics/parsers/json.d.ts +8 -0
- package/dist/metrics/parsers/json.d.ts.map +1 -0
- package/dist/metrics/parsers/json.js +42 -0
- package/dist/metrics/parsers/json.js.map +1 -0
- package/dist/metrics/parsers/stdout.d.ts +3 -0
- package/dist/metrics/parsers/stdout.d.ts.map +1 -0
- package/dist/metrics/parsers/stdout.js +79 -0
- package/dist/metrics/parsers/stdout.js.map +1 -0
- package/dist/metrics/parsers/tensorboard.d.ts +16 -0
- package/dist/metrics/parsers/tensorboard.d.ts.map +1 -0
- package/dist/metrics/parsers/tensorboard.js +22 -0
- package/dist/metrics/parsers/tensorboard.js.map +1 -0
- package/dist/metrics/store.d.ts +33 -0
- package/dist/metrics/store.d.ts.map +1 -0
- package/dist/metrics/store.js +141 -0
- package/dist/metrics/store.js.map +1 -0
- package/dist/providers/auth/auth-manager.d.ts +22 -0
- package/dist/providers/auth/auth-manager.d.ts.map +1 -0
- package/dist/providers/auth/auth-manager.js +77 -0
- package/dist/providers/auth/auth-manager.js.map +1 -0
- package/dist/providers/auth/token-store.d.ts +14 -0
- package/dist/providers/auth/token-store.d.ts.map +1 -0
- package/dist/providers/auth/token-store.js +55 -0
- package/dist/providers/auth/token-store.js.map +1 -0
- package/dist/providers/claude/provider.d.ts +46 -0
- package/dist/providers/claude/provider.d.ts.map +1 -0
- package/dist/providers/claude/provider.js +606 -0
- package/dist/providers/claude/provider.js.map +1 -0
- package/dist/providers/openai/callback-server.d.ts +11 -0
- package/dist/providers/openai/callback-server.d.ts.map +1 -0
- package/dist/providers/openai/callback-server.js +58 -0
- package/dist/providers/openai/callback-server.js.map +1 -0
- package/dist/providers/openai/oauth.d.ts +16 -0
- package/dist/providers/openai/oauth.d.ts.map +1 -0
- package/dist/providers/openai/oauth.js +110 -0
- package/dist/providers/openai/oauth.js.map +1 -0
- package/dist/providers/openai/provider.d.ts +29 -0
- package/dist/providers/openai/provider.d.ts.map +1 -0
- package/dist/providers/openai/provider.js +363 -0
- package/dist/providers/openai/provider.js.map +1 -0
- package/dist/providers/retry.d.ts +6 -0
- package/dist/providers/retry.d.ts.map +1 -0
- package/dist/providers/retry.js +13 -0
- package/dist/providers/retry.js.map +1 -0
- package/dist/providers/sse.d.ts +6 -0
- package/dist/providers/sse.d.ts.map +1 -0
- package/dist/providers/sse.js +35 -0
- package/dist/providers/sse.js.map +1 -0
- package/dist/providers/types.d.ts +102 -0
- package/dist/providers/types.d.ts.map +1 -0
- package/dist/providers/types.js +11 -0
- package/dist/providers/types.js.map +1 -0
- package/dist/remote/config.d.ts +14 -0
- package/dist/remote/config.d.ts.map +1 -0
- package/dist/remote/config.js +81 -0
- package/dist/remote/config.js.map +1 -0
- package/dist/remote/connection-pool.d.ts +26 -0
- package/dist/remote/connection-pool.d.ts.map +1 -0
- package/dist/remote/connection-pool.js +268 -0
- package/dist/remote/connection-pool.js.map +1 -0
- package/dist/remote/executor.d.ts +23 -0
- package/dist/remote/executor.d.ts.map +1 -0
- package/dist/remote/executor.js +53 -0
- package/dist/remote/executor.js.map +1 -0
- package/dist/remote/file-sync.d.ts +19 -0
- package/dist/remote/file-sync.d.ts.map +1 -0
- package/dist/remote/file-sync.js +44 -0
- package/dist/remote/file-sync.js.map +1 -0
- package/dist/remote/types.d.ts +33 -0
- package/dist/remote/types.d.ts.map +1 -0
- package/dist/remote/types.js +2 -0
- package/dist/remote/types.js.map +1 -0
- package/dist/scheduler/sleep-manager.d.ts +33 -0
- package/dist/scheduler/sleep-manager.d.ts.map +1 -0
- package/dist/scheduler/sleep-manager.js +132 -0
- package/dist/scheduler/sleep-manager.js.map +1 -0
- package/dist/scheduler/ssh-batcher.d.ts +23 -0
- package/dist/scheduler/ssh-batcher.d.ts.map +1 -0
- package/dist/scheduler/ssh-batcher.js +98 -0
- package/dist/scheduler/ssh-batcher.js.map +1 -0
- package/dist/scheduler/state-store.d.ts +8 -0
- package/dist/scheduler/state-store.d.ts.map +1 -0
- package/dist/scheduler/state-store.js +48 -0
- package/dist/scheduler/state-store.js.map +1 -0
- package/dist/scheduler/trigger-scheduler.d.ts +24 -0
- package/dist/scheduler/trigger-scheduler.d.ts.map +1 -0
- package/dist/scheduler/trigger-scheduler.js +142 -0
- package/dist/scheduler/trigger-scheduler.js.map +1 -0
- package/dist/scheduler/triggers/file.d.ts +4 -0
- package/dist/scheduler/triggers/file.d.ts.map +1 -0
- package/dist/scheduler/triggers/file.js +39 -0
- package/dist/scheduler/triggers/file.js.map +1 -0
- package/dist/scheduler/triggers/metric.d.ts +4 -0
- package/dist/scheduler/triggers/metric.d.ts.map +1 -0
- package/dist/scheduler/triggers/metric.js +74 -0
- package/dist/scheduler/triggers/metric.js.map +1 -0
- package/dist/scheduler/triggers/process-exit.d.ts +4 -0
- package/dist/scheduler/triggers/process-exit.d.ts.map +1 -0
- package/dist/scheduler/triggers/process-exit.js +13 -0
- package/dist/scheduler/triggers/process-exit.js.map +1 -0
- package/dist/scheduler/triggers/resource.d.ts +4 -0
- package/dist/scheduler/triggers/resource.d.ts.map +1 -0
- package/dist/scheduler/triggers/resource.js +54 -0
- package/dist/scheduler/triggers/resource.js.map +1 -0
- package/dist/scheduler/triggers/timer.d.ts +3 -0
- package/dist/scheduler/triggers/timer.d.ts.map +1 -0
- package/dist/scheduler/triggers/timer.js +4 -0
- package/dist/scheduler/triggers/timer.js.map +1 -0
- package/dist/scheduler/triggers/types.d.ts +95 -0
- package/dist/scheduler/triggers/types.d.ts.map +1 -0
- package/dist/scheduler/triggers/types.js +11 -0
- package/dist/scheduler/triggers/types.js.map +1 -0
- package/dist/store/database.d.ts +5 -0
- package/dist/store/database.d.ts.map +1 -0
- package/dist/store/database.js +29 -0
- package/dist/store/database.js.map +1 -0
- package/dist/store/migrations.d.ts +3 -0
- package/dist/store/migrations.d.ts.map +1 -0
- package/dist/store/migrations.js +124 -0
- package/dist/store/migrations.js.map +1 -0
- package/dist/store/preferences.d.ts +7 -0
- package/dist/store/preferences.d.ts.map +1 -0
- package/dist/store/preferences.js +26 -0
- package/dist/store/preferences.js.map +1 -0
- package/dist/store/session-store.d.ts +29 -0
- package/dist/store/session-store.d.ts.map +1 -0
- package/dist/store/session-store.js +103 -0
- package/dist/store/session-store.js.map +1 -0
- package/dist/tools/clear-metrics.d.ts +5 -0
- package/dist/tools/clear-metrics.d.ts.map +1 -0
- package/dist/tools/clear-metrics.js +27 -0
- package/dist/tools/clear-metrics.js.map +1 -0
- package/dist/tools/compare-runs.d.ts +4 -0
- package/dist/tools/compare-runs.d.ts.map +1 -0
- package/dist/tools/compare-runs.js +77 -0
- package/dist/tools/compare-runs.js.map +1 -0
- package/dist/tools/consult.d.ts +3 -0
- package/dist/tools/consult.d.ts.map +1 -0
- package/dist/tools/consult.js +62 -0
- package/dist/tools/consult.js.map +1 -0
- package/dist/tools/file-ops.d.ts +6 -0
- package/dist/tools/file-ops.d.ts.map +1 -0
- package/dist/tools/file-ops.js +158 -0
- package/dist/tools/file-ops.js.map +1 -0
- package/dist/tools/kill-task.d.ts +6 -0
- package/dist/tools/kill-task.d.ts.map +1 -0
- package/dist/tools/kill-task.js +55 -0
- package/dist/tools/kill-task.js.map +1 -0
- package/dist/tools/list-machines.d.ts +4 -0
- package/dist/tools/list-machines.d.ts.map +1 -0
- package/dist/tools/list-machines.js +24 -0
- package/dist/tools/list-machines.js.map +1 -0
- package/dist/tools/memory-tools.d.ts +8 -0
- package/dist/tools/memory-tools.d.ts.map +1 -0
- package/dist/tools/memory-tools.js +122 -0
- package/dist/tools/memory-tools.js.map +1 -0
- package/dist/tools/metrics-query.d.ts +4 -0
- package/dist/tools/metrics-query.d.ts.map +1 -0
- package/dist/tools/metrics-query.js +69 -0
- package/dist/tools/metrics-query.js.map +1 -0
- package/dist/tools/monitor.d.ts +5 -0
- package/dist/tools/monitor.d.ts.map +1 -0
- package/dist/tools/monitor.js +64 -0
- package/dist/tools/monitor.js.map +1 -0
- package/dist/tools/remote-exec.d.ts +6 -0
- package/dist/tools/remote-exec.d.ts.map +1 -0
- package/dist/tools/remote-exec.js +98 -0
- package/dist/tools/remote-exec.js.map +1 -0
- package/dist/tools/remote-sync.d.ts +5 -0
- package/dist/tools/remote-sync.d.ts.map +1 -0
- package/dist/tools/remote-sync.js +57 -0
- package/dist/tools/remote-sync.js.map +1 -0
- package/dist/tools/show-metrics.d.ts +4 -0
- package/dist/tools/show-metrics.d.ts.map +1 -0
- package/dist/tools/show-metrics.js +69 -0
- package/dist/tools/show-metrics.js.map +1 -0
- package/dist/tools/sleep.d.ts +4 -0
- package/dist/tools/sleep.d.ts.map +1 -0
- package/dist/tools/sleep.js +116 -0
- package/dist/tools/sleep.js.map +1 -0
- package/dist/tools/task-output.d.ts +5 -0
- package/dist/tools/task-output.d.ts.map +1 -0
- package/dist/tools/task-output.js +51 -0
- package/dist/tools/task-output.js.map +1 -0
- package/dist/tools/web-fetch.d.ts +3 -0
- package/dist/tools/web-fetch.d.ts.map +1 -0
- package/dist/tools/web-fetch.js +112 -0
- package/dist/tools/web-fetch.js.map +1 -0
- package/dist/ui/commands.d.ts +7 -0
- package/dist/ui/commands.d.ts.map +1 -0
- package/dist/ui/commands.js +21 -0
- package/dist/ui/commands.js.map +1 -0
- package/dist/ui/components/gradient-edge.d.ts +7 -0
- package/dist/ui/components/gradient-edge.d.ts.map +1 -0
- package/dist/ui/components/gradient-edge.js +33 -0
- package/dist/ui/components/gradient-edge.js.map +1 -0
- package/dist/ui/components/input-bar.d.ts +8 -0
- package/dist/ui/components/input-bar.d.ts.map +1 -0
- package/dist/ui/components/input-bar.js +139 -0
- package/dist/ui/components/input-bar.js.map +1 -0
- package/dist/ui/components/key-hint-rule.d.ts +2 -0
- package/dist/ui/components/key-hint-rule.d.ts.map +1 -0
- package/dist/ui/components/key-hint-rule.js +19 -0
- package/dist/ui/components/key-hint-rule.js.map +1 -0
- package/dist/ui/components/overlay-header.d.ts +6 -0
- package/dist/ui/components/overlay-header.d.ts.map +1 -0
- package/dist/ui/components/overlay-header.js +10 -0
- package/dist/ui/components/overlay-header.js.map +1 -0
- package/dist/ui/components/status-bar.d.ts +11 -0
- package/dist/ui/components/status-bar.d.ts.map +1 -0
- package/dist/ui/components/status-bar.js +55 -0
- package/dist/ui/components/status-bar.js.map +1 -0
- package/dist/ui/format.d.ts +12 -0
- package/dist/ui/format.d.ts.map +1 -0
- package/dist/ui/format.js +36 -0
- package/dist/ui/format.js.map +1 -0
- package/dist/ui/layout.d.ts +48 -0
- package/dist/ui/layout.d.ts.map +1 -0
- package/dist/ui/layout.js +857 -0
- package/dist/ui/layout.js.map +1 -0
- package/dist/ui/markdown.d.ts +7 -0
- package/dist/ui/markdown.d.ts.map +1 -0
- package/dist/ui/markdown.js +67 -0
- package/dist/ui/markdown.js.map +1 -0
- package/dist/ui/mouse-filter.d.ts +21 -0
- package/dist/ui/mouse-filter.d.ts.map +1 -0
- package/dist/ui/mouse-filter.js +66 -0
- package/dist/ui/mouse-filter.js.map +1 -0
- package/dist/ui/overlays/metrics-overlay.d.ts +11 -0
- package/dist/ui/overlays/metrics-overlay.d.ts.map +1 -0
- package/dist/ui/overlays/metrics-overlay.js +86 -0
- package/dist/ui/overlays/metrics-overlay.js.map +1 -0
- package/dist/ui/overlays/task-overlay.d.ts +12 -0
- package/dist/ui/overlays/task-overlay.d.ts.map +1 -0
- package/dist/ui/overlays/task-overlay.js +77 -0
- package/dist/ui/overlays/task-overlay.js.map +1 -0
- package/dist/ui/panels/conversation.d.ts +8 -0
- package/dist/ui/panels/conversation.d.ts.map +1 -0
- package/dist/ui/panels/conversation.js +216 -0
- package/dist/ui/panels/conversation.js.map +1 -0
- package/dist/ui/panels/metrics-dashboard.d.ts +8 -0
- package/dist/ui/panels/metrics-dashboard.d.ts.map +1 -0
- package/dist/ui/panels/metrics-dashboard.js +56 -0
- package/dist/ui/panels/metrics-dashboard.js.map +1 -0
- package/dist/ui/panels/sleep-panel.d.ts +7 -0
- package/dist/ui/panels/sleep-panel.d.ts.map +1 -0
- package/dist/ui/panels/sleep-panel.js +48 -0
- package/dist/ui/panels/sleep-panel.js.map +1 -0
- package/dist/ui/panels/sticky-notes.d.ts +8 -0
- package/dist/ui/panels/sticky-notes.d.ts.map +1 -0
- package/dist/ui/panels/sticky-notes.js +49 -0
- package/dist/ui/panels/sticky-notes.js.map +1 -0
- package/dist/ui/panels/task-list.d.ts +8 -0
- package/dist/ui/panels/task-list.d.ts.map +1 -0
- package/dist/ui/panels/task-list.js +39 -0
- package/dist/ui/panels/task-list.js.map +1 -0
- package/dist/ui/theme.d.ts +25 -0
- package/dist/ui/theme.d.ts.map +1 -0
- package/dist/ui/theme.js +50 -0
- package/dist/ui/theme.js.map +1 -0
- package/dist/version.d.ts +9 -0
- package/dist/version.d.ts.map +1 -0
- package/dist/version.js +62 -0
- package/dist/version.js.map +1 -0
- package/package.json +59 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.d.ts","sourceRoot":"","sources":["../../src/providers/sse.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,wBAAuB,aAAa,CAAC,IAAI,EAAE,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,CA4B5E"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Shared SSE (Server-Sent Events) stream parser.
|
|
3
|
+
* Both Claude and OpenAI providers use SSE framing with identical structure.
|
|
4
|
+
*/
|
|
5
|
+
export async function* parseSSELines(resp) {
|
|
6
|
+
const reader = resp.body?.getReader();
|
|
7
|
+
if (!reader)
|
|
8
|
+
throw new Error("No response body");
|
|
9
|
+
const decoder = new TextDecoder();
|
|
10
|
+
let buffer = "";
|
|
11
|
+
while (true) {
|
|
12
|
+
const { done, value } = await reader.read();
|
|
13
|
+
if (done)
|
|
14
|
+
break;
|
|
15
|
+
buffer += decoder.decode(value, { stream: true });
|
|
16
|
+
const lines = buffer.split("\n");
|
|
17
|
+
buffer = lines.pop() ?? "";
|
|
18
|
+
for (const line of lines) {
|
|
19
|
+
if (!line.startsWith("data: "))
|
|
20
|
+
continue;
|
|
21
|
+
const data = line.slice(6).trim();
|
|
22
|
+
if (!data || data === "[DONE]")
|
|
23
|
+
continue;
|
|
24
|
+
try {
|
|
25
|
+
yield JSON.parse(data);
|
|
26
|
+
}
|
|
27
|
+
catch (e) {
|
|
28
|
+
// Skip malformed/partial JSON, re-throw anything else
|
|
29
|
+
if (!(e instanceof SyntaxError))
|
|
30
|
+
throw e;
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=sse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.js","sourceRoot":"","sources":["../../src/providers/sse.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,aAAa,CAAC,IAAc;IACjD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IACtC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAEjD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,IAAI;YAAE,MAAM;QAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAAE,SAAS;YACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,IAAI,IAAI,KAAK,QAAQ;gBAAE,SAAS;YAEzC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,sDAAsD;gBACtD,IAAI,CAAC,CAAC,CAAC,YAAY,WAAW,CAAC;oBAAE,MAAM,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,102 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
export interface TextEvent {
|
|
3
|
+
type: "text";
|
|
4
|
+
text: string;
|
|
5
|
+
/** Incremental delta for streaming */
|
|
6
|
+
delta?: string;
|
|
7
|
+
}
|
|
8
|
+
export interface ToolCallEvent {
|
|
9
|
+
type: "tool_call";
|
|
10
|
+
id: string;
|
|
11
|
+
name: string;
|
|
12
|
+
args: Record<string, unknown>;
|
|
13
|
+
}
|
|
14
|
+
export interface ToolResultEvent {
|
|
15
|
+
type: "tool_result";
|
|
16
|
+
callId: string;
|
|
17
|
+
result: string;
|
|
18
|
+
isError?: boolean;
|
|
19
|
+
}
|
|
20
|
+
export interface ErrorEvent {
|
|
21
|
+
type: "error";
|
|
22
|
+
error: Error;
|
|
23
|
+
recoverable: boolean;
|
|
24
|
+
}
|
|
25
|
+
export interface DoneEvent {
|
|
26
|
+
type: "done";
|
|
27
|
+
usage?: TokenUsage;
|
|
28
|
+
}
|
|
29
|
+
export type AgentEvent = TextEvent | ToolCallEvent | ToolResultEvent | ErrorEvent | DoneEvent;
|
|
30
|
+
export interface TokenUsage {
|
|
31
|
+
inputTokens: number;
|
|
32
|
+
outputTokens: number;
|
|
33
|
+
costUsd?: number;
|
|
34
|
+
}
|
|
35
|
+
export interface Session {
|
|
36
|
+
id: string;
|
|
37
|
+
providerId: string;
|
|
38
|
+
providerSessionId?: string;
|
|
39
|
+
createdAt: number;
|
|
40
|
+
lastActiveAt: number;
|
|
41
|
+
}
|
|
42
|
+
export interface SessionConfig {
|
|
43
|
+
systemPrompt?: string;
|
|
44
|
+
model?: string;
|
|
45
|
+
temperature?: number;
|
|
46
|
+
maxTokens?: number;
|
|
47
|
+
}
|
|
48
|
+
export declare const ToolParameterSchema: z.ZodObject<{
|
|
49
|
+
type: z.ZodLiteral<"object">;
|
|
50
|
+
properties: z.ZodRecord<z.ZodString, z.ZodAny>;
|
|
51
|
+
required: z.ZodOptional<z.ZodArray<z.ZodString>>;
|
|
52
|
+
}, z.core.$strip>;
|
|
53
|
+
export interface ToolDefinition {
|
|
54
|
+
name: string;
|
|
55
|
+
description: string;
|
|
56
|
+
parameters: z.infer<typeof ToolParameterSchema>;
|
|
57
|
+
execute: (args: Record<string, unknown>) => Promise<string>;
|
|
58
|
+
}
|
|
59
|
+
export interface ModelProvider {
|
|
60
|
+
readonly name: "claude" | "openai";
|
|
61
|
+
readonly displayName: string;
|
|
62
|
+
/** Check if the provider is authenticated */
|
|
63
|
+
isAuthenticated(): Promise<boolean>;
|
|
64
|
+
/** Initiate authentication flow */
|
|
65
|
+
authenticate(): Promise<void>;
|
|
66
|
+
/** Create a new conversation session */
|
|
67
|
+
createSession(config: SessionConfig): Promise<Session>;
|
|
68
|
+
/** Resume an existing session */
|
|
69
|
+
resumeSession(id: string): Promise<Session>;
|
|
70
|
+
/** Send a message and stream events */
|
|
71
|
+
send(session: Session, message: string, tools: ToolDefinition[]): AsyncGenerator<AgentEvent>;
|
|
72
|
+
/** Interrupt the current generation */
|
|
73
|
+
interrupt(session: Session): void;
|
|
74
|
+
/** Replace conversation history with a checkpoint briefing (context window management). */
|
|
75
|
+
resetHistory(session: Session, briefingMessage: string): void;
|
|
76
|
+
/** Clean up session resources */
|
|
77
|
+
closeSession(session: Session): Promise<void>;
|
|
78
|
+
/** Fetch available models (optional) */
|
|
79
|
+
fetchModels?(): Promise<ModelInfo[]>;
|
|
80
|
+
/** Get/set the current model */
|
|
81
|
+
currentModel: string;
|
|
82
|
+
/** Get/set reasoning effort */
|
|
83
|
+
reasoningEffort: ReasoningEffort;
|
|
84
|
+
}
|
|
85
|
+
export type ReasoningEffort = "none" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max";
|
|
86
|
+
export interface ModelInfo {
|
|
87
|
+
id: string;
|
|
88
|
+
name: string;
|
|
89
|
+
description?: string;
|
|
90
|
+
}
|
|
91
|
+
/** Faux-assistant response used after a context checkpoint reset. */
|
|
92
|
+
export declare const CHECKPOINT_ACK = "Understood. I have my memory tree and will continue working. Let me check what needs to be done.";
|
|
93
|
+
export type AuthMethod = "api_key" | "oauth";
|
|
94
|
+
export interface AuthCredentials {
|
|
95
|
+
method: AuthMethod;
|
|
96
|
+
provider: "claude" | "openai";
|
|
97
|
+
accessToken?: string;
|
|
98
|
+
refreshToken?: string;
|
|
99
|
+
apiKey?: string;
|
|
100
|
+
expiresAt?: number;
|
|
101
|
+
}
|
|
102
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/providers/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAIxB,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,sCAAsC;IACtC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,WAAW,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB;AAED,MAAM,MAAM,UAAU,GAClB,SAAS,GACT,aAAa,GACb,eAAe,GACf,UAAU,GACV,SAAS,CAAC;AAId,MAAM,WAAW,UAAU;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAID,MAAM,WAAW,OAAO;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,aAAa;IAC5B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAID,eAAO,MAAM,mBAAmB;;;;iBAI9B,CAAC;AAEH,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;IAChD,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CAC7D;AAID,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,QAAQ,CAAC;IACnC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAE7B,6CAA6C;IAC7C,eAAe,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAEpC,mCAAmC;IACnC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAE9B,wCAAwC;IACxC,aAAa,CAAC,MAAM,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAEvD,iCAAiC;IACjC,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAE5C,uCAAuC;IACvC,IAAI,CACF,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,cAAc,EAAE,GACtB,cAAc,CAAC,UAAU,CAAC,CAAC;IAE9B,uCAAuC;IACvC,SAAS,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAAC;IAElC,2FAA2F;IAC3F,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAE9D,iCAAiC;IACjC,YAAY,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE9C,wCAAwC;IACxC,WAAW,CAAC,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IAErC,gCAAgC;IAChC,YAAY,EAAE,MAAM,CAAC;IAErB,+BAA+B;IAC/B,eAAe,EAAE,eAAe,CAAC;CAClC;AAID,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,SAAS,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;AAE/F,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAID,qEAAqE;AACrE,eAAO,MAAM,cAAc,qGAAqG,CAAC;AAIjI,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,OAAO,CAAC;AAE7C,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
// --- Tool Definition ---
|
|
3
|
+
export const ToolParameterSchema = z.object({
|
|
4
|
+
type: z.literal("object"),
|
|
5
|
+
properties: z.record(z.string(), z.any()),
|
|
6
|
+
required: z.array(z.string()).optional(),
|
|
7
|
+
});
|
|
8
|
+
// --- Constants ---
|
|
9
|
+
/** Faux-assistant response used after a context checkpoint reset. */
|
|
10
|
+
export const CHECKPOINT_ACK = "Understood. I have my memory tree and will continue working. Let me check what needs to be done.";
|
|
11
|
+
//# sourceMappingURL=types.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/providers/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAoExB,0BAA0B;AAE1B,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1C,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;IACzB,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;IACzC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACzC,CAAC,CAAC;AA+DH,oBAAoB;AAEpB,qEAAqE;AACrE,MAAM,CAAC,MAAM,cAAc,GAAG,kGAAkG,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { RemoteMachine } from "./types.js";
|
|
2
|
+
export declare function loadMachines(): RemoteMachine[];
|
|
3
|
+
export declare function saveMachines(machines: RemoteMachine[]): void;
|
|
4
|
+
export declare function addMachine(machine: RemoteMachine): void;
|
|
5
|
+
export declare function removeMachine(id: string): boolean;
|
|
6
|
+
/**
|
|
7
|
+
* Parse a machine spec string into a RemoteMachine.
|
|
8
|
+
* Format: user@host[:port] [--key path] [--auth agent|key|password]
|
|
9
|
+
*/
|
|
10
|
+
export declare function parseMachineSpec(id: string, spec: string, options?: {
|
|
11
|
+
key?: string;
|
|
12
|
+
auth?: string;
|
|
13
|
+
}): RemoteMachine;
|
|
14
|
+
//# sourceMappingURL=config.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/remote/config.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAWhD,wBAAgB,YAAY,IAAI,aAAa,EAAE,CAU9C;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAI,CAG5D;AAED,wBAAgB,UAAU,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,CASvD;AAED,wBAAgB,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAMjD;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAC9B,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE;IAAE,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAO,GAC5C,aAAa,CA6Bf"}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { readFileSync, writeFileSync, existsSync, mkdirSync } from "node:fs";
|
|
2
|
+
import { join } from "node:path";
|
|
3
|
+
import { homedir } from "node:os";
|
|
4
|
+
const CONFIG_DIR = join(homedir(), ".helios");
|
|
5
|
+
const MACHINES_FILE = join(CONFIG_DIR, "machines.json");
|
|
6
|
+
function ensureConfigDir() {
|
|
7
|
+
if (!existsSync(CONFIG_DIR)) {
|
|
8
|
+
mkdirSync(CONFIG_DIR, { recursive: true });
|
|
9
|
+
}
|
|
10
|
+
}
|
|
11
|
+
export function loadMachines() {
|
|
12
|
+
try {
|
|
13
|
+
if (!existsSync(MACHINES_FILE))
|
|
14
|
+
return [];
|
|
15
|
+
const data = readFileSync(MACHINES_FILE, "utf-8");
|
|
16
|
+
const parsed = JSON.parse(data);
|
|
17
|
+
if (!Array.isArray(parsed))
|
|
18
|
+
return [];
|
|
19
|
+
return parsed;
|
|
20
|
+
}
|
|
21
|
+
catch {
|
|
22
|
+
return [];
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
export function saveMachines(machines) {
|
|
26
|
+
ensureConfigDir();
|
|
27
|
+
writeFileSync(MACHINES_FILE, JSON.stringify(machines, null, 2), "utf-8");
|
|
28
|
+
}
|
|
29
|
+
export function addMachine(machine) {
|
|
30
|
+
const machines = loadMachines();
|
|
31
|
+
const existing = machines.findIndex((m) => m.id === machine.id);
|
|
32
|
+
if (existing >= 0) {
|
|
33
|
+
machines[existing] = machine;
|
|
34
|
+
}
|
|
35
|
+
else {
|
|
36
|
+
machines.push(machine);
|
|
37
|
+
}
|
|
38
|
+
saveMachines(machines);
|
|
39
|
+
}
|
|
40
|
+
export function removeMachine(id) {
|
|
41
|
+
const machines = loadMachines();
|
|
42
|
+
const filtered = machines.filter((m) => m.id !== id);
|
|
43
|
+
if (filtered.length === machines.length)
|
|
44
|
+
return false;
|
|
45
|
+
saveMachines(filtered);
|
|
46
|
+
return true;
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Parse a machine spec string into a RemoteMachine.
|
|
50
|
+
* Format: user@host[:port] [--key path] [--auth agent|key|password]
|
|
51
|
+
*/
|
|
52
|
+
export function parseMachineSpec(id, spec, options = {}) {
|
|
53
|
+
// Parse user@host[:port]
|
|
54
|
+
const atIdx = spec.indexOf("@");
|
|
55
|
+
if (atIdx < 0)
|
|
56
|
+
throw new Error("Format: user@host[:port]");
|
|
57
|
+
const username = spec.slice(0, atIdx);
|
|
58
|
+
const hostPart = spec.slice(atIdx + 1);
|
|
59
|
+
let host;
|
|
60
|
+
let port = 22;
|
|
61
|
+
const colonIdx = hostPart.lastIndexOf(":");
|
|
62
|
+
if (colonIdx >= 0) {
|
|
63
|
+
host = hostPart.slice(0, colonIdx);
|
|
64
|
+
port = parseInt(hostPart.slice(colonIdx + 1), 10);
|
|
65
|
+
if (isNaN(port))
|
|
66
|
+
throw new Error("Invalid port number");
|
|
67
|
+
}
|
|
68
|
+
else {
|
|
69
|
+
host = hostPart;
|
|
70
|
+
}
|
|
71
|
+
const authMethod = options.auth ?? (options.key ? "key" : "agent");
|
|
72
|
+
return {
|
|
73
|
+
id,
|
|
74
|
+
host,
|
|
75
|
+
port,
|
|
76
|
+
username,
|
|
77
|
+
authMethod,
|
|
78
|
+
keyPath: options.key,
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/remote/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAGlC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,CAAC,CAAC;AAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;AAExD,SAAS,eAAe;IACtB,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,IAAI,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC;YAAE,OAAO,EAAE,CAAC;QAC1C,MAAM,IAAI,GAAG,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,CAAC;QACtC,OAAO,MAAyB,CAAC;IACnC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,QAAyB;IACpD,eAAe,EAAE,CAAC;IAClB,aAAa,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;AAC3E,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,OAAsB;IAC/C,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CAAC;IAChE,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;QAClB,QAAQ,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC;IACD,YAAY,CAAC,QAAQ,CAAC,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,EAAU;IACtC,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACrD,IAAI,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM;QAAE,OAAO,KAAK,CAAC;IACtD,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvB,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,EAAU,EACV,IAAY,EACZ,UAA2C,EAAE;IAE7C,yBAAyB;IACzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAChC,IAAI,KAAK,GAAG,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAE3D,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IAEvC,IAAI,IAAY,CAAC;IACjB,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC3C,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;QAClB,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QACnC,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAClD,IAAI,KAAK,CAAC,IAAI,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;IAC1D,CAAC;SAAM,CAAC;QACN,IAAI,GAAG,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,UAAU,GAAI,OAAO,CAAC,IAAqC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAErG,OAAO;QACL,EAAE;QACF,IAAI;QACJ,IAAI;QACJ,QAAQ;QACR,UAAU;QACV,OAAO,EAAE,OAAO,CAAC,GAAG;KACrB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import type { RemoteMachine, ExecResult, ConnectionStatus } from "./types.js";
|
|
2
|
+
export declare const LOCAL_MACHINE: RemoteMachine;
|
|
3
|
+
export declare class ConnectionPool {
|
|
4
|
+
private connections;
|
|
5
|
+
private machines;
|
|
6
|
+
constructor();
|
|
7
|
+
addMachine(machine: RemoteMachine): void;
|
|
8
|
+
removeMachine(id: string): void;
|
|
9
|
+
connect(machineId: string): Promise<void>;
|
|
10
|
+
exec(machineId: string, command: string): Promise<ExecResult>;
|
|
11
|
+
private execLocal;
|
|
12
|
+
execBackground(machineId: string, command: string, logPath?: string): Promise<{
|
|
13
|
+
pid: number;
|
|
14
|
+
logPath: string;
|
|
15
|
+
}>;
|
|
16
|
+
private execBackgroundLocal;
|
|
17
|
+
isProcessRunning(machineId: string, pid: number): Promise<boolean>;
|
|
18
|
+
tailFile(machineId: string, path: string, lines?: number): Promise<string>;
|
|
19
|
+
getStatus(machineId: string): ConnectionStatus;
|
|
20
|
+
getAllStatuses(): ConnectionStatus[];
|
|
21
|
+
getMachineIds(): string[];
|
|
22
|
+
disconnect(machineId: string): void;
|
|
23
|
+
disconnectAll(): void;
|
|
24
|
+
private getConnection;
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=connection-pool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"connection-pool.d.ts","sourceRoot":"","sources":["../../src/remote/connection-pool.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,aAAa,EACb,UAAU,EACV,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAEpB,eAAO,MAAM,aAAa,EAAE,aAO3B,CAAC;AAWF,qBAAa,cAAc;IACzB,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,QAAQ,CAAoC;;IAcpD,UAAU,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAIxC,aAAa,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAMzB,OAAO,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiFzC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC;IA8BnE,OAAO,CAAC,SAAS;IAwBX,cAAc,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAoB5C,OAAO,CAAC,mBAAmB;IAuCrB,gBAAgB,CACpB,SAAS,EAAE,MAAM,EACjB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,OAAO,CAAC;IAQb,QAAQ,CACZ,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,EACZ,KAAK,SAAK,GACT,OAAO,CAAC,MAAM,CAAC;IAKlB,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,gBAAgB;IAU9C,cAAc,IAAI,gBAAgB,EAAE;IAMpC,aAAa,IAAI,MAAM,EAAE;IAIzB,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IASnC,aAAa,IAAI,IAAI;YAMP,aAAa;CAa5B"}
|
|
@@ -0,0 +1,268 @@
|
|
|
1
|
+
import { Client } from "ssh2";
|
|
2
|
+
import { exec as cpExec, spawn } from "node:child_process";
|
|
3
|
+
import { readFileSync, openSync, closeSync } from "node:fs";
|
|
4
|
+
export const LOCAL_MACHINE = {
|
|
5
|
+
id: "local",
|
|
6
|
+
host: "localhost",
|
|
7
|
+
port: 0,
|
|
8
|
+
username: process.env.USER ?? "local",
|
|
9
|
+
authMethod: "local",
|
|
10
|
+
labels: ["local"],
|
|
11
|
+
};
|
|
12
|
+
export class ConnectionPool {
|
|
13
|
+
connections = new Map();
|
|
14
|
+
machines = new Map();
|
|
15
|
+
constructor() {
|
|
16
|
+
// Always register the local machine
|
|
17
|
+
this.machines.set("local", LOCAL_MACHINE);
|
|
18
|
+
this.connections.set("local", {
|
|
19
|
+
client: null,
|
|
20
|
+
machine: LOCAL_MACHINE,
|
|
21
|
+
connected: true,
|
|
22
|
+
lastUsedAt: Date.now(),
|
|
23
|
+
reconnectAttempts: 0,
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
addMachine(machine) {
|
|
27
|
+
this.machines.set(machine.id, machine);
|
|
28
|
+
}
|
|
29
|
+
removeMachine(id) {
|
|
30
|
+
if (id === "local")
|
|
31
|
+
return; // Can't remove local
|
|
32
|
+
this.disconnect(id);
|
|
33
|
+
this.machines.delete(id);
|
|
34
|
+
}
|
|
35
|
+
async connect(machineId) {
|
|
36
|
+
if (machineId === "local")
|
|
37
|
+
return; // Always connected
|
|
38
|
+
const machine = this.machines.get(machineId);
|
|
39
|
+
if (!machine)
|
|
40
|
+
throw new Error(`Unknown machine: ${machineId}`);
|
|
41
|
+
const existing = this.connections.get(machineId);
|
|
42
|
+
if (existing?.connected)
|
|
43
|
+
return;
|
|
44
|
+
// Clean up any previous failed client before creating a new one
|
|
45
|
+
if (existing?.client) {
|
|
46
|
+
try {
|
|
47
|
+
existing.client.end();
|
|
48
|
+
}
|
|
49
|
+
catch { /* best effort */ }
|
|
50
|
+
}
|
|
51
|
+
const client = new Client();
|
|
52
|
+
return new Promise((resolve, reject) => {
|
|
53
|
+
client.on("ready", () => {
|
|
54
|
+
this.connections.set(machineId, {
|
|
55
|
+
client,
|
|
56
|
+
machine,
|
|
57
|
+
connected: true,
|
|
58
|
+
lastUsedAt: Date.now(),
|
|
59
|
+
reconnectAttempts: 0,
|
|
60
|
+
});
|
|
61
|
+
resolve();
|
|
62
|
+
});
|
|
63
|
+
client.on("error", (err) => {
|
|
64
|
+
const errMsg = err instanceof Error ? err.message : String(err);
|
|
65
|
+
const conn = this.connections.get(machineId);
|
|
66
|
+
if (conn) {
|
|
67
|
+
conn.connected = false;
|
|
68
|
+
conn.lastError = errMsg;
|
|
69
|
+
}
|
|
70
|
+
else {
|
|
71
|
+
this.connections.set(machineId, {
|
|
72
|
+
client,
|
|
73
|
+
machine,
|
|
74
|
+
connected: false,
|
|
75
|
+
lastUsedAt: Date.now(),
|
|
76
|
+
reconnectAttempts: 0,
|
|
77
|
+
lastError: errMsg,
|
|
78
|
+
});
|
|
79
|
+
}
|
|
80
|
+
reject(err);
|
|
81
|
+
});
|
|
82
|
+
client.on("close", () => {
|
|
83
|
+
const conn = this.connections.get(machineId);
|
|
84
|
+
if (conn)
|
|
85
|
+
conn.connected = false;
|
|
86
|
+
});
|
|
87
|
+
const connectConfig = {
|
|
88
|
+
host: machine.host,
|
|
89
|
+
port: machine.port,
|
|
90
|
+
username: machine.username,
|
|
91
|
+
keepaliveInterval: 30_000,
|
|
92
|
+
keepaliveCountMax: 3,
|
|
93
|
+
readyTimeout: 30_000,
|
|
94
|
+
};
|
|
95
|
+
if (machine.authMethod === "key" && machine.keyPath) {
|
|
96
|
+
try {
|
|
97
|
+
connectConfig.privateKey = readFileSync(machine.keyPath);
|
|
98
|
+
}
|
|
99
|
+
catch (err) {
|
|
100
|
+
return reject(new Error(`Cannot read SSH key at ${machine.keyPath}: ${err instanceof Error ? err.message : String(err)}`));
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
else if (machine.authMethod === "agent") {
|
|
104
|
+
const sock = process.env.SSH_AUTH_SOCK;
|
|
105
|
+
if (!sock) {
|
|
106
|
+
return reject(new Error("SSH_AUTH_SOCK not set — cannot use agent auth. Start ssh-agent or use key auth."));
|
|
107
|
+
}
|
|
108
|
+
connectConfig.agent = sock;
|
|
109
|
+
}
|
|
110
|
+
else if (machine.authMethod === "password" && machine.password) {
|
|
111
|
+
connectConfig.password = machine.password;
|
|
112
|
+
}
|
|
113
|
+
client.connect(connectConfig);
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
async exec(machineId, command) {
|
|
117
|
+
if (machineId === "local") {
|
|
118
|
+
return this.execLocal(command);
|
|
119
|
+
}
|
|
120
|
+
const conn = await this.getConnection(machineId);
|
|
121
|
+
conn.lastUsedAt = Date.now();
|
|
122
|
+
return new Promise((resolve, reject) => {
|
|
123
|
+
conn.client.exec(command, (err, stream) => {
|
|
124
|
+
if (err)
|
|
125
|
+
return reject(err);
|
|
126
|
+
let stdout = "";
|
|
127
|
+
let stderr = "";
|
|
128
|
+
stream.on("data", (data) => {
|
|
129
|
+
stdout += data.toString();
|
|
130
|
+
});
|
|
131
|
+
stream.stderr.on("data", (data) => {
|
|
132
|
+
stderr += data.toString();
|
|
133
|
+
});
|
|
134
|
+
stream.on("error", (streamErr) => {
|
|
135
|
+
reject(streamErr);
|
|
136
|
+
});
|
|
137
|
+
stream.on("close", (code) => {
|
|
138
|
+
resolve({ stdout, stderr, exitCode: code ?? 0 });
|
|
139
|
+
});
|
|
140
|
+
});
|
|
141
|
+
});
|
|
142
|
+
}
|
|
143
|
+
execLocal(command, timeoutMs = 300_000) {
|
|
144
|
+
return new Promise((resolve) => {
|
|
145
|
+
cpExec(command, { maxBuffer: 10 * 1024 * 1024, timeout: timeoutMs, shell: "/bin/bash" }, (err, stdout, stderr) => {
|
|
146
|
+
let exitCode = 0;
|
|
147
|
+
if (err) {
|
|
148
|
+
// err.code can be number (exit code), string (error code like ETIMEDOUT), or null (signal kill)
|
|
149
|
+
if (typeof err.code === "number") {
|
|
150
|
+
exitCode = err.code;
|
|
151
|
+
}
|
|
152
|
+
else if (err.signal) {
|
|
153
|
+
// Process killed by signal — report as non-zero
|
|
154
|
+
exitCode = 128 + ({ SIGHUP: 1, SIGINT: 2, SIGQUIT: 3, SIGTERM: 15, SIGKILL: 9 }[err.signal] ?? 1);
|
|
155
|
+
}
|
|
156
|
+
else {
|
|
157
|
+
exitCode = 1;
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
resolve({
|
|
161
|
+
stdout: stdout ?? "",
|
|
162
|
+
stderr: stderr ?? "",
|
|
163
|
+
exitCode,
|
|
164
|
+
});
|
|
165
|
+
});
|
|
166
|
+
});
|
|
167
|
+
}
|
|
168
|
+
async execBackground(machineId, command, logPath) {
|
|
169
|
+
const log = logPath ?? `/tmp/helios-${Date.now()}.log`;
|
|
170
|
+
if (machineId === "local") {
|
|
171
|
+
return this.execBackgroundLocal(command, log);
|
|
172
|
+
}
|
|
173
|
+
// Remote: use nohup + & over SSH (SSH channels close cleanly)
|
|
174
|
+
// PYTHONUNBUFFERED ensures Python flushes stdout immediately for live metric capture
|
|
175
|
+
const wrappedCmd = `PYTHONUNBUFFERED=1 nohup ${command} > ${log} 2>&1 & echo $!`;
|
|
176
|
+
const result = await this.exec(machineId, wrappedCmd);
|
|
177
|
+
const pid = parseInt(result.stdout.trim().split("\n").pop() ?? "", 10);
|
|
178
|
+
if (isNaN(pid)) {
|
|
179
|
+
throw new Error(`Failed to get PID. stdout: ${result.stdout}, stderr: ${result.stderr}`);
|
|
180
|
+
}
|
|
181
|
+
return { pid, logPath: log };
|
|
182
|
+
}
|
|
183
|
+
execBackgroundLocal(command, logPath) {
|
|
184
|
+
return new Promise((resolve, reject) => {
|
|
185
|
+
let logFd = null;
|
|
186
|
+
try {
|
|
187
|
+
// Open the log file for writing
|
|
188
|
+
logFd = openSync(logPath, "w");
|
|
189
|
+
// Spawn detached with stdio redirected to log file
|
|
190
|
+
// PYTHONUNBUFFERED ensures Python flushes stdout immediately for live metric capture
|
|
191
|
+
const child = spawn("/bin/bash", ["-c", command], {
|
|
192
|
+
detached: true,
|
|
193
|
+
stdio: ["ignore", logFd, logFd],
|
|
194
|
+
env: { ...process.env, PYTHONUNBUFFERED: "1" },
|
|
195
|
+
});
|
|
196
|
+
const pid = child.pid;
|
|
197
|
+
if (pid === undefined) {
|
|
198
|
+
closeSync(logFd);
|
|
199
|
+
reject(new Error("Failed to spawn background process"));
|
|
200
|
+
return;
|
|
201
|
+
}
|
|
202
|
+
// Fully detach — don't let Node wait for this child
|
|
203
|
+
child.unref();
|
|
204
|
+
closeSync(logFd);
|
|
205
|
+
resolve({ pid, logPath });
|
|
206
|
+
}
|
|
207
|
+
catch (err) {
|
|
208
|
+
if (logFd !== null) {
|
|
209
|
+
try {
|
|
210
|
+
closeSync(logFd);
|
|
211
|
+
}
|
|
212
|
+
catch { /* best effort */ }
|
|
213
|
+
}
|
|
214
|
+
reject(err);
|
|
215
|
+
}
|
|
216
|
+
});
|
|
217
|
+
}
|
|
218
|
+
async isProcessRunning(machineId, pid) {
|
|
219
|
+
const result = await this.exec(machineId, `kill -0 ${pid} 2>/dev/null && echo "running" || echo "stopped"`);
|
|
220
|
+
return result.stdout.trim() === "running";
|
|
221
|
+
}
|
|
222
|
+
async tailFile(machineId, path, lines = 50) {
|
|
223
|
+
const result = await this.exec(machineId, `tail -n ${lines} '${path.replace(/'/g, "'\\''")}'`);
|
|
224
|
+
return result.stdout;
|
|
225
|
+
}
|
|
226
|
+
getStatus(machineId) {
|
|
227
|
+
const conn = this.connections.get(machineId);
|
|
228
|
+
return {
|
|
229
|
+
machineId,
|
|
230
|
+
connected: conn?.connected ?? false,
|
|
231
|
+
lastConnectedAt: conn?.lastUsedAt,
|
|
232
|
+
error: conn?.lastError,
|
|
233
|
+
};
|
|
234
|
+
}
|
|
235
|
+
getAllStatuses() {
|
|
236
|
+
return Array.from(this.machines.keys()).map((id) => this.getStatus(id));
|
|
237
|
+
}
|
|
238
|
+
getMachineIds() {
|
|
239
|
+
return Array.from(this.machines.keys());
|
|
240
|
+
}
|
|
241
|
+
disconnect(machineId) {
|
|
242
|
+
if (machineId === "local")
|
|
243
|
+
return;
|
|
244
|
+
const conn = this.connections.get(machineId);
|
|
245
|
+
if (conn) {
|
|
246
|
+
conn.client?.end();
|
|
247
|
+
this.connections.delete(machineId);
|
|
248
|
+
}
|
|
249
|
+
}
|
|
250
|
+
disconnectAll() {
|
|
251
|
+
for (const [id] of this.connections) {
|
|
252
|
+
if (id !== "local")
|
|
253
|
+
this.disconnect(id);
|
|
254
|
+
}
|
|
255
|
+
}
|
|
256
|
+
async getConnection(machineId) {
|
|
257
|
+
let conn = this.connections.get(machineId);
|
|
258
|
+
if (!conn?.connected) {
|
|
259
|
+
await this.connect(machineId);
|
|
260
|
+
conn = this.connections.get(machineId);
|
|
261
|
+
}
|
|
262
|
+
if (!conn?.connected) {
|
|
263
|
+
throw new Error(`Cannot connect to ${machineId}`);
|
|
264
|
+
}
|
|
265
|
+
return conn;
|
|
266
|
+
}
|
|
267
|
+
}
|
|
268
|
+
//# sourceMappingURL=connection-pool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"connection-pool.js","sourceRoot":"","sources":["../../src/remote/connection-pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,IAAI,IAAI,MAAM,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAO5D,MAAM,CAAC,MAAM,aAAa,GAAkB;IAC1C,EAAE,EAAE,OAAO;IACX,IAAI,EAAE,WAAW;IACjB,IAAI,EAAE,CAAC;IACP,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,OAAO;IACrC,UAAU,EAAE,OAAc;IAC1B,MAAM,EAAE,CAAC,OAAO,CAAC;CAClB,CAAC;AAWF,MAAM,OAAO,cAAc;IACjB,WAAW,GAAG,IAAI,GAAG,EAA4B,CAAC;IAClD,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;IAEpD;QACE,oCAAoC;QACpC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAC1C,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE;YAC5B,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,aAAa;YACtB,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;YACtB,iBAAiB,EAAE,CAAC;SACrB,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAAsB;QAC/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,aAAa,CAAC,EAAU;QACtB,IAAI,EAAE,KAAK,OAAO;YAAE,OAAO,CAAC,qBAAqB;QACjD,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB;QAC7B,IAAI,SAAS,KAAK,OAAO;YAAE,OAAO,CAAC,mBAAmB;QAEtD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAC;QAE/D,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,QAAQ,EAAE,SAAS;YAAE,OAAO;QAEhC,gEAAgE;QAChE,IAAI,QAAQ,EAAE,MAAM,EAAE,CAAC;YACrB,IAAI,CAAC;gBAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAE5B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE;oBAC9B,MAAM;oBACN,OAAO;oBACP,SAAS,EAAE,IAAI;oBACf,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;oBACtB,iBAAiB,EAAE,CAAC;iBACrB,CAAC,CAAC;gBACH,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzB,MAAM,MAAM,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAChE,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC7C,IAAI,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;oBACvB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC;gBAC1B,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE;wBAC9B,MAAM;wBACN,OAAO;wBACP,SAAS,EAAE,KAAK;wBAChB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;wBACtB,iBAAiB,EAAE,CAAC;wBACpB,SAAS,EAAE,MAAM;qBAClB,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACtB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC7C,IAAI,IAAI;oBAAE,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACnC,CAAC,CAAC,CAAC;YAEH,MAAM,aAAa,GAA4B;gBAC7C,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,iBAAiB,EAAE,MAAM;gBACzB,iBAAiB,EAAE,CAAC;gBACpB,YAAY,EAAE,MAAM;aACrB,CAAC;YAEF,IAAI,OAAO,CAAC,UAAU,KAAK,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpD,IAAI,CAAC;oBACH,aAAa,CAAC,UAAU,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC3D,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,0BAA0B,OAAO,CAAC,OAAO,KAAK,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC7H,CAAC;YACH,CAAC;iBAAM,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBAC1C,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;gBACvC,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,iFAAiF,CAAC,CAAC,CAAC;gBAC9G,CAAC;gBACD,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC;YAC7B,CAAC;iBAAM,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACjE,aAAa,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;YAC5C,CAAC;YAED,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,OAAe;QAC3C,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjC,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,MAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE;gBACzC,IAAI,GAAG;oBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;gBAE5B,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,IAAI,MAAM,GAAG,EAAE,CAAC;gBAEhB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;oBACjC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;oBACxC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC5B,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,SAAgB,EAAE,EAAE;oBACtC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACpB,CAAC,CAAC,CAAC;gBACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAY,EAAE,EAAE;oBAClC,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC;gBACnD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,SAAS,CAAC,OAAe,EAAE,SAAS,GAAG,OAAO;QACpD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;gBAC/G,IAAI,QAAQ,GAAG,CAAC,CAAC;gBACjB,IAAI,GAAG,EAAE,CAAC;oBACR,gGAAgG;oBAChG,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;wBACjC,QAAQ,GAAG,GAAG,CAAC,IAAI,CAAC;oBACtB,CAAC;yBAAM,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;wBACtB,gDAAgD;wBAChD,QAAQ,GAAG,GAAG,GAAG,CAAE,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,EAA6B,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;oBAChI,CAAC;yBAAM,CAAC;wBACN,QAAQ,GAAG,CAAC,CAAC;oBACf,CAAC;gBACH,CAAC;gBACD,OAAO,CAAC;oBACN,MAAM,EAAE,MAAM,IAAI,EAAE;oBACpB,MAAM,EAAE,MAAM,IAAI,EAAE;oBACpB,QAAQ;iBACT,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,SAAiB,EACjB,OAAe,EACf,OAAgB;QAEhB,MAAM,GAAG,GAAG,OAAO,IAAI,eAAe,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;QAEvD,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAChD,CAAC;QAED,8DAA8D;QAC9D,qFAAqF;QACrF,MAAM,UAAU,GAAG,4BAA4B,OAAO,MAAM,GAAG,iBAAiB,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QACtD,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,8BAA8B,MAAM,CAAC,MAAM,aAAa,MAAM,CAAC,MAAM,EAAE,CACxE,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;IAC/B,CAAC;IAEO,mBAAmB,CACzB,OAAe,EACf,OAAe;QAEf,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,KAAK,GAAkB,IAAI,CAAC;YAChC,IAAI,CAAC;gBACH,gCAAgC;gBAChC,KAAK,GAAG,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;gBAE/B,mDAAmD;gBACnD,qFAAqF;gBACrF,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE;oBAChD,QAAQ,EAAE,IAAI;oBACd,KAAK,EAAE,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC;oBAC/B,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,gBAAgB,EAAE,GAAG,EAAE;iBAC/C,CAAC,CAAC;gBAEH,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;gBACtB,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,SAAS,CAAC,KAAK,CAAC,CAAC;oBACjB,MAAM,CAAC,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC,CAAC;oBACxD,OAAO;gBACT,CAAC;gBAED,oDAAoD;gBACpD,KAAK,CAAC,KAAK,EAAE,CAAC;gBACd,SAAS,CAAC,KAAK,CAAC,CAAC;gBAEjB,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;YAC5B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;oBACnB,IAAI,CAAC;wBAAC,SAAS,CAAC,KAAK,CAAC,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,iBAAiB,CAAC,CAAC;gBACvD,CAAC;gBACD,MAAM,CAAC,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,SAAiB,EACjB,GAAW;QAEX,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAC5B,SAAS,EACT,WAAW,GAAG,kDAAkD,CACjE,CAAC;QACF,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,SAAS,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,SAAiB,EACjB,IAAY,EACZ,KAAK,GAAG,EAAE;QAEV,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,WAAW,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QAC/F,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,SAAS,CAAC,SAAiB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,OAAO;YACL,SAAS;YACT,SAAS,EAAE,IAAI,EAAE,SAAS,IAAI,KAAK;YACnC,eAAe,EAAE,IAAI,EAAE,UAAU;YACjC,KAAK,EAAE,IAAI,EAAE,SAAS;SACvB,CAAC;IACJ,CAAC;IAED,cAAc;QACZ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CACjD,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CACnB,CAAC;IACJ,CAAC;IAED,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,IAAI,SAAS,KAAK,OAAO;YAAE,OAAO;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,aAAa;QACX,KAAK,MAAM,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACpC,IAAI,EAAE,KAAK,OAAO;gBAAE,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,SAAiB;QAEjB,IAAI,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,qBAAqB,SAAS,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { ConnectionPool } from "./connection-pool.js";
|
|
2
|
+
import type { ExecResult, BackgroundProcess } from "./types.js";
|
|
3
|
+
/**
|
|
4
|
+
* High-level remote execution interface.
|
|
5
|
+
* Wraps ConnectionPool with task tracking and convenience methods.
|
|
6
|
+
*/
|
|
7
|
+
export declare class RemoteExecutor {
|
|
8
|
+
private pool;
|
|
9
|
+
private backgroundProcesses;
|
|
10
|
+
constructor(pool: ConnectionPool);
|
|
11
|
+
exec(machineId: string, command: string, timeoutMs?: number): Promise<ExecResult>;
|
|
12
|
+
execBackground(machineId: string, command: string, logPath?: string, opts?: {
|
|
13
|
+
metricNames?: string[];
|
|
14
|
+
metricPatterns?: Record<string, string>;
|
|
15
|
+
}): Promise<BackgroundProcess>;
|
|
16
|
+
isRunning(machineId: string, pid: number): Promise<boolean>;
|
|
17
|
+
tail(machineId: string, path: string, lines?: number): Promise<string>;
|
|
18
|
+
gpuStatus(machineId: string): Promise<string>;
|
|
19
|
+
getBackgroundProcesses(): BackgroundProcess[];
|
|
20
|
+
getBackgroundProcess(machineId: string, pid: number): BackgroundProcess | undefined;
|
|
21
|
+
removeBackgroundProcess(key: string): void;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../src/remote/executor.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,KAAK,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEhE;;;GAGG;AACH,qBAAa,cAAc;IAGb,OAAO,CAAC,IAAI;IAFxB,OAAO,CAAC,mBAAmB,CAAwC;gBAE/C,IAAI,EAAE,cAAc;IAElC,IAAI,CACR,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,UAAU,CAAC;IAQhB,cAAc,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,GACzE,OAAO,CAAC,iBAAiB,CAAC;IAsBvB,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAI3D,IAAI,CACR,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,EACZ,KAAK,SAAK,GACT,OAAO,CAAC,MAAM,CAAC;IAIZ,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAQnD,sBAAsB,IAAI,iBAAiB,EAAE;IAI7C,oBAAoB,CAClB,SAAS,EAAE,MAAM,EACjB,GAAG,EAAE,MAAM,GACV,iBAAiB,GAAG,SAAS;IAIhC,uBAAuB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;CAG3C"}
|