@code-yeongyu/senpi 2026.9.12-2 → 2026.9.12
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +0 -17
- package/dist/core/agent-session-runtime.d.ts.map +1 -1
- package/dist/core/agent-session-runtime.js +1 -5
- package/dist/core/agent-session-runtime.js.map +1 -1
- package/dist/core/agent-session.d.ts +9 -13
- package/dist/core/agent-session.d.ts.map +1 -1
- package/dist/core/agent-session.js +121 -54
- package/dist/core/agent-session.js.map +1 -1
- package/dist/core/extensions/builtin/cursor-cli-oauth/models.d.ts.map +1 -1
- package/dist/core/extensions/builtin/cursor-cli-oauth/models.js +1 -2
- package/dist/core/extensions/builtin/cursor-cli-oauth/models.js.map +1 -1
- package/dist/core/session-manager.d.ts +0 -2
- package/dist/core/session-manager.d.ts.map +1 -1
- package/dist/core/session-manager.js +13 -22
- package/dist/core/session-manager.js.map +1 -1
- package/dist/core/session-write-reservation.d.ts +0 -9
- package/dist/core/session-write-reservation.d.ts.map +1 -1
- package/dist/core/session-write-reservation.js +0 -33
- package/dist/core/session-write-reservation.js.map +1 -1
- package/dist/modes/interactive/components/ask-user-async-widget.d.ts +0 -2
- package/dist/modes/interactive/components/ask-user-async-widget.d.ts.map +1 -1
- package/dist/modes/interactive/components/ask-user-async-widget.js +1 -17
- package/dist/modes/interactive/components/ask-user-async-widget.js.map +1 -1
- package/dist/modes/interactive/interactive-mode.js +2 -2
- package/dist/modes/interactive/interactive-mode.js.map +1 -1
- package/dist/modes/rpc/rpc-types.d.ts +1 -2
- package/dist/modes/rpc/rpc-types.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-types.js +0 -1
- package/dist/modes/rpc/rpc-types.js.map +1 -1
- package/dist/modes/rpc/session-registry.d.ts +1 -3
- package/dist/modes/rpc/session-registry.d.ts.map +1 -1
- package/dist/modes/rpc/session-registry.js.map +1 -1
- package/dist/modes/rpc/session-worker-client.d.ts +8 -10
- package/dist/modes/rpc/session-worker-client.d.ts.map +1 -1
- package/dist/modes/rpc/session-worker-client.js +23 -14
- package/dist/modes/rpc/session-worker-client.js.map +1 -1
- package/dist/modes/rpc/session-worker-protocol.d.ts +0 -10
- package/dist/modes/rpc/session-worker-protocol.d.ts.map +1 -1
- package/dist/modes/rpc/session-worker-protocol.js +0 -2
- package/dist/modes/rpc/session-worker-protocol.js.map +1 -1
- package/dist/modes/rpc/session-worker.js +16 -12
- package/dist/modes/rpc/session-worker.js.map +1 -1
- package/dist/modes/rpc/worker-session-registry.d.ts +0 -12
- package/dist/modes/rpc/worker-session-registry.d.ts.map +1 -1
- package/dist/modes/rpc/worker-session-registry.js +24 -51
- package/dist/modes/rpc/worker-session-registry.js.map +1 -1
- package/docs/rpc.md +6 -14
- package/node_modules/@code-yeongyu/senpi-codemode/CHANGELOG.md +0 -12
- package/node_modules/@code-yeongyu/senpi-codemode/package.json +4 -4
- package/node_modules/@earendil-works/pi-agent-core/package.json +3 -3
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent/measure.d.ts +0 -11
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent/measure.d.ts.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent/measure.js +0 -18
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent/measure.js.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent.d.ts +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent.d.ts.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent.js +1 -5
- package/node_modules/@earendil-works/pi-ai/dist/api/cursor-agent.js.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/cursor/store-migration.d.ts.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/cursor/store-migration.js +1 -2
- package/node_modules/@earendil-works/pi-ai/dist/cursor/store-migration.js.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/index.d.ts +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/index.d.ts.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/index.js +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/index.js.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/providers/cursor.d.ts.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/providers/cursor.js +1 -2
- package/node_modules/@earendil-works/pi-ai/dist/providers/cursor.js.map +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/providers/data/.manifest.json +1 -1
- package/node_modules/@earendil-works/pi-ai/dist/providers/data/openrouter.json +1 -1
- package/node_modules/@earendil-works/pi-ai/package.json +2 -2
- package/node_modules/@earendil-works/pi-pty/package.json +1 -1
- package/node_modules/@earendil-works/pi-telemetry/package.json +1 -1
- package/node_modules/@earendil-works/pi-tui/package.json +1 -1
- package/package.json +7 -7
- package/dist/core/cursor-history-admission.d.ts +0 -50
- package/dist/core/cursor-history-admission.d.ts.map +0 -1
- package/dist/core/cursor-history-admission.js +0 -130
- package/dist/core/cursor-history-admission.js.map +0 -1
- package/dist/modes/rpc/session-path-reservations.d.ts +0 -30
- package/dist/modes/rpc/session-path-reservations.d.ts.map +0 -1
- package/dist/modes/rpc/session-path-reservations.js +0 -58
- package/dist/modes/rpc/session-path-reservations.js.map +0 -1
- package/dist/modes/rpc/session-worker-credit.d.ts +0 -17
- package/dist/modes/rpc/session-worker-credit.d.ts.map +0 -1
- package/dist/modes/rpc/session-worker-credit.js +0 -39
- package/dist/modes/rpc/session-worker-credit.js.map +0 -1
- package/dist/modes/rpc/session-worker-signals.d.ts +0 -7
- package/dist/modes/rpc/session-worker-signals.d.ts.map +0 -1
- package/dist/modes/rpc/session-worker-signals.js +0 -19
- package/dist/modes/rpc/session-worker-signals.js.map +0 -1
- package/node_modules/@earendil-works/pi-ai/dist/cursor/context-limit-store.d.ts +0 -32
- package/node_modules/@earendil-works/pi-ai/dist/cursor/context-limit-store.d.ts.map +0 -1
- package/node_modules/@earendil-works/pi-ai/dist/cursor/context-limit-store.js +0 -62
- package/node_modules/@earendil-works/pi-ai/dist/cursor/context-limit-store.js.map +0 -1
- package/node_modules/@earendil-works/pi-ai/dist/utils/cursor-context-limit.d.ts +0 -12
- package/node_modules/@earendil-works/pi-ai/dist/utils/cursor-context-limit.d.ts.map +0 -1
- package/node_modules/@earendil-works/pi-ai/dist/utils/cursor-context-limit.js +0 -93
- package/node_modules/@earendil-works/pi-ai/dist/utils/cursor-context-limit.js.map +0 -1
|
@@ -2,7 +2,6 @@ import { fileURLToPath } from "node:url";
|
|
|
2
2
|
import { Worker } from "node:worker_threads";
|
|
3
3
|
import { isBunBinary } from "../../config.js";
|
|
4
4
|
import { SessionWorkerRequests } from "./session-worker-requests.js";
|
|
5
|
-
import { acknowledge, acknowledgeGrant, respondDisplay } from "./session-worker-signals.js";
|
|
6
5
|
const compiledWorkerEntry = typeof SENPI_RPC_SESSION_WORKER_ENTRY === "string"
|
|
7
6
|
? SENPI_RPC_SESSION_WORKER_ENTRY
|
|
8
7
|
: "./src/modes/rpc/session-worker.ts";
|
|
@@ -115,10 +114,23 @@ export class SessionWorkerClient {
|
|
|
115
114
|
}
|
|
116
115
|
this.worker.postMessage(message);
|
|
117
116
|
}
|
|
117
|
+
acknowledge(signal, granted) {
|
|
118
|
+
const state = new Int32Array(signal);
|
|
119
|
+
Atomics.store(state, 0, granted ? 1 : 2);
|
|
120
|
+
Atomics.notify(state, 0);
|
|
121
|
+
}
|
|
122
|
+
respondDisplay(signal) {
|
|
123
|
+
const display = this.display();
|
|
124
|
+
const values = new Float64Array(signal);
|
|
125
|
+
values[1] = display.width;
|
|
126
|
+
values[2] = display.revision;
|
|
127
|
+
Atomics.store(new Int32Array(signal), 1, display.rendered ? 1 : 0);
|
|
128
|
+
this.acknowledge(signal, true);
|
|
129
|
+
}
|
|
118
130
|
receive(message) {
|
|
119
131
|
if (this.stopped) {
|
|
120
132
|
if ("signal" in message)
|
|
121
|
-
acknowledge(message.signal, false);
|
|
133
|
+
this.acknowledge(message.signal, false);
|
|
122
134
|
return;
|
|
123
135
|
}
|
|
124
136
|
switch (message.type) {
|
|
@@ -129,12 +141,11 @@ export class SessionWorkerClient {
|
|
|
129
141
|
return;
|
|
130
142
|
}
|
|
131
143
|
case "reserve":
|
|
132
|
-
|
|
144
|
+
this.acknowledge(message.signal, !this.stopped && this.callbacks.reserve(message.path));
|
|
133
145
|
return;
|
|
134
146
|
case "snapshot":
|
|
135
147
|
this.snapshot = message.snapshot;
|
|
136
|
-
this.
|
|
137
|
-
acknowledge(message.signal, true);
|
|
148
|
+
this.acknowledge(message.signal, !this.stopped);
|
|
138
149
|
if (message.settled)
|
|
139
150
|
for (const listener of [...this.listeners])
|
|
140
151
|
listener();
|
|
@@ -152,14 +163,12 @@ export class SessionWorkerClient {
|
|
|
152
163
|
const writer = this.writer;
|
|
153
164
|
const sessionId = this.sessionId;
|
|
154
165
|
if (!writer || !sessionId || this.stopped) {
|
|
155
|
-
acknowledge(message.signal, false);
|
|
166
|
+
this.acknowledge(message.signal, false);
|
|
156
167
|
return;
|
|
157
168
|
}
|
|
158
169
|
// Identity and activity commit before publication; clients may attach or disconnect on that event.
|
|
159
|
-
if (message.snapshot)
|
|
170
|
+
if (message.snapshot)
|
|
160
171
|
this.snapshot = message.snapshot;
|
|
161
|
-
this.callbacks.reconcile(message.snapshot.liveSessionPaths);
|
|
162
|
-
}
|
|
163
172
|
else if (this.snapshot)
|
|
164
173
|
this.snapshot = {
|
|
165
174
|
...this.snapshot,
|
|
@@ -168,15 +177,15 @@ export class SessionWorkerClient {
|
|
|
168
177
|
};
|
|
169
178
|
const enqueue = () => {
|
|
170
179
|
if (!writer.enqueue(sessionId, message.record)) {
|
|
171
|
-
acknowledge(message.signal, false);
|
|
180
|
+
this.acknowledge(message.signal, false);
|
|
172
181
|
this.fail("session_output_overflow_or_closed");
|
|
173
182
|
return Promise.reject(new Error("session_output_overflow_or_closed"));
|
|
174
183
|
}
|
|
175
184
|
return writer.waitForSessionBackpressure(sessionId);
|
|
176
185
|
};
|
|
177
186
|
const consumed = message.connection === undefined ? enqueue() : writer.withConnection(message.connection, enqueue);
|
|
178
|
-
void consumed.then(() => acknowledge(message.signal, true), (cause) => {
|
|
179
|
-
acknowledge(message.signal, false);
|
|
187
|
+
void consumed.then(() => this.acknowledge(message.signal, true), (cause) => {
|
|
188
|
+
this.acknowledge(message.signal, false);
|
|
180
189
|
this.fail(cause instanceof Error ? cause.message : String(cause));
|
|
181
190
|
});
|
|
182
191
|
return;
|
|
@@ -184,11 +193,11 @@ export class SessionWorkerClient {
|
|
|
184
193
|
case "width":
|
|
185
194
|
this.options.sharedWidth?.setWidth(message.connection, message.width);
|
|
186
195
|
this.options.sharedWidth?.onChange?.();
|
|
187
|
-
respondDisplay(message.signal
|
|
196
|
+
this.respondDisplay(message.signal);
|
|
188
197
|
return;
|
|
189
198
|
case "capabilities":
|
|
190
199
|
this.options.sharedWidth?.setCapabilities?.(message.connection, message.capabilities);
|
|
191
|
-
respondDisplay(message.signal
|
|
200
|
+
this.respondDisplay(message.signal);
|
|
192
201
|
return;
|
|
193
202
|
case "request_close":
|
|
194
203
|
this.requestClose?.();
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"session-worker-client.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAc9C,OAAO,EAAE,qBAAqB,EAA2B,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAa5F,MAAM,mBAAmB,GACxB,OAAO,8BAA8B,KAAK,QAAQ;IACjD,CAAC,CAAC,8BAA8B;IAChC,CAAC,CAAC,mCAAmC,CAAC;AAExC,MAAM,OAAO,mBAAmB;IA2B/B,YAAY,SAAiC;QA1BpC,WAAM,GAAG,IAAI,MAAM,CAC3B,WAAW;YACV,CAAC,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,mBAAmB,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YACpF,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,qBAAqB,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAC5G,CAAC;QAGF,wEAAwE;QACxE,iBAAY,GAAG,KAAK,CAAC;QACJ,aAAQ,GAAG,IAAI,qBAAqB,CACpD,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAC7C,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gCAAgC,CAAC,CACjD,CAAC;QACM,YAAO,GAAG,KAAK,CAAC;QAChB,oBAAe,GAAG,CAAC,CAAC;QAKpB,YAAO,GAA+D,EAAE,CAAC;QAChE,cAAS,GAAG,IAAI,GAAG,EAAc,CAAC;QAClC,aAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;QAM9D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU;oBAAE,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBAC1E,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;gBACpB,IAAI,IAAI,CAAC,UAAU;oBAAE,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnD,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBACxD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;gBACvB,SAAS,CAAC,IAAI,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC;YACX,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAA4B,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAChH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,aAAsC,EAAE,OAAgC;QACrF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/E,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACnF,OAAO,MAAM,CAAC,WAAW,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM;QACX,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACtD,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAC/E,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,OAAO,MAAM,CAAC,QAAQ,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,IAAI,CACT,SAAiB,EACjB,MAA0B,EAC1B,YAAwB,EACxB,OAAmE;QAEnE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;QACjH,OAAO;YACN,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBACzB,MAAM,IAAI,CAAC,OAAO,CAAC;oBAClB,IAAI,EAAE,SAAS;oBACf,OAAO;oBACP,UAAU,EAAE,MAAM,CAAC,iBAAiB,EAAE;oBACtC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;iBACvB,CAAC,CAAC;YACJ,CAAC;YACD,gCAAgC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;YACxE,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACjF,OAAO,EAAE,KAAK,IAAI,EAAE;gBACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAClC,CAAC;SACD,CAAC;IACH,CAAC;IAED,IAAI,IAAI;QACP,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,IAAI,KAAK,IAAI,CAAC;IACtE,CAAC;IAED,gBAAgB,CAAC,QAAoB;QACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,2FAA2F;IAC3F,KAAK,CAAC,OAAe;QACpB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,UAAU;QACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAClD,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IAC9B,CAAC;IAEO,OAAO;QACd,OAAO;YACN,QAAQ,EAAE,EAAE,IAAI,CAAC,eAAe;YAChC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE;YACjD,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,qBAAqB,EAAE,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,IAAI,KAAK;YAC1F,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE;SAC7C,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,OAA2B;QAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAEO,IAAI,CAAC,OAA4B;QACxC,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAChE,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS;oBAAE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;gBAC7D,OAAO;YACR,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAEO,OAAO,CAAC,OAA4B;QAC3C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,OAAO;gBAAE,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC5D,OAAO;QACR,CAAC;QACD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,QAAQ,EAAE,CAAC;gBACf,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC/B,OAAO;YACR,CAAC;YACD,KAAK,SAAS;gBACb,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;gBACvE,OAAO;YACR,KAAK,UAAU;gBACd,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;gBACjC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;gBAC5D,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBAClC,IAAI,OAAO,CAAC,OAAO;oBAAE,KAAK,MAAM,QAAQ,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;wBAAE,QAAQ,EAAE,CAAC;gBAC5E,OAAO;YACR,KAAK,cAAc,EAAE,CAAC;gBACrB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;oBAClC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;oBAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnB,CAAC;gBACD,OAAO;YACR,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;gBAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;gBACjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBAC3C,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBACnC,OAAO;gBACR,CAAC;gBACD,mGAAmG;gBACnG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;oBACtB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;oBACjC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;gBAC7D,CAAC;qBAAM,IAAI,IAAI,CAAC,QAAQ;oBACvB,IAAI,CAAC,QAAQ,GAAG;wBACf,GAAG,IAAI,CAAC,QAAQ;wBAChB,GAAG,OAAO,CAAC,QAAQ;wBACnB,KAAK,EAAE,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;qBAC1E,CAAC;gBACH,MAAM,OAAO,GAAG,GAAG,EAAE;oBACpB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBAChD,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;wBACnC,IAAI,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;wBAC/C,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC,CAAC;oBACvE,CAAC;oBACD,OAAO,MAAM,CAAC,0BAA0B,CAAC,SAAS,CAAC,CAAC;gBACrD,CAAC,CAAC;gBACF,MAAM,QAAQ,GACb,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACnG,KAAK,QAAQ,CAAC,IAAI,CACjB,GAAG,EAAE,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,EACvC,CAAC,KAAc,EAAE,EAAE;oBAClB,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBACnC,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACnE,CAAC,CACD,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,OAAO;gBACX,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBACtE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,EAAE,CAAC;gBACvC,cAAc,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC/C,OAAO;YACR,KAAK,cAAc;gBAClB,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,eAAe,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBACtF,cAAc,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC/C,OAAO;YACR,KAAK,eAAe;gBACnB,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;gBACtB,OAAO;YACR,KAAK,SAAS;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzB,OAAO;QACT,CAAC;IACF,CAAC;IAEO,IAAI,CAAC,KAAa;QACzB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE;gBACxC,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,eAAe;gBACxB,OAAO,EAAE,KAAK;gBACd,KAAK;aACL,CAAC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;IACnB,CAAC;CACD","sourcesContent":["import { fileURLToPath } from \"node:url\";\nimport { Worker } from \"node:worker_threads\";\nimport { isBunBinary } from \"../../config.ts\";\nimport type { CliRuntimeConfiguration } from \"../../main.ts\";\nimport type { RpcConnectionOptions } from \"./connection-handler.ts\";\nimport type { RpcSessionBinding } from \"./session-binding.ts\";\nimport type { SessionEventWriter } from \"./session-event-writer.ts\";\nimport type { RpcSessionLaunchProfile } from \"./session-registry.ts\";\nimport type {\n\tHostToSessionWorker,\n\tSessionWorkerToHost,\n\tSessionWriteGrant,\n\tWorkerDisplay,\n\tWorkerSnapshot,\n} from \"./session-worker-protocol.ts\";\n\nimport { SessionWorkerRequests, type WorkerRequestInput } from \"./session-worker-requests.ts\";\nimport { acknowledge, acknowledgeGrant, respondDisplay } from \"./session-worker-signals.ts\";\n\n/** Lifecycle hooks the owning registry installs on every worker it allocates. */\nexport interface SessionWorkerCallbacks {\n\treserve: (path: string) => SessionWriteGrant;\n\t/** Every snapshot republishes which paths this worker still writes. */\n\treconcile: (livePaths: readonly string[]) => void;\n\texit: () => void;\n\tfailure: (error: string) => void;\n}\n\n/** Bun wrapper builders define the worker entry name relative to their explicit --root. */\ndeclare const SENPI_RPC_SESSION_WORKER_ENTRY: string | undefined;\nconst compiledWorkerEntry =\n\ttypeof SENPI_RPC_SESSION_WORKER_ENTRY === \"string\"\n\t\t? SENPI_RPC_SESSION_WORKER_ENTRY\n\t\t: \"./src/modes/rpc/session-worker.ts\";\n\nexport class SessionWorkerClient {\n\treadonly worker = new Worker(\n\t\tisBunBinary\n\t\t\t? fileURLToPath(new URL(compiledWorkerEntry, import.meta.url)).replaceAll(\"\\\\\", \"/\")\n\t\t\t: new URL(import.meta.url.endsWith(\".ts\") ? \"./session-worker.ts\" : \"./session-worker.js\", import.meta.url),\n\t);\n\treadonly exited: Promise<void>;\n\tsnapshot?: WorkerSnapshot;\n\t/** Main marks this only after installing the unique routing binding. */\n\tbindingReady = false;\n\tprivate readonly requests = new SessionWorkerRequests(\n\t\t(message) => this.worker.postMessage(message),\n\t\t() => this.fail(\"session_worker_request_timeout\"),\n\t);\n\tprivate stopped = false;\n\tprivate displayRevision = 0;\n\tprivate closeTimer?: ReturnType<typeof setTimeout>;\n\tprivate writer?: SessionEventWriter;\n\tprivate sessionId?: string;\n\tprivate requestClose?: () => void;\n\tprivate options: Pick<RpcConnectionOptions, \"capabilities\" | \"sharedWidth\"> = {};\n\tprivate readonly listeners = new Set<() => void>();\n\tprivate readonly controls = new Set<\"display\" | \"cancel_ui\">();\n\tprivate latestDisplay?: Extract<HostToSessionWorker, { type: \"display\" }>;\n\n\tprivate readonly callbacks: SessionWorkerCallbacks;\n\n\tconstructor(callbacks: SessionWorkerCallbacks) {\n\t\tthis.callbacks = callbacks;\n\t\tthis.exited = new Promise((resolve) => {\n\t\t\tthis.worker.once(\"exit\", () => {\n\t\t\t\tif (!this.stopped && !this.closeTimer) this.fail(\"session_worker_exited\");\n\t\t\t\tthis.stopped = true;\n\t\t\t\tif (this.closeTimer) clearTimeout(this.closeTimer);\n\t\t\t\tthis.requests.close(new Error(\"session_worker_exited\"));\n\t\t\t\tthis.listeners.clear();\n\t\t\t\tcallbacks.exit();\n\t\t\t\tresolve();\n\t\t\t});\n\t\t});\n\t\tthis.worker.on(\"message\", (message: SessionWorkerToHost) => this.receive(message));\n\t\tthis.worker.on(\"error\", (error: unknown) => this.fail(error instanceof Error ? error.message : String(error)));\n\t}\n\n\tasync prepare(configuration: CliRuntimeConfiguration, profile: RpcSessionLaunchProfile): Promise<string> {\n\t\tconst result = await this.request({ type: \"prepare\", configuration, profile });\n\t\tif (result.type !== \"prepared\") throw new Error(\"Invalid worker prepare response\");\n\t\treturn result.sessionPath;\n\t}\n\n\tasync commit(): Promise<WorkerSnapshot> {\n\t\tconst result = await this.request({ type: \"commit\" });\n\t\tif (result.type !== \"ready\") throw new Error(\"Invalid worker commit response\");\n\t\tthis.snapshot = result.snapshot;\n\t\treturn result.snapshot;\n\t}\n\n\tasync bind(\n\t\tsessionId: string,\n\t\twriter: SessionEventWriter,\n\t\trequestClose: () => void,\n\t\toptions: Pick<RpcConnectionOptions, \"capabilities\" | \"sharedWidth\">,\n\t): Promise<RpcSessionBinding> {\n\t\tthis.sessionId = sessionId;\n\t\tthis.writer = writer;\n\t\tthis.requestClose = requestClose;\n\t\tthis.options = options;\n\t\tawait this.request({ type: \"bind\", sessionId, display: this.display(), connection: writer.currentConnection() });\n\t\treturn {\n\t\t\thandle: async (command) => {\n\t\t\t\tawait this.request({\n\t\t\t\t\ttype: \"command\",\n\t\t\t\t\tcommand,\n\t\t\t\t\tconnection: writer.currentConnection(),\n\t\t\t\t\tdisplay: this.display(),\n\t\t\t\t});\n\t\t\t},\n\t\t\tcancelPendingExtensionUiRequests: () => this.post({ type: \"cancel_ui\" }),\n\t\t\trerenderComponents: () => this.post({ type: \"display\", display: this.display() }),\n\t\t\tdispose: async () => {\n\t\t\t\tthis.post({ type: \"cancel_ui\" });\n\t\t\t},\n\t\t};\n\t}\n\n\tget busy(): boolean {\n\t\treturn this.requests.activeCount > 0 || this.snapshot?.busy === true;\n\t}\n\n\tsubscribeSettled(listener: () => void): () => void {\n\t\tthis.listeners.add(listener);\n\t\treturn () => this.listeners.delete(listener);\n\t}\n\n\t/** Deadline requests termination, but only the actual exit callback releases ownership. */\n\tclose(graceMs: number): Promise<void> {\n\t\tif (this.stopped || this.closeTimer) return this.exited;\n\t\tthis.post({ type: \"close\" });\n\t\tthis.closeTimer = setTimeout(() => this.quarantine(), graceMs);\n\t\treturn this.exited;\n\t}\n\n\tquarantine(): void {\n\t\tthis.stopped = true;\n\t\tthis.requests.close(new Error(\"session_closing\"));\n\t\tvoid this.worker.terminate();\n\t}\n\n\tprivate display(): WorkerDisplay {\n\t\treturn {\n\t\t\trevision: ++this.displayRevision,\n\t\t\twidth: this.options.sharedWidth?.getWidth() ?? 80,\n\t\t\trendered: this.options.sharedWidth?.hasRenderedComponents?.(this.sessionId ?? \"\") ?? false,\n\t\t\tcapabilities: this.options.capabilities ?? [],\n\t\t};\n\t}\n\n\tprivate request(message: WorkerRequestInput): ReturnType<SessionWorkerRequests[\"request\"]> {\n\t\treturn this.requests.request(message);\n\t}\n\n\tprivate post(message: HostToSessionWorker): void {\n\t\tif (this.stopped) return;\n\t\tif (message.type === \"display\" || message.type === \"cancel_ui\") {\n\t\t\tif (this.controls.has(message.type)) {\n\t\t\t\tif (message.type === \"display\") this.latestDisplay = message;\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tthis.controls.add(message.type);\n\t\t}\n\t\tthis.worker.postMessage(message);\n\t}\n\n\tprivate receive(message: SessionWorkerToHost): void {\n\t\tif (this.stopped) {\n\t\t\tif (\"signal\" in message) acknowledge(message.signal, false);\n\t\t\treturn;\n\t\t}\n\t\tswitch (message.type) {\n\t\t\tcase \"prepared\":\n\t\t\tcase \"ready\":\n\t\t\tcase \"result\": {\n\t\t\t\tthis.requests.receive(message);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"reserve\":\n\t\t\t\tacknowledgeGrant(message.signal, this.callbacks.reserve(message.path));\n\t\t\t\treturn;\n\t\t\tcase \"snapshot\":\n\t\t\t\tthis.snapshot = message.snapshot;\n\t\t\t\tthis.callbacks.reconcile(message.snapshot.liveSessionPaths);\n\t\t\t\tacknowledge(message.signal, true);\n\t\t\t\tif (message.settled) for (const listener of [...this.listeners]) listener();\n\t\t\t\treturn;\n\t\t\tcase \"control_done\": {\n\t\t\t\tthis.controls.delete(message.control);\n\t\t\t\tif (message.control === \"display\" && this.latestDisplay) {\n\t\t\t\t\tconst latest = this.latestDisplay;\n\t\t\t\t\tthis.latestDisplay = undefined;\n\t\t\t\t\tthis.post(latest);\n\t\t\t\t}\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"output\": {\n\t\t\t\tconst writer = this.writer;\n\t\t\t\tconst sessionId = this.sessionId;\n\t\t\t\tif (!writer || !sessionId || this.stopped) {\n\t\t\t\t\tacknowledge(message.signal, false);\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\t// Identity and activity commit before publication; clients may attach or disconnect on that event.\n\t\t\t\tif (message.snapshot) {\n\t\t\t\t\tthis.snapshot = message.snapshot;\n\t\t\t\t\tthis.callbacks.reconcile(message.snapshot.liveSessionPaths);\n\t\t\t\t} else if (this.snapshot)\n\t\t\t\t\tthis.snapshot = {\n\t\t\t\t\t\t...this.snapshot,\n\t\t\t\t\t\t...message.activity,\n\t\t\t\t\t\tstate: { ...this.snapshot.state, isStreaming: message.activity.streaming },\n\t\t\t\t\t};\n\t\t\t\tconst enqueue = () => {\n\t\t\t\t\tif (!writer.enqueue(sessionId, message.record)) {\n\t\t\t\t\t\tacknowledge(message.signal, false);\n\t\t\t\t\t\tthis.fail(\"session_output_overflow_or_closed\");\n\t\t\t\t\t\treturn Promise.reject(new Error(\"session_output_overflow_or_closed\"));\n\t\t\t\t\t}\n\t\t\t\t\treturn writer.waitForSessionBackpressure(sessionId);\n\t\t\t\t};\n\t\t\t\tconst consumed =\n\t\t\t\t\tmessage.connection === undefined ? enqueue() : writer.withConnection(message.connection, enqueue);\n\t\t\t\tvoid consumed.then(\n\t\t\t\t\t() => acknowledge(message.signal, true),\n\t\t\t\t\t(cause: unknown) => {\n\t\t\t\t\t\tacknowledge(message.signal, false);\n\t\t\t\t\t\tthis.fail(cause instanceof Error ? cause.message : String(cause));\n\t\t\t\t\t},\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"width\":\n\t\t\t\tthis.options.sharedWidth?.setWidth(message.connection, message.width);\n\t\t\t\tthis.options.sharedWidth?.onChange?.();\n\t\t\t\trespondDisplay(message.signal, this.display());\n\t\t\t\treturn;\n\t\t\tcase \"capabilities\":\n\t\t\t\tthis.options.sharedWidth?.setCapabilities?.(message.connection, message.capabilities);\n\t\t\t\trespondDisplay(message.signal, this.display());\n\t\t\t\treturn;\n\t\t\tcase \"request_close\":\n\t\t\t\tthis.requestClose?.();\n\t\t\t\treturn;\n\t\t\tcase \"failure\":\n\t\t\t\tthis.fail(message.error);\n\t\t\t\treturn;\n\t\t}\n\t}\n\n\tprivate fail(error: string): void {\n\t\tif (this.stopped) return;\n\t\tthis.callbacks.failure(error);\n\t\tif (this.writer && this.sessionId) {\n\t\t\tthis.writer.enqueue(this.sessionId, { type: \"session_error\", error });\n\t\t\tthis.writer.closeSession(this.sessionId, {\n\t\t\t\ttype: \"response\",\n\t\t\t\tcommand: \"close_session\",\n\t\t\t\tsuccess: false,\n\t\t\t\terror,\n\t\t\t});\n\t\t}\n\t\tthis.quarantine();\n\t}\n}\n"]}
|
|
1
|
+
{"version":3,"file":"session-worker-client.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAa9C,OAAO,EAAE,qBAAqB,EAA2B,MAAM,8BAA8B,CAAC;AAI9F,MAAM,mBAAmB,GACxB,OAAO,8BAA8B,KAAK,QAAQ;IACjD,CAAC,CAAC,8BAA8B;IAChC,CAAC,CAAC,mCAAmC,CAAC;AAExC,MAAM,OAAO,mBAAmB;IA+B/B,YAAY,SAAqG;QA9BxG,WAAM,GAAG,IAAI,MAAM,CAC3B,WAAW;YACV,CAAC,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,mBAAmB,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;YACpF,CAAC,CAAC,IAAI,GAAG,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,qBAAqB,EAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAC5G,CAAC;QAGF,wEAAwE;QACxE,iBAAY,GAAG,KAAK,CAAC;QACJ,aAAQ,GAAG,IAAI,qBAAqB,CACpD,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAC7C,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gCAAgC,CAAC,CACjD,CAAC;QACM,YAAO,GAAG,KAAK,CAAC;QAChB,oBAAe,GAAG,CAAC,CAAC;QAKpB,YAAO,GAA+D,EAAE,CAAC;QAChE,cAAS,GAAG,IAAI,GAAG,EAAc,CAAC;QAClC,aAAQ,GAAG,IAAI,GAAG,EAA2B,CAAC;QAU9D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU;oBAAE,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBAC1E,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;gBACpB,IAAI,IAAI,CAAC,UAAU;oBAAE,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnD,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBACxD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;gBACvB,SAAS,CAAC,IAAI,EAAE,CAAC;gBACjB,OAAO,EAAE,CAAC;YACX,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAA4B,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAc,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAChH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,aAAsC,EAAE,OAAgC;QACrF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/E,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACnF,OAAO,MAAM,CAAC,WAAW,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM;QACX,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACtD,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QAC/E,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,OAAO,MAAM,CAAC,QAAQ,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,IAAI,CACT,SAAiB,EACjB,MAA0B,EAC1B,YAAwB,EACxB,OAAmE;QAEnE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;QACjH,OAAO;YACN,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBACzB,MAAM,IAAI,CAAC,OAAO,CAAC;oBAClB,IAAI,EAAE,SAAS;oBACf,OAAO;oBACP,UAAU,EAAE,MAAM,CAAC,iBAAiB,EAAE;oBACtC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;iBACvB,CAAC,CAAC;YACJ,CAAC;YACD,gCAAgC,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;YACxE,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACjF,OAAO,EAAE,KAAK,IAAI,EAAE;gBACnB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;YAClC,CAAC;SACD,CAAC;IACH,CAAC;IAED,IAAI,IAAI;QACP,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,IAAI,KAAK,IAAI,CAAC;IACtE,CAAC;IAED,gBAAgB,CAAC,QAAoB;QACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,2FAA2F;IAC3F,KAAK,CAAC,OAAe;QACpB,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;IAED,UAAU;QACT,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAClD,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IAC9B,CAAC;IAEO,OAAO;QACd,OAAO;YACN,QAAQ,EAAE,EAAE,IAAI,CAAC,eAAe;YAChC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE;YACjD,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,qBAAqB,EAAE,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,IAAI,KAAK;YAC1F,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE;SAC7C,CAAC;IACH,CAAC;IAEO,OAAO,CAAC,OAA2B;QAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAEO,IAAI,CAAC,OAA4B;QACxC,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAChE,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBACrC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS;oBAAE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC;gBAC7D,OAAO;YACR,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAEO,WAAW,CAAC,MAAyB,EAAE,OAAgB;QAC9D,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;QACrC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC;IAEO,cAAc,CAAC,MAAyB;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QAC1B,MAAM,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC7B,OAAO,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC;IAEO,OAAO,CAAC,OAA4B;QAC3C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,QAAQ,IAAI,OAAO;gBAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACjE,OAAO;QACR,CAAC;QACD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,QAAQ,EAAE,CAAC;gBACf,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC/B,OAAO;YACR,CAAC;YACD,KAAK,SAAS;gBACb,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;gBACxF,OAAO;YACR,KAAK,UAAU;gBACd,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;gBACjC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAChD,IAAI,OAAO,CAAC,OAAO;oBAAE,KAAK,MAAM,QAAQ,IAAI,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;wBAAE,QAAQ,EAAE,CAAC;gBAC5E,OAAO;YACR,KAAK,cAAc,EAAE,CAAC;gBACrB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACzD,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC;oBAClC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;oBAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnB,CAAC;gBACD,OAAO;YACR,CAAC;YACD,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;gBAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;gBACjC,IAAI,CAAC,MAAM,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;oBAC3C,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBACxC,OAAO;gBACR,CAAC;gBACD,mGAAmG;gBACnG,IAAI,OAAO,CAAC,QAAQ;oBAAE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;qBAClD,IAAI,IAAI,CAAC,QAAQ;oBACrB,IAAI,CAAC,QAAQ,GAAG;wBACf,GAAG,IAAI,CAAC,QAAQ;wBAChB,GAAG,OAAO,CAAC,QAAQ;wBACnB,KAAK,EAAE,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE;qBAC1E,CAAC;gBACH,MAAM,OAAO,GAAG,GAAG,EAAE;oBACpB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBAChD,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;wBACxC,IAAI,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;wBAC/C,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC,CAAC;oBACvE,CAAC;oBACD,OAAO,MAAM,CAAC,0BAA0B,CAAC,SAAS,CAAC,CAAC;gBACrD,CAAC,CAAC;gBACF,MAAM,QAAQ,GACb,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACnG,KAAK,QAAQ,CAAC,IAAI,CACjB,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,EAC5C,CAAC,KAAc,EAAE,EAAE;oBAClB,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;oBACxC,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACnE,CAAC,CACD,CAAC;gBACF,OAAO;YACR,CAAC;YACD,KAAK,OAAO;gBACX,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBACtE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,EAAE,EAAE,CAAC;gBACvC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpC,OAAO;YACR,KAAK,cAAc;gBAClB,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,eAAe,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;gBACtF,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpC,OAAO;YACR,KAAK,eAAe;gBACnB,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;gBACtB,OAAO;YACR,KAAK,SAAS;gBACb,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACzB,OAAO;QACT,CAAC;IACF,CAAC;IAEO,IAAI,CAAC,KAAa;QACzB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE;gBACxC,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,eAAe;gBACxB,OAAO,EAAE,KAAK;gBACd,KAAK;aACL,CAAC,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;IACnB,CAAC;CACD","sourcesContent":["import { fileURLToPath } from \"node:url\";\nimport { Worker } from \"node:worker_threads\";\nimport { isBunBinary } from \"../../config.ts\";\nimport type { CliRuntimeConfiguration } from \"../../main.ts\";\nimport type { RpcConnectionOptions } from \"./connection-handler.ts\";\nimport type { RpcSessionBinding } from \"./session-binding.ts\";\nimport type { SessionEventWriter } from \"./session-event-writer.ts\";\nimport type { RpcSessionLaunchProfile } from \"./session-registry.ts\";\nimport type {\n\tHostToSessionWorker,\n\tSessionWorkerToHost,\n\tWorkerDisplay,\n\tWorkerSnapshot,\n} from \"./session-worker-protocol.ts\";\n\nimport { SessionWorkerRequests, type WorkerRequestInput } from \"./session-worker-requests.ts\";\n\n/** Bun wrapper builders define the worker entry name relative to their explicit --root. */\ndeclare const SENPI_RPC_SESSION_WORKER_ENTRY: string | undefined;\nconst compiledWorkerEntry =\n\ttypeof SENPI_RPC_SESSION_WORKER_ENTRY === \"string\"\n\t\t? SENPI_RPC_SESSION_WORKER_ENTRY\n\t\t: \"./src/modes/rpc/session-worker.ts\";\n\nexport class SessionWorkerClient {\n\treadonly worker = new Worker(\n\t\tisBunBinary\n\t\t\t? fileURLToPath(new URL(compiledWorkerEntry, import.meta.url)).replaceAll(\"\\\\\", \"/\")\n\t\t\t: new URL(import.meta.url.endsWith(\".ts\") ? \"./session-worker.ts\" : \"./session-worker.js\", import.meta.url),\n\t);\n\treadonly exited: Promise<void>;\n\tsnapshot?: WorkerSnapshot;\n\t/** Main marks this only after installing the unique routing binding. */\n\tbindingReady = false;\n\tprivate readonly requests = new SessionWorkerRequests(\n\t\t(message) => this.worker.postMessage(message),\n\t\t() => this.fail(\"session_worker_request_timeout\"),\n\t);\n\tprivate stopped = false;\n\tprivate displayRevision = 0;\n\tprivate closeTimer?: ReturnType<typeof setTimeout>;\n\tprivate writer?: SessionEventWriter;\n\tprivate sessionId?: string;\n\tprivate requestClose?: () => void;\n\tprivate options: Pick<RpcConnectionOptions, \"capabilities\" | \"sharedWidth\"> = {};\n\tprivate readonly listeners = new Set<() => void>();\n\tprivate readonly controls = new Set<\"display\" | \"cancel_ui\">();\n\tprivate latestDisplay?: Extract<HostToSessionWorker, { type: \"display\" }>;\n\n\tprivate readonly callbacks: {\n\t\treserve: (path: string) => boolean;\n\t\texit: () => void;\n\t\tfailure: (error: string) => void;\n\t};\n\n\tconstructor(callbacks: { reserve: (path: string) => boolean; exit: () => void; failure: (error: string) => void }) {\n\t\tthis.callbacks = callbacks;\n\t\tthis.exited = new Promise((resolve) => {\n\t\t\tthis.worker.once(\"exit\", () => {\n\t\t\t\tif (!this.stopped && !this.closeTimer) this.fail(\"session_worker_exited\");\n\t\t\t\tthis.stopped = true;\n\t\t\t\tif (this.closeTimer) clearTimeout(this.closeTimer);\n\t\t\t\tthis.requests.close(new Error(\"session_worker_exited\"));\n\t\t\t\tthis.listeners.clear();\n\t\t\t\tcallbacks.exit();\n\t\t\t\tresolve();\n\t\t\t});\n\t\t});\n\t\tthis.worker.on(\"message\", (message: SessionWorkerToHost) => this.receive(message));\n\t\tthis.worker.on(\"error\", (error: unknown) => this.fail(error instanceof Error ? error.message : String(error)));\n\t}\n\n\tasync prepare(configuration: CliRuntimeConfiguration, profile: RpcSessionLaunchProfile): Promise<string> {\n\t\tconst result = await this.request({ type: \"prepare\", configuration, profile });\n\t\tif (result.type !== \"prepared\") throw new Error(\"Invalid worker prepare response\");\n\t\treturn result.sessionPath;\n\t}\n\n\tasync commit(): Promise<WorkerSnapshot> {\n\t\tconst result = await this.request({ type: \"commit\" });\n\t\tif (result.type !== \"ready\") throw new Error(\"Invalid worker commit response\");\n\t\tthis.snapshot = result.snapshot;\n\t\treturn result.snapshot;\n\t}\n\n\tasync bind(\n\t\tsessionId: string,\n\t\twriter: SessionEventWriter,\n\t\trequestClose: () => void,\n\t\toptions: Pick<RpcConnectionOptions, \"capabilities\" | \"sharedWidth\">,\n\t): Promise<RpcSessionBinding> {\n\t\tthis.sessionId = sessionId;\n\t\tthis.writer = writer;\n\t\tthis.requestClose = requestClose;\n\t\tthis.options = options;\n\t\tawait this.request({ type: \"bind\", sessionId, display: this.display(), connection: writer.currentConnection() });\n\t\treturn {\n\t\t\thandle: async (command) => {\n\t\t\t\tawait this.request({\n\t\t\t\t\ttype: \"command\",\n\t\t\t\t\tcommand,\n\t\t\t\t\tconnection: writer.currentConnection(),\n\t\t\t\t\tdisplay: this.display(),\n\t\t\t\t});\n\t\t\t},\n\t\t\tcancelPendingExtensionUiRequests: () => this.post({ type: \"cancel_ui\" }),\n\t\t\trerenderComponents: () => this.post({ type: \"display\", display: this.display() }),\n\t\t\tdispose: async () => {\n\t\t\t\tthis.post({ type: \"cancel_ui\" });\n\t\t\t},\n\t\t};\n\t}\n\n\tget busy(): boolean {\n\t\treturn this.requests.activeCount > 0 || this.snapshot?.busy === true;\n\t}\n\n\tsubscribeSettled(listener: () => void): () => void {\n\t\tthis.listeners.add(listener);\n\t\treturn () => this.listeners.delete(listener);\n\t}\n\n\t/** Deadline requests termination, but only the actual exit callback releases ownership. */\n\tclose(graceMs: number): Promise<void> {\n\t\tif (this.stopped || this.closeTimer) return this.exited;\n\t\tthis.post({ type: \"close\" });\n\t\tthis.closeTimer = setTimeout(() => this.quarantine(), graceMs);\n\t\treturn this.exited;\n\t}\n\n\tquarantine(): void {\n\t\tthis.stopped = true;\n\t\tthis.requests.close(new Error(\"session_closing\"));\n\t\tvoid this.worker.terminate();\n\t}\n\n\tprivate display(): WorkerDisplay {\n\t\treturn {\n\t\t\trevision: ++this.displayRevision,\n\t\t\twidth: this.options.sharedWidth?.getWidth() ?? 80,\n\t\t\trendered: this.options.sharedWidth?.hasRenderedComponents?.(this.sessionId ?? \"\") ?? false,\n\t\t\tcapabilities: this.options.capabilities ?? [],\n\t\t};\n\t}\n\n\tprivate request(message: WorkerRequestInput): ReturnType<SessionWorkerRequests[\"request\"]> {\n\t\treturn this.requests.request(message);\n\t}\n\n\tprivate post(message: HostToSessionWorker): void {\n\t\tif (this.stopped) return;\n\t\tif (message.type === \"display\" || message.type === \"cancel_ui\") {\n\t\t\tif (this.controls.has(message.type)) {\n\t\t\t\tif (message.type === \"display\") this.latestDisplay = message;\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tthis.controls.add(message.type);\n\t\t}\n\t\tthis.worker.postMessage(message);\n\t}\n\n\tprivate acknowledge(signal: SharedArrayBuffer, granted: boolean): void {\n\t\tconst state = new Int32Array(signal);\n\t\tAtomics.store(state, 0, granted ? 1 : 2);\n\t\tAtomics.notify(state, 0);\n\t}\n\n\tprivate respondDisplay(signal: SharedArrayBuffer): void {\n\t\tconst display = this.display();\n\t\tconst values = new Float64Array(signal);\n\t\tvalues[1] = display.width;\n\t\tvalues[2] = display.revision;\n\t\tAtomics.store(new Int32Array(signal), 1, display.rendered ? 1 : 0);\n\t\tthis.acknowledge(signal, true);\n\t}\n\n\tprivate receive(message: SessionWorkerToHost): void {\n\t\tif (this.stopped) {\n\t\t\tif (\"signal\" in message) this.acknowledge(message.signal, false);\n\t\t\treturn;\n\t\t}\n\t\tswitch (message.type) {\n\t\t\tcase \"prepared\":\n\t\t\tcase \"ready\":\n\t\t\tcase \"result\": {\n\t\t\t\tthis.requests.receive(message);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"reserve\":\n\t\t\t\tthis.acknowledge(message.signal, !this.stopped && this.callbacks.reserve(message.path));\n\t\t\t\treturn;\n\t\t\tcase \"snapshot\":\n\t\t\t\tthis.snapshot = message.snapshot;\n\t\t\t\tthis.acknowledge(message.signal, !this.stopped);\n\t\t\t\tif (message.settled) for (const listener of [...this.listeners]) listener();\n\t\t\t\treturn;\n\t\t\tcase \"control_done\": {\n\t\t\t\tthis.controls.delete(message.control);\n\t\t\t\tif (message.control === \"display\" && this.latestDisplay) {\n\t\t\t\t\tconst latest = this.latestDisplay;\n\t\t\t\t\tthis.latestDisplay = undefined;\n\t\t\t\t\tthis.post(latest);\n\t\t\t\t}\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"output\": {\n\t\t\t\tconst writer = this.writer;\n\t\t\t\tconst sessionId = this.sessionId;\n\t\t\t\tif (!writer || !sessionId || this.stopped) {\n\t\t\t\t\tthis.acknowledge(message.signal, false);\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\t// Identity and activity commit before publication; clients may attach or disconnect on that event.\n\t\t\t\tif (message.snapshot) this.snapshot = message.snapshot;\n\t\t\t\telse if (this.snapshot)\n\t\t\t\t\tthis.snapshot = {\n\t\t\t\t\t\t...this.snapshot,\n\t\t\t\t\t\t...message.activity,\n\t\t\t\t\t\tstate: { ...this.snapshot.state, isStreaming: message.activity.streaming },\n\t\t\t\t\t};\n\t\t\t\tconst enqueue = () => {\n\t\t\t\t\tif (!writer.enqueue(sessionId, message.record)) {\n\t\t\t\t\t\tthis.acknowledge(message.signal, false);\n\t\t\t\t\t\tthis.fail(\"session_output_overflow_or_closed\");\n\t\t\t\t\t\treturn Promise.reject(new Error(\"session_output_overflow_or_closed\"));\n\t\t\t\t\t}\n\t\t\t\t\treturn writer.waitForSessionBackpressure(sessionId);\n\t\t\t\t};\n\t\t\t\tconst consumed =\n\t\t\t\t\tmessage.connection === undefined ? enqueue() : writer.withConnection(message.connection, enqueue);\n\t\t\t\tvoid consumed.then(\n\t\t\t\t\t() => this.acknowledge(message.signal, true),\n\t\t\t\t\t(cause: unknown) => {\n\t\t\t\t\t\tthis.acknowledge(message.signal, false);\n\t\t\t\t\t\tthis.fail(cause instanceof Error ? cause.message : String(cause));\n\t\t\t\t\t},\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tcase \"width\":\n\t\t\t\tthis.options.sharedWidth?.setWidth(message.connection, message.width);\n\t\t\t\tthis.options.sharedWidth?.onChange?.();\n\t\t\t\tthis.respondDisplay(message.signal);\n\t\t\t\treturn;\n\t\t\tcase \"capabilities\":\n\t\t\t\tthis.options.sharedWidth?.setCapabilities?.(message.connection, message.capabilities);\n\t\t\t\tthis.respondDisplay(message.signal);\n\t\t\t\treturn;\n\t\t\tcase \"request_close\":\n\t\t\t\tthis.requestClose?.();\n\t\t\t\treturn;\n\t\t\tcase \"failure\":\n\t\t\t\tthis.fail(message.error);\n\t\t\t\treturn;\n\t\t}\n\t}\n\n\tprivate fail(error: string): void {\n\t\tif (this.stopped) return;\n\t\tthis.callbacks.failure(error);\n\t\tif (this.writer && this.sessionId) {\n\t\t\tthis.writer.enqueue(this.sessionId, { type: \"session_error\", error });\n\t\t\tthis.writer.closeSession(this.sessionId, {\n\t\t\t\ttype: \"response\",\n\t\t\t\tcommand: \"close_session\",\n\t\t\t\tsuccess: false,\n\t\t\t\terror,\n\t\t\t});\n\t\t}\n\t\tthis.quarantine();\n\t}\n}\n"]}
|
|
@@ -12,20 +12,10 @@ export declare const SESSION_WORKER_LIMITS: {
|
|
|
12
12
|
readonly openMs: 30000;
|
|
13
13
|
readonly controlMs: 5000;
|
|
14
14
|
};
|
|
15
|
-
/** Wire values the host writes into a worker's wait signal; `conflict` is the generic denial. */
|
|
16
|
-
export declare const WORKER_CREDIT_CODES: {
|
|
17
|
-
readonly granted: 1;
|
|
18
|
-
readonly conflict: 2;
|
|
19
|
-
readonly limit: 3;
|
|
20
|
-
};
|
|
21
|
-
/** Host decision on a worker's session-write path request. */
|
|
22
|
-
export type SessionWriteGrant = keyof typeof WORKER_CREDIT_CODES;
|
|
23
15
|
export interface WorkerSnapshot {
|
|
24
16
|
state: RpcSessionState;
|
|
25
17
|
/** Canonicalized by the owning worker, never by the transport thread. */
|
|
26
18
|
sessionPath?: string;
|
|
27
|
-
/** Every canonical path this worker's live session writers still own. */
|
|
28
|
-
liveSessionPaths: readonly string[];
|
|
29
19
|
busy: boolean;
|
|
30
20
|
streaming: boolean;
|
|
31
21
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"session-worker-protocol.d.ts","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAErE,eAAO,MAAM,qBAAqB;aACjC,OAAO,EAAE,EAAE;aACX,QAAQ,EAAE,EAAE;aACZ,eAAe,EAAE,CAAC;aAClB,YAAY;aACZ,YAAY;aACZ,WAAW;aACX,YAAY,EAAE,EAAE;aAChB,MAAM,EAAE,KAAM;aACd,SAAS,EAAE,IAAK;CACP,CAAC;AAEX,
|
|
1
|
+
{"version":3,"file":"session-worker-protocol.d.ts","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-protocol.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAErE,eAAO,MAAM,qBAAqB;aACjC,OAAO,EAAE,EAAE;aACX,QAAQ,EAAE,EAAE;aACZ,eAAe,EAAE,CAAC;aAClB,YAAY;aACZ,YAAY;aACZ,WAAW;aACX,YAAY,EAAE,EAAE;aAChB,MAAM,EAAE,KAAM;aACd,SAAS,EAAE,IAAK;CACP,CAAC;AAEX,MAAM,WAAW,cAAc;IAC9B,KAAK,EAAE,eAAe,CAAC;IACvB,yEAAyE;IACzE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,CAAC;IAClB,YAAY,EAAE,SAAS,MAAM,EAAE,CAAC;CAChC;AAED,MAAM,MAAM,mBAAmB,GAC5B;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,uBAAuB,CAAC;IAAC,OAAO,EAAE,uBAAuB,CAAA;CAAE,GAC9G;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,aAAa,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,GACjG;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,aAAa,CAAA;CAAE,GAClG;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,aAAa,CAAA;CAAE,GAC3C;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,GACrB;IAAE,IAAI,EAAE,OAAO,CAAA;CAAE,CAAC;AAErB,MAAM,MAAM,mBAAmB,GAC5B;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,GAC1D;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,cAAc,CAAA;CAAE,GAC5D;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,GACnD;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,iBAAiB,CAAA;CAAE,GAC5D;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,QAAQ,EAAE,cAAc,CAAC;IAAC,MAAM,EAAE,iBAAiB,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,GAC5F;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,OAAO,EAAE,SAAS,GAAG,WAAW,CAAA;CAAE,GAC1D;IACA,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,iBAAiB,CAAC;IAC1B,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,MAAM,GAAG,WAAW,CAAC,CAAC;IACrD,QAAQ,CAAC,EAAE,cAAc,CAAC;CACzB,GACD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,iBAAiB,CAAA;CAAE,GAChF;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,SAAS,MAAM,EAAE,CAAC;IAAC,MAAM,EAAE,iBAAiB,CAAA;CAAE,GACzG;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC"}
|
|
@@ -9,6 +9,4 @@ export const SESSION_WORKER_LIMITS = {
|
|
|
9
9
|
openMs: 30_000,
|
|
10
10
|
controlMs: 5_000,
|
|
11
11
|
};
|
|
12
|
-
/** Wire values the host writes into a worker's wait signal; `conflict` is the generic denial. */
|
|
13
|
-
export const WORKER_CREDIT_CODES = { granted: 1, conflict: 2, limit: 3 };
|
|
14
12
|
//# sourceMappingURL=session-worker-protocol.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"session-worker-protocol.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-protocol.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACpC,OAAO,EAAE,EAAE;IACX,QAAQ,EAAE,EAAE;IACZ,eAAe,EAAE,CAAC;IAClB,YAAY,EAAE,IAAI,GAAG,IAAI;IACzB,YAAY,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;IAC9B,WAAW,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;IAC7B,YAAY,EAAE,EAAE;IAChB,MAAM,EAAE,MAAM;IACd,SAAS,EAAE,KAAK;CACP,CAAC
|
|
1
|
+
{"version":3,"file":"session-worker-protocol.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker-protocol.ts"],"names":[],"mappings":"AAIA,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACpC,OAAO,EAAE,EAAE;IACX,QAAQ,EAAE,EAAE;IACZ,eAAe,EAAE,CAAC;IAClB,YAAY,EAAE,IAAI,GAAG,IAAI;IACzB,YAAY,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;IAC9B,WAAW,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;IAC7B,YAAY,EAAE,EAAE;IAChB,MAAM,EAAE,MAAM;IACd,SAAS,EAAE,KAAK;CACP,CAAC","sourcesContent":["import type { CliRuntimeConfiguration } from \"../../main.ts\";\nimport type { RpcSessionState } from \"./rpc-types.ts\";\nimport type { RpcSessionLaunchProfile } from \"./session-registry.ts\";\n\nexport const SESSION_WORKER_LIMITS = {\n\tworkers: 20,\n\trequests: 64,\n\tcontrolRequests: 4,\n\tcontrolBytes: 1024 * 1024,\n\trequestBytes: 16 * 1024 * 1024,\n\toutputBytes: 16 * 1024 * 1024,\n\treservations: 64,\n\topenMs: 30_000,\n\tcontrolMs: 5_000,\n} as const;\n\nexport interface WorkerSnapshot {\n\tstate: RpcSessionState;\n\t/** Canonicalized by the owning worker, never by the transport thread. */\n\tsessionPath?: string;\n\tbusy: boolean;\n\tstreaming: boolean;\n}\n\nexport interface WorkerDisplay {\n\trevision: number;\n\twidth: number;\n\trendered: boolean;\n\tcapabilities: readonly string[];\n}\n\nexport type HostToSessionWorker =\n\t| { type: \"prepare\"; request: number; configuration: CliRuntimeConfiguration; profile: RpcSessionLaunchProfile }\n\t| { type: \"commit\"; request: number }\n\t| { type: \"bind\"; request: number; sessionId: string; display: WorkerDisplay; connection?: string }\n\t| { type: \"command\"; request: number; command: object; connection?: string; display: WorkerDisplay }\n\t| { type: \"display\"; display: WorkerDisplay }\n\t| { type: \"cancel_ui\" }\n\t| { type: \"close\" };\n\nexport type SessionWorkerToHost =\n\t| { type: \"prepared\"; request: number; sessionPath: string }\n\t| { type: \"ready\"; request: number; snapshot: WorkerSnapshot }\n\t| { type: \"result\"; request: number; error?: string }\n\t| { type: \"reserve\"; path: string; signal: SharedArrayBuffer }\n\t| { type: \"snapshot\"; snapshot: WorkerSnapshot; signal: SharedArrayBuffer; settled?: boolean }\n\t| { type: \"control_done\"; control: \"display\" | \"cancel_ui\" }\n\t| {\n\t\t\ttype: \"output\";\n\t\t\trecord: object;\n\t\t\tconnection?: string;\n\t\t\tsignal: SharedArrayBuffer;\n\t\t\tactivity: Pick<WorkerSnapshot, \"busy\" | \"streaming\">;\n\t\t\tsnapshot?: WorkerSnapshot;\n\t }\n\t| { type: \"width\"; connection?: string; width: number; signal: SharedArrayBuffer }\n\t| { type: \"capabilities\"; connection?: string; capabilities: readonly string[]; signal: SharedArrayBuffer }\n\t| { type: \"request_close\" }\n\t| { type: \"failure\"; error: string };\n"]}
|
|
@@ -6,7 +6,7 @@ import { runWithProviderScope } from "@earendil-works/pi-ai/node/provider-scope"
|
|
|
6
6
|
import { WAKE_SOURCE_STATE_EVENT } from "../../core/extensions/builtin/monitor-state-event.js";
|
|
7
7
|
import { takeOverStdout } from "../../core/output-guard.js";
|
|
8
8
|
import { getDefaultSessionDir } from "../../core/session-manager.js";
|
|
9
|
-
import {
|
|
9
|
+
import { installSessionWriteReservation } from "../../core/session-write-reservation.js";
|
|
10
10
|
import { SettingsManager } from "../../core/settings-manager.js";
|
|
11
11
|
import { createCliRuntimeFactory } from "../../main.js";
|
|
12
12
|
import { initTheme } from "../interactive/theme/theme.js";
|
|
@@ -14,7 +14,6 @@ import { buildRpcSessionState } from "./connection-handler.js";
|
|
|
14
14
|
import { createRpcSessionBinding } from "./session-binding.js";
|
|
15
15
|
import { SessionEventWriter } from "./session-event-writer.js";
|
|
16
16
|
import { RpcSessionRegistry } from "./session-registry.js";
|
|
17
|
-
import { createWorkerCredit } from "./session-worker-credit.js";
|
|
18
17
|
import { SESSION_WORKER_LIMITS, } from "./session-worker-protocol.js";
|
|
19
18
|
takeOverStdout();
|
|
20
19
|
const port = parentPort;
|
|
@@ -29,8 +28,20 @@ function canonicalPath(path) {
|
|
|
29
28
|
const absolute = resolve(path);
|
|
30
29
|
return existsSync(absolute) ? realpathSync(absolute) : join(realpathSync(dirname(absolute)), basename(absolute));
|
|
31
30
|
}
|
|
32
|
-
|
|
33
|
-
|
|
31
|
+
function exchange(message, deniedError = "session_worker_output_denied", signal = new SharedArrayBuffer(4)) {
|
|
32
|
+
const state = new Int32Array(signal);
|
|
33
|
+
send(message(signal));
|
|
34
|
+
Atomics.wait(state, 0, 0, SESSION_WORKER_LIMITS.controlMs);
|
|
35
|
+
const result = Atomics.load(state, 0);
|
|
36
|
+
if (result === 2) {
|
|
37
|
+
if (deniedError === "session_path_in_use")
|
|
38
|
+
throw new Error(deniedError);
|
|
39
|
+
failWorker(deniedError);
|
|
40
|
+
}
|
|
41
|
+
if (result !== 1)
|
|
42
|
+
failWorker("session_worker_credit_timeout");
|
|
43
|
+
}
|
|
44
|
+
installSessionWriteReservation((path) => exchange((signal) => ({ type: "reserve", path: canonicalPath(path), signal }), "session_path_in_use"));
|
|
34
45
|
class WorkerEventWriter extends SessionEventWriter {
|
|
35
46
|
constructor() {
|
|
36
47
|
super(() => { });
|
|
@@ -104,16 +115,9 @@ function snapshot() {
|
|
|
104
115
|
if (!entry?.runtime)
|
|
105
116
|
throw new Error("Session runtime is not ready");
|
|
106
117
|
const session = entry.runtime.session;
|
|
107
|
-
const sessionPath = session.sessionFile ? canonicalPath(session.sessionFile) : undefined;
|
|
108
|
-
// The host releases every granted path this list omits, so it must name each writer
|
|
109
|
-
// still alive in this isolate, plus the session the runtime currently writes.
|
|
110
|
-
const live = new Set(liveSessionWritePaths().map(canonicalPath));
|
|
111
|
-
if (sessionPath)
|
|
112
|
-
live.add(sessionPath);
|
|
113
118
|
return {
|
|
114
119
|
state: buildRpcSessionState(session),
|
|
115
|
-
sessionPath,
|
|
116
|
-
liveSessionPaths: [...live],
|
|
120
|
+
sessionPath: session.sessionFile ? canonicalPath(session.sessionFile) : undefined,
|
|
117
121
|
busy: session.isSessionBusy,
|
|
118
122
|
streaming: session.isStreaming,
|
|
119
123
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"session-worker.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,uBAAuB,EAAE,MAAM,sDAAsD,CAAC;AAC/F,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,yCAAyC,CAAC;AAChF,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAA0B,MAAM,sBAAsB,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAwB,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,EAEN,qBAAqB,GAIrB,MAAM,8BAA8B,CAAC;AAEtC,cAAc,EAAE,CAAC;AACjB,MAAM,IAAI,GAAG,UAAU,CAAC;AACxB,IAAI,CAAC,IAAI;IAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;AACpE,MAAM,IAAI,GAAG,CAAC,OAA4B,EAAQ,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AAE/E,SAAS,UAAU,CAAC,KAAa;IAChC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;IACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,SAAS,aAAa,CAAC,IAAY;IAClC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/B,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;AAClH,CAAC;AAED,MAAM,EAAE,QAAQ,EAAE,uBAAuB,EAAE,GAAG,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;AACnF,uBAAuB,CAAC,aAAa,CAAC,CAAC;AAEvC,MAAM,iBAAkB,SAAQ,kBAAkB;IACjD;QACC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACjB,CAAC;IACQ,OAAO,CAAC,UAAkB,EAAE,MAAc;QAClD,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;YAChF,UAAU,CAAC,6BAA6B,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC;QACxC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1E,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,aAAa,EAAE,SAAS,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;QACjF,MAAM,WAAW,GAChB,MAAM,IAAI,MAAM;YAChB,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB;gBAClC,MAAM,CAAC,IAAI,KAAK,mBAAmB;gBACnC,MAAM,CAAC,IAAI,KAAK,kBAAkB;gBAClC,CAAC,MAAM,CAAC,IAAI,KAAK,sBAAsB,IAAI,QAAQ,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC;YAC/F,CAAC,CAAC,QAAQ,EAAE;YACZ,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,WAAW,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;YACpG,UAAU,CAAC,+BAA+B,CAAC,CAAC;QAC7C,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACrB,IAAI,EAAE,QAAQ;YACd,MAAM;YACN,UAAU,EAAE,IAAI,CAAC,iBAAiB,EAAE;YACpC,MAAM;YACN,QAAQ;YACR,QAAQ,EAAE,WAAW;SACrB,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACb,CAAC;CACD;AAED,MAAM,MAAM,GAAG,IAAI,iBAAiB,EAAE,CAAC;AACvC,IAAI,QAAuE,CAAC;AAC5E,IAAI,QAAwC,CAAC;AAC7C,IAAI,KAAkC,CAAC;AACvC,IAAI,OAAsC,CAAC;AAC3C,IAAI,OAAO,GAAkB,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;AAC3F,IAAI,OAAO,GAAG,KAAK,CAAC;AACpB,IAAI,WAAqC,CAAC;AAC1C,IAAI,eAAyC,CAAC;AAE9C,SAAS,YAAY,CAAC,IAAmB;IACxC,IAAI,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IACnD,OAAO,GAAG,IAAI,CAAC;IACf,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,aAAa,CAAC,OAA2D;IACjF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACzC,QAAQ,CAAC,OAAO,EAAE,+BAA+B,EAAE,MAAM,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACxC,YAAY,CAAC;QACZ,GAAG,OAAO;QACV,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QACnB,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC;KACvD,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAO,GAAG,KAAK;IACvC,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;IACzB,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;QAC/E,UAAU,CAAC,+BAA+B,CAAC,CAAC;IAC7C,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,gBAAgB;IACxB,WAAW,EAAE,EAAE,CAAC;IAChB,eAAe,EAAE,EAAE,CAAC;IACpB,eAAe,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAClG,eAAe,EAAE,CACjB,CAAC;IACF,WAAW,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CACzD,eAAe,CAAC,KAAK,CAAC,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAC9E,CAAC;AACH,CAAC;AAED,SAAS,QAAQ;IAChB,IAAI,CAAC,KAAK,EAAE,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IACrE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;IACtC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACzF,oFAAoF;IACpF,8EAA8E;IAC9E,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,qBAAqB,EAAE,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;IACjE,IAAI,WAAW;QAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACvC,OAAO;QACN,KAAK,EAAE,oBAAoB,CAAC,OAAO,CAAC;QACpC,WAAW;QACX,gBAAgB,EAAE,CAAC,GAAG,IAAI,CAAC;QAC3B,IAAI,EAAE,OAAO,CAAC,aAAa;QAC3B,SAAS,EAAE,OAAO,CAAC,WAAW;KAC9B,CAAC;AACH,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,OAA4B;IACjD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACtB,KAAK,SAAS,EAAE,CAAC;YAChB,IAAI,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACjE,MAAM,IAAI,GACT,OAAO,CAAC,OAAO,CAAC,WAAW;gBAC3B,IAAI,CACH,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,EACzE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,UAAU,EAAE,QAAQ,CACzE,CAAC;YACH,QAAQ,GAAG,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YAC7F,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvF,OAAO;QACR,CAAC;QACD,KAAK,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,IAAI,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACrE,MAAM,sBAAsB,GAAG,eAAe,CAAC,MAAM,CACpD,QAAQ,CAAC,aAAa,CAAC,GAAG,EAC1B,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAC/B,CAAC;YACF,MAAM,OAAO,GAAG,uBAAuB,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,sBAAsB,EAAE,CAAC,CAAC;YAC5F,SAAS,CAAC,sBAAsB,CAAC,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;YACpD,QAAQ,GAAG,IAAI,kBAAkB,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;YACzG,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC5D,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YACjE,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACxE,OAAO;QACR,CAAC;QACD,KAAK,MAAM,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,IAAI,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAClE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YAC1B,MAAM,YAAY,GAAG,KAAK,CAAC;YAC3B,MAAM,aAAa,GAAG,GAAG,EAAE,CAC1B,uBAAuB,CAAC,OAAO,CAAC,SAAS,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE;gBACvG,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,WAAW,EAAE;oBACZ,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK;oBAC7B,QAAQ,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAC/B,aAAa,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC1E,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,kBAAkB,EAAE,EAAE;oBAC/C,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;oBACpB,YAAY,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,EAAE;oBAC9C,qBAAqB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,QAAQ;oBAC7C,eAAe,EAAE,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE;wBAC7C,aAAa,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;wBACxF,OAAO,EAAE,kBAAkB,EAAE,EAAE,CAAC;oBACjC,CAAC;iBACD;aACD,CAAC,CAAC;YACJ,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS;gBAChD,CAAC,CAAC,aAAa,EAAE;gBACjB,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC;YACnC,KAAK,CAAC,aAAa,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE;gBACvC,MAAM,MAAM,EAAE,CAAC,OAAO,CAAC,CAAC;gBACxB,gBAAgB,EAAE,CAAC;YACpB,CAAC,CAAC;YACF,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACrD,gBAAgB,EAAE,CAAC;YACnB,eAAe,EAAE,CAAC;YAClB,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,CAAC;QACD,KAAK,SAAS,EAAE,CAAC;YAChB,MAAM,UAAU,GACf,MAAM,IAAI,OAAO,CAAC,OAAO;gBACzB,CAAC,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,uBAAuB,CAAC,CAAC,QAAQ,CACjF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAC5B,CAAC;YACH,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,UAAU,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC7E,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,aAAa,GAAG,OAAO,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS;gBACtC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;gBACvC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3F,eAAe,EAAE,CAAC;YAClB,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,CAAC;QACD,KAAK,SAAS;YACb,IAAI,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;gBAAE,OAAO,EAAE,kBAAkB,EAAE,EAAE,CAAC;YACnE,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,KAAK,WAAW;YACf,OAAO,EAAE,gCAAgC,EAAE,EAAE,CAAC;YAC9C,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YACrD,OAAO;QACR,KAAK,OAAO;YACX,OAAO,GAAG,IAAI,CAAC;YACf,WAAW,EAAE,EAAE,CAAC;YAChB,eAAe,EAAE,EAAE,CAAC;YACpB,MAAM,OAAO,EAAE,OAAO,EAAE,CAAC;YACzB,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,KAAK,CAAC;gBACtB,MAAM,oBAAoB,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;oBACpD,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;oBACvC,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;oBAC7C,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBACjC,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC7B,CAAC,CAAC,CAAC;YACJ,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACF,CAAC;AAED,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAA4B,EAAE,EAAE;IACnD,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QAC7C,MAAM,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACrE,IAAI,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QACD,IAAI,SAAS,IAAI,OAAO;YAAE,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;aAC/E,CAAC;YACL,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport { existsSync, realpathSync } from \"node:fs\";\nimport { basename, dirname, join, resolve } from \"node:path\";\nimport { parentPort } from \"node:worker_threads\";\nimport { runWithProviderScope } from \"@earendil-works/pi-ai/node/provider-scope\";\nimport { WAKE_SOURCE_STATE_EVENT } from \"../../core/extensions/builtin/monitor-state-event.ts\";\nimport { takeOverStdout } from \"../../core/output-guard.ts\";\nimport { getDefaultSessionDir } from \"../../core/session-manager.ts\";\nimport { liveSessionWritePaths } from \"../../core/session-write-reservation.ts\";\nimport { SettingsManager } from \"../../core/settings-manager.ts\";\nimport { createCliRuntimeFactory } from \"../../main.ts\";\nimport { initTheme } from \"../interactive/theme/theme.ts\";\nimport { buildRpcSessionState } from \"./connection-handler.ts\";\nimport { createRpcSessionBinding, type RpcSessionBinding } from \"./session-binding.ts\";\nimport { SessionEventWriter } from \"./session-event-writer.ts\";\nimport { type RpcSessionEntry, RpcSessionRegistry } from \"./session-registry.ts\";\nimport { createWorkerCredit } from \"./session-worker-credit.ts\";\nimport {\n\ttype HostToSessionWorker,\n\tSESSION_WORKER_LIMITS,\n\ttype SessionWorkerToHost,\n\ttype WorkerDisplay,\n\ttype WorkerSnapshot,\n} from \"./session-worker-protocol.ts\";\n\ntakeOverStdout();\nconst port = parentPort;\nif (!port) throw new Error(\"Session worker requires a parent port\");\nconst send = (message: SessionWorkerToHost): void => port.postMessage(message);\n\nfunction failWorker(error: string): never {\n\tsend({ type: \"failure\", error });\n\tprocess.exit(1);\n}\n\nfunction canonicalPath(path: string): string {\n\tconst absolute = resolve(path);\n\treturn existsSync(absolute) ? realpathSync(absolute) : join(realpathSync(dirname(absolute)), basename(absolute));\n}\n\nconst { exchange, installWriteReservation } = createWorkerCredit(send, failWorker);\ninstallWriteReservation(canonicalPath);\n\nclass WorkerEventWriter extends SessionEventWriter {\n\tconstructor() {\n\t\tsuper(() => {});\n\t}\n\toverride enqueue(_sessionId: string, record: object): boolean {\n\t\tif (Buffer.byteLength(JSON.stringify(record)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\t\tfailWorker(\"session_worker_output_limit\");\n\t\tconst session = entry?.runtime?.session;\n\t\tif (!session) throw new Error(\"Session output preceded runtime creation\");\n\t\tconst activity = { busy: session.isSessionBusy, streaming: session.isStreaming };\n\t\tconst replacement =\n\t\t\t\"type\" in record &&\n\t\t\t(record.type === \"session_replaced\" ||\n\t\t\t\trecord.type === \"question_resolved\" ||\n\t\t\t\trecord.type === \"question_updated\" ||\n\t\t\t\t(record.type === \"extension_ui_request\" && \"method\" in record && record.method === \"question\"))\n\t\t\t\t? snapshot()\n\t\t\t\t: undefined;\n\t\tif (replacement && Buffer.byteLength(JSON.stringify(replacement)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\t\tfailWorker(\"session_worker_snapshot_limit\");\n\t\texchange((signal) => ({\n\t\t\ttype: \"output\",\n\t\t\trecord,\n\t\t\tconnection: this.currentConnection(),\n\t\t\tsignal,\n\t\t\tactivity,\n\t\t\tsnapshot: replacement,\n\t\t}));\n\t\treturn true;\n\t}\n}\n\nconst writer = new WorkerEventWriter();\nlet prepared: Extract<HostToSessionWorker, { type: \"prepare\" }> | undefined;\nlet registry: RpcSessionRegistry | undefined;\nlet entry: RpcSessionEntry | undefined;\nlet binding: RpcSessionBinding | undefined;\nlet display: WorkerDisplay = { revision: 0, width: 80, rendered: false, capabilities: [] };\nlet closing = false;\nlet unsubscribe: (() => void) | undefined;\nlet unsubscribeWake: (() => void) | undefined;\n\nfunction applyDisplay(next: WorkerDisplay): boolean {\n\tif (next.revision < display.revision) return false;\n\tdisplay = next;\n\treturn true;\n}\n\nfunction updateDisplay(message: (signal: SharedArrayBuffer) => SessionWorkerToHost): void {\n\tconst signal = new SharedArrayBuffer(24);\n\texchange(message, \"session_worker_display_denied\", signal);\n\tconst values = new Float64Array(signal);\n\tapplyDisplay({\n\t\t...display,\n\t\twidth: values[1],\n\t\trevision: values[2],\n\t\trendered: Atomics.load(new Int32Array(signal), 1) === 1,\n\t});\n}\n\nfunction publishSnapshot(settled = false): void {\n\tconst value = snapshot();\n\tif (Buffer.byteLength(JSON.stringify(value)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\tfailWorker(\"session_worker_snapshot_limit\");\n\texchange((signal) => ({ type: \"snapshot\", snapshot: value, signal, settled }));\n}\n\nfunction subscribeSession(): void {\n\tunsubscribe?.();\n\tunsubscribeWake?.();\n\tunsubscribeWake = entry?.runtime?.session.extensionRunner.onBusEvent(WAKE_SOURCE_STATE_EVENT, () =>\n\t\tpublishSnapshot(),\n\t);\n\tunsubscribe = entry?.runtime?.session.subscribe((event) =>\n\t\tpublishSnapshot(event.type === \"agent_settled\" || event.type === \"agent_idle\"),\n\t);\n}\n\nfunction snapshot(): WorkerSnapshot {\n\tif (!entry?.runtime) throw new Error(\"Session runtime is not ready\");\n\tconst session = entry.runtime.session;\n\tconst sessionPath = session.sessionFile ? canonicalPath(session.sessionFile) : undefined;\n\t// The host releases every granted path this list omits, so it must name each writer\n\t// still alive in this isolate, plus the session the runtime currently writes.\n\tconst live = new Set(liveSessionWritePaths().map(canonicalPath));\n\tif (sessionPath) live.add(sessionPath);\n\treturn {\n\t\tstate: buildRpcSessionState(session),\n\t\tsessionPath,\n\t\tliveSessionPaths: [...live],\n\t\tbusy: session.isSessionBusy,\n\t\tstreaming: session.isStreaming,\n\t};\n}\n\nasync function handle(message: HostToSessionWorker): Promise<void> {\n\tswitch (message.type) {\n\t\tcase \"prepare\": {\n\t\t\tif (prepared) throw new Error(\"Session worker already prepared\");\n\t\t\tconst path =\n\t\t\t\tmessage.profile.sessionPath ??\n\t\t\t\tjoin(\n\t\t\t\t\tgetDefaultSessionDir(message.profile.cwd, message.configuration.agentDir),\n\t\t\t\t\t`${new Date().toISOString().replace(/[:.]/g, \"-\")}_${randomUUID()}.jsonl`,\n\t\t\t\t);\n\t\t\tprepared = { ...message, profile: { ...message.profile, sessionPath: canonicalPath(path) } };\n\t\t\tsend({ type: \"prepared\", request: message.request, sessionPath: canonicalPath(path) });\n\t\t\treturn;\n\t\t}\n\t\tcase \"commit\": {\n\t\t\tif (!prepared || registry) throw new Error(\"Invalid session commit\");\n\t\t\tconst startupSettingsManager = SettingsManager.create(\n\t\t\t\tprepared.configuration.cwd,\n\t\t\t\tprepared.configuration.agentDir,\n\t\t\t);\n\t\t\tconst factory = createCliRuntimeFactory(prepared.configuration, { startupSettingsManager });\n\t\t\tinitTheme(startupSettingsManager.getTheme(), false);\n\t\t\tregistry = new RpcSessionRegistry({ agentDir: prepared.configuration.agentDir, createRuntime: factory });\n\t\t\tconst opened = await registry.openSession(prepared.profile);\n\t\t\tentry = registry.getForCommand(opened.sessionId, \"open_session\");\n\t\t\tsend({ type: \"ready\", request: message.request, snapshot: snapshot() });\n\t\t\treturn;\n\t\t}\n\t\tcase \"bind\": {\n\t\t\tif (!entry || binding) throw new Error(\"Invalid session binding\");\n\t\t\tdisplay = message.display;\n\t\t\tconst bindingEntry = entry;\n\t\t\tconst createBinding = () =>\n\t\t\t\tcreateRpcSessionBinding(message.sessionId, bindingEntry, writer, () => send({ type: \"request_close\" }), {\n\t\t\t\t\tcapabilities: display.capabilities,\n\t\t\t\t\tsharedWidth: {\n\t\t\t\t\t\tgetWidth: () => display.width,\n\t\t\t\t\t\tsetWidth: (connection, width) =>\n\t\t\t\t\t\t\tupdateDisplay((signal) => ({ type: \"width\", connection, width, signal })),\n\t\t\t\t\t\tonChange: () => binding?.rerenderComponents?.(),\n\t\t\t\t\t\tclearWidth: () => {},\n\t\t\t\t\t\tconnectionId: () => writer.currentConnection(),\n\t\t\t\t\t\thasRenderedComponents: () => display.rendered,\n\t\t\t\t\t\tsetCapabilities: (connection, capabilities) => {\n\t\t\t\t\t\t\tupdateDisplay((signal) => ({ type: \"capabilities\", connection, capabilities, signal }));\n\t\t\t\t\t\t\tbinding?.rerenderComponents?.();\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\tbinding = await (message.connection === undefined\n\t\t\t\t? createBinding()\n\t\t\t\t: writer.withConnection(message.connection, createBinding));\n\t\t\tconst rebind = entry.rebindSession;\n\t\t\tentry.rebindSession = async (session) => {\n\t\t\t\tawait rebind?.(session);\n\t\t\t\tsubscribeSession();\n\t\t\t};\n\t\t\tentry.runtime?.setRebindSession(entry.rebindSession);\n\t\t\tsubscribeSession();\n\t\t\tpublishSnapshot();\n\t\t\tsend({ type: \"result\", request: message.request });\n\t\t\treturn;\n\t\t}\n\t\tcase \"command\": {\n\t\t\tconst privileged =\n\t\t\t\t\"type\" in message.command &&\n\t\t\t\t[\"abort\", \"abort_bash\", \"extension_ui_response\", \"extension_ui_progress\"].includes(\n\t\t\t\t\tString(message.command.type),\n\t\t\t\t);\n\t\t\tif (!binding || (closing && !privileged)) throw new Error(\"session_closing\");\n\t\t\tapplyDisplay(message.display);\n\t\t\tconst activeBinding = binding;\n\t\t\tawait (message.connection === undefined\n\t\t\t\t? activeBinding.handle(message.command)\n\t\t\t\t: writer.withConnection(message.connection, () => activeBinding.handle(message.command)));\n\t\t\tpublishSnapshot();\n\t\t\tsend({ type: \"result\", request: message.request });\n\t\t\treturn;\n\t\t}\n\t\tcase \"display\":\n\t\t\tif (applyDisplay(message.display)) binding?.rerenderComponents?.();\n\t\t\tsend({ type: \"control_done\", control: \"display\" });\n\t\t\treturn;\n\t\tcase \"cancel_ui\":\n\t\t\tbinding?.cancelPendingExtensionUiRequests?.();\n\t\t\tsend({ type: \"control_done\", control: \"cancel_ui\" });\n\t\t\treturn;\n\t\tcase \"close\":\n\t\t\tclosing = true;\n\t\t\tunsubscribe?.();\n\t\t\tunsubscribeWake?.();\n\t\t\tawait binding?.dispose();\n\t\t\tif (entry?.runtime) {\n\t\t\t\tconst current = entry;\n\t\t\t\tawait runWithProviderScope(current.scope, async () => {\n\t\t\t\t\tawait current.runtime?.session.abort();\n\t\t\t\t\tawait current.runtime?.session.waitForIdle();\n\t\t\t\t\tawait current.runtime?.dispose();\n\t\t\t\t\tawait current.scope.close();\n\t\t\t\t});\n\t\t\t}\n\t\t\tprocess.exit(0);\n\t}\n}\n\nport.on(\"message\", (message: HostToSessionWorker) => {\n\tvoid handle(message).catch((cause: unknown) => {\n\t\tconst error = cause instanceof Error ? cause.message : String(cause);\n\t\tif (error.startsWith(\"session_worker_\")) {\n\t\t\tsend({ type: \"failure\", error });\n\t\t\tprocess.exit(1);\n\t\t}\n\t\tif (\"request\" in message) send({ type: \"result\", request: message.request, error });\n\t\telse {\n\t\t\tsend({ type: \"failure\", error });\n\t\t\tprocess.exit(1);\n\t\t}\n\t});\n});\n"]}
|
|
1
|
+
{"version":3,"file":"session-worker.js","sourceRoot":"","sources":["../../../src/modes/rpc/session-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,uBAAuB,EAAE,MAAM,sDAAsD,CAAC;AAC/F,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAE,8BAA8B,EAAE,MAAM,yCAAyC,CAAC;AACzF,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAC/D,OAAO,EAAE,uBAAuB,EAA0B,MAAM,sBAAsB,CAAC;AACvF,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAwB,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AACjF,OAAO,EAEN,qBAAqB,GAIrB,MAAM,8BAA8B,CAAC;AAEtC,cAAc,EAAE,CAAC;AACjB,MAAM,IAAI,GAAG,UAAU,CAAC;AACxB,IAAI,CAAC,IAAI;IAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;AACpE,MAAM,IAAI,GAAG,CAAC,OAA4B,EAAQ,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;AAE/E,SAAS,UAAU,CAAC,KAAa;IAChC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;IACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,SAAS,aAAa,CAAC,IAAY;IAClC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/B,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;AAClH,CAAC;AAED,SAAS,QAAQ,CAChB,OAA2D,EAC3D,WAAW,GAAG,8BAA8B,EAC5C,MAAM,GAAG,IAAI,iBAAiB,CAAC,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;IACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IACtB,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACtC,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;QAClB,IAAI,WAAW,KAAK,qBAAqB;YAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;QACxE,UAAU,CAAC,WAAW,CAAC,CAAC;IACzB,CAAC;IACD,IAAI,MAAM,KAAK,CAAC;QAAE,UAAU,CAAC,+BAA+B,CAAC,CAAC;AAC/D,CAAC;AAED,8BAA8B,CAAC,CAAC,IAAI,EAAE,EAAE,CACvC,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,qBAAqB,CAAC,CACrG,CAAC;AAEF,MAAM,iBAAkB,SAAQ,kBAAkB;IACjD;QACC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IACjB,CAAC;IACQ,OAAO,CAAC,UAAkB,EAAE,MAAc;QAClD,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;YAChF,UAAU,CAAC,6BAA6B,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC;QACxC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1E,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,aAAa,EAAE,SAAS,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;QACjF,MAAM,WAAW,GAChB,MAAM,IAAI,MAAM;YAChB,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB;gBAClC,MAAM,CAAC,IAAI,KAAK,mBAAmB;gBACnC,MAAM,CAAC,IAAI,KAAK,kBAAkB;gBAClC,CAAC,MAAM,CAAC,IAAI,KAAK,sBAAsB,IAAI,QAAQ,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC;YAC/F,CAAC,CAAC,QAAQ,EAAE;YACZ,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,WAAW,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;YACpG,UAAU,CAAC,+BAA+B,CAAC,CAAC;QAC7C,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACrB,IAAI,EAAE,QAAQ;YACd,MAAM;YACN,UAAU,EAAE,IAAI,CAAC,iBAAiB,EAAE;YACpC,MAAM;YACN,QAAQ;YACR,QAAQ,EAAE,WAAW;SACrB,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC;IACb,CAAC;CACD;AAED,MAAM,MAAM,GAAG,IAAI,iBAAiB,EAAE,CAAC;AACvC,IAAI,QAAuE,CAAC;AAC5E,IAAI,QAAwC,CAAC;AAC7C,IAAI,KAAkC,CAAC;AACvC,IAAI,OAAsC,CAAC;AAC3C,IAAI,OAAO,GAAkB,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;AAC3F,IAAI,OAAO,GAAG,KAAK,CAAC;AACpB,IAAI,WAAqC,CAAC;AAC1C,IAAI,eAAyC,CAAC;AAE9C,SAAS,YAAY,CAAC,IAAmB;IACxC,IAAI,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IACnD,OAAO,GAAG,IAAI,CAAC;IACf,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,aAAa,CAAC,OAA2D;IACjF,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACzC,QAAQ,CAAC,OAAO,EAAE,+BAA+B,EAAE,MAAM,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IACxC,YAAY,CAAC;QACZ,GAAG,OAAO;QACV,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;QACnB,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC;KACvD,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAO,GAAG,KAAK;IACvC,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;IACzB,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,qBAAqB,CAAC,WAAW;QAC/E,UAAU,CAAC,+BAA+B,CAAC,CAAC;IAC7C,QAAQ,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,gBAAgB;IACxB,WAAW,EAAE,EAAE,CAAC;IAChB,eAAe,EAAE,EAAE,CAAC;IACpB,eAAe,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAClG,eAAe,EAAE,CACjB,CAAC;IACF,WAAW,GAAG,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CACzD,eAAe,CAAC,KAAK,CAAC,IAAI,KAAK,eAAe,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAC9E,CAAC;AACH,CAAC;AAED,SAAS,QAAQ;IAChB,IAAI,CAAC,KAAK,EAAE,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IACrE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;IACtC,OAAO;QACN,KAAK,EAAE,oBAAoB,CAAC,OAAO,CAAC;QACpC,WAAW,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;QACjF,IAAI,EAAE,OAAO,CAAC,aAAa;QAC3B,SAAS,EAAE,OAAO,CAAC,WAAW;KAC9B,CAAC;AACH,CAAC;AAED,KAAK,UAAU,MAAM,CAAC,OAA4B;IACjD,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACtB,KAAK,SAAS,EAAE,CAAC;YAChB,IAAI,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACjE,MAAM,IAAI,GACT,OAAO,CAAC,OAAO,CAAC,WAAW;gBAC3B,IAAI,CACH,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,EACzE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,UAAU,EAAE,QAAQ,CACzE,CAAC;YACH,QAAQ,GAAG,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YAC7F,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvF,OAAO;QACR,CAAC;QACD,KAAK,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,IAAI,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YACrE,MAAM,sBAAsB,GAAG,eAAe,CAAC,MAAM,CACpD,QAAQ,CAAC,aAAa,CAAC,GAAG,EAC1B,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAC/B,CAAC;YACF,MAAM,OAAO,GAAG,uBAAuB,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,sBAAsB,EAAE,CAAC,CAAC;YAC5F,SAAS,CAAC,sBAAsB,CAAC,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;YACpD,QAAQ,GAAG,IAAI,kBAAkB,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;YACzG,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC5D,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YACjE,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACxE,OAAO;QACR,CAAC;QACD,KAAK,MAAM,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,IAAI,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAClE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YAC1B,MAAM,YAAY,GAAG,KAAK,CAAC;YAC3B,MAAM,aAAa,GAAG,GAAG,EAAE,CAC1B,uBAAuB,CAAC,OAAO,CAAC,SAAS,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE;gBACvG,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,WAAW,EAAE;oBACZ,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK;oBAC7B,QAAQ,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,CAC/B,aAAa,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC1E,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,kBAAkB,EAAE,EAAE;oBAC/C,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;oBACpB,YAAY,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,iBAAiB,EAAE;oBAC9C,qBAAqB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,QAAQ;oBAC7C,eAAe,EAAE,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE;wBAC7C,aAAa,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;wBACxF,OAAO,EAAE,kBAAkB,EAAE,EAAE,CAAC;oBACjC,CAAC;iBACD;aACD,CAAC,CAAC;YACJ,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS;gBAChD,CAAC,CAAC,aAAa,EAAE;gBACjB,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC;YACnC,KAAK,CAAC,aAAa,GAAG,KAAK,EAAE,OAAO,EAAE,EAAE;gBACvC,MAAM,MAAM,EAAE,CAAC,OAAO,CAAC,CAAC;gBACxB,gBAAgB,EAAE,CAAC;YACpB,CAAC,CAAC;YACF,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACrD,gBAAgB,EAAE,CAAC;YACnB,eAAe,EAAE,CAAC;YAClB,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,CAAC;QACD,KAAK,SAAS,EAAE,CAAC;YAChB,MAAM,UAAU,GACf,MAAM,IAAI,OAAO,CAAC,OAAO;gBACzB,CAAC,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,uBAAuB,CAAC,CAAC,QAAQ,CACjF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAC5B,CAAC;YACH,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,UAAU,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC7E,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,aAAa,GAAG,OAAO,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS;gBACtC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;gBACvC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC3F,eAAe,EAAE,CAAC;YAClB,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,CAAC;QACD,KAAK,SAAS;YACb,IAAI,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC;gBAAE,OAAO,EAAE,kBAAkB,EAAE,EAAE,CAAC;YACnE,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YACnD,OAAO;QACR,KAAK,WAAW;YACf,OAAO,EAAE,gCAAgC,EAAE,EAAE,CAAC;YAC9C,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YACrD,OAAO;QACR,KAAK,OAAO;YACX,OAAO,GAAG,IAAI,CAAC;YACf,WAAW,EAAE,EAAE,CAAC;YAChB,eAAe,EAAE,EAAE,CAAC;YACpB,MAAM,OAAO,EAAE,OAAO,EAAE,CAAC;YACzB,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,KAAK,CAAC;gBACtB,MAAM,oBAAoB,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;oBACpD,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;oBACvC,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;oBAC7C,MAAM,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBACjC,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC7B,CAAC,CAAC,CAAC;YACJ,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACF,CAAC;AAED,IAAI,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAA4B,EAAE,EAAE;IACnD,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QAC7C,MAAM,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACrE,IAAI,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;QACD,IAAI,SAAS,IAAI,OAAO;YAAE,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;aAC/E,CAAC;YACL,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport { existsSync, realpathSync } from \"node:fs\";\nimport { basename, dirname, join, resolve } from \"node:path\";\nimport { parentPort } from \"node:worker_threads\";\nimport { runWithProviderScope } from \"@earendil-works/pi-ai/node/provider-scope\";\nimport { WAKE_SOURCE_STATE_EVENT } from \"../../core/extensions/builtin/monitor-state-event.ts\";\nimport { takeOverStdout } from \"../../core/output-guard.ts\";\nimport { getDefaultSessionDir } from \"../../core/session-manager.ts\";\nimport { installSessionWriteReservation } from \"../../core/session-write-reservation.ts\";\nimport { SettingsManager } from \"../../core/settings-manager.ts\";\nimport { createCliRuntimeFactory } from \"../../main.ts\";\nimport { initTheme } from \"../interactive/theme/theme.ts\";\nimport { buildRpcSessionState } from \"./connection-handler.ts\";\nimport { createRpcSessionBinding, type RpcSessionBinding } from \"./session-binding.ts\";\nimport { SessionEventWriter } from \"./session-event-writer.ts\";\nimport { type RpcSessionEntry, RpcSessionRegistry } from \"./session-registry.ts\";\nimport {\n\ttype HostToSessionWorker,\n\tSESSION_WORKER_LIMITS,\n\ttype SessionWorkerToHost,\n\ttype WorkerDisplay,\n\ttype WorkerSnapshot,\n} from \"./session-worker-protocol.ts\";\n\ntakeOverStdout();\nconst port = parentPort;\nif (!port) throw new Error(\"Session worker requires a parent port\");\nconst send = (message: SessionWorkerToHost): void => port.postMessage(message);\n\nfunction failWorker(error: string): never {\n\tsend({ type: \"failure\", error });\n\tprocess.exit(1);\n}\n\nfunction canonicalPath(path: string): string {\n\tconst absolute = resolve(path);\n\treturn existsSync(absolute) ? realpathSync(absolute) : join(realpathSync(dirname(absolute)), basename(absolute));\n}\n\nfunction exchange(\n\tmessage: (signal: SharedArrayBuffer) => SessionWorkerToHost,\n\tdeniedError = \"session_worker_output_denied\",\n\tsignal = new SharedArrayBuffer(4),\n): void {\n\tconst state = new Int32Array(signal);\n\tsend(message(signal));\n\tAtomics.wait(state, 0, 0, SESSION_WORKER_LIMITS.controlMs);\n\tconst result = Atomics.load(state, 0);\n\tif (result === 2) {\n\t\tif (deniedError === \"session_path_in_use\") throw new Error(deniedError);\n\t\tfailWorker(deniedError);\n\t}\n\tif (result !== 1) failWorker(\"session_worker_credit_timeout\");\n}\n\ninstallSessionWriteReservation((path) =>\n\texchange((signal) => ({ type: \"reserve\", path: canonicalPath(path), signal }), \"session_path_in_use\"),\n);\n\nclass WorkerEventWriter extends SessionEventWriter {\n\tconstructor() {\n\t\tsuper(() => {});\n\t}\n\toverride enqueue(_sessionId: string, record: object): boolean {\n\t\tif (Buffer.byteLength(JSON.stringify(record)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\t\tfailWorker(\"session_worker_output_limit\");\n\t\tconst session = entry?.runtime?.session;\n\t\tif (!session) throw new Error(\"Session output preceded runtime creation\");\n\t\tconst activity = { busy: session.isSessionBusy, streaming: session.isStreaming };\n\t\tconst replacement =\n\t\t\t\"type\" in record &&\n\t\t\t(record.type === \"session_replaced\" ||\n\t\t\t\trecord.type === \"question_resolved\" ||\n\t\t\t\trecord.type === \"question_updated\" ||\n\t\t\t\t(record.type === \"extension_ui_request\" && \"method\" in record && record.method === \"question\"))\n\t\t\t\t? snapshot()\n\t\t\t\t: undefined;\n\t\tif (replacement && Buffer.byteLength(JSON.stringify(replacement)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\t\tfailWorker(\"session_worker_snapshot_limit\");\n\t\texchange((signal) => ({\n\t\t\ttype: \"output\",\n\t\t\trecord,\n\t\t\tconnection: this.currentConnection(),\n\t\t\tsignal,\n\t\t\tactivity,\n\t\t\tsnapshot: replacement,\n\t\t}));\n\t\treturn true;\n\t}\n}\n\nconst writer = new WorkerEventWriter();\nlet prepared: Extract<HostToSessionWorker, { type: \"prepare\" }> | undefined;\nlet registry: RpcSessionRegistry | undefined;\nlet entry: RpcSessionEntry | undefined;\nlet binding: RpcSessionBinding | undefined;\nlet display: WorkerDisplay = { revision: 0, width: 80, rendered: false, capabilities: [] };\nlet closing = false;\nlet unsubscribe: (() => void) | undefined;\nlet unsubscribeWake: (() => void) | undefined;\n\nfunction applyDisplay(next: WorkerDisplay): boolean {\n\tif (next.revision < display.revision) return false;\n\tdisplay = next;\n\treturn true;\n}\n\nfunction updateDisplay(message: (signal: SharedArrayBuffer) => SessionWorkerToHost): void {\n\tconst signal = new SharedArrayBuffer(24);\n\texchange(message, \"session_worker_display_denied\", signal);\n\tconst values = new Float64Array(signal);\n\tapplyDisplay({\n\t\t...display,\n\t\twidth: values[1],\n\t\trevision: values[2],\n\t\trendered: Atomics.load(new Int32Array(signal), 1) === 1,\n\t});\n}\n\nfunction publishSnapshot(settled = false): void {\n\tconst value = snapshot();\n\tif (Buffer.byteLength(JSON.stringify(value)) > SESSION_WORKER_LIMITS.outputBytes)\n\t\tfailWorker(\"session_worker_snapshot_limit\");\n\texchange((signal) => ({ type: \"snapshot\", snapshot: value, signal, settled }));\n}\n\nfunction subscribeSession(): void {\n\tunsubscribe?.();\n\tunsubscribeWake?.();\n\tunsubscribeWake = entry?.runtime?.session.extensionRunner.onBusEvent(WAKE_SOURCE_STATE_EVENT, () =>\n\t\tpublishSnapshot(),\n\t);\n\tunsubscribe = entry?.runtime?.session.subscribe((event) =>\n\t\tpublishSnapshot(event.type === \"agent_settled\" || event.type === \"agent_idle\"),\n\t);\n}\n\nfunction snapshot(): WorkerSnapshot {\n\tif (!entry?.runtime) throw new Error(\"Session runtime is not ready\");\n\tconst session = entry.runtime.session;\n\treturn {\n\t\tstate: buildRpcSessionState(session),\n\t\tsessionPath: session.sessionFile ? canonicalPath(session.sessionFile) : undefined,\n\t\tbusy: session.isSessionBusy,\n\t\tstreaming: session.isStreaming,\n\t};\n}\n\nasync function handle(message: HostToSessionWorker): Promise<void> {\n\tswitch (message.type) {\n\t\tcase \"prepare\": {\n\t\t\tif (prepared) throw new Error(\"Session worker already prepared\");\n\t\t\tconst path =\n\t\t\t\tmessage.profile.sessionPath ??\n\t\t\t\tjoin(\n\t\t\t\t\tgetDefaultSessionDir(message.profile.cwd, message.configuration.agentDir),\n\t\t\t\t\t`${new Date().toISOString().replace(/[:.]/g, \"-\")}_${randomUUID()}.jsonl`,\n\t\t\t\t);\n\t\t\tprepared = { ...message, profile: { ...message.profile, sessionPath: canonicalPath(path) } };\n\t\t\tsend({ type: \"prepared\", request: message.request, sessionPath: canonicalPath(path) });\n\t\t\treturn;\n\t\t}\n\t\tcase \"commit\": {\n\t\t\tif (!prepared || registry) throw new Error(\"Invalid session commit\");\n\t\t\tconst startupSettingsManager = SettingsManager.create(\n\t\t\t\tprepared.configuration.cwd,\n\t\t\t\tprepared.configuration.agentDir,\n\t\t\t);\n\t\t\tconst factory = createCliRuntimeFactory(prepared.configuration, { startupSettingsManager });\n\t\t\tinitTheme(startupSettingsManager.getTheme(), false);\n\t\t\tregistry = new RpcSessionRegistry({ agentDir: prepared.configuration.agentDir, createRuntime: factory });\n\t\t\tconst opened = await registry.openSession(prepared.profile);\n\t\t\tentry = registry.getForCommand(opened.sessionId, \"open_session\");\n\t\t\tsend({ type: \"ready\", request: message.request, snapshot: snapshot() });\n\t\t\treturn;\n\t\t}\n\t\tcase \"bind\": {\n\t\t\tif (!entry || binding) throw new Error(\"Invalid session binding\");\n\t\t\tdisplay = message.display;\n\t\t\tconst bindingEntry = entry;\n\t\t\tconst createBinding = () =>\n\t\t\t\tcreateRpcSessionBinding(message.sessionId, bindingEntry, writer, () => send({ type: \"request_close\" }), {\n\t\t\t\t\tcapabilities: display.capabilities,\n\t\t\t\t\tsharedWidth: {\n\t\t\t\t\t\tgetWidth: () => display.width,\n\t\t\t\t\t\tsetWidth: (connection, width) =>\n\t\t\t\t\t\t\tupdateDisplay((signal) => ({ type: \"width\", connection, width, signal })),\n\t\t\t\t\t\tonChange: () => binding?.rerenderComponents?.(),\n\t\t\t\t\t\tclearWidth: () => {},\n\t\t\t\t\t\tconnectionId: () => writer.currentConnection(),\n\t\t\t\t\t\thasRenderedComponents: () => display.rendered,\n\t\t\t\t\t\tsetCapabilities: (connection, capabilities) => {\n\t\t\t\t\t\t\tupdateDisplay((signal) => ({ type: \"capabilities\", connection, capabilities, signal }));\n\t\t\t\t\t\t\tbinding?.rerenderComponents?.();\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\tbinding = await (message.connection === undefined\n\t\t\t\t? createBinding()\n\t\t\t\t: writer.withConnection(message.connection, createBinding));\n\t\t\tconst rebind = entry.rebindSession;\n\t\t\tentry.rebindSession = async (session) => {\n\t\t\t\tawait rebind?.(session);\n\t\t\t\tsubscribeSession();\n\t\t\t};\n\t\t\tentry.runtime?.setRebindSession(entry.rebindSession);\n\t\t\tsubscribeSession();\n\t\t\tpublishSnapshot();\n\t\t\tsend({ type: \"result\", request: message.request });\n\t\t\treturn;\n\t\t}\n\t\tcase \"command\": {\n\t\t\tconst privileged =\n\t\t\t\t\"type\" in message.command &&\n\t\t\t\t[\"abort\", \"abort_bash\", \"extension_ui_response\", \"extension_ui_progress\"].includes(\n\t\t\t\t\tString(message.command.type),\n\t\t\t\t);\n\t\t\tif (!binding || (closing && !privileged)) throw new Error(\"session_closing\");\n\t\t\tapplyDisplay(message.display);\n\t\t\tconst activeBinding = binding;\n\t\t\tawait (message.connection === undefined\n\t\t\t\t? activeBinding.handle(message.command)\n\t\t\t\t: writer.withConnection(message.connection, () => activeBinding.handle(message.command)));\n\t\t\tpublishSnapshot();\n\t\t\tsend({ type: \"result\", request: message.request });\n\t\t\treturn;\n\t\t}\n\t\tcase \"display\":\n\t\t\tif (applyDisplay(message.display)) binding?.rerenderComponents?.();\n\t\t\tsend({ type: \"control_done\", control: \"display\" });\n\t\t\treturn;\n\t\tcase \"cancel_ui\":\n\t\t\tbinding?.cancelPendingExtensionUiRequests?.();\n\t\t\tsend({ type: \"control_done\", control: \"cancel_ui\" });\n\t\t\treturn;\n\t\tcase \"close\":\n\t\t\tclosing = true;\n\t\t\tunsubscribe?.();\n\t\t\tunsubscribeWake?.();\n\t\t\tawait binding?.dispose();\n\t\t\tif (entry?.runtime) {\n\t\t\t\tconst current = entry;\n\t\t\t\tawait runWithProviderScope(current.scope, async () => {\n\t\t\t\t\tawait current.runtime?.session.abort();\n\t\t\t\t\tawait current.runtime?.session.waitForIdle();\n\t\t\t\t\tawait current.runtime?.dispose();\n\t\t\t\t\tawait current.scope.close();\n\t\t\t\t});\n\t\t\t}\n\t\t\tprocess.exit(0);\n\t}\n}\n\nport.on(\"message\", (message: HostToSessionWorker) => {\n\tvoid handle(message).catch((cause: unknown) => {\n\t\tconst error = cause instanceof Error ? cause.message : String(cause);\n\t\tif (error.startsWith(\"session_worker_\")) {\n\t\t\tsend({ type: \"failure\", error });\n\t\t\tprocess.exit(1);\n\t\t}\n\t\tif (\"request\" in message) send({ type: \"result\", request: message.request, error });\n\t\telse {\n\t\t\tsend({ type: \"failure\", error });\n\t\t\tprocess.exit(1);\n\t\t}\n\t});\n});\n"]}
|
|
@@ -31,19 +31,7 @@ export declare class WorkerSessionRegistry {
|
|
|
31
31
|
/** Only compare spellings already tied to a granted identity; do not inspect caller paths here. */
|
|
32
32
|
private knownReservationKey;
|
|
33
33
|
private attach;
|
|
34
|
-
/** Granted paths currently held by a handle; the per-worker budget is bounded by it. */
|
|
35
|
-
reservationCount(handle: string): number;
|
|
36
34
|
private reserve;
|
|
37
|
-
/**
|
|
38
|
-
* Releases the grants a worker's latest snapshot no longer claims.
|
|
39
|
-
*
|
|
40
|
-
* Only a fully open entry reports live writers. An entry still opening, closing or
|
|
41
|
-
* quarantined keeps every grant until its real exit, because a worker stuck in a
|
|
42
|
-
* syscall can still be writing a path it can no longer report.
|
|
43
|
-
*/
|
|
44
|
-
private reconcile;
|
|
45
|
-
/** The live view a full budget is reconciled against; absent while the entry cannot report one. */
|
|
46
|
-
private liveWorkerPaths;
|
|
47
35
|
private openResult;
|
|
48
36
|
}
|
|
49
37
|
//# sourceMappingURL=worker-session-registry.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"worker-session-registry.d.ts","sourceRoot":"","sources":["../../../src/modes/rpc/worker-session-registry.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;
|
|
1
|
+
{"version":3,"file":"worker-session-registry.d.ts","sourceRoot":"","sources":["../../../src/modes/rpc/worker-session-registry.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EACN,KAAK,cAAc,EACnB,KAAK,eAAe,EACpB,KAAK,uBAAuB,EAE5B,MAAM,uBAAuB,CAAC;AAI/B,2FAA2F;AAC3F,qBAAa,qBAAqB;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAsC;IAC9D,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA6B;IAC1D,OAAO,CAAC,MAAM,CAAK;IACnB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAe;IAEnC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAsF;IAE9G,YAAY,OAAO,EAAE;QAAE,aAAa,EAAE,uBAAuB,CAAC;QAAC,YAAY,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,MAAM,CAAA;KAAE,EAIvG;IAED,IAAI,IAAI,IAAI,MAAM,CAEjB;IAEK,WAAW,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC,cAAc,CAAC,CA0D3E;IAED,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS,CAEhD;IAED,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,eAAe,CAY9D;IAED,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,KAAK,IAAI,GAAG,eAAe,CAgBjF;IAED,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAGnC;IAEK,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAiB/C;IAED,IAAI,IAAI,KAAK,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,GAAG,EAAE,MAAM,CAAC;QACZ,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,CAAC;KACzD,CAAC,CAYD;IAED,mGAAmG;IACnG,OAAO,CAAC,mBAAmB;IAU3B,OAAO,CAAC,MAAM;IAUd,OAAO,CAAC,OAAO;IAaf,OAAO,CAAC,UAAU;CAKlB"}
|
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
import { isAbsolute } from "node:path";
|
|
2
2
|
import { ProviderScope } from "@earendil-works/pi-ai/node/provider-scope";
|
|
3
|
-
import { RESERVATION_DENIAL_CODES, SessionPathReservations, } from "./session-path-reservations.js";
|
|
4
3
|
import { RpcSessionRegistryError, } from "./session-registry.js";
|
|
5
4
|
import { SessionWorkerClient } from "./session-worker-client.js";
|
|
6
5
|
import { SESSION_WORKER_LIMITS } from "./session-worker-protocol.js";
|
|
@@ -8,7 +7,7 @@ import { SESSION_WORKER_LIMITS } from "./session-worker-protocol.js";
|
|
|
8
7
|
export class WorkerSessionRegistry {
|
|
9
8
|
constructor(options) {
|
|
10
9
|
this.entries = new Map();
|
|
11
|
-
this.reservations = new
|
|
10
|
+
this.reservations = new Map();
|
|
12
11
|
this.serial = 0;
|
|
13
12
|
this.options = options;
|
|
14
13
|
this.closeGraceMs = options.closeGraceMs;
|
|
@@ -22,7 +21,7 @@ export class WorkerSessionRegistry {
|
|
|
22
21
|
throw new RpcSessionRegistryError("invalid_path");
|
|
23
22
|
if (profile.sessionPath) {
|
|
24
23
|
const key = this.knownReservationKey(profile.sessionPath);
|
|
25
|
-
const owner = key ? this.reservations.
|
|
24
|
+
const owner = key ? this.reservations.get(key) : undefined;
|
|
26
25
|
if (key && owner)
|
|
27
26
|
return this.attach(owner, key);
|
|
28
27
|
}
|
|
@@ -40,13 +39,14 @@ export class WorkerSessionRegistry {
|
|
|
40
39
|
};
|
|
41
40
|
const worker = new SessionWorkerClient({
|
|
42
41
|
reserve: (path) => this.reserve(handle, path),
|
|
43
|
-
reconcile: (livePaths) => this.reconcile(handle, livePaths),
|
|
44
42
|
exit: () => {
|
|
45
43
|
if (this.entries.get(handle) !== entry)
|
|
46
44
|
return;
|
|
47
45
|
entry.state = "closed";
|
|
48
46
|
this.entries.delete(handle);
|
|
49
|
-
this.reservations
|
|
47
|
+
for (const [path, owner] of this.reservations)
|
|
48
|
+
if (owner === handle)
|
|
49
|
+
this.reservations.delete(path);
|
|
50
50
|
entry.closeResolve?.();
|
|
51
51
|
},
|
|
52
52
|
failure: (error) => {
|
|
@@ -58,18 +58,16 @@ export class WorkerSessionRegistry {
|
|
|
58
58
|
this.entries.set(handle, entry);
|
|
59
59
|
try {
|
|
60
60
|
const path = await worker.prepare(this.options.configuration, profile);
|
|
61
|
-
const owner = this.reservations.
|
|
61
|
+
const owner = this.reservations.get(path);
|
|
62
62
|
if (owner) {
|
|
63
63
|
const attached = this.attach(owner, path);
|
|
64
64
|
entry.state = "quarantined";
|
|
65
65
|
worker.quarantine();
|
|
66
66
|
return attached;
|
|
67
67
|
}
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
throw new RpcSessionRegistryError(RESERVATION_DENIAL_CODES[grant]);
|
|
68
|
+
if (!this.reserve(handle, path))
|
|
69
|
+
throw new RpcSessionRegistryError("session_path_in_use");
|
|
71
70
|
entry.reservationKey = path;
|
|
72
|
-
entry.requestedPathKey = profile.sessionPath ? path : undefined;
|
|
73
71
|
entry.sessionPath = path;
|
|
74
72
|
const snapshot = await worker.commit();
|
|
75
73
|
if (entry.state !== "opening")
|
|
@@ -162,15 +160,11 @@ export class WorkerSessionRegistry {
|
|
|
162
160
|
}
|
|
163
161
|
/** Only compare spellings already tied to a granted identity; do not inspect caller paths here. */
|
|
164
162
|
knownReservationKey(path) {
|
|
165
|
-
if (this.reservations.
|
|
163
|
+
if (this.reservations.has(path))
|
|
166
164
|
return path;
|
|
167
165
|
for (const entry of this.entries.values()) {
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
if (entry.profile.sessionPath === path &&
|
|
171
|
-
entry.requestedPathKey &&
|
|
172
|
-
this.reservations.owner(entry.requestedPathKey))
|
|
173
|
-
return entry.requestedPathKey;
|
|
166
|
+
if (entry.profile.sessionPath === path && entry.reservationKey)
|
|
167
|
+
return entry.reservationKey;
|
|
174
168
|
const snapshot = entry.worker?.snapshot;
|
|
175
169
|
if (snapshot?.state.sessionFile === path)
|
|
176
170
|
return snapshot.sessionPath;
|
|
@@ -186,44 +180,23 @@ export class WorkerSessionRegistry {
|
|
|
186
180
|
entry.lastCommandAt = this.now();
|
|
187
181
|
return { ...result, attached: true };
|
|
188
182
|
}
|
|
189
|
-
/** Granted paths currently held by a handle; the per-worker budget is bounded by it. */
|
|
190
|
-
reservationCount(handle) {
|
|
191
|
-
return this.reservations.count(handle);
|
|
192
|
-
}
|
|
193
183
|
reserve(handle, path) {
|
|
194
184
|
const entry = this.entries.get(handle);
|
|
195
185
|
if (!entry)
|
|
196
|
-
return
|
|
197
|
-
|
|
198
|
-
|
|
186
|
+
return false;
|
|
187
|
+
const owner = this.reservations.get(path);
|
|
188
|
+
if (owner)
|
|
189
|
+
return owner === handle;
|
|
199
190
|
if (entry.state !== "opening" && entry.state !== "open")
|
|
200
|
-
return
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
*/
|
|
210
|
-
reconcile(handle, livePaths) {
|
|
211
|
-
const entry = this.entries.get(handle);
|
|
212
|
-
if (entry?.state !== "open")
|
|
213
|
-
return;
|
|
214
|
-
const sessionPath = entry.worker?.snapshot?.sessionPath;
|
|
215
|
-
this.reservations.reconcile(handle, { livePaths, sessionPath });
|
|
216
|
-
if (!sessionPath)
|
|
217
|
-
return;
|
|
218
|
-
entry.reservationKey = sessionPath;
|
|
219
|
-
entry.sessionPath = sessionPath;
|
|
220
|
-
}
|
|
221
|
-
/** The live view a full budget is reconciled against; absent while the entry cannot report one. */
|
|
222
|
-
liveWorkerPaths(entry) {
|
|
223
|
-
const snapshot = entry.state === "open" ? entry.worker?.snapshot : undefined;
|
|
224
|
-
if (!snapshot)
|
|
225
|
-
return undefined;
|
|
226
|
-
return { livePaths: snapshot.liveSessionPaths, sessionPath: snapshot.sessionPath };
|
|
191
|
+
return false;
|
|
192
|
+
let count = 0;
|
|
193
|
+
for (const current of this.reservations.values())
|
|
194
|
+
if (current === handle)
|
|
195
|
+
count++;
|
|
196
|
+
if (count >= SESSION_WORKER_LIMITS.reservations)
|
|
197
|
+
return false;
|
|
198
|
+
this.reservations.set(path, handle);
|
|
199
|
+
return true;
|
|
227
200
|
}
|
|
228
201
|
openResult(handle, entry) {
|
|
229
202
|
const state = entry.worker?.snapshot?.state;
|