@walkhi/code-relax 0.1.0-beta.23 → 0.1.0-beta.25

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.
@@ -1,204 +1,208 @@
1
- import { randomUUID } from 'node:crypto';
2
- import { EventEmitter } from 'node:events';
3
- import fs from 'node:fs';
4
- import WebSocket from 'ws';
5
- import { decode, loopbackUrl, relayAddress, send } from './wire.mjs';
1
+ import { randomUUID } from 'node:crypto';
2
+ import { EventEmitter } from 'node:events';
3
+ import fs from 'node:fs';
4
+ import WebSocket from 'ws';
5
+ import { decode, loopbackUrl, relayAddress, send } from './wire.mjs';
6
6
  import { createP2pProbe, validProbeSignal } from './p2p-probe.mjs';
7
7
  import { createWindowsRouteHelper } from './windows-route-helper.mjs';
8
8
 
9
9
  // Reuse the actual LAN dispatcher, including its normal/shared capability boundary.
10
10
  // The LAN token never travels to the relay. No requests are replayed after a close.
11
- export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken, identity, bridgeConnection, usageCacheFile,
12
- hostName = 'Codex-PC', lanCandidates = [], p2pProbe = false, iceServers = [], p2pWindowsRouteHelper = false, p2pWindowsRouteHelperToken = '' }) {
11
+ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken, identity, bridgeConnection, usageCacheFile,
12
+ hostName = 'Codex-PC', lanCandidates = [], p2pProbe = false, iceServers = [], p2pWindowsRouteHelper = false, p2pWindowsRouteHelperToken = '' }) {
13
13
  relayAddress(relayUrl); loopbackUrl(bridgeUrl, '/api/socket');
14
- if (!bridgeToken) throw new Error('Desktop service credential required');
14
+ if (!bridgeToken) throw new Error('Desktop service credential required');
15
15
  const events = new EventEmitter();
16
16
  const remote = new WebSocket(relayUrl, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: {
17
17
  serverNoContextTakeover: true, clientNoContextTakeover: true,
18
18
  threshold: 1024, zlibDeflateOptions: { level: 3 },
19
19
  } });
20
- const locals = new Map();
21
- const fileLocals = new Map();
22
- const activeTurns = new Map();
23
- const completedTurns = new Map();
24
- const trackedTurns = new Map();
25
- if (usageCacheFile) {
26
- try {
27
- const saved = JSON.parse(fs.readFileSync(usageCacheFile, 'utf8'));
28
- if (saved.version === 1 && Array.isArray(saved.turns)) {
29
- for (const item of saved.turns.slice(0, 128)) {
30
- if (typeof item.threadId === 'string' && item.threadId.length <= 80 &&
31
- typeof item.turnId === 'string' && item.turnId.length <= 80) trackedTurns.set(item.threadId, item.turnId);
32
- }
33
- }
34
- } catch (error) { if (error.code !== 'ENOENT') events.emit('diagnostic', `Usage task cache: ${error.message}`); }
35
- }
36
- function saveTrackedTurns() {
37
- if (!usageCacheFile) return;
38
- try {
39
- const temp = `${usageCacheFile}.tmp`;
40
- fs.writeFileSync(temp, JSON.stringify({ version: 1, turns: [...trackedTurns].map(([threadId, turnId]) => ({ threadId, turnId })) }), { mode: 0o600 });
41
- fs.renameSync(temp, usageCacheFile);
42
- } catch (error) { events.emit('diagnostic', `Usage task cache: ${error.message}`); }
43
- }
44
- let fileRemote, fileRetryTimer, turnLocal, turnRetryTimer, turnPulseTimer, turnObserverReady = false,
45
- queueTurnResync = () => {}, verifyActiveTurns = () => {}, heartbeat, operation, lastClose, closing = false;
20
+ const locals = new Map();
21
+ const fileLocals = new Map();
22
+ const activeTurns = new Map();
23
+ const completedTurns = new Map();
24
+ const trackedTurns = new Map();
25
+ if (usageCacheFile) {
26
+ try {
27
+ const saved = JSON.parse(fs.readFileSync(usageCacheFile, 'utf8'));
28
+ if (saved.version === 1 && Array.isArray(saved.turns)) {
29
+ for (const item of saved.turns.slice(0, 128)) {
30
+ if (typeof item.threadId === 'string' && item.threadId.length <= 80 &&
31
+ typeof item.turnId === 'string' && item.turnId.length <= 80) trackedTurns.set(item.threadId, item.turnId);
32
+ }
33
+ }
34
+ } catch (error) { if (error.code !== 'ENOENT') events.emit('diagnostic', `Usage task cache: ${error.message}`); }
35
+ }
36
+ function saveTrackedTurns() {
37
+ if (!usageCacheFile) return;
38
+ try {
39
+ const temp = `${usageCacheFile}.tmp`;
40
+ fs.writeFileSync(temp, JSON.stringify({ version: 1, turns: [...trackedTurns].map(([threadId, turnId]) => ({ threadId, turnId })) }), { mode: 0o600 });
41
+ fs.renameSync(temp, usageCacheFile);
42
+ } catch (error) { events.emit('diagnostic', `Usage task cache: ${error.message}`); }
43
+ }
44
+ let fileRemote, fileRetryTimer, turnLocal, turnRetryTimer, turnPulseTimer, turnObserverReady = false,
45
+ queueTurnResync = () => {}, verifyActiveTurns = () => {}, heartbeat, operation, lastClose, closing = false;
46
46
  const alive = () => { clearTimeout(heartbeat); heartbeat = setTimeout(() => remote.terminate(), 75000); };
