@myagentroam/node 0.1.2 → 0.1.4
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/claude-agent-sdk.d.ts +1 -15
- package/dist/claude-agent-sdk.js +9 -77
- package/dist/claude-agent-sdk.js.map +1 -1
- package/dist/codex-app-server.d.ts +4 -0
- package/dist/codex-app-server.js +18 -4
- package/dist/codex-app-server.js.map +1 -1
- package/dist/connector/node-connector-options.d.ts +23 -0
- package/dist/connector/node-connector-options.js +2 -0
- package/dist/connector/node-connector-options.js.map +1 -0
- package/dist/connector/node-operation-router.d.ts +9 -0
- package/dist/connector/node-operation-router.js +23 -0
- package/dist/connector/node-operation-router.js.map +1 -0
- package/dist/connector.d.ts +59 -353
- package/dist/connector.js +614 -5936
- package/dist/connector.js.map +1 -1
- package/dist/control-outbound-scheduler.d.ts +26 -0
- package/dist/control-outbound-scheduler.js +122 -0
- package/dist/control-outbound-scheduler.js.map +1 -0
- package/dist/database.js +1 -1
- package/dist/database.js.map +1 -1
- package/dist/native-jsonl-reader.d.ts +14 -0
- package/dist/native-jsonl-reader.js +98 -0
- package/dist/native-jsonl-reader.js.map +1 -0
- package/dist/native-session-history.d.ts +22 -3
- package/dist/native-session-history.js +281 -157
- package/dist/native-session-history.js.map +1 -1
- package/dist/runner/abstract-runner.d.ts +148 -0
- package/dist/runner/abstract-runner.js +139 -0
- package/dist/runner/abstract-runner.js.map +1 -0
- package/dist/runner/claude/managed-run-controller.d.ts +52 -0
- package/dist/runner/claude/managed-run-controller.js +307 -0
- package/dist/runner/claude/managed-run-controller.js.map +1 -0
- package/dist/runner/claude-code-runner.d.ts +68 -0
- package/dist/runner/claude-code-runner.js +268 -0
- package/dist/runner/claude-code-runner.js.map +1 -0
- package/dist/runner/codex/conversation-parser.d.ts +61 -0
- package/dist/runner/codex/conversation-parser.js +676 -0
- package/dist/runner/codex/conversation-parser.js.map +1 -0
- package/dist/runner/codex/managed-run-controller.d.ts +79 -0
- package/dist/runner/codex/managed-run-controller.js +379 -0
- package/dist/runner/codex/managed-run-controller.js.map +1 -0
- package/dist/runner/codex-runner.d.ts +81 -0
- package/dist/runner/codex-runner.js +342 -0
- package/dist/runner/codex-runner.js.map +1 -0
- package/dist/runner/fake-test-runner.d.ts +26 -0
- package/dist/runner/fake-test-runner.js +48 -0
- package/dist/runner/fake-test-runner.js.map +1 -0
- package/dist/runner/runner-registry.d.ts +17 -0
- package/dist/runner/runner-registry.js +66 -0
- package/dist/runner/runner-registry.js.map +1 -0
- package/dist/runner-command-engine.d.ts +9 -0
- package/dist/runner-command-engine.js +24 -0
- package/dist/runner-command-engine.js.map +1 -1
- package/dist/runner-profiles.js +12 -0
- package/dist/runner-profiles.js.map +1 -1
- package/dist/runner-usage.d.ts +2 -2
- package/dist/runner-usage.js +3 -5
- package/dist/runner-usage.js.map +1 -1
- package/dist/runtime-state.js +6 -5
- package/dist/runtime-state.js.map +1 -1
- package/dist/service/attachment-domain-state.d.ts +5 -0
- package/dist/service/attachment-domain-state.js +6 -0
- package/dist/service/attachment-domain-state.js.map +1 -0
- package/dist/service/conversation-history-service.d.ts +34 -0
- package/dist/service/conversation-history-service.js +159 -0
- package/dist/service/conversation-history-service.js.map +1 -0
- package/dist/service/external-session-resume-service.d.ts +32 -0
- package/dist/service/external-session-resume-service.js +99 -0
- package/dist/service/external-session-resume-service.js.map +1 -0
- package/dist/service/fake-managed-run-service.d.ts +29 -0
- package/dist/service/fake-managed-run-service.js +63 -0
- package/dist/service/fake-managed-run-service.js.map +1 -0
- package/dist/service/managed-runner-session-service.d.ts +10 -0
- package/dist/service/managed-runner-session-service.js +31 -0
- package/dist/service/managed-runner-session-service.js.map +1 -0
- package/dist/service/native-session-projection-service.d.ts +50 -0
- package/dist/service/native-session-projection-service.js +268 -0
- package/dist/service/native-session-projection-service.js.map +1 -0
- package/dist/service/native-session-watch-service.d.ts +37 -0
- package/dist/service/native-session-watch-service.js +151 -0
- package/dist/service/native-session-watch-service.js.map +1 -0
- package/dist/service/node-connection-lifecycle-service.d.ts +37 -0
- package/dist/service/node-connection-lifecycle-service.js +84 -0
- package/dist/service/node-connection-lifecycle-service.js.map +1 -0
- package/dist/service/node-control-channel.d.ts +32 -0
- package/dist/service/node-control-channel.js +84 -0
- package/dist/service/node-control-channel.js.map +1 -0
- package/dist/service/node-control-message-service.d.ts +28 -0
- package/dist/service/node-control-message-service.js +105 -0
- package/dist/service/node-control-message-service.js.map +1 -0
- package/dist/service/node-request-service.d.ts +14 -0
- package/dist/service/node-request-service.js +51 -0
- package/dist/service/node-request-service.js.map +1 -0
- package/dist/service/node-run-event-bridge.d.ts +38 -0
- package/dist/service/node-run-event-bridge.js +80 -0
- package/dist/service/node-run-event-bridge.js.map +1 -0
- package/dist/service/node-terminal-channel.d.ts +29 -0
- package/dist/service/node-terminal-channel.js +176 -0
- package/dist/service/node-terminal-channel.js.map +1 -0
- package/dist/service/node-workspace-coordinator.d.ts +32 -0
- package/dist/service/node-workspace-coordinator.js +81 -0
- package/dist/service/node-workspace-coordinator.js.map +1 -0
- package/dist/service/run-attachment-service.d.ts +37 -0
- package/dist/service/run-attachment-service.js +161 -0
- package/dist/service/run-attachment-service.js.map +1 -0
- package/dist/service/run-domain-state.d.ts +4 -0
- package/dist/service/run-domain-state.js +5 -0
- package/dist/service/run-domain-state.js.map +1 -0
- package/dist/service/run-event-service.d.ts +31 -0
- package/dist/service/run-event-service.js +103 -0
- package/dist/service/run-event-service.js.map +1 -0
- package/dist/service/run-workbench-service.d.ts +27 -0
- package/dist/service/run-workbench-service.js +114 -0
- package/dist/service/run-workbench-service.js.map +1 -0
- package/dist/service/runner-channel-message-service.d.ts +13 -0
- package/dist/service/runner-channel-message-service.js +21 -0
- package/dist/service/runner-channel-message-service.js.map +1 -0
- package/dist/service/runner-interaction-service.d.ts +40 -0
- package/dist/service/runner-interaction-service.js +245 -0
- package/dist/service/runner-interaction-service.js.map +1 -0
- package/dist/service/runner-service.d.ts +43 -0
- package/dist/service/runner-service.js +55 -0
- package/dist/service/runner-service.js.map +1 -0
- package/dist/service/session-catalog-service.d.ts +30 -0
- package/dist/service/session-catalog-service.js +145 -0
- package/dist/service/session-catalog-service.js.map +1 -0
- package/dist/service/session-command-service.d.ts +23 -0
- package/dist/service/session-command-service.js +70 -0
- package/dist/service/session-command-service.js.map +1 -0
- package/dist/service/session-context-service.d.ts +34 -0
- package/dist/service/session-context-service.js +88 -0
- package/dist/service/session-context-service.js.map +1 -0
- package/dist/service/session-domain-state.d.ts +8 -0
- package/dist/service/session-domain-state.js +9 -0
- package/dist/service/session-domain-state.js.map +1 -0
- package/dist/service/session-identity-service.d.ts +24 -0
- package/dist/service/session-identity-service.js +87 -0
- package/dist/service/session-identity-service.js.map +1 -0
- package/dist/service/session-lifecycle-service.d.ts +31 -0
- package/dist/service/session-lifecycle-service.js +236 -0
- package/dist/service/session-lifecycle-service.js.map +1 -0
- package/dist/service/session-message-service.d.ts +33 -0
- package/dist/service/session-message-service.js +158 -0
- package/dist/service/session-message-service.js.map +1 -0
- package/dist/service/session-presentation-service.d.ts +33 -0
- package/dist/service/session-presentation-service.js +37 -0
- package/dist/service/session-presentation-service.js.map +1 -0
- package/dist/service/session-query-service.d.ts +31 -0
- package/dist/service/session-query-service.js +78 -0
- package/dist/service/session-query-service.js.map +1 -0
- package/dist/service/terminal-relay-state.d.ts +6 -0
- package/dist/service/terminal-relay-state.js +4 -0
- package/dist/service/terminal-relay-state.js.map +1 -0
- package/dist/service/terminal-service.d.ts +12 -0
- package/dist/service/terminal-service.js +53 -0
- package/dist/service/terminal-service.js.map +1 -0
- package/dist/service/workbench-event-service.d.ts +21 -0
- package/dist/service/workbench-event-service.js +54 -0
- package/dist/service/workbench-event-service.js.map +1 -0
- package/dist/service/workbench-manifest-service.d.ts +88 -0
- package/dist/service/workbench-manifest-service.js +69 -0
- package/dist/service/workbench-manifest-service.js.map +1 -0
- package/dist/service/workspace-change-service.d.ts +9 -0
- package/dist/service/workspace-change-service.js +86 -0
- package/dist/service/workspace-change-service.js.map +1 -0
- package/dist/service/workspace-domain-state.d.ts +75 -0
- package/dist/service/workspace-domain-state.js +41 -0
- package/dist/service/workspace-domain-state.js.map +1 -0
- package/dist/service/workspace-file-service.d.ts +14 -0
- package/dist/service/workspace-file-service.js +77 -0
- package/dist/service/workspace-file-service.js.map +1 -0
- package/dist/service/workspace-queue-workbench-service.d.ts +55 -0
- package/dist/service/workspace-queue-workbench-service.js +227 -0
- package/dist/service/workspace-queue-workbench-service.js.map +1 -0
- package/dist/service/workspace-service.d.ts +15 -0
- package/dist/service/workspace-service.js +54 -0
- package/dist/service/workspace-service.js.map +1 -0
- package/dist/service/workspace-session-service.d.ts +23 -0
- package/dist/service/workspace-session-service.js +140 -0
- package/dist/service/workspace-session-service.js.map +1 -0
- package/dist/service/workspace-upload-service.d.ts +37 -0
- package/dist/service/workspace-upload-service.js +241 -0
- package/dist/service/workspace-upload-service.js.map +1 -0
- package/dist/service/workspace-watch-service.d.ts +18 -0
- package/dist/service/workspace-watch-service.js +121 -0
- package/dist/service/workspace-watch-service.js.map +1 -0
- package/dist/service/workspace-workbench-service.d.ts +38 -0
- package/dist/service/workspace-workbench-service.js +156 -0
- package/dist/service/workspace-workbench-service.js.map +1 -0
- package/dist/terminal.d.ts +1 -0
- package/dist/terminal.js +1 -1
- package/dist/terminal.js.map +1 -1
- package/dist/util/node-operation-parsers.d.ts +91 -0
- package/dist/util/node-operation-parsers.js +380 -0
- package/dist/util/node-operation-parsers.js.map +1 -0
- package/dist/util/runner-native-session-parsers.d.ts +72 -0
- package/dist/util/runner-native-session-parsers.js +337 -0
- package/dist/util/runner-native-session-parsers.js.map +1 -0
- package/dist/util/secret-environment.d.ts +1 -0
- package/dist/util/secret-environment.js +26 -0
- package/dist/util/secret-environment.js.map +1 -0
- package/dist/workspace.d.ts +9 -2
- package/dist/workspace.js +41 -0
- package/dist/workspace.js.map +1 -1
- package/package.json +2 -2
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
import { isPlainRecord } from '../runner/codex/conversation-parser.js';
|
|
2
|
+
const INTERVAL_MS = 2_000;
|
|
3
|
+
const TTL_MS = 45_000;
|
|
4
|
+
const MAX_FAILURES = 3;
|
|
5
|
+
const INITIAL_TURN_LIMIT = 30;
|
|
6
|
+
export class NativeSessionWatchService {
|
|
7
|
+
options;
|
|
8
|
+
watches = new Map();
|
|
9
|
+
constructor(options) {
|
|
10
|
+
this.options = options;
|
|
11
|
+
}
|
|
12
|
+
operations() {
|
|
13
|
+
return {
|
|
14
|
+
'session.watch': (data) => this.executeWatch(isPlainRecord(data) ? data : {})
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
async executeWatch(input) {
|
|
18
|
+
if (typeof input.sessionId !== 'string' || (input.mode !== 'initial' && input.mode !== 'renew'))
|
|
19
|
+
throw new Error('SESSION_INVALID');
|
|
20
|
+
const session = await this.options.resolve(input.sessionId);
|
|
21
|
+
if (session === undefined || session.externalSessionId === null)
|
|
22
|
+
throw new Error('SESSION_NOT_FOUND');
|
|
23
|
+
if (input.mode === 'renew') {
|
|
24
|
+
const watch = this.renew(session);
|
|
25
|
+
if (watch === undefined)
|
|
26
|
+
throw new Error('SESSION_WATCH_NOT_FOUND');
|
|
27
|
+
return { renewed: true };
|
|
28
|
+
}
|
|
29
|
+
const watch = this.watch(session);
|
|
30
|
+
const snapshot = watch === undefined ? undefined : await this.refresh(session.id, INITIAL_TURN_LIMIT);
|
|
31
|
+
if (watch !== undefined && snapshot === undefined)
|
|
32
|
+
throw new Error('NATIVE_TRANSCRIPT_UNAVAILABLE');
|
|
33
|
+
return {
|
|
34
|
+
session: this.options.present(session),
|
|
35
|
+
watch: watch === undefined ? null : { expiresAt: watch.expiresAt },
|
|
36
|
+
...(snapshot === undefined ? {} : { snapshot })
|
|
37
|
+
};
|
|
38
|
+
}
|
|
39
|
+
watch(session) {
|
|
40
|
+
if (this.options.managedActive(session)) {
|
|
41
|
+
this.stop(session.id);
|
|
42
|
+
return undefined;
|
|
43
|
+
}
|
|
44
|
+
const existing = this.watches.get(session.id);
|
|
45
|
+
if (existing !== undefined) {
|
|
46
|
+
existing.expiresAt = Date.now() + TTL_MS;
|
|
47
|
+
return { expiresAt: existing.expiresAt };
|
|
48
|
+
}
|
|
49
|
+
const watch = {
|
|
50
|
+
expiresAt: Date.now() + TTL_MS,
|
|
51
|
+
refresh: undefined,
|
|
52
|
+
sessionSignature: undefined,
|
|
53
|
+
signature: undefined,
|
|
54
|
+
failures: 0,
|
|
55
|
+
timer: undefined
|
|
56
|
+
};
|
|
57
|
+
watch.timer = setInterval(() => void this.refresh(session.id), INTERVAL_MS);
|
|
58
|
+
watch.timer.unref();
|
|
59
|
+
this.watches.set(session.id, watch);
|
|
60
|
+
return { expiresAt: watch.expiresAt };
|
|
61
|
+
}
|
|
62
|
+
renew(session) {
|
|
63
|
+
if (this.options.managedActive(session)) {
|
|
64
|
+
this.stop(session.id);
|
|
65
|
+
return undefined;
|
|
66
|
+
}
|
|
67
|
+
const watch = this.watches.get(session.id);
|
|
68
|
+
if (watch === undefined || watch.expiresAt <= Date.now()) {
|
|
69
|
+
this.stop(session.id);
|
|
70
|
+
return undefined;
|
|
71
|
+
}
|
|
72
|
+
watch.expiresAt = Date.now() + TTL_MS;
|
|
73
|
+
return { expiresAt: watch.expiresAt };
|
|
74
|
+
}
|
|
75
|
+
stop(sessionId) {
|
|
76
|
+
const watch = this.watches.get(sessionId);
|
|
77
|
+
if (watch === undefined)
|
|
78
|
+
return;
|
|
79
|
+
clearInterval(watch.timer);
|
|
80
|
+
this.watches.delete(sessionId);
|
|
81
|
+
}
|
|
82
|
+
clear() {
|
|
83
|
+
for (const watch of this.watches.values())
|
|
84
|
+
clearInterval(watch.timer);
|
|
85
|
+
this.watches.clear();
|
|
86
|
+
}
|
|
87
|
+
has(sessionId) {
|
|
88
|
+
return this.watches.has(sessionId);
|
|
89
|
+
}
|
|
90
|
+
entry(sessionId) {
|
|
91
|
+
return this.watches.get(sessionId);
|
|
92
|
+
}
|
|
93
|
+
get size() {
|
|
94
|
+
return this.watches.size;
|
|
95
|
+
}
|
|
96
|
+
async refresh(sessionId, limit = 10) {
|
|
97
|
+
const watch = this.watches.get(sessionId);
|
|
98
|
+
if (watch === undefined)
|
|
99
|
+
return undefined;
|
|
100
|
+
if (watch.refresh !== undefined)
|
|
101
|
+
return watch.refresh;
|
|
102
|
+
if (watch.expiresAt <= Date.now()) {
|
|
103
|
+
this.stop(sessionId);
|
|
104
|
+
return undefined;
|
|
105
|
+
}
|
|
106
|
+
const refresh = this.performRefresh(sessionId, watch, limit);
|
|
107
|
+
watch.refresh = refresh;
|
|
108
|
+
try {
|
|
109
|
+
return await refresh;
|
|
110
|
+
}
|
|
111
|
+
finally {
|
|
112
|
+
if (this.watches.get(sessionId) === watch)
|
|
113
|
+
watch.refresh = undefined;
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
async performRefresh(sessionId, watch, limit) {
|
|
117
|
+
try {
|
|
118
|
+
const session = await this.options.resolve(sessionId);
|
|
119
|
+
if (session === undefined ||
|
|
120
|
+
session.externalSessionId === null ||
|
|
121
|
+
this.options.managedActive(session)) {
|
|
122
|
+
this.stop(sessionId);
|
|
123
|
+
return undefined;
|
|
124
|
+
}
|
|
125
|
+
await this.options.refreshActivity(session);
|
|
126
|
+
const presented = this.options.present(session);
|
|
127
|
+
const sessionSignature = JSON.stringify(presented);
|
|
128
|
+
if (sessionSignature !== watch.sessionSignature) {
|
|
129
|
+
watch.sessionSignature = sessionSignature;
|
|
130
|
+
this.options.emitSession(presented);
|
|
131
|
+
}
|
|
132
|
+
const page = await this.options.readPage(session, limit);
|
|
133
|
+
watch.failures = 0;
|
|
134
|
+
const turns = page.turns.length <= 10 ? page.turns : page.turns.slice(0, 10);
|
|
135
|
+
const signature = JSON.stringify(turns);
|
|
136
|
+
if (signature === watch.signature)
|
|
137
|
+
return page;
|
|
138
|
+
watch.signature = signature;
|
|
139
|
+
for (const turn of turns)
|
|
140
|
+
this.options.emitTurn(turn);
|
|
141
|
+
return page;
|
|
142
|
+
}
|
|
143
|
+
catch {
|
|
144
|
+
watch.failures += 1;
|
|
145
|
+
if (watch.failures >= MAX_FAILURES)
|
|
146
|
+
this.stop(sessionId);
|
|
147
|
+
return undefined;
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
//# sourceMappingURL=native-session-watch-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"native-session-watch-service.js","sourceRoot":"","sources":["../../src/service/native-session-watch-service.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,wCAAwC,CAAC;AAEvE,MAAM,WAAW,GAAG,KAAK,CAAC;AAC1B,MAAM,MAAM,GAAG,MAAM,CAAC;AACtB,MAAM,YAAY,GAAG,CAAC,CAAC;AACvB,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAqB9B,MAAM,OAAO,yBAAyB;IAGP;IAFZ,OAAO,GAAG,IAAI,GAAG,EAAqC,CAAC;IAExE,YAA6B,OAAyC;QAAzC,YAAO,GAAP,OAAO,CAAkC;IAAG,CAAC;IAE1E,UAAU;QACR,OAAO;YACL,eAAe,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9E,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,KAA8B;QACvD,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC;YAC7F,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC5D,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,iBAAiB,KAAK,IAAI;YAC7D,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,KAAK,KAAK,SAAS;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YACpE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,QAAQ,GACZ,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,kBAAkB,CAAC,CAAC;QACvF,IAAI,KAAK,KAAK,SAAS,IAAI,QAAQ,KAAK,SAAS;YAC/C,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;YACtC,KAAK,EAAE,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE;YAClE,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;SAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAyB;QAC7B,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC;YACzC,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC;QAC3C,CAAC;QACD,MAAM,KAAK,GAA8B;YACvC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM;YAC9B,OAAO,EAAE,SAAS;YAClB,gBAAgB,EAAE,SAAS;YAC3B,SAAS,EAAE,SAAS;YACpB,QAAQ,EAAE,CAAC;YACX,KAAK,EAAE,SAAsC;SAC9C,CAAC;QACF,KAAK,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC;QAC5E,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACpC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,OAAyB;QAC7B,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YACzD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC;QACtC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC;IAED,IAAI,CAAC,SAAiB;QACpB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO;QAChC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAED,KAAK;QACH,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;YAAE,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,GAAG,CAAC,SAAiB;QACnB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB,EAAE,KAAK,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAC1C,IAAI,KAAK,CAAC,OAAO,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC,OAAO,CAAC;QACtD,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAClC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC7D,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,CAAC;QACvB,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,KAAK;gBAAE,KAAK,CAAC,OAAO,GAAG,SAAS,CAAC;QACvE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,SAAiB,EACjB,KAAgC,EAChC,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACtD,IACE,OAAO,KAAK,SAAS;gBACrB,OAAO,CAAC,iBAAiB,KAAK,IAAI;gBAClC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,EACnC,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACrB,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACnD,IAAI,gBAAgB,KAAK,KAAK,CAAC,gBAAgB,EAAE,CAAC;gBAChD,KAAK,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;gBAC1C,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YACtC,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACzD,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC;YACnB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7E,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,SAAS,KAAK,KAAK,CAAC,SAAS;gBAAE,OAAO,IAAI,CAAC;YAC/C,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;YAC5B,KAAK,MAAM,IAAI,IAAI,KAAK;gBAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC;YACpB,IAAI,KAAK,CAAC,QAAQ,IAAI,YAAY;gBAAE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACzD,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import type { NodeCapabilities } from '@myagentroam/protocol';
|
|
2
|
+
import { type NodeConfig } from '../config.js';
|
|
3
|
+
import type { NodeMetrics } from '../operational.js';
|
|
4
|
+
import type { NodeControlChannel } from './node-control-channel.js';
|
|
5
|
+
import type { NodeTerminalChannel } from './node-terminal-channel.js';
|
|
6
|
+
import type { WorkbenchEventService } from './workbench-event-service.js';
|
|
7
|
+
interface NodeConnectionLifecycleServiceOptions {
|
|
8
|
+
readonly config: () => NodeConfig | undefined;
|
|
9
|
+
readonly setConfig: (config: NodeConfig) => void;
|
|
10
|
+
readonly configPath: string | undefined;
|
|
11
|
+
readonly capabilities: () => NodeCapabilities;
|
|
12
|
+
readonly detectCapabilities: () => NodeCapabilities | Promise<NodeCapabilities>;
|
|
13
|
+
readonly capabilitiesProvided: boolean;
|
|
14
|
+
readonly stopped: () => boolean;
|
|
15
|
+
readonly setCapabilities: (capabilities: NodeCapabilities) => void;
|
|
16
|
+
readonly control: NodeControlChannel;
|
|
17
|
+
readonly terminal: NodeTerminalChannel;
|
|
18
|
+
readonly events: WorkbenchEventService;
|
|
19
|
+
readonly metrics: NodeMetrics;
|
|
20
|
+
readonly heartbeatMs: number;
|
|
21
|
+
readonly refreshMs: number;
|
|
22
|
+
}
|
|
23
|
+
/** Owns registration completion, heartbeat and background capability refresh timers. */
|
|
24
|
+
export declare class NodeConnectionLifecycleService {
|
|
25
|
+
private readonly options;
|
|
26
|
+
private heartbeat;
|
|
27
|
+
private refreshTimer;
|
|
28
|
+
private refreshing;
|
|
29
|
+
constructor(options: NodeConnectionLifecycleServiceOptions);
|
|
30
|
+
start(): void;
|
|
31
|
+
startCapabilityRefresh(): void;
|
|
32
|
+
stop(): void;
|
|
33
|
+
register(nodeId: string, credential: string): Promise<void>;
|
|
34
|
+
connected(): void;
|
|
35
|
+
refreshCapabilities(): Promise<void>;
|
|
36
|
+
}
|
|
37
|
+
export {};
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import { saveNodeConfig } from '../config.js';
|
|
2
|
+
/** Owns registration completion, heartbeat and background capability refresh timers. */
|
|
3
|
+
export class NodeConnectionLifecycleService {
|
|
4
|
+
options;
|
|
5
|
+
heartbeat;
|
|
6
|
+
refreshTimer;
|
|
7
|
+
refreshing = false;
|
|
8
|
+
constructor(options) {
|
|
9
|
+
this.options = options;
|
|
10
|
+
}
|
|
11
|
+
start() {
|
|
12
|
+
this.startCapabilityRefresh();
|
|
13
|
+
}
|
|
14
|
+
startCapabilityRefresh() {
|
|
15
|
+
if (this.refreshTimer !== undefined)
|
|
16
|
+
clearInterval(this.refreshTimer);
|
|
17
|
+
if (!this.options.capabilitiesProvided)
|
|
18
|
+
void this.refreshCapabilities();
|
|
19
|
+
this.refreshTimer = setInterval(() => void this.refreshCapabilities(), this.options.refreshMs);
|
|
20
|
+
}
|
|
21
|
+
stop() {
|
|
22
|
+
if (this.heartbeat !== undefined)
|
|
23
|
+
clearInterval(this.heartbeat);
|
|
24
|
+
if (this.refreshTimer !== undefined)
|
|
25
|
+
clearInterval(this.refreshTimer);
|
|
26
|
+
this.heartbeat = undefined;
|
|
27
|
+
this.refreshTimer = undefined;
|
|
28
|
+
}
|
|
29
|
+
async register(nodeId, credential) {
|
|
30
|
+
const config = this.options.config();
|
|
31
|
+
if (config === undefined)
|
|
32
|
+
return;
|
|
33
|
+
const registered = {
|
|
34
|
+
serverUrl: config.serverUrl,
|
|
35
|
+
allowedRoots: config.allowedRoots,
|
|
36
|
+
...(config.databasePath === undefined ? {} : { databasePath: config.databasePath }),
|
|
37
|
+
nodeId,
|
|
38
|
+
credential
|
|
39
|
+
};
|
|
40
|
+
this.options.setConfig(registered);
|
|
41
|
+
await saveNodeConfig(registered, this.options.configPath);
|
|
42
|
+
this.connected();
|
|
43
|
+
}
|
|
44
|
+
connected() {
|
|
45
|
+
if (this.heartbeat !== undefined)
|
|
46
|
+
clearInterval(this.heartbeat);
|
|
47
|
+
this.options.control.send('capabilities', this.options.capabilities());
|
|
48
|
+
this.options.terminal.connect();
|
|
49
|
+
this.options.control.send('heartbeat', {});
|
|
50
|
+
this.options.metrics.heartbeatSent();
|
|
51
|
+
this.options.events.replay();
|
|
52
|
+
this.heartbeat = setInterval(() => {
|
|
53
|
+
this.options.metrics.heartbeatSent();
|
|
54
|
+
this.options.control.send('heartbeat', {});
|
|
55
|
+
}, this.options.heartbeatMs);
|
|
56
|
+
}
|
|
57
|
+
async refreshCapabilities() {
|
|
58
|
+
if (this.options.stopped() || this.refreshing)
|
|
59
|
+
return;
|
|
60
|
+
this.refreshing = true;
|
|
61
|
+
try {
|
|
62
|
+
let next;
|
|
63
|
+
try {
|
|
64
|
+
next = await this.options.detectCapabilities();
|
|
65
|
+
}
|
|
66
|
+
catch {
|
|
67
|
+
next = {
|
|
68
|
+
...this.options.capabilities(),
|
|
69
|
+
codex: { available: false },
|
|
70
|
+
claudeCode: { available: false }
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
if (this.options.stopped() ||
|
|
74
|
+
JSON.stringify(next) === JSON.stringify(this.options.capabilities()))
|
|
75
|
+
return;
|
|
76
|
+
this.options.setCapabilities(next);
|
|
77
|
+
this.options.control.send('capabilities', next);
|
|
78
|
+
}
|
|
79
|
+
finally {
|
|
80
|
+
this.refreshing = false;
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
//# sourceMappingURL=node-connection-lifecycle-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-connection-lifecycle-service.js","sourceRoot":"","sources":["../../src/service/node-connection-lifecycle-service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAmB,MAAM,cAAc,CAAC;AAuB/D,wFAAwF;AACxF,MAAM,OAAO,8BAA8B;IAKZ;IAJrB,SAAS,CAA6B;IACtC,YAAY,CAA6B;IACzC,UAAU,GAAG,KAAK,CAAC;IAE3B,YAA6B,OAA8C;QAA9C,YAAO,GAAP,OAAO,CAAuC;IAAG,CAAC;IAE/E,KAAK;QACH,IAAI,CAAC,sBAAsB,EAAE,CAAC;IAChC,CAAC;IAED,sBAAsB;QACpB,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB;YAAE,KAAK,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACxE,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,mBAAmB,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACjG,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAc,EAAE,UAAkB;QAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACrC,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO;QACjC,MAAM,UAAU,GAAe;YAC7B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,GAAG,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC;YACnF,MAAM;YACN,UAAU;SACX,CAAC;QACF,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACnC,MAAM,cAAc,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,EAAE,CAAC;IACnB,CAAC;IAED,SAAS;QACP,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;YAChC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QACtD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,IAAsB,CAAC;YAC3B,IAAI,CAAC;gBACH,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACjD,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,GAAG;oBACL,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE;oBAC9B,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE;oBAC3B,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE;iBACjC,CAAC;YACJ,CAAC;YACD,IACE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;gBACtB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;gBAEpE,OAAO;YACT,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACnC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
import WebSocket from 'ws';
|
|
2
|
+
import { type NodeCapabilities, type NodeEnvelope } from '@myagentroam/protocol';
|
|
3
|
+
import type { NodeConfig } from '../config.js';
|
|
4
|
+
interface NodeControlChannelOptions {
|
|
5
|
+
readonly config: () => NodeConfig | undefined;
|
|
6
|
+
readonly capabilities: () => NodeCapabilities;
|
|
7
|
+
readonly stopped: () => boolean;
|
|
8
|
+
readonly reconnect: boolean;
|
|
9
|
+
readonly reconnectDelay: (attempt: number) => number;
|
|
10
|
+
readonly onConnecting: (attempt: number) => void;
|
|
11
|
+
readonly onReconnectScheduled: () => void;
|
|
12
|
+
readonly onMessage: (raw: string) => void;
|
|
13
|
+
readonly onClose: (code: number) => void;
|
|
14
|
+
}
|
|
15
|
+
export declare class NodeControlChannel {
|
|
16
|
+
private readonly options;
|
|
17
|
+
private socket;
|
|
18
|
+
private readonly outbound;
|
|
19
|
+
private reconnectTimer;
|
|
20
|
+
private attempts;
|
|
21
|
+
constructor(options: NodeControlChannelOptions);
|
|
22
|
+
connect(): void;
|
|
23
|
+
connected(): boolean;
|
|
24
|
+
/** Keeps transport injection at the transport boundary for contract tests. */
|
|
25
|
+
replaceSocketForTesting(socket: WebSocket | undefined): void;
|
|
26
|
+
close(code?: number, reason?: string): void;
|
|
27
|
+
stop(): void;
|
|
28
|
+
send(type: string, payload: unknown, replyTo?: string): void;
|
|
29
|
+
sendEnvelope(envelope: NodeEnvelope): void;
|
|
30
|
+
private scheduleReconnect;
|
|
31
|
+
}
|
|
32
|
+
export {};
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import WebSocket from 'ws';
|
|
2
|
+
import { createEnvelope } from '@myagentroam/protocol';
|
|
3
|
+
import { ControlOutboundScheduler } from '../control-outbound-scheduler.js';
|
|
4
|
+
import { nodeConnectEndpoint } from '../runner/codex/conversation-parser.js';
|
|
5
|
+
export class NodeControlChannel {
|
|
6
|
+
options;
|
|
7
|
+
socket;
|
|
8
|
+
outbound = new ControlOutboundScheduler(() => this.socket);
|
|
9
|
+
reconnectTimer;
|
|
10
|
+
attempts = 0;
|
|
11
|
+
constructor(options) {
|
|
12
|
+
this.options = options;
|
|
13
|
+
}
|
|
14
|
+
connect() {
|
|
15
|
+
const config = this.options.config();
|
|
16
|
+
if (config === undefined ||
|
|
17
|
+
this.options.stopped() ||
|
|
18
|
+
(this.socket !== undefined &&
|
|
19
|
+
(this.socket.readyState === WebSocket.OPEN ||
|
|
20
|
+
this.socket.readyState === WebSocket.CONNECTING)))
|
|
21
|
+
return;
|
|
22
|
+
this.options.onConnecting(this.attempts);
|
|
23
|
+
const socket = new WebSocket(nodeConnectEndpoint(config.serverUrl));
|
|
24
|
+
this.socket = socket;
|
|
25
|
+
socket.once('open', () => {
|
|
26
|
+
this.attempts = 0;
|
|
27
|
+
if (config.nodeId !== undefined && config.credential !== undefined)
|
|
28
|
+
this.send('authenticate', { nodeId: config.nodeId, credential: config.credential });
|
|
29
|
+
else if (config.registrationToken !== undefined)
|
|
30
|
+
this.send('register', {
|
|
31
|
+
registrationToken: config.registrationToken,
|
|
32
|
+
capabilities: this.options.capabilities()
|
|
33
|
+
});
|
|
34
|
+
else
|
|
35
|
+
socket.close(4000, 'NODE_CREDENTIAL_MISSING');
|
|
36
|
+
});
|
|
37
|
+
socket.on('message', (raw) => this.options.onMessage(raw.toString()));
|
|
38
|
+
socket.once('close', (code) => {
|
|
39
|
+
if (this.socket !== socket)
|
|
40
|
+
return;
|
|
41
|
+
this.socket = undefined;
|
|
42
|
+
this.outbound.reset();
|
|
43
|
+
this.options.onClose(code);
|
|
44
|
+
this.scheduleReconnect();
|
|
45
|
+
});
|
|
46
|
+
socket.once('error', () => undefined);
|
|
47
|
+
}
|
|
48
|
+
connected() {
|
|
49
|
+
return this.socket?.readyState === WebSocket.OPEN;
|
|
50
|
+
}
|
|
51
|
+
/** Keeps transport injection at the transport boundary for contract tests. */
|
|
52
|
+
replaceSocketForTesting(socket) {
|
|
53
|
+
this.socket = socket;
|
|
54
|
+
}
|
|
55
|
+
close(code, reason) {
|
|
56
|
+
this.socket?.close(code, reason);
|
|
57
|
+
}
|
|
58
|
+
stop() {
|
|
59
|
+
if (this.reconnectTimer !== undefined)
|
|
60
|
+
clearTimeout(this.reconnectTimer);
|
|
61
|
+
this.reconnectTimer = undefined;
|
|
62
|
+
this.outbound.reset();
|
|
63
|
+
this.socket?.close();
|
|
64
|
+
this.socket = undefined;
|
|
65
|
+
}
|
|
66
|
+
send(type, payload, replyTo) {
|
|
67
|
+
this.sendEnvelope(createEnvelope(type, payload, { ...(replyTo === undefined ? {} : { replyTo }) }));
|
|
68
|
+
}
|
|
69
|
+
sendEnvelope(envelope) {
|
|
70
|
+
if (!this.connected())
|
|
71
|
+
return;
|
|
72
|
+
this.outbound.enqueue(envelope, envelope.type === 'workbench.event' ? 'normal' : 'high');
|
|
73
|
+
}
|
|
74
|
+
scheduleReconnect() {
|
|
75
|
+
if (this.options.stopped() || !this.options.reconnect || this.reconnectTimer !== undefined)
|
|
76
|
+
return;
|
|
77
|
+
this.options.onReconnectScheduled();
|
|
78
|
+
this.reconnectTimer = setTimeout(() => {
|
|
79
|
+
this.reconnectTimer = undefined;
|
|
80
|
+
this.connect();
|
|
81
|
+
}, this.options.reconnectDelay(this.attempts++));
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
//# sourceMappingURL=node-control-channel.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-control-channel.js","sourceRoot":"","sources":["../../src/service/node-control-channel.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,cAAc,EAA4C,MAAM,uBAAuB,CAAC;AAEjG,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,MAAM,wCAAwC,CAAC;AAc7E,MAAM,OAAO,kBAAkB;IAMA;IALrB,MAAM,CAAwB;IACrB,QAAQ,GAAG,IAAI,wBAAwB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpE,cAAc,CAA6B;IAC3C,QAAQ,GAAG,CAAC,CAAC;IAErB,YAA6B,OAAkC;QAAlC,YAAO,GAAP,OAAO,CAA2B;IAAG,CAAC;IAEnE,OAAO;QACL,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACrC,IACE,MAAM,KAAK,SAAS;YACpB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YACtB,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS;gBACxB,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI;oBACxC,IAAI,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,UAAU,CAAC,CAAC;YAErD,OAAO;QACT,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;YACvB,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;YAClB,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS;gBAChE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;iBACjF,IAAI,MAAM,CAAC,iBAAiB,KAAK,SAAS;gBAC7C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;oBACpB,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;oBAC3C,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE;iBAC1C,CAAC,CAAC;;gBACA,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YAC5B,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM;gBAAE,OAAO;YACnC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,EAAE,UAAU,KAAK,SAAS,CAAC,IAAI,CAAC;IACpD,CAAC;IACD,8EAA8E;IAC9E,uBAAuB,CAAC,MAA6B;QACnD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IACD,KAAK,CAAC,IAAa,EAAE,MAAe;QAClC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACnC,CAAC;IACD,IAAI;QACF,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS;YAAE,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzE,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1B,CAAC;IAED,IAAI,CAAC,IAAY,EAAE,OAAgB,EAAE,OAAgB;QACnD,IAAI,CAAC,YAAY,CACf,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CACjF,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,QAAsB;QACjC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;YAAE,OAAO;QAC9B,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IAC3F,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS;YACxF,OAAO;QACT,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;QACpC,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAChC,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { type NodeEnvelope } from '@myagentroam/protocol';
|
|
2
|
+
import type { NodeConfig } from '../config.js';
|
|
3
|
+
import type { RunnerRegistry } from '../runner/runner-registry.js';
|
|
4
|
+
interface NodeControlMessageServiceOptions {
|
|
5
|
+
readonly config: () => NodeConfig | undefined;
|
|
6
|
+
readonly register: (nodeId: string, credential: string) => Promise<void>;
|
|
7
|
+
readonly connected: () => void;
|
|
8
|
+
readonly close: (code: number, reason: string) => void;
|
|
9
|
+
readonly inspectWorkspace: (envelope: NodeEnvelope) => Promise<void>;
|
|
10
|
+
readonly handleNodeRequest: (envelope: NodeEnvelope) => Promise<void>;
|
|
11
|
+
readonly acknowledgeWorkbenchEvent: (replyTo: string | undefined) => void;
|
|
12
|
+
readonly reconcileRuns: (envelope: NodeEnvelope) => void;
|
|
13
|
+
readonly runners: RunnerRegistry;
|
|
14
|
+
readonly deliverSessionChannelMessage: (envelope: NodeEnvelope) => void;
|
|
15
|
+
readonly completeUncontrolledCancellation: (runId: string) => void;
|
|
16
|
+
readonly send: (type: string, payload: unknown, replyTo?: string) => void;
|
|
17
|
+
}
|
|
18
|
+
/** Owns parsing and dispatch of the closed Node control-message protocol. */
|
|
19
|
+
export declare class NodeControlMessageService {
|
|
20
|
+
private readonly options;
|
|
21
|
+
constructor(options: NodeControlMessageServiceOptions);
|
|
22
|
+
handle(raw: string): void;
|
|
23
|
+
private dispatch;
|
|
24
|
+
private register;
|
|
25
|
+
private decideApproval;
|
|
26
|
+
private cancelRun;
|
|
27
|
+
}
|
|
28
|
+
export {};
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
import { nodeEnvelopeSchema } from '@myagentroam/protocol';
|
|
2
|
+
/** Owns parsing and dispatch of the closed Node control-message protocol. */
|
|
3
|
+
export class NodeControlMessageService {
|
|
4
|
+
options;
|
|
5
|
+
constructor(options) {
|
|
6
|
+
this.options = options;
|
|
7
|
+
}
|
|
8
|
+
handle(raw) {
|
|
9
|
+
let envelope;
|
|
10
|
+
try {
|
|
11
|
+
envelope = nodeEnvelopeSchema.parse(JSON.parse(raw));
|
|
12
|
+
}
|
|
13
|
+
catch {
|
|
14
|
+
this.options.close(4002, 'MESSAGE_INVALID');
|
|
15
|
+
return;
|
|
16
|
+
}
|
|
17
|
+
this.dispatch(envelope);
|
|
18
|
+
}
|
|
19
|
+
dispatch(envelope) {
|
|
20
|
+
if (envelope.type === 'registered') {
|
|
21
|
+
this.register(envelope);
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
if (envelope.type === 'authenticated') {
|
|
25
|
+
this.options.connected();
|
|
26
|
+
return;
|
|
27
|
+
}
|
|
28
|
+
if (envelope.type === 'workspace.inspect') {
|
|
29
|
+
void this.options.inspectWorkspace(envelope);
|
|
30
|
+
return;
|
|
31
|
+
}
|
|
32
|
+
if (envelope.type === 'node.request') {
|
|
33
|
+
void this.options.handleNodeRequest(envelope);
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
if (envelope.type === 'ack') {
|
|
37
|
+
const payload = envelope.payload;
|
|
38
|
+
if (payload.type === 'workbench.event')
|
|
39
|
+
this.options.acknowledgeWorkbenchEvent(envelope.replyTo);
|
|
40
|
+
return;
|
|
41
|
+
}
|
|
42
|
+
if (envelope.type === 'runs.reconcile') {
|
|
43
|
+
this.options.reconcileRuns(envelope);
|
|
44
|
+
return;
|
|
45
|
+
}
|
|
46
|
+
if (envelope.type === 'run.approval.decision') {
|
|
47
|
+
this.decideApproval(envelope);
|
|
48
|
+
return;
|
|
49
|
+
}
|
|
50
|
+
if (envelope.type === 'run.start') {
|
|
51
|
+
this.options.send('ack', { type: envelope.type }, envelope.id);
|
|
52
|
+
this.options.runners.startManagedRun(envelope);
|
|
53
|
+
return;
|
|
54
|
+
}
|
|
55
|
+
if (envelope.type === 'session.channel.message') {
|
|
56
|
+
this.options.deliverSessionChannelMessage(envelope);
|
|
57
|
+
return;
|
|
58
|
+
}
|
|
59
|
+
if (envelope.type === 'run.cancel' || envelope.type === 'run.interrupt') {
|
|
60
|
+
this.cancelRun(envelope);
|
|
61
|
+
return;
|
|
62
|
+
}
|
|
63
|
+
if (envelope.type === 'session.resume' || envelope.type === 'lease.reconcile') {
|
|
64
|
+
this.options.send('ack', { type: envelope.type }, envelope.id);
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
if (envelope.type !== 'error')
|
|
68
|
+
this.options.send('error', { code: 'MESSAGE_TYPE_UNKNOWN' }, envelope.id);
|
|
69
|
+
}
|
|
70
|
+
register(envelope) {
|
|
71
|
+
const payload = envelope.payload;
|
|
72
|
+
if (typeof payload.nodeId !== 'string' ||
|
|
73
|
+
typeof payload.credential !== 'string' ||
|
|
74
|
+
this.options.config() === undefined) {
|
|
75
|
+
this.options.close(4002, 'MESSAGE_INVALID');
|
|
76
|
+
return;
|
|
77
|
+
}
|
|
78
|
+
void this.options.register(payload.nodeId, payload.credential);
|
|
79
|
+
}
|
|
80
|
+
decideApproval(envelope) {
|
|
81
|
+
const payload = envelope.payload;
|
|
82
|
+
if (typeof payload.runId !== 'string' ||
|
|
83
|
+
typeof payload.approvalId !== 'string' ||
|
|
84
|
+
(payload.decision !== 'APPROVED' && payload.decision !== 'REJECTED')) {
|
|
85
|
+
this.options.send('error', { code: 'MESSAGE_INVALID' }, envelope.id);
|
|
86
|
+
return;
|
|
87
|
+
}
|
|
88
|
+
this.options.runners.decideApproval({
|
|
89
|
+
runId: payload.runId,
|
|
90
|
+
approvalId: payload.approvalId,
|
|
91
|
+
decision: payload.decision
|
|
92
|
+
});
|
|
93
|
+
this.options.send('ack', { type: envelope.type }, envelope.id);
|
|
94
|
+
}
|
|
95
|
+
cancelRun(envelope) {
|
|
96
|
+
const payload = envelope.payload;
|
|
97
|
+
if (typeof payload.runId === 'string') {
|
|
98
|
+
const handled = this.options.runners.cancelManagedRun(payload.runId, envelope.type === 'run.cancel' ? 'CANCEL' : 'INTERRUPT');
|
|
99
|
+
if (!handled)
|
|
100
|
+
this.options.completeUncontrolledCancellation(payload.runId);
|
|
101
|
+
}
|
|
102
|
+
this.options.send('ack', { type: envelope.type }, envelope.id);
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
//# sourceMappingURL=node-control-message-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-control-message-service.js","sourceRoot":"","sources":["../../src/service/node-control-message-service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAqB,MAAM,uBAAuB,CAAC;AAmB9E,6EAA6E;AAC7E,MAAM,OAAO,yBAAyB;IACP;IAA7B,YAA6B,OAAyC;QAAzC,YAAO,GAAP,OAAO,CAAkC;IAAG,CAAC;IAE1E,MAAM,CAAC,GAAW;QAChB,IAAI,QAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,QAAQ,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC1B,CAAC;IAEO,QAAQ,CAAC,QAAsB;QACrC,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACnC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YACxB,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YAC1C,KAAK,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;YACrC,KAAK,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,QAAQ,CAAC,OAA6B,CAAC;YACvD,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB;gBACpC,IAAI,CAAC,OAAO,CAAC,yBAAyB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC3D,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YACrC,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;YAC9C,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAC9B,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAClC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC/C,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,yBAAyB,EAAE,CAAC;YAChD,IAAI,CAAC,OAAO,CAAC,4BAA4B,CAAC,QAAQ,CAAC,CAAC;YACpD,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,IAAI,QAAQ,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACxE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACzB,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,gBAAgB,IAAI,QAAQ,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YAC9E,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,IAAI,KAAK,OAAO;YAC3B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC9E,CAAC;IAEO,QAAQ,CAAC,QAAsB;QACrC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAqD,CAAC;QAC/E,IACE,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ;YAClC,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ;YACtC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,SAAS,EACnC,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QACD,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IACjE,CAAC;IAEO,cAAc,CAAC,QAAsB;QAC3C,MAAM,OAAO,GAAG,QAAQ,CAAC,OAIxB,CAAC;QACF,IACE,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ;YACjC,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ;YACtC,CAAC,OAAO,CAAC,QAAQ,KAAK,UAAU,IAAI,OAAO,CAAC,QAAQ,KAAK,UAAU,CAAC,EACpE,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YACrE,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC;YAClC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC3B,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAEO,SAAS,CAAC,QAAsB;QACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAA8B,CAAC;QACxD,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CACnD,OAAO,CAAC,KAAK,EACb,QAAQ,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CACxD,CAAC;YACF,IAAI,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,CAAC,gCAAgC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { NodeEnvelope } from '@myagentroam/protocol';
|
|
2
|
+
interface NodeRequestServiceOptions {
|
|
3
|
+
readonly nodeId: () => string | undefined;
|
|
4
|
+
readonly databaseAvailable: () => boolean;
|
|
5
|
+
readonly execute: (operation: string, data: unknown) => Promise<unknown>;
|
|
6
|
+
readonly send: (type: string, payload: unknown, replyTo?: string) => void;
|
|
7
|
+
}
|
|
8
|
+
/** Validates and executes the closed Admin-facing Node operation envelope. */
|
|
9
|
+
export declare class NodeRequestService {
|
|
10
|
+
private readonly options;
|
|
11
|
+
constructor(options: NodeRequestServiceOptions);
|
|
12
|
+
handle(envelope: NodeEnvelope): Promise<void>;
|
|
13
|
+
}
|
|
14
|
+
export {};
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import { nodeLog } from '../operational.js';
|
|
2
|
+
import { validUserAuthorization } from '../runner/codex/conversation-parser.js';
|
|
3
|
+
/** Validates and executes the closed Admin-facing Node operation envelope. */
|
|
4
|
+
export class NodeRequestService {
|
|
5
|
+
options;
|
|
6
|
+
constructor(options) {
|
|
7
|
+
this.options = options;
|
|
8
|
+
}
|
|
9
|
+
async handle(envelope) {
|
|
10
|
+
const payload = envelope.payload;
|
|
11
|
+
if (typeof payload.operation !== 'string' || !this.options.databaseAvailable()) {
|
|
12
|
+
this.options.send('node.response', { error: 'MESSAGE_INVALID' }, envelope.id);
|
|
13
|
+
return;
|
|
14
|
+
}
|
|
15
|
+
if (!validUserAuthorization(payload.userAuthorization, envelope, payload.operation, this.options.nodeId())) {
|
|
16
|
+
this.options.send('node.response', { error: 'USER_AUTHORIZATION_REQUIRED' }, envelope.id);
|
|
17
|
+
return;
|
|
18
|
+
}
|
|
19
|
+
const startedAt = performance.now();
|
|
20
|
+
try {
|
|
21
|
+
const data = payload.data !== null && typeof payload.data === 'object'
|
|
22
|
+
? {
|
|
23
|
+
...payload.data,
|
|
24
|
+
__workspaceWatchSessionHash: payload.userAuthorization
|
|
25
|
+
.sessionHash
|
|
26
|
+
}
|
|
27
|
+
: payload.data;
|
|
28
|
+
const result = await this.options.execute(payload.operation, data);
|
|
29
|
+
this.options.send('node.response', { result }, envelope.id);
|
|
30
|
+
const durationMs = Math.round(performance.now() - startedAt);
|
|
31
|
+
if (durationMs >= 250)
|
|
32
|
+
nodeLog('node.operation.completed', {
|
|
33
|
+
nodeId: this.options.nodeId(),
|
|
34
|
+
operation: payload.operation,
|
|
35
|
+
requestId: envelope.id,
|
|
36
|
+
durationMs
|
|
37
|
+
});
|
|
38
|
+
}
|
|
39
|
+
catch (error) {
|
|
40
|
+
nodeLog('node.operation.failed', {
|
|
41
|
+
nodeId: this.options.nodeId(),
|
|
42
|
+
operation: payload.operation,
|
|
43
|
+
requestId: envelope.id,
|
|
44
|
+
durationMs: Math.round(performance.now() - startedAt),
|
|
45
|
+
error: error instanceof Error ? error.message : 'INTERNAL_ERROR'
|
|
46
|
+
});
|
|
47
|
+
this.options.send('node.response', { error: error instanceof Error ? error.message : 'INTERNAL_ERROR' }, envelope.id);
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
//# sourceMappingURL=node-request-service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"node-request-service.js","sourceRoot":"","sources":["../../src/service/node-request-service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,sBAAsB,EAAE,MAAM,wCAAwC,CAAC;AAShF,8EAA8E;AAC9E,MAAM,OAAO,kBAAkB;IACA;IAA7B,YAA6B,OAAkC;QAAlC,YAAO,GAAP,OAAO,CAA2B;IAAG,CAAC;IAEnE,KAAK,CAAC,MAAM,CAAC,QAAsB;QACjC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAIxB,CAAC;QACF,IAAI,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC;YAC/E,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9E,OAAO;QACT,CAAC;QACD,IACE,CAAC,sBAAsB,CACrB,OAAO,CAAC,iBAAiB,EACzB,QAAQ,EACR,OAAO,CAAC,SAAS,EACjB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CACtB,EACD,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,KAAK,EAAE,6BAA6B,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC1F,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,IAAI,GACR,OAAO,CAAC,IAAI,KAAK,IAAI,IAAI,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;gBACvD,CAAC,CAAC;oBACE,GAAI,OAAO,CAAC,IAAgC;oBAC5C,2BAA2B,EAAG,OAAO,CAAC,iBAA6C;yBAChF,WAAW;iBACf;gBACH,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;YACnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACnE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC7D,IAAI,UAAU,IAAI,GAAG;gBACnB,OAAO,CAAC,0BAA0B,EAAE;oBAClC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;oBAC7B,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,SAAS,EAAE,QAAQ,CAAC,EAAE;oBACtB,UAAU;iBACX,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,uBAAuB,EAAE;gBAC/B,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;gBAC7B,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,SAAS,EAAE,QAAQ,CAAC,EAAE;gBACtB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBACrD,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB;aACjE,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,IAAI,CACf,eAAe,EACf,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,EAAE,EACpE,QAAQ,CAAC,EAAE,CACZ,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}
|