@walkhi/code-relax 0.1.0-beta.22 → 0.1.0-beta.24
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/bin/cleanup-accounts.mjs +63 -0
- package/dist/bin/self-relay-server.mjs +20 -19
- package/dist/shared/app-server-events.cjs +107 -100
- package/dist/shared/service-endpoints.cjs +21 -0
- package/dist/shared/timeline-sync.cjs +94 -5
- package/dist/skills/code-relax/SKILL.md +21 -18
- package/dist/src/lan-socket-server.mjs +50 -10
- package/dist/src/self-relay/account-cleanup.mjs +30 -0
- package/dist/src/self-relay/admin-page.mjs +8 -14
- package/dist/src/self-relay/admin-server.mjs +17 -9
- package/dist/src/self-relay/client.mjs +45 -45
- package/dist/src/self-relay/connector.mjs +293 -289
- package/dist/src/self-relay/diagnostics.mjs +54 -54
- package/dist/src/self-relay/email-accounts.mjs +158 -141
- package/dist/src/self-relay/lifecycle.mjs +160 -160
- package/dist/src/self-relay/privacy-page.mjs +26 -0
- package/dist/src/self-relay/server.mjs +153 -67
- package/dist/src/self-relay/usage-meter.mjs +141 -136
- package/dist/src/server.mjs +102 -203
- package/dist/src/shared-app-server.mjs +132 -116
- package/dist/src/shared-thread-stream.mjs +56 -50
- package/dist/src/turn-model-history.mjs +47 -0
- package/package.json +1 -1
|
@@ -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'].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
|
}
|