47
47
  remote.on('open', alive); remote.on('ping', alive); remote.on('message', alive);
48
- function dropLocal(session) {
49
- const current = locals.get(session);
50
- if (!current) return;
51
- locals.delete(session); clearTimeout(current.timer);
52
- if (current.ready && current.socket?.readyState === WebSocket.OPEN && current.usageRequests.size &&
53
- remote.readyState === WebSocket.OPEN && !closing) {
54
- // Keep only the in-flight submission receipt; no late response goes to the phone.
55
- current.detached = true;
56
- current.detachedTimer = setTimeout(() => current.socket?.terminate(), 35000);
57
- } else current.socket?.terminate();
58
- current.probe?.then(probe => probe.close());
59
- current.diagnosticProbe?.then(probe => probe.close());
48
+ function dropLocal(session) {
49
+ const current = locals.get(session);
50
+ if (!current) return;
51
+ locals.delete(session); clearTimeout(current.timer);
52
+ if (current.ready && current.socket?.readyState === WebSocket.OPEN && current.usageRequests.size &&
53
+ remote.readyState === WebSocket.OPEN && !closing) {
54
+ // Keep only the in-flight submission receipt; no late response goes to the phone.
55
+ current.detached = true;
56
+ current.detachedTimer = setTimeout(() => current.socket?.terminate(), 35000);
57
+ } else current.socket?.terminate();
58
+ current.probe?.then(probe => probe.close());
59
+ current.diagnosticProbe?.then(probe => probe.close());
60
+ }
61
+ function dropLocals() {
62
+ for (const session of [...locals.keys()]) dropLocal(session);
63
+ for (const socket of fileLocals.values()) socket.terminate();
64
+ fileLocals.clear(); clearTimeout(fileRetryTimer); fileRemote?.terminate(); fileRemote = undefined;
65
+ clearTimeout(turnRetryTimer); clearInterval(turnPulseTimer); turnLocal?.terminate(); turnLocal = undefined;
66
+ turnObserverReady = false; activeTurns.clear(); completedTurns.clear(); verifyActiveTurns = () => {};
67
+ }
68
+ function startTurnReceiver() {
69
+ if (closing || remote.readyState !== WebSocket.OPEN) return;
70
+ let connection;
71
+ try { connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken }; loopbackUrl(connection.url, '/api/socket'); }
72
+ catch { turnRetryTimer = setTimeout(startTurnReceiver, 3000); return; }
73
+ const socket = turnLocal = new WebSocket(connection.url, { maxPayload: 64 * 1024, perMessageDeflate: false });
74
+ const resyncPending = new Map();
75
+ const pulsePending = new Map();
76
+ let resyncId = 1, resyncQueue = [];
77
+ function resyncNext() {
78
+ if (!turnObserverReady || turnLocal !== socket) return;
79
+ while (resyncPending.size < 8 && resyncQueue.length) {
80
+ const [threadId, turnId] = resyncQueue.shift();
81
+ if (trackedTurns.get(threadId) !== turnId) continue;
82
+ const id = ++resyncId;
83
+ resyncPending.set(id, { threadId, turnId });
84
+ send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
85
+ }
86
+ }
87
+ queueTurnResync = (threadId, turnId) => { resyncQueue.push([threadId, turnId]); resyncNext(); };
88
+ verifyActiveTurns = () => {
89
+ if (!turnObserverReady || turnLocal !== socket) return;
90
+ for (const [threadId, turnId] of trackedTurns) {
91
+ if (activeTurns.get(threadId) !== turnId) continue;
92
+ if (pulsePending.size >= 8) break;
93
+ if ([...pulsePending.values()].some(pending => pending.threadId === threadId)) continue;
94
+ const id = ++resyncId;
95
+ pulsePending.set(id, { threadId, turnId });
96
+ send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
97
+ }
98
+ };
99
+ socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
100
+ socket.on('error', error => events.emit('diagnostic', `Local turn observer: ${error.message}`));
101
+ socket.on('close', () => {
102
+ if (turnLocal !== socket) return;
103
+ turnLocal = undefined;
104
+ if (turnObserverReady) send(remote, { type: 'usage-observer-lost' });
105
+ turnObserverReady = false;
106
+ queueTurnResync = () => {};
107
+ verifyActiveTurns = () => {};
108
+ activeTurns.clear(); completedTurns.clear();
109
+ if (!closing && remote.readyState === WebSocket.OPEN) turnRetryTimer = setTimeout(startTurnReceiver, 3000);
110
+ });
111
+ socket.on('message', (bytes, binary) => {
112
+ if (binary || turnLocal !== socket) return socket.terminate();
113
+ try {
114
+ const message = decode(bytes, false);
115
+ if (message.type === 'ready') { send(socket, { type: 'subscribe', id: 1, path: '/api/turn-events' }); return; }
116
+ if (message.type === 'subscribed' && message.id === 1) {
117
+ turnObserverReady = true; resyncQueue = [...trackedTurns]; resyncNext(); return;
118
+ }
119
+ if (message.type === 'response' && pulsePending.has(message.id)) {
120
+ const { threadId, turnId } = pulsePending.get(message.id);
121
+ pulsePending.delete(message.id);
122
+ if (trackedTurns.get(threadId) !== turnId || activeTurns.get(threadId) !== turnId) return;
123
+ if (message.status === 200 && message.body?.turnId === turnId) {
124
+ send(remote, { type: 'usage-turn', method: 'heartbeat', threadId, turnId });
125
+ } else if (message.status === 200 || message.status === 404) {
126
+ activeTurns.delete(threadId);
127
+ if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
128
+ send(remote, { type: 'usage-turn', method: 'inactive', threadId, turnId });
129
+ } else socket.close();
130
+ return;
131
+ }
132
+ if (message.type === 'response' && resyncPending.has(message.id)) {
133
+ const { threadId, turnId } = resyncPending.get(message.id);
134
+ resyncPending.delete(message.id);
135
+ if (trackedTurns.get(threadId) === turnId) {
136
+ if (message.status === 200 && message.body?.turnId === turnId) {
137
+ activeTurns.set(threadId, turnId);
138
+ send(remote, { type: 'usage-turn', method: 'resume', threadId, turnId });
139
+ } else if (message.status === 200 || message.status === 404) {
140
+ trackedTurns.delete(threadId); saveTrackedTurns();
141
+ } else { socket.close(); return; }
142
+ }
143
+ resyncNext(); return;
144
+ }
145
+ if (message.type === 'event' && message.id === 1 && message.event === 'turn') {
146
+ const { method, threadId, turnId } = message.data || {};
147
+ if (!['turn/started', 'turn/completed'].includes(method) || typeof threadId !== 'string' || typeof turnId !== 'string') return;
148
+ const key = `${threadId}:${turnId}`;
149
+ const tracked = trackedTurns.get(threadId) === turnId;
150
+ // Keep local event state for a phone submission receipt arriving later.
151
+ // Only registered usage turns need cloud updates or periodic verification.
152
+ if (method === 'turn/started') activeTurns.set(threadId, turnId);
153
+ else {
154
+ if (activeTurns.get(threadId) === turnId) activeTurns.delete(threadId);
155
+ if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
156
+ completedTurns.set(key, Date.now());
157
+ if (completedTurns.size > 100) completedTurns.delete(completedTurns.keys().next().value);
158
+ }
159
+ if (tracked) send(remote, { type: 'usage-turn', method, threadId, turnId });
160
+ } else if (message.type === 'end' || (message.type === 'response' && message.status >= 400)) socket.close();
161
+ } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
162
+ });
163
+ }
164
+ function startFileReceiver() {
165
+ if (!identity || closing || remote.readyState !== WebSocket.OPEN) return;
166
+ const receiver = fileRemote = new WebSocket(relayUrl, { maxPayload: 512 * 1024, perMessageDeflate: false });
167
+ receiver.on('open', () => send(receiver, { type: 'file-host', deviceId: identity.id, deviceSecret: identity.secret }));
168
+ receiver.on('error', error => events.emit('diagnostic', `File receiver: ${error.message}`));
169
+ receiver.on('close', () => {
170
+ for (const socket of fileLocals.values()) socket.terminate();
171
+ fileLocals.clear();
172
+ if (fileRemote === receiver) {
173
+ fileRemote = undefined;
174
+ if (!closing && remote.readyState === WebSocket.OPEN) fileRetryTimer = setTimeout(startFileReceiver, 3000);
175
+ }
176
+ });
177
+ receiver.on('message', (bytes, binary) => {
178
+ try {
179
+ const message = decode(bytes, binary);
180
+ if (message.type === 'file-host-ready') return;
181
+ if (message.type === 'file-open') {
182
+ const connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
183
+ const url = new URL(connection.url); url.pathname = '/api/file-transfer';
184
+ loopbackUrl(url.href, '/api/file-transfer');
185
+ const socket = new WebSocket(url.href, { maxPayload: 512 * 1024, perMessageDeflate: false });
186
+ fileLocals.set(message.session, socket);
187
+ socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
188
+ socket.on('error', error => events.emit('diagnostic', `Local file receiver: ${error.message}`));
189
+ socket.on('close', () => {
190
+ if (fileLocals.get(message.session) !== socket) return;
191
+ fileLocals.delete(message.session);
192
+ send(receiver, { type: 'file-close', session: message.session });
193
+ });
194
+ socket.on('message', (data, isBinary) => {
195
+ if (isBinary || fileLocals.get(message.session) !== socket) return socket.terminate();
196
+ send(receiver, { type: 'file-frame', session: message.session, payload: data.toString() });
197
+ });
198
+ } else if (message.type === 'file-frame') {
199
+ const socket = fileLocals.get(message.session);
200
+ if (socket?.readyState === WebSocket.OPEN) socket.send(message.payload);
201
+ } else if (message.type === 'file-close') fileLocals.get(message.session)?.terminate();
202
+ else throw new Error('Invalid file receiver frame');
203
+ } catch (error) { events.emit('diagnostic', error.message); receiver.close(1008, 'Invalid file frame'); }
204
+ });
60
205
  }
61
- function dropLocals() {
62
- for (const session of [...locals.keys()]) dropLocal(session);
63
- for (const socket of fileLocals.values()) socket.terminate();
64
- fileLocals.clear(); clearTimeout(fileRetryTimer); fileRemote?.terminate(); fileRemote = undefined;
65
- clearTimeout(turnRetryTimer); clearInterval(turnPulseTimer); turnLocal?.terminate(); turnLocal = undefined;
66
- turnObserverReady = false; activeTurns.clear(); completedTurns.clear(); verifyActiveTurns = () => {};
67
- }
68
- function startTurnReceiver() {
69
- if (closing || remote.readyState !== WebSocket.OPEN) return;
70
- let connection;
71
- try { connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken }; loopbackUrl(connection.url, '/api/socket'); }
72
- catch { turnRetryTimer = setTimeout(startTurnReceiver, 3000); return; }
73
- const socket = turnLocal = new WebSocket(connection.url, { maxPayload: 64 * 1024, perMessageDeflate: false });
74
- const resyncPending = new Map();
75
- const pulsePending = new Map();
76
- let resyncId = 1, resyncQueue = [];
77
- function resyncNext() {
78
- if (!turnObserverReady || turnLocal !== socket) return;
79
- while (resyncPending.size < 8 && resyncQueue.length) {
80
- const [threadId, turnId] = resyncQueue.shift();
81
- if (trackedTurns.get(threadId) !== turnId) continue;
82
- const id = ++resyncId;
83
- resyncPending.set(id, { threadId, turnId });
84
- send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
85
- }
86
- }
87
- queueTurnResync = (threadId, turnId) => { resyncQueue.push([threadId, turnId]); resyncNext(); };
88
- verifyActiveTurns = () => {
89
- if (!turnObserverReady || turnLocal !== socket) return;
90
- for (const [threadId, turnId] of activeTurns) {
91
- if (pulsePending.size >= 8) break;
92
- if ([...pulsePending.values()].some(pending => pending.threadId === threadId)) continue;
93
- const id = ++resyncId;
94
- pulsePending.set(id, { threadId, turnId });
95
- send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
96
- }
97
- };
98
- socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
99
- socket.on('error', error => events.emit('diagnostic', `Local turn observer: ${error.message}`));
100
- socket.on('close', () => {
101
- if (turnLocal !== socket) return;
102
- turnLocal = undefined;
103
- if (turnObserverReady) send(remote, { type: 'usage-observer-lost' });
104
- turnObserverReady = false;
105
- queueTurnResync = () => {};
106
- verifyActiveTurns = () => {};
107
- activeTurns.clear(); completedTurns.clear();
108
- if (!closing && remote.readyState === WebSocket.OPEN) turnRetryTimer = setTimeout(startTurnReceiver, 3000);
109
- });
110
- socket.on('message', (bytes, binary) => {
111
- if (binary || turnLocal !== socket) return socket.terminate();
112
- try {
113
- const message = decode(bytes, false);
114
- if (message.type === 'ready') { send(socket, { type: 'subscribe', id: 1, path: '/api/turn-events' }); return; }
115
- if (message.type === 'subscribed' && message.id === 1) {
116
- turnObserverReady = true; resyncQueue = [...trackedTurns]; resyncNext(); return;
117
- }
118
- if (message.type === 'response' && pulsePending.has(message.id)) {
119
- const { threadId, turnId } = pulsePending.get(message.id);
120
- pulsePending.delete(message.id);
121
- if (activeTurns.get(threadId) !== turnId) return;
122
- if (message.status === 200 && message.body?.turnId === turnId) {
123
- send(remote, { type: 'usage-turn', method: 'heartbeat', threadId, turnId });
124
- } else if (message.status === 200 || message.status === 404) {
125
- activeTurns.delete(threadId);
126
- if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
127
- send(remote, { type: 'usage-turn', method: 'inactive', threadId, turnId });
128
- } else socket.close();
129
- return;
130
- }
131
- if (message.type === 'response' && resyncPending.has(message.id)) {
132
- const { threadId, turnId } = resyncPending.get(message.id);
133
- resyncPending.delete(message.id);
134
- if (trackedTurns.get(threadId) === turnId) {
135
- if (message.status === 200 && message.body?.turnId === turnId) {
136
- activeTurns.set(threadId, turnId);
137
- send(remote, { type: 'usage-turn', method: 'resume', threadId, turnId });
138
- } else if (message.status === 200 || message.status === 404) {
139
- trackedTurns.delete(threadId); saveTrackedTurns();
140
- } else { socket.close(); return; }
141
- }
142
- resyncNext(); return;
143
- }
144
- if (message.type === 'event' && message.id === 1 && message.event === 'turn') {
145
- const { method, threadId, turnId } = message.data || {};
146
- if (!['turn/started', 'turn/completed'].includes(method) || typeof threadId !== 'string' || typeof turnId !== 'string') return;
147
- const key = `${threadId}:${turnId}`;
148
- if (method === 'turn/started') activeTurns.set(threadId, turnId);
149
- else {
150
- if (activeTurns.get(threadId) === turnId) activeTurns.delete(threadId);
151
- if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
152
- completedTurns.set(key, Date.now());
153
- if (completedTurns.size > 100) completedTurns.delete(completedTurns.keys().next().value);
154
- }
155
- send(remote, { type: 'usage-turn', method, threadId, turnId });
156
- } else if (message.type === 'end' || (message.type === 'response' && message.status >= 400)) socket.close();
157
- } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
158
- });
159
- }
160
- function startFileReceiver() {
161
- if (!identity || closing || remote.readyState !== WebSocket.OPEN) return;
162
- const receiver = fileRemote = new WebSocket(relayUrl, { maxPayload: 512 * 1024, perMessageDeflate: false });
163
- receiver.on('open', () => send(receiver, { type: 'file-host', deviceId: identity.id, deviceSecret: identity.secret }));
164
- receiver.on('error', error => events.emit('diagnostic', `File receiver: ${error.message}`));
165
- receiver.on('close', () => {
166
- for (const socket of fileLocals.values()) socket.terminate();
167
- fileLocals.clear();
168
- if (fileRemote === receiver) {
169
- fileRemote = undefined;
170
- if (!closing && remote.readyState === WebSocket.OPEN) fileRetryTimer = setTimeout(startFileReceiver, 3000);
171
- }
172
- });
173
- receiver.on('message', (bytes, binary) => {
174
- try {
175
- const message = decode(bytes, binary);
176
- if (message.type === 'file-host-ready') return;
177
- if (message.type === 'file-open') {
178
- const connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
179
- const url = new URL(connection.url); url.pathname = '/api/file-transfer';
180
- loopbackUrl(url.href, '/api/file-transfer');
181
- const socket = new WebSocket(url.href, { maxPayload: 512 * 1024, perMessageDeflate: false });
182
- fileLocals.set(message.session, socket);
183
- socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
184
- socket.on('error', error => events.emit('diagnostic', `Local file receiver: ${error.message}`));
185
- socket.on('close', () => {
186
- if (fileLocals.get(message.session) !== socket) return;
187
- fileLocals.delete(message.session);
188
- send(receiver, { type: 'file-close', session: message.session });
189
- });
190
- socket.on('message', (data, isBinary) => {
191
- if (isBinary || fileLocals.get(message.session) !== socket) return socket.terminate();
192
- send(receiver, { type: 'file-frame', session: message.session, payload: data.toString() });
193
- });
194
- } else if (message.type === 'file-frame') {
195
- const socket = fileLocals.get(message.session);
196
- if (socket?.readyState === WebSocket.OPEN) socket.send(message.payload);
197
- } else if (message.type === 'file-close') fileLocals.get(message.session)?.terminate();
198
- else throw new Error('Invalid file receiver frame');
199
- } catch (error) { events.emit('diagnostic', error.message); receiver.close(1008, 'Invalid file frame'); }
200
- });
201
- }
202
206
  const ready = new Promise((resolve, reject) => {
203
207
  let registered = false;
204
208
  const timer = setTimeout(() => { reject(new Error('Relay registration timeout')); remote.terminate(); }, 5000);
@@ -215,13 +219,13 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
215
219
  remote.on('message', (bytes, binary) => {
216
220
  try {
217
221
  const message = decode(bytes, binary);
218
- if (message.type === 'registered' && !registered) {
219
- registered = true; clearTimeout(timer); startFileReceiver(); startTurnReceiver();
220
- turnPulseTimer = setInterval(() => {
221
- verifyActiveTurns();
222
- }, 10000);
223
- resolve(message); return;
224
- }
222
+ if (message.type === 'registered' && !registered) {
223
+ registered = true; clearTimeout(timer); startFileReceiver(); startTurnReceiver();
224
+ turnPulseTimer = setInterval(() => {
225
+ verifyActiveTurns();
226
+ }, 10000);
227
+ resolve(message); return;
228
+ }
225
229
  if (!registered) throw new Error('Relay not registered');
226
230
  if (message.type === 'pair-code' || message.type === 'revoked' || message.type === 'status') {
227
231
  if (operation && message.commandId === operation.id && message.type === operation.responseType) {
@@ -236,8 +240,8 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
236
240
  if (message.type === 'attach') {
237
241
  events.emit('attached', { session: message.session, clientId: message.clientId });
238
242
  dropLocal(message.session);
239
- const current = { socket: undefined, ready: false, timer: undefined, p2pProbe: p2pProbe && message.p2pProbe === 1,
240
- bridgeOutbound: Promise.resolve(), usageRequests: new Map(), detached: false, detachedTimer: undefined };
243
+ const current = { socket: undefined, ready: false, timer: undefined, p2pProbe: p2pProbe && message.p2pProbe === 1,
244
+ bridgeOutbound: Promise.resolve(), usageRequests: new Map(), detached: false, detachedTimer: undefined };
241
245
  locals.set(message.session, current);
242
246
  let connection;
243
247
  try {
@@ -245,105 +249,105 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
245
249
  loopbackUrl(connection.url, '/api/socket');
246
250
  } catch {
247
251
  send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
248
- events.emit('diagnostic', 'Local desktop service connection not ready'); return;
252
+ events.emit('diagnostic', 'Local desktop service connection not ready'); return;
249
253
  }
250
- const socket = current.socket = new WebSocket(connection.url, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: false });
251
- current.forwardToBridge = payload => {
252
- if (locals.get(message.session) !== current || !current.ready) throw new Error('Local desktop service connection not ready');
253
- const request = decode(payload, false);
254
- if (!['request', 'subscribe', 'unsubscribe', 'response-ack', 'cancel'].includes(request.type)) throw new Error('Unsupported LAN message');
255
- if (request.type === 'request' && request.method === 'POST' && Number.isSafeInteger(request.id)) {
256
- const path = (request.path || '').split('?')[0];
257
- const existing = /^\/api\/threads\/([^/?]+)\/(?:messages|queue)$/.exec(path);
258
- const created = /^\/api\/projects\/[^/?]+\/threads$/.test(path);
259
- if (existing || created) {
260
- const threadId = existing ? decodeURIComponent(existing[1]) : null;
261
- const messageKey = typeof request.body?.clientUserMessageId === 'string' && request.body.clientUserMessageId.length <= 64
262
- ? request.body.clientUserMessageId : `${message.session}:${request.id}`;
263
- current.usageRequests.set(request.id, { threadId, messageKey, created });
264
- send(remote, { type: 'usage-arm', session: message.session, requestId: request.id, threadId });
265
- }
266
- }
267
- send(socket, request);
268
- };
254
+ const socket = current.socket = new WebSocket(connection.url, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: false });
255
+ current.forwardToBridge = payload => {
256
+ if (locals.get(message.session) !== current || !current.ready) throw new Error('Local desktop service connection not ready');
257
+ const request = decode(payload, false);
258
+ if (!['request', 'subscribe', 'unsubscribe', 'response-ack', 'cancel'].includes(request.type)) throw new Error('Unsupported LAN message');
259
+ if (request.type === 'request' && request.method === 'POST' && Number.isSafeInteger(request.id)) {
260
+ const path = (request.path || '').split('?')[0];
261
+ const existing = /^\/api\/threads\/([^/?]+)\/(?:messages|queue)$/.exec(path);
262
+ const created = /^\/api\/projects\/[^/?]+\/threads$/.test(path);
263
+ if (existing || created) {
264
+ const threadId = existing ? decodeURIComponent(existing[1]) : null;
265
+ const messageKey = typeof request.body?.clientUserMessageId === 'string' && request.body.clientUserMessageId.length <= 64
266
+ ? request.body.clientUserMessageId : `${message.session}:${request.id}`;
267
+ current.usageRequests.set(request.id, { threadId, messageKey, created });
268
+ send(remote, { type: 'usage-arm', session: message.session, requestId: request.id, threadId });
269
+ }
270
+ }
271
+ send(socket, request);
272
+ };
269
273
  current.timer = setTimeout(() => socket.terminate(), 5000);
270
274
  socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
271
- socket.on('error', error => events.emit('diagnostic', `Local desktop service: ${error.message}`));
272
- socket.on('close', () => {
273
- clearTimeout(current.detachedTimer);
274
- if (locals.get(message.session)?.socket !== socket) return;
275
+ socket.on('error', error => events.emit('diagnostic', `Local desktop service: ${error.message}`));
276
+ socket.on('close', () => {
277
+ clearTimeout(current.detachedTimer);
278
+ if (locals.get(message.session)?.socket !== socket) return;
275
279
  send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
276
280
  });
277
- socket.on('message', (data, isBinary) => {
278
- if (locals.get(message.session)?.socket !== socket && !current.detached) return;
281
+ socket.on('message', (data, isBinary) => {
282
+ if (locals.get(message.session)?.socket !== socket && !current.detached) return;
279
283
  try {
280
284
  const response = decode(data, isBinary);
281
285
  if (!current.ready) {
282
- if (response.type !== 'ready') throw new Error('Local desktop service authentication failed');
286
+ if (response.type !== 'ready') throw new Error('Local desktop service authentication failed');
283
287
  clearTimeout(current.timer); current.ready = true; send(remote, { type: 'ready', session: message.session });
284
- } else {
285
- const usage = response.type === 'response' ? current.usageRequests.get(response.id) : null;
286
- if (usage) {
287
- current.usageRequests.delete(response.id);
288
- const threadId = usage.threadId || response.body?.threadId;
289
- if (response.status >= 200 && response.status < 300 && typeof threadId === 'string') {
290
- const turnId = usage.created ? response.body?.turnId : response.body?.usage?.turnId;
291
- const turnKind = usage.created ? 'start' : response.body?.usage?.kind;
292
- send(remote, { type: 'usage-message', session: message.session, requestId: response.id,
293
- threadId, messageKey: usage.messageKey, turnId, turnKind,
294
- completed: completedTurns.has(`${threadId}:${turnId}`), observerReady: turnObserverReady,
295
- active: activeTurns.get(threadId) === turnId });
296
- if (['start', 'steer'].includes(turnKind) && typeof turnId === 'string' &&
297
- !completedTurns.has(`${threadId}:${turnId}`)) {
298
- trackedTurns.set(threadId, turnId); saveTrackedTurns();
299
- if (turnObserverReady && !activeTurns.has(threadId)) queueTurnResync(threadId, turnId);
300
- }
301
- } else send(remote, { type: 'usage-cancel', session: message.session, requestId: response.id });
302
- }
303
- if (current.detached) {
304
- if (!current.usageRequests.size) socket.terminate();
305
- return;
306
- }
307
- const payload = data.toString();
308
- current.bridgeOutbound = current.bridgeOutbound.then(async () => {
309
- const probe = await current.probe;
310
- if (locals.get(message.session) !== current || !current.ready) return;
311
- if (!probe?.send?.(payload)) send(remote, { type: 'data', session: message.session, payload });
312
- });
313
- void current.bridgeOutbound.catch(error => { events.emit('diagnostic', error.message); socket.terminate(); });
314
- }
288
+ } else {
289
+ const usage = response.type === 'response' ? current.usageRequests.get(response.id) : null;
290
+ if (usage) {
291
+ current.usageRequests.delete(response.id);
292
+ const threadId = usage.threadId || response.body?.threadId;
293
+ if (response.status >= 200 && response.status < 300 && typeof threadId === 'string') {
294
+ const turnId = usage.created ? response.body?.turnId : response.body?.usage?.turnId;
295
+ const turnKind = usage.created ? 'start' : response.body?.usage?.kind;
296
+ send(remote, { type: 'usage-message', session: message.session, requestId: response.id,
297
+ threadId, messageKey: usage.messageKey, turnId, turnKind,
298
+ completed: completedTurns.has(`${threadId}:${turnId}`), observerReady: turnObserverReady,
299
+ active: activeTurns.get(threadId) === turnId });
300
+ if (['start', 'steer'].includes(turnKind) && typeof turnId === 'string' &&
301
+ !completedTurns.has(`${threadId}:${turnId}`)) {
302
+ trackedTurns.set(threadId, turnId); saveTrackedTurns();
303
+ if (turnObserverReady && !activeTurns.has(threadId)) queueTurnResync(threadId, turnId);
304
+ }
305
+ } else send(remote, { type: 'usage-cancel', session: message.session, requestId: response.id });
306
+ }
307
+ if (current.detached) {
308
+ if (!current.usageRequests.size) socket.terminate();
309
+ return;
310
+ }
311
+ const payload = data.toString();
312
+ current.bridgeOutbound = current.bridgeOutbound.then(async () => {
313
+ const probe = await current.probe;
314
+ if (locals.get(message.session) !== current || !current.ready) return;
315
+ if (!probe?.send?.(payload)) send(remote, { type: 'data', session: message.session, payload });
316
+ });
317
+ void current.bridgeOutbound.catch(error => { events.emit('diagnostic', error.message); socket.terminate(); });
318
+ }
315
319
  } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
316
320
  });
317
- } else if (message.type === 'p2p-probe' && locals.has(message.session)) {
318
- const current = locals.get(message.session);
319
- if (!current.ready || !current.p2pProbe || !validProbeSignal(message)) throw new Error('Invalid probe signal');
320
- if (message.purpose === 'diagnostic') {
321
- if (message.kind === 'start' && current.diagnosticProbeId !== message.probeId) {
322
- void current.diagnosticProbe?.then(probe => probe.close());
323
- current.diagnosticProbeId = message.probeId;
324
- const probeId = message.probeId;
325
- current.diagnosticProbe = createP2pProbe({ iceServers,
326
- signal: signal => {
327
- if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
328
- send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
329
- version: 1, probeId, ...signal });
330
- }
331
- },
332
- report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
333
- }).catch(() => {
334
- if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
335
- send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
336
- version: 1, probeId, kind: 'close' });
337
- }
338
- return { receive() {}, close() {} };
339
- });
340
- } else if (current.diagnosticProbe && current.diagnosticProbeId === message.probeId) {
341
- void current.diagnosticProbe.then(probe => probe.receive(message));
342
- }
343
- return;
344
- }
345
- if (message.purpose) throw new Error('Invalid probe purpose');
346
- if (message.kind === 'start' && current.probeId !== message.probeId) {
321
+ } else if (message.type === 'p2p-probe' && locals.has(message.session)) {
322
+ const current = locals.get(message.session);
323
+ if (!current.ready || !current.p2pProbe || !validProbeSignal(message)) throw new Error('Invalid probe signal');
324
+ if (message.purpose === 'diagnostic') {
325
+ if (message.kind === 'start' && current.diagnosticProbeId !== message.probeId) {
326
+ void current.diagnosticProbe?.then(probe => probe.close());
327
+ current.diagnosticProbeId = message.probeId;
328
+ const probeId = message.probeId;
329
+ current.diagnosticProbe = createP2pProbe({ iceServers,
330
+ signal: signal => {
331
+ if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
332
+ send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
333
+ version: 1, probeId, ...signal });
334
+ }
335
+ },
336
+ report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
337
+ }).catch(() => {
338
+ if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
339
+ send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
340
+ version: 1, probeId, kind: 'close' });
341
+ }
342
+ return { receive() {}, close() {} };
343
+ });
344
+ } else if (current.diagnosticProbe && current.diagnosticProbeId === message.probeId) {
345
+ void current.diagnosticProbe.then(probe => probe.receive(message));
346
+ }
347
+ return;
348
+ }
349
+ if (message.purpose) throw new Error('Invalid probe purpose');
350
+ if (message.kind === 'start' && current.probeId !== message.probeId) {
347
351
  void current.probe?.then(probe => probe.close());
348
352
  current.probeId = message.probeId;
349
353
  current.routeSignals = Promise.resolve();
@@ -354,9 +358,9 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
354
358
  signal: signal => {
355
359
  if (locals.get(message.session) === current && current.probeId === probeId) send(remote, { type: 'p2p-probe', session: message.session,
356
360
  version: 1, probeId, ...signal });
357
- },
358
- onData: payload => current.forwardToBridge(payload),
359
- report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
361
+ },
362
+ onData: payload => current.forwardToBridge(payload),
363
+ report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
360
364
  })).catch(() => {
361
365
  events.emit('diagnostic', p2pWindowsRouteHelper ? 'Windows RTC 路由辅助不可用' : 'WebRTC probe initialization failed');
362
366
  return { receive() {}, close() {} };
@@ -376,7 +380,7 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
376
380
  else if (message.type === 'data' && locals.has(message.session)) {
377
381
  const current = locals.get(message.session);
378
382
  if (!current.ready) { send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); return; }
379
- current.forwardToBridge(message.payload);
383
+ current.forwardToBridge(message.payload);
380
384
  }
381
385
  } catch (error) { events.emit('diagnostic', error.message); remote.close(1008, 'Invalid relay message'); }
382
386
  });
@@ -395,6 +399,6 @@ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken,
395
399
  updateLanCandidates(candidates) {
396
400
  if (!send(remote, { type: 'lan-candidates', candidates })) throw new Error('中继未连接');
397
401
  },
398
- close() { closing = true; clearTimeout(heartbeat); dropLocals(); remote.terminate(); }
402
+ close() { closing = true; clearTimeout(heartbeat); dropLocals(); remote.terminate(); }
399
403
  };
400
404
  }