@walkhi/code-relax 0.1.0-beta.30 → 0.1.0-beta.32

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.
Files changed (187) hide show
  1. package/README.md +95 -89
  2. package/dist/bin/cleanup-accounts.mjs +63 -63
  3. package/dist/bin/codex-remote.mjs +374 -367
  4. package/dist/bin/manage-relay-admin.mjs +20 -20
  5. package/dist/bin/migrate-relay-db.mjs +109 -109
  6. package/dist/bin/publish-android-release.mjs +63 -58
  7. package/dist/bin/publish-harmony-release.mjs +46 -0
  8. package/dist/bin/release-upload-server.mjs +174 -0
  9. package/dist/bin/self-relay-admin.mjs +8 -8
  10. package/dist/bin/self-relay-demo.mjs +33 -33
  11. package/dist/bin/self-relay-server.mjs +66 -66
  12. package/dist/build-platform.json +1 -1
  13. package/dist/get-lan-networks.ps1 +40 -40
  14. package/dist/get-shared-service-url.ps1 +6 -6
  15. package/dist/shared/app-server-events.cjs +422 -412
  16. package/dist/shared/code-browser.cjs +159 -159
  17. package/dist/shared/file-changes.cjs +31 -31
  18. package/dist/shared/file-citations.cjs +117 -117
  19. package/dist/shared/file-mentions.cjs +92 -92
  20. package/dist/shared/message-delivery.cjs +54 -54
  21. package/dist/shared/p2p-data-channel.cjs +72 -72
  22. package/dist/shared/p2p-diagnostics.cjs +28 -28
  23. package/dist/shared/p2p-file-channel.cjs +78 -74
  24. package/dist/shared/plan-pricing.cjs +32 -32
  25. package/dist/shared/question-attention.cjs +69 -0
  26. package/dist/shared/server-requests.cjs +219 -219
  27. package/dist/shared/timeline-sync.cjs +173 -173
  28. package/dist/skills/code-relax/SKILL.md +22 -22
  29. package/dist/src/archived-threads.mjs +22 -0
  30. package/dist/src/background-process.mjs +18 -18
  31. package/dist/src/bounded-process-log.mjs +63 -63
  32. package/dist/src/celsius-client.mjs +107 -107
  33. package/dist/src/code-browser.mjs +100 -100
  34. package/dist/src/desktop-launcher.mjs +46 -46
  35. package/dist/src/dots-approval-timeline.mjs +99 -99
  36. package/dist/src/dots-avatar.mjs +42 -42
  37. package/dist/src/dots-chat-projection.mjs +50 -50
  38. package/dist/src/dots-chat-server.mjs +461 -461
  39. package/dist/src/dots-message-images.mjs +319 -319
  40. package/dist/src/dots-profile.mjs +77 -77
  41. package/dist/src/dots-room-activity.mjs +94 -94
  42. package/dist/src/dots-room-client.mjs +135 -135
  43. package/dist/src/dots-room-projection.mjs +94 -94
  44. package/dist/src/dots-schedules.mjs +66 -66
  45. package/dist/src/dots-summary-tasks.mjs +32 -32
  46. package/dist/src/dots-task-catalog.mjs +430 -430
  47. package/dist/src/dots-task-files.mjs +52 -52
  48. package/dist/src/execution-health.mjs +26 -26
  49. package/dist/src/file-transfer-range.mjs +22 -22
  50. package/dist/src/file-upload-session.mjs +130 -130
  51. package/dist/src/hosted-history-app-server.mjs +103 -103
  52. package/dist/src/hosted-history-cache.mjs +113 -113
  53. package/dist/src/lan-socket-server.mjs +265 -263
  54. package/dist/src/managed-app-server.mjs +244 -244
  55. package/dist/src/message-images.mjs +9 -9
  56. package/dist/src/message-upload-files.mjs +74 -74
  57. package/dist/src/onboarding.mjs +12 -12
  58. package/dist/src/plan-pricing.mjs +47 -47
  59. package/dist/src/platform/README.md +12 -12
  60. package/dist/src/platform/codex-protocol.mjs +46 -46
  61. package/dist/src/platform/codex-runtime.mjs +3 -3
  62. package/dist/src/platform/credential-store.mjs +3 -3
  63. package/dist/src/platform/desktop-host.mjs +3 -3
  64. package/dist/src/platform/desktop-project-registration.mjs +3 -3
  65. package/dist/src/platform/desktop-project-store.mjs +208 -149
  66. package/dist/src/platform/file-opener.mjs +9 -0
  67. package/dist/src/platform/linux/README.md +3 -3
  68. package/dist/src/platform/linux/codex-runtime.mjs +56 -0
  69. package/dist/src/platform/linux/credential-store.mjs +28 -0
  70. package/dist/src/platform/linux/desktop-host.mjs +83 -0
  71. package/dist/src/platform/linux/desktop-installation.mjs +36 -0
  72. package/dist/src/platform/linux/desktop-project-registration.mjs +11 -0
  73. package/dist/src/platform/linux/network-info.mjs +11 -0
  74. package/dist/src/platform/linux/process.mjs +42 -0
  75. package/dist/src/platform/linux/project-locations.mjs +17 -0
  76. package/dist/src/platform/linux/resident-startup.mjs +52 -0
  77. package/dist/src/platform/linux/retained-runtime.mjs +76 -0
  78. package/dist/src/platform/linux/service-host.mjs +18 -0
  79. package/dist/src/platform/linux/system.mjs +49 -0
  80. package/dist/src/platform/linux/user-environment.mjs +7 -0
  81. package/dist/src/platform/macos/README.md +3 -3
  82. package/dist/src/platform/macos/codex-runtime.mjs +125 -125
  83. package/dist/src/platform/macos/credential-store.mjs +10 -10
  84. package/dist/src/platform/macos/desktop-host.mjs +82 -82
  85. package/dist/src/platform/macos/desktop-project-registration.mjs +11 -10
  86. package/dist/src/platform/macos/network-info.mjs +8 -28
  87. package/dist/src/platform/macos/project-locations.mjs +8 -8
  88. package/dist/src/platform/macos/resident-startup.mjs +84 -84
  89. package/dist/src/platform/macos/retained-runtime.mjs +127 -127
  90. package/dist/src/platform/macos/service-host.mjs +18 -18
  91. package/dist/src/platform/macos/system.mjs +29 -29
  92. package/dist/src/platform/macos/user-environment.mjs +7 -7
  93. package/dist/src/platform/native-package.mjs +58 -46
  94. package/dist/src/platform/network-info.mjs +3 -3
  95. package/dist/src/platform/path-identity.mjs +5 -5
  96. package/dist/src/platform/posix/network-candidates.mjs +23 -0
  97. package/dist/src/platform/posix/runtime-paths.mjs +25 -25
  98. package/dist/src/platform/project-locations.mjs +3 -3
  99. package/dist/src/platform/runtime-paths.mjs +8 -8
  100. package/dist/src/platform/select.mjs +5 -5
  101. package/dist/src/platform/service-host.mjs +3 -3
  102. package/dist/src/platform/user-environment.mjs +3 -3
  103. package/dist/src/platform/windows/IsolatedProcess.cs +79 -79
  104. package/dist/src/platform/windows/activate-desktop.ps1 +48 -48
  105. package/dist/src/platform/windows/background-process.mjs +30 -30
  106. package/dist/src/platform/windows/codex-runtime.mjs +215 -215
  107. package/dist/src/platform/windows/credential-store.mjs +27 -27
  108. package/dist/src/platform/windows/credential-store.ps1 +49 -49
  109. package/dist/src/platform/windows/desktop-helper.mjs +179 -179
  110. package/dist/src/platform/windows/desktop-host.mjs +58 -58
  111. package/dist/src/platform/windows/desktop-info.ps1 +14 -14
  112. package/dist/src/platform/windows/desktop-project-registration.mjs +4 -4
  113. package/dist/src/platform/windows/desktop-runtime.ps1 +10 -10
  114. package/dist/src/platform/windows/desktop-shortcuts.ps1 +21 -21
  115. package/dist/src/platform/windows/dots-profile.mjs +49 -49
  116. package/dist/src/platform/windows/launch-worker.mjs +41 -41
  117. package/dist/src/platform/windows/network-info.mjs +34 -34
  118. package/dist/src/platform/windows/project-locations.mjs +16 -16
  119. package/dist/src/platform/windows/resident-start.js +20 -20
  120. package/dist/src/platform/windows/resident-start.ps1 +7 -7
  121. package/dist/src/platform/windows/resident-startup.ps1 +83 -83
  122. package/dist/src/platform/windows/resolve-state-root.ps1 +14 -14
  123. package/dist/src/platform/windows/runtime-paths.mjs +20 -20
  124. package/dist/src/platform/windows/rustdesk-video-probe.mjs +167 -167
  125. package/dist/src/platform/windows/service-host.mjs +31 -31
  126. package/dist/src/platform/windows/start-hidden-console.ps1 +49 -49
  127. package/dist/src/platform/windows/stop-desktop.ps1 +22 -22
  128. package/dist/src/platform/windows/user-environment.mjs +50 -50
  129. package/dist/src/platform/windows/user-environment.ps1 +41 -41
  130. package/dist/src/project-archive.mjs +30 -0
  131. package/dist/src/project-create.mjs +121 -119
  132. package/dist/src/project-management.mjs +59 -69
  133. package/dist/src/project-remove.mjs +55 -0
  134. package/dist/src/project-update.mjs +56 -0
  135. package/dist/src/runtime-discovery.mjs +47 -47
  136. package/dist/src/self-relay/account-cleanup.mjs +30 -30
  137. package/dist/src/self-relay/account-connections.mjs +52 -52
  138. package/dist/src/self-relay/admin-cloud.mjs +101 -101
  139. package/dist/src/self-relay/admin-page.mjs +80 -80
  140. package/dist/src/self-relay/admin-server.mjs +74 -74
  141. package/dist/src/self-relay/admin-state.mjs +127 -127
  142. package/dist/src/self-relay/admin-style.mjs +14 -14
  143. package/dist/src/self-relay/android-releases.mjs +61 -61
  144. package/dist/src/self-relay/client.mjs +61 -61
  145. package/dist/src/self-relay/connector.mjs +439 -438
  146. package/dist/src/self-relay/debug-allowlist.mjs +33 -33
  147. package/dist/src/self-relay/demo.mjs +37 -37
  148. package/dist/src/self-relay/device-info.mjs +27 -0
  149. package/dist/src/self-relay/device-store.mjs +92 -84
  150. package/dist/src/self-relay/diagnostics.mjs +54 -54
  151. package/dist/src/self-relay/email-accounts.mjs +183 -183
  152. package/dist/src/self-relay/file-wire.mjs +13 -13
  153. package/dist/src/self-relay/lan-authorizations.mjs +71 -42
  154. package/dist/src/self-relay/lifecycle.mjs +264 -250
  155. package/dist/src/self-relay/long-pairing.mjs +40 -40
  156. package/dist/src/self-relay/mysql-state.mjs +118 -105
  157. package/dist/src/self-relay/p2p-probe.mjs +159 -159
  158. package/dist/src/self-relay/privacy-page.mjs +26 -26
  159. package/dist/src/self-relay/public-releases.mjs +52 -0
  160. package/dist/src/self-relay/rtc-file-bridge.mjs +41 -41
  161. package/dist/src/self-relay/server.mjs +1006 -932
  162. package/dist/src/self-relay/usage-daily.mjs +55 -55
  163. package/dist/src/self-relay/usage-meter.mjs +141 -141
  164. package/dist/src/self-relay/windows-route-helper.mjs +75 -75
  165. package/dist/src/self-relay/wire.mjs +36 -36
  166. package/dist/src/server.mjs +1861 -1778
  167. package/dist/src/service-control.mjs +82 -82
  168. package/dist/src/service-doctor.mjs +110 -110
  169. package/dist/src/service-health.mjs +33 -33
  170. package/dist/src/service-install.mjs +91 -91
  171. package/dist/src/service-lifecycle.mjs +138 -138
  172. package/dist/src/service-restart.mjs +81 -81
  173. package/dist/src/shared-app-server.mjs +615 -611
  174. package/dist/src/shared-catalog.mjs +132 -132
  175. package/dist/src/shared-recovery.mjs +146 -146
  176. package/dist/src/shared-thread-stream.mjs +164 -164
  177. package/dist/src/skill-install.mjs +51 -51
  178. package/dist/src/thread-title-generation.mjs +150 -150
  179. package/dist/src/turn-model-history.mjs +47 -47
  180. package/dist/src/usage-reset.mjs +17 -17
  181. package/dist/src/workstation-downloads.mjs +120 -120
  182. package/dist/tools/rtc-route-helper.ps1 +99 -99
  183. package/dist/tools/start-rtc-route-helper.ps1 +15 -15
  184. package/dist/web/vendor/markdown-it.LICENSE.txt +22 -22
  185. package/dist/web/vendor/markdown-it.umd.min.js +13 -13
  186. package/package.json +7 -6
  187. package/tools/postinstall.mjs +127 -127
@@ -1,438 +1,439 @@
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
- import { routedFileBytes, parseFileBytes } from './file-wire.mjs';
7
- import { attachRtcFile } from './rtc-file-bridge.mjs';
8
- import { createP2pProbe, validProbeSignal } from './p2p-probe.mjs';
9
- import { createWindowsRouteHelper } from './windows-route-helper.mjs';
10
-
11
- // Reuse the actual LAN dispatcher, including its normal/shared capability boundary.
12
- // The LAN token never travels to the relay. No requests are replayed after a close.
13
- export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken, identity, bridgeConnection, usageCacheFile,
14
- hostName = 'Codex-PC', lanCandidates = [], p2pProbe = false, iceServers = [], p2pWindowsRouteHelper = false, p2pWindowsRouteHelperToken = '' }) {
15
- relayAddress(relayUrl); loopbackUrl(bridgeUrl, '/api/socket');
16
- if (!bridgeToken) throw new Error('Desktop service credential required');
17
- const events = new EventEmitter();
18
- const remote = new WebSocket(relayUrl, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: {
19
- serverNoContextTakeover: true, clientNoContextTakeover: true,
20
- threshold: 1024, zlibDeflateOptions: { level: 3 },
21
- } });
22
- const locals = new Map();
23
- const fileLocals = new Map();
24
- const activeTurns = new Map();
25
- const completedTurns = new Map();
26
- const trackedTurns = new Map();
27
- if (usageCacheFile) {
28
- try {
29
- const saved = JSON.parse(fs.readFileSync(usageCacheFile, 'utf8'));
30
- if (saved.version === 1 && Array.isArray(saved.turns)) {
31
- for (const item of saved.turns.slice(0, 128)) {
32
- if (typeof item.threadId === 'string' && item.threadId.length <= 80 &&
33
- typeof item.turnId === 'string' && item.turnId.length <= 80) trackedTurns.set(item.threadId, item.turnId);
34
- }
35
- }
36
- } catch (error) { if (error.code !== 'ENOENT') events.emit('diagnostic', `Usage task cache: ${error.message}`); }
37
- }
38
- function saveTrackedTurns() {
39
- if (!usageCacheFile) return;
40
- try {
41
- const temp = `${usageCacheFile}.tmp`;
42
- fs.writeFileSync(temp, JSON.stringify({ version: 1, turns: [...trackedTurns].map(([threadId, turnId]) => ({ threadId, turnId })) }), { mode: 0o600 });
43
- fs.renameSync(temp, usageCacheFile);
44
- } catch (error) { events.emit('diagnostic', `Usage task cache: ${error.message}`); }
45
- }
46
- let fileRemote, fileRetryTimer, turnLocal, turnRetryTimer, turnPulseTimer, turnObserverReady = false,
47
- queueTurnResync = () => {}, verifyActiveTurns = () => {}, heartbeat, operation, lastClose, closing = false;
48
- const alive = () => { clearTimeout(heartbeat); heartbeat = setTimeout(() => remote.terminate(), 75000); };
49
- remote.on('open', alive); remote.on('ping', alive); remote.on('message', alive);
50
- function dropLocal(session) {
51
- const current = locals.get(session);
52
- if (!current) return;
53
- locals.delete(session); clearTimeout(current.timer);
54
- if (current.ready && current.socket?.readyState === WebSocket.OPEN && current.usageRequests.size &&
55
- remote.readyState === WebSocket.OPEN && !closing) {
56
- // Keep only the in-flight submission receipt; no late response goes to the phone.
57
- current.detached = true;
58
- current.detachedTimer = setTimeout(() => current.socket?.terminate(), 35000);
59
- } else current.socket?.terminate();
60
- current.probe?.then(probe => probe.close());
61
- current.fileProbe?.then(probe => probe.close());
62
- current.diagnosticProbe?.then(probe => probe.close());
63
- }
64
- function dropLocals() {
65
- for (const session of [...locals.keys()]) dropLocal(session);
66
- for (const socket of fileLocals.values()) socket.terminate();
67
- fileLocals.clear(); clearTimeout(fileRetryTimer); fileRemote?.terminate(); fileRemote = undefined;
68
- clearTimeout(turnRetryTimer); clearInterval(turnPulseTimer); turnLocal?.terminate(); turnLocal = undefined;
69
- turnObserverReady = false; activeTurns.clear(); completedTurns.clear(); verifyActiveTurns = () => {};
70
- }
71
- function startTurnReceiver() {
72
- if (closing || remote.readyState !== WebSocket.OPEN) return;
73
- let connection;
74
- try { connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken }; loopbackUrl(connection.url, '/api/socket'); }
75
- catch { turnRetryTimer = setTimeout(startTurnReceiver, 3000); return; }
76
- const socket = turnLocal = new WebSocket(connection.url, { maxPayload: 64 * 1024, perMessageDeflate: false });
77
- const resyncPending = new Map();
78
- const pulsePending = new Map();
79
- let resyncId = 1, resyncQueue = [];
80
- function resyncNext() {
81
- if (!turnObserverReady || turnLocal !== socket) return;
82
- while (resyncPending.size < 8 && resyncQueue.length) {
83
- const [threadId, turnId] = resyncQueue.shift();
84
- if (trackedTurns.get(threadId) !== turnId) continue;
85
- const id = ++resyncId;
86
- resyncPending.set(id, { threadId, turnId });
87
- send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
88
- }
89
- }
90
- queueTurnResync = (threadId, turnId) => { resyncQueue.push([threadId, turnId]); resyncNext(); };
91
- verifyActiveTurns = () => {
92
- if (!turnObserverReady || turnLocal !== socket) return;
93
- for (const [threadId, turnId] of trackedTurns) {
94
- if (activeTurns.get(threadId) !== turnId) continue;
95
- if (pulsePending.size >= 8) break;
96
- if ([...pulsePending.values()].some(pending => pending.threadId === threadId)) continue;
97
- const id = ++resyncId;
98
- pulsePending.set(id, { threadId, turnId });
99
- send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
100
- }
101
- };
102
- socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
103
- socket.on('error', error => events.emit('diagnostic', `Local turn observer: ${error.message}`));
104
- socket.on('close', () => {
105
- if (turnLocal !== socket) return;
106
- turnLocal = undefined;
107
- if (turnObserverReady) send(remote, { type: 'usage-observer-lost' });
108
- turnObserverReady = false;
109
- queueTurnResync = () => {};
110
- verifyActiveTurns = () => {};
111
- activeTurns.clear(); completedTurns.clear();
112
- if (!closing && remote.readyState === WebSocket.OPEN) turnRetryTimer = setTimeout(startTurnReceiver, 3000);
113
- });
114
- socket.on('message', (bytes, binary) => {
115
- if (binary || turnLocal !== socket) return socket.terminate();
116
- try {
117
- const message = decode(bytes, false);
118
- if (message.type === 'ready') { send(socket, { type: 'subscribe', id: 1, path: '/api/turn-events' }); return; }
119
- if (message.type === 'subscribed' && message.id === 1) {
120
- turnObserverReady = true; resyncQueue = [...trackedTurns]; resyncNext(); return;
121
- }
122
- if (message.type === 'response' && pulsePending.has(message.id)) {
123
- const { threadId, turnId } = pulsePending.get(message.id);
124
- pulsePending.delete(message.id);
125
- if (trackedTurns.get(threadId) !== turnId || activeTurns.get(threadId) !== turnId) return;
126
- if (message.status === 200 && message.body?.turnId === turnId) {
127
- send(remote, { type: 'usage-turn', method: 'heartbeat', threadId, turnId });
128
- } else if (message.status === 200 || message.status === 404) {
129
- activeTurns.delete(threadId);
130
- if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
131
- send(remote, { type: 'usage-turn', method: 'inactive', threadId, turnId });
132
- } else socket.close();
133
- return;
134
- }
135
- if (message.type === 'response' && resyncPending.has(message.id)) {
136
- const { threadId, turnId } = resyncPending.get(message.id);
137
- resyncPending.delete(message.id);
138
- if (trackedTurns.get(threadId) === turnId) {
139
- if (message.status === 200 && message.body?.turnId === turnId) {
140
- activeTurns.set(threadId, turnId);
141
- send(remote, { type: 'usage-turn', method: 'resume', threadId, turnId });
142
- } else if (message.status === 200 || message.status === 404) {
143
- trackedTurns.delete(threadId); saveTrackedTurns();
144
- } else { socket.close(); return; }
145
- }
146
- resyncNext(); return;
147
- }
148
- if (message.type === 'event' && message.id === 1 && message.event === 'turn') {
149
- const { method, threadId, turnId } = message.data || {};
150
- if (!['turn/started', 'turn/completed'].includes(method) || typeof threadId !== 'string' || typeof turnId !== 'string') return;
151
- const key = `${threadId}:${turnId}`;
152
- const tracked = trackedTurns.get(threadId) === turnId;
153
- // Keep local event state for a phone submission receipt arriving later.
154
- // Only registered usage turns need cloud updates or periodic verification.
155
- if (method === 'turn/started') activeTurns.set(threadId, turnId);
156
- else {
157
- if (activeTurns.get(threadId) === turnId) activeTurns.delete(threadId);
158
- if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
159
- completedTurns.set(key, Date.now());
160
- if (completedTurns.size > 100) completedTurns.delete(completedTurns.keys().next().value);
161
- }
162
- if (tracked) send(remote, { type: 'usage-turn', method, threadId, turnId });
163
- } else if (message.type === 'end' || (message.type === 'response' && message.status >= 400)) socket.close();
164
- } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
165
- });
166
- }
167
- function startFileReceiver() {
168
- if (!identity || closing || remote.readyState !== WebSocket.OPEN) return;
169
- const receiver = fileRemote = new WebSocket(relayUrl, { maxPayload: 512 * 1024, perMessageDeflate: false });
170
- receiver.on('open', () => send(receiver, { type: 'file-host', deviceId: identity.id, deviceSecret: identity.secret }));
171
- receiver.on('error', error => events.emit('diagnostic', `File receiver: ${error.message}`));
172
- receiver.on('close', () => {
173
- for (const socket of fileLocals.values()) socket.terminate();
174
- fileLocals.clear();
175
- if (fileRemote === receiver) {
176
- fileRemote = undefined;
177
- if (!closing && remote.readyState === WebSocket.OPEN) fileRetryTimer = setTimeout(startFileReceiver, 3000);
178
- }
179
- });
180
- receiver.on('message', (bytes, binary) => {
181
- try {
182
- if (binary) {
183
- const { session, data } = parseFileBytes(bytes);
184
- const socket = fileLocals.get(session);
185
- if (socket?.readyState === WebSocket.OPEN) {
186
- if (socket.bufferedAmount > 1024 * 1024) { socket.close(1013, 'File receiver too slow'); return; }
187
- socket.send(data);
188
- }
189
- return;
190
- }
191
- const message = decode(bytes, binary);
192
- if (message.type === 'file-host-ready') return;
193
- if (message.type === 'file-open') {
194
- const connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
195
- const url = new URL(connection.url); url.pathname = '/api/file-transfer';
196
- loopbackUrl(url.href, '/api/file-transfer');
197
- const socket = new WebSocket(url.href, { maxPayload: 512 * 1024, perMessageDeflate: false });
198
- fileLocals.set(message.session, socket);
199
- socket.on('open', () => send(socket, { type: 'auth', token: connection.token, purpose: message.purpose }));
200
- socket.on('error', error => events.emit('diagnostic', `Local file receiver: ${error.message}`));
201
- socket.on('close', () => {
202
- if (fileLocals.get(message.session) !== socket) return;
203
- fileLocals.delete(message.session);
204
- send(receiver, { type: 'file-close', session: message.session });
205
- });
206
- socket.on('message', (data, isBinary) => {
207
- if (fileLocals.get(message.session) !== socket) return socket.terminate();
208
- if (receiver.readyState !== WebSocket.OPEN || receiver.bufferedAmount > 1024 * 1024) return socket.close(1013, 'File receiver too slow');
209
- if (isBinary) receiver.send(routedFileBytes(message.session, data));
210
- else send(receiver, { type: 'file-frame', session: message.session, payload: data.toString() });
211
- });
212
- } else if (message.type === 'file-frame') {
213
- const socket = fileLocals.get(message.session);
214
- if (socket?.readyState === WebSocket.OPEN) {
215
- if (socket.bufferedAmount > 1024 * 1024) { socket.close(1013, 'File receiver too slow'); return; }
216
- socket.send(message.payload);
217
- }
218
- } else if (message.type === 'file-close') fileLocals.get(message.session)?.terminate();
219
- else throw new Error('Invalid file receiver frame');
220
- } catch (error) { events.emit('diagnostic', error.message); receiver.close(1008, 'Invalid file frame'); }
221
- });
222
- }
223
- const ready = new Promise((resolve, reject) => {
224
- let registered = false;
225
- const timer = setTimeout(() => { reject(new Error('Relay registration timeout')); remote.terminate(); }, 5000);
226
- remote.on('open', () => send(remote, { type: 'host', key: hostKey, hostName, lanCandidates, p2pProbe: p2pProbe ? 1 : 0, p2pProbeRepeat: true,
227
- ...(identity ? { deviceId: identity.id, deviceSecret: identity.secret } : {}) }));
228
- remote.on('error', error => { reject(error); events.emit('diagnostic', error.message); });
229
- remote.on('close', (code, reason) => {
230
- clearTimeout(timer); clearTimeout(heartbeat); dropLocals();
231
- const error = lastClose = Object.assign(new Error(`Relay closed (${code}): ${reason.toString()}`), { code });
232
- if (!registered) reject(error);
233
- if (operation) { clearTimeout(operation.timer); operation.reject(error); operation = undefined; }
234
- events.emit('disconnected', { code, reason: reason.toString() });
235
- });
236
- remote.on('message', (bytes, binary) => {
237
- try {
238
- const message = decode(bytes, binary);
239
- if (message.type === 'registered' && !registered) {
240
- registered = true; clearTimeout(timer); startFileReceiver(); startTurnReceiver();
241
- turnPulseTimer = setInterval(() => {
242
- verifyActiveTurns();
243
- }, 10000);
244
- resolve(message); return;
245
- }
246
- if (!registered) throw new Error('Relay not registered');
247
- if (['pair-code', 'revoked', 'status', 'invite-create', 'invite-list', 'invite-revoke'].includes(message.type)) {
248
- if (operation && message.commandId === operation.id && message.type === operation.responseType) {
249
- clearTimeout(operation.timer); const pending = operation; operation = undefined;
250
- message.error ? pending.reject(new Error(message.error)) : pending.resolve(message);
251
- }
252
- return;
253
- }
254
- if (message.type === 'authorization-changed' && Array.isArray(message.clients) && Number.isSafeInteger(message.revision)) {
255
- events.emit('authorization-changed', { clients: message.clients, revision: message.revision }); return;
256
- }
257
- if (message.type === 'attach') {
258
- events.emit('attached', { session: message.session, clientId: message.clientId });
259
- dropLocal(message.session);
260
- const current = { socket: undefined, ready: false, timer: undefined, p2pProbe: p2pProbe && message.p2pProbe === 1,
261
- bridgeOutbound: Promise.resolve(), healthRequests: new Set(), usageRequests: new Map(), detached: false, detachedTimer: undefined };
262
- locals.set(message.session, current);
263
- let connection;
264
- try {
265
- connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
266
- loopbackUrl(connection.url, '/api/socket');
267
- } catch {
268
- send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
269
- events.emit('diagnostic', 'Local desktop service connection not ready'); return;
270
- }
271
- const socket = current.socket = new WebSocket(connection.url, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: false });
272
- current.forwardToBridge = payload => {
273
- if (locals.get(message.session) !== current || !current.ready) throw new Error('Local desktop service connection not ready');
274
- const request = decode(payload, false);
275
- if (!['request', 'subscribe', 'unsubscribe', 'response-ack', 'cancel'].includes(request.type)) throw new Error('Unsupported LAN message');
276
- if (request.type === 'request' && request.method === 'GET' && request.path === '/api/health' && Number.isSafeInteger(request.id)) {
277
- current.healthRequests.add(request.id);
278
- }
279
- if (request.type === 'cancel') current.healthRequests.delete(request.id);
280
- if (request.type === 'request' && request.method === 'POST' && Number.isSafeInteger(request.id)) {
281
- const path = (request.path || '').split('?')[0];
282
- const existing = /^\/api\/threads\/([^/?]+)\/(?:messages|queue)$/.exec(path);
283
- const created = /^\/api\/projects\/[^/?]+\/threads$/.test(path);
284
- if (existing || created) {
285
- const threadId = existing ? decodeURIComponent(existing[1]) : null;
286
- const messageKey = typeof request.body?.clientUserMessageId === 'string' && request.body.clientUserMessageId.length <= 64
287
- ? request.body.clientUserMessageId : `${message.session}:${request.id}`;
288
- current.usageRequests.set(request.id, { threadId, messageKey, created });
289
- send(remote, { type: 'usage-arm', session: message.session, requestId: request.id, threadId });
290
- }
291
- }
292
- send(socket, request);
293
- };
294
- current.timer = setTimeout(() => socket.terminate(), 5000);
295
- socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
296
- socket.on('error', error => events.emit('diagnostic', `Local desktop service: ${error.message}`));
297
- socket.on('close', () => {
298
- clearTimeout(current.detachedTimer);
299
- if (locals.get(message.session)?.socket !== socket) return;
300
- send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
301
- });
302
- socket.on('message', (data, isBinary) => {
303
- if (locals.get(message.session)?.socket !== socket && !current.detached) return;
304
- try {
305
- const response = decode(data, isBinary);
306
- if (!current.ready) {
307
- if (response.type !== 'ready') throw new Error('Local desktop service authentication failed');
308
- clearTimeout(current.timer); current.ready = true; send(remote, { type: 'ready', session: message.session });
309
- } else {
310
- const usage = response.type === 'response' ? current.usageRequests.get(response.id) : null;
311
- if (usage) {
312
- current.usageRequests.delete(response.id);
313
- const threadId = usage.threadId || response.body?.threadId;
314
- if (response.status >= 200 && response.status < 300 && typeof threadId === 'string') {
315
- const turnId = usage.created ? response.body?.turnId : response.body?.usage?.turnId;
316
- const turnKind = usage.created ? 'start' : response.body?.usage?.kind;
317
- send(remote, { type: 'usage-message', session: message.session, requestId: response.id,
318
- threadId, messageKey: usage.messageKey, turnId, turnKind,
319
- completed: completedTurns.has(`${threadId}:${turnId}`), observerReady: turnObserverReady,
320
- active: activeTurns.get(threadId) === turnId });
321
- if (['start', 'steer'].includes(turnKind) && typeof turnId === 'string' &&
322
- !completedTurns.has(`${threadId}:${turnId}`)) {
323
- trackedTurns.set(threadId, turnId); saveTrackedTurns();
324
- if (turnObserverReady && !activeTurns.has(threadId)) queueTurnResync(threadId, turnId);
325
- }
326
- } else send(remote, { type: 'usage-cancel', session: message.session, requestId: response.id });
327
- }
328
- if (current.detached) {
329
- if (!current.usageRequests.size) socket.terminate();
330
- return;
331
- }
332
- const payload = data.toString();
333
- // Native health probes use the relay socket. Keep their replies on that
334
- // route, outside the RTC queue shared by chat and file downloads.
335
- if (response.type === 'response' && current.healthRequests.delete(response.id)) {
336
- send(remote, { type: 'data', session: message.session, payload });
337
- return;
338
- }
339
- current.bridgeOutbound = current.bridgeOutbound.then(async () => {
340
- const probe = await current.probe;
341
- if (locals.get(message.session) !== current || !current.ready) return;
342
- if (!probe?.send?.(payload)) send(remote, { type: 'data', session: message.session, payload });
343
- });
344
- void current.bridgeOutbound.catch(error => { events.emit('diagnostic', error.message); socket.terminate(); });
345
- }
346
- } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
347
- });
348
- } else if (message.type === 'p2p-probe' && locals.has(message.session)) {
349
- const current = locals.get(message.session);
350
- if (!current.ready || !current.p2pProbe || !validProbeSignal(message)) throw new Error('Invalid probe signal');
351
- if (message.purpose === 'diagnostic') {
352
- if (message.kind === 'start' && current.diagnosticProbeId !== message.probeId) {
353
- void current.diagnosticProbe?.then(probe => probe.close());
354
- current.diagnosticProbeId = message.probeId;
355
- const probeId = message.probeId;
356
- current.diagnosticProbe = createP2pProbe({ iceServers,
357
- signal: signal => {
358
- if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
359
- send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
360
- version: 1, probeId, ...signal });
361
- }
362
- },
363
- report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
364
- }).catch(() => {
365
- if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
366
- send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
367
- version: 1, probeId, kind: 'close' });
368
- }
369
- return { receive() {}, close() {} };
370
- });
371
- } else if (current.diagnosticProbe && current.diagnosticProbeId === message.probeId) {
372
- void current.diagnosticProbe.then(probe => probe.receive(message));
373
- }
374
- return;
375
- }
376
- if (message.purpose && message.purpose !== 'file') throw new Error('Invalid probe purpose');
377
- const file = message.purpose === 'file';
378
- const probeField = file ? 'fileProbe' : 'probe', idField = file ? 'fileProbeId' : 'probeId';
379
- const signalsField = file ? 'fileRouteSignals' : 'routeSignals', helperField = file ? 'fileRouteHelper' : 'routeHelper';
380
- if (message.kind === 'start' && current[idField] !== message.probeId) {
381
- void current[probeField]?.then(probe => probe.close());
382
- current[idField] = message.probeId;
383
- current[signalsField] = Promise.resolve();
384
- const probeId = message.probeId;
385
- current[helperField] = createWindowsRouteHelper({ enabled: p2pWindowsRouteHelper, token: p2pWindowsRouteHelperToken, iceServers,
386
- report: (event, details) => events.emit('diagnostic', `${event};count=${details.count}`) });
387
- current[probeField] = current[helperField].prepare().then(() => createP2pProbe({ iceServers, purpose: file ? 'file' : '',
388
- signal: signal => {
389
- if (locals.get(message.session) === current && current[idField] === probeId) send(remote, { type: 'p2p-probe', session: message.session, ...(file ? { purpose: 'file' } : {}),
390
- version: 1, probeId, ...signal });
391
- },
392
- onData: file ? undefined : payload => current.forwardToBridge(payload),
393
- onFileChannel: file ? channel => attachRtcFile(channel, bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken },
394
- () => locals.get(message.session) === current && current.ready && current[idField] === probeId && remote.readyState === WebSocket.OPEN) : undefined,
395
- report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic, purpose: file ? 'file' : 'chat' })
396
- })).catch(() => {
397
- events.emit('diagnostic', p2pWindowsRouteHelper ? 'Windows RTC 路由辅助不可用' : 'WebRTC probe initialization failed');
398
- if (locals.get(message.session) === current && current[idField] === probeId) send(remote, {
399
- type: 'p2p-probe', session: message.session, version: 1, probeId, ...(file ? { purpose: 'file' } : {}), kind: 'close' });
400
- return { receive() {}, close() {} };
401
- });
402
- } else if (current[probeField] && current[idField] === message.probeId) {
403
- const pending = current[probeField];
404
- current[signalsField] ||= Promise.resolve();
405
- current[signalsField] = current[signalsField].then(async () => {
406
- await current[helperField]?.include(message);
407
- const probe = await pending;
408
- if (locals.get(message.session) === current && current[idField] === message.probeId) probe.receive(message);
409
- });
410
- void current[signalsField]
411
- .catch(() => { events.emit('diagnostic', 'WebRTC probe negotiation failed'); void pending.then(probe => probe.close()); });
412
- }
413
- } else if (message.type === 'detach' && locals.has(message.session)) { dropLocal(message.session); events.emit('detached', { session: message.session }); }
414
- else if (message.type === 'data' && locals.has(message.session)) {
415
- const current = locals.get(message.session);
416
- if (!current.ready) { send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); return; }
417
- current.forwardToBridge(message.payload);
418
- }
419
- } catch (error) { events.emit('diagnostic', error.message); remote.close(1008, 'Invalid relay message'); }
420
- });
421
- });
422
- return { ready, events,
423
- assertConnected() { if (remote.readyState !== WebSocket.OPEN) throw lastClose || new Error('Relay disconnected during registration'); },
424
- command(type, details = {}) {
425
- if (operation) return Promise.reject(new Error('已有设备管理操作正在执行'));
426
- return new Promise((resolve, reject) => {
427
- const timer = setTimeout(() => { operation = undefined; reject(new Error('设备管理请求超时,结果未确认;未自动重发')); }, 7000);
428
- const id = randomUUID();
429
- operation = { resolve, reject, timer, id, responseType: type === 'revoke' ? 'revoked' : type };
430
- if (!send(remote, { type, commandId: id, ...details })) { clearTimeout(timer); operation = undefined; reject(new Error('中继未连接')); }
431
- });
432
- },
433
- updateLanCandidates(candidates) {
434
- if (!send(remote, { type: 'lan-candidates', candidates })) throw new Error('中继未连接');
435
- },
436
- close() { closing = true; clearTimeout(heartbeat); dropLocals(); remote.terminate(); }
437
- };
438
- }
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
+ import { routedFileBytes, parseFileBytes } from './file-wire.mjs';
7
+ import { attachRtcFile } from './rtc-file-bridge.mjs';
8
+ import { createP2pProbe, validProbeSignal } from './p2p-probe.mjs';
9
+ import { createWindowsRouteHelper } from './windows-route-helper.mjs';
10
+
11
+ // Reuse the actual LAN dispatcher, including its normal/shared capability boundary.
12
+ // The LAN token never travels to the relay. No requests are replayed after a close.
13
+ export function connectBridgeRelay({ relayUrl, hostKey, bridgeUrl, bridgeToken, identity, bridgeConnection, usageCacheFile,
14
+ hostName = 'Codex-PC', lanCandidates = [], p2pProbe = false, iceServers = [], p2pWindowsRouteHelper = false, p2pWindowsRouteHelperToken = '' }) {
15
+ relayAddress(relayUrl); loopbackUrl(bridgeUrl, '/api/socket');
16
+ if (!bridgeToken) throw new Error('Desktop service credential required');
17
+ const events = new EventEmitter();
18
+ const remote = new WebSocket(relayUrl, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: {
19
+ serverNoContextTakeover: true, clientNoContextTakeover: true,
20
+ threshold: 1024, zlibDeflateOptions: { level: 3 },
21
+ } });
22
+ const locals = new Map();
23
+ const fileLocals = new Map();
24
+ const activeTurns = new Map();
25
+ const completedTurns = new Map();
26
+ const trackedTurns = new Map();
27
+ if (usageCacheFile) {
28
+ try {
29
+ const saved = JSON.parse(fs.readFileSync(usageCacheFile, 'utf8'));
30
+ if (saved.version === 1 && Array.isArray(saved.turns)) {
31
+ for (const item of saved.turns.slice(0, 128)) {
32
+ if (typeof item.threadId === 'string' && item.threadId.length <= 80 &&
33
+ typeof item.turnId === 'string' && item.turnId.length <= 80) trackedTurns.set(item.threadId, item.turnId);
34
+ }
35
+ }
36
+ } catch (error) { if (error.code !== 'ENOENT') events.emit('diagnostic', `Usage task cache: ${error.message}`); }
37
+ }
38
+ function saveTrackedTurns() {
39
+ if (!usageCacheFile) return;
40
+ try {
41
+ const temp = `${usageCacheFile}.tmp`;
42
+ fs.writeFileSync(temp, JSON.stringify({ version: 1, turns: [...trackedTurns].map(([threadId, turnId]) => ({ threadId, turnId })) }), { mode: 0o600 });
43
+ fs.renameSync(temp, usageCacheFile);
44
+ } catch (error) { events.emit('diagnostic', `Usage task cache: ${error.message}`); }
45
+ }
46
+ let fileRemote, fileRetryTimer, turnLocal, turnRetryTimer, turnPulseTimer, turnObserverReady = false,
47
+ queueTurnResync = () => {}, verifyActiveTurns = () => {}, heartbeat, operation, lastClose, closing = false;
48
+ const alive = () => { clearTimeout(heartbeat); heartbeat = setTimeout(() => remote.terminate(), 75000); };
49
+ remote.on('open', alive); remote.on('ping', alive); remote.on('message', alive);
50
+ function dropLocal(session) {
51
+ const current = locals.get(session);
52
+ if (!current) return;
53
+ locals.delete(session); clearTimeout(current.timer);
54
+ if (current.ready && current.socket?.readyState === WebSocket.OPEN && current.usageRequests.size &&
55
+ remote.readyState === WebSocket.OPEN && !closing) {
56
+ // Keep only the in-flight submission receipt; no late response goes to the phone.
57
+ current.detached = true;
58
+ current.detachedTimer = setTimeout(() => current.socket?.terminate(), 35000);
59
+ } else current.socket?.terminate();
60
+ current.probe?.then(probe => probe.close());
61
+ current.fileProbe?.then(probe => probe.close());
62
+ current.diagnosticProbe?.then(probe => probe.close());
63
+ }
64
+ function dropLocals() {
65
+ for (const session of [...locals.keys()]) dropLocal(session);
66
+ for (const socket of fileLocals.values()) socket.terminate();
67
+ fileLocals.clear(); clearTimeout(fileRetryTimer); fileRemote?.terminate(); fileRemote = undefined;
68
+ clearTimeout(turnRetryTimer); clearInterval(turnPulseTimer); turnLocal?.terminate(); turnLocal = undefined;
69
+ turnObserverReady = false; activeTurns.clear(); completedTurns.clear(); verifyActiveTurns = () => {};
70
+ }
71
+ function startTurnReceiver() {
72
+ if (closing || remote.readyState !== WebSocket.OPEN) return;
73
+ let connection;
74
+ try { connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken }; loopbackUrl(connection.url, '/api/socket'); }
75
+ catch { turnRetryTimer = setTimeout(startTurnReceiver, 3000); return; }
76
+ const socket = turnLocal = new WebSocket(connection.url, { maxPayload: 64 * 1024, perMessageDeflate: false });
77
+ const resyncPending = new Map();
78
+ const pulsePending = new Map();
79
+ let resyncId = 1, resyncQueue = [];
80
+ function resyncNext() {
81
+ if (!turnObserverReady || turnLocal !== socket) return;
82
+ while (resyncPending.size < 8 && resyncQueue.length) {
83
+ const [threadId, turnId] = resyncQueue.shift();
84
+ if (trackedTurns.get(threadId) !== turnId) continue;
85
+ const id = ++resyncId;
86
+ resyncPending.set(id, { threadId, turnId });
87
+ send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
88
+ }
89
+ }
90
+ queueTurnResync = (threadId, turnId) => { resyncQueue.push([threadId, turnId]); resyncNext(); };
91
+ verifyActiveTurns = () => {
92
+ if (!turnObserverReady || turnLocal !== socket) return;
93
+ for (const [threadId, turnId] of trackedTurns) {
94
+ if (activeTurns.get(threadId) !== turnId) continue;
95
+ if (pulsePending.size >= 8) break;
96
+ if ([...pulsePending.values()].some(pending => pending.threadId === threadId)) continue;
97
+ const id = ++resyncId;
98
+ pulsePending.set(id, { threadId, turnId });
99
+ send(socket, { type: 'request', id, method: 'GET', path: `/api/threads/${encodeURIComponent(threadId)}/active-turn` });
100
+ }
101
+ };
102
+ socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
103
+ socket.on('error', error => events.emit('diagnostic', `Local turn observer: ${error.message}`));
104
+ socket.on('close', () => {
105
+ if (turnLocal !== socket) return;
106
+ turnLocal = undefined;
107
+ if (turnObserverReady) send(remote, { type: 'usage-observer-lost' });
108
+ turnObserverReady = false;
109
+ queueTurnResync = () => {};
110
+ verifyActiveTurns = () => {};
111
+ activeTurns.clear(); completedTurns.clear();
112
+ if (!closing && remote.readyState === WebSocket.OPEN) turnRetryTimer = setTimeout(startTurnReceiver, 3000);
113
+ });
114
+ socket.on('message', (bytes, binary) => {
115
+ if (binary || turnLocal !== socket) return socket.terminate();
116
+ try {
117
+ const message = decode(bytes, false);
118
+ if (message.type === 'ready') { send(socket, { type: 'subscribe', id: 1, path: '/api/turn-events' }); return; }
119
+ if (message.type === 'subscribed' && message.id === 1) {
120
+ turnObserverReady = true; resyncQueue = [...trackedTurns]; resyncNext(); return;
121
+ }
122
+ if (message.type === 'response' && pulsePending.has(message.id)) {
123
+ const { threadId, turnId } = pulsePending.get(message.id);
124
+ pulsePending.delete(message.id);
125
+ if (trackedTurns.get(threadId) !== turnId || activeTurns.get(threadId) !== turnId) return;
126
+ if (message.status === 200 && message.body?.turnId === turnId) {
127
+ send(remote, { type: 'usage-turn', method: 'heartbeat', threadId, turnId });
128
+ } else if (message.status === 200 || message.status === 404) {
129
+ activeTurns.delete(threadId);
130
+ if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
131
+ send(remote, { type: 'usage-turn', method: 'inactive', threadId, turnId });
132
+ } else socket.close();
133
+ return;
134
+ }
135
+ if (message.type === 'response' && resyncPending.has(message.id)) {
136
+ const { threadId, turnId } = resyncPending.get(message.id);
137
+ resyncPending.delete(message.id);
138
+ if (trackedTurns.get(threadId) === turnId) {
139
+ if (message.status === 200 && message.body?.turnId === turnId) {
140
+ activeTurns.set(threadId, turnId);
141
+ send(remote, { type: 'usage-turn', method: 'resume', threadId, turnId });
142
+ } else if (message.status === 200 || message.status === 404) {
143
+ trackedTurns.delete(threadId); saveTrackedTurns();
144
+ } else { socket.close(); return; }
145
+ }
146
+ resyncNext(); return;
147
+ }
148
+ if (message.type === 'event' && message.id === 1 && message.event === 'turn') {
149
+ const { method, threadId, turnId } = message.data || {};
150
+ if (!['turn/started', 'turn/completed'].includes(method) || typeof threadId !== 'string' || typeof turnId !== 'string') return;
151
+ const key = `${threadId}:${turnId}`;
152
+ const tracked = trackedTurns.get(threadId) === turnId;
153
+ // Keep local event state for a phone submission receipt arriving later.
154
+ // Only registered usage turns need cloud updates or periodic verification.
155
+ if (method === 'turn/started') activeTurns.set(threadId, turnId);
156
+ else {
157
+ if (activeTurns.get(threadId) === turnId) activeTurns.delete(threadId);
158
+ if (trackedTurns.get(threadId) === turnId) { trackedTurns.delete(threadId); saveTrackedTurns(); }
159
+ completedTurns.set(key, Date.now());
160
+ if (completedTurns.size > 100) completedTurns.delete(completedTurns.keys().next().value);
161
+ }
162
+ if (tracked) send(remote, { type: 'usage-turn', method, threadId, turnId });
163
+ } else if (message.type === 'end' || (message.type === 'response' && message.status >= 400)) socket.close();
164
+ } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
165
+ });
166
+ }
167
+ function startFileReceiver() {
168
+ if (!identity || closing || remote.readyState !== WebSocket.OPEN) return;
169
+ const receiver = fileRemote = new WebSocket(relayUrl, { maxPayload: 512 * 1024, perMessageDeflate: false });
170
+ receiver.on('open', () => send(receiver, { type: 'file-host', deviceId: identity.id, deviceSecret: identity.secret }));
171
+ receiver.on('error', error => events.emit('diagnostic', `File receiver: ${error.message}`));
172
+ receiver.on('close', () => {
173
+ for (const socket of fileLocals.values()) socket.terminate();
174
+ fileLocals.clear();
175
+ if (fileRemote === receiver) {
176
+ fileRemote = undefined;
177
+ if (!closing && remote.readyState === WebSocket.OPEN) fileRetryTimer = setTimeout(startFileReceiver, 3000);
178
+ }
179
+ });
180
+ receiver.on('message', (bytes, binary) => {
181
+ try {
182
+ if (binary) {
183
+ const { session, data } = parseFileBytes(bytes);
184
+ const socket = fileLocals.get(session);
185
+ if (socket?.readyState === WebSocket.OPEN) {
186
+ if (socket.bufferedAmount > 1024 * 1024) { socket.close(1013, 'File receiver too slow'); return; }
187
+ socket.send(data);
188
+ }
189
+ return;
190
+ }
191
+ const message = decode(bytes, binary);
192
+ if (message.type === 'file-host-ready') return;
193
+ if (message.type === 'file-open') {
194
+ const connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
195
+ const url = new URL(connection.url); url.pathname = '/api/file-transfer';
196
+ loopbackUrl(url.href, '/api/file-transfer');
197
+ const socket = new WebSocket(url.href, { maxPayload: 512 * 1024, perMessageDeflate: false });
198
+ fileLocals.set(message.session, socket);
199
+ socket.on('open', () => send(socket, { type: 'auth', token: connection.token, purpose: message.purpose }));
200
+ socket.on('error', error => events.emit('diagnostic', `Local file receiver: ${error.message}`));
201
+ socket.on('close', () => {
202
+ if (fileLocals.get(message.session) !== socket) return;
203
+ fileLocals.delete(message.session);
204
+ send(receiver, { type: 'file-close', session: message.session });
205
+ });
206
+ socket.on('message', (data, isBinary) => {
207
+ if (fileLocals.get(message.session) !== socket) return socket.terminate();
208
+ if (receiver.readyState !== WebSocket.OPEN || receiver.bufferedAmount > 1024 * 1024) return socket.close(1013, 'File receiver too slow');
209
+ if (isBinary) receiver.send(routedFileBytes(message.session, data));
210
+ else send(receiver, { type: 'file-frame', session: message.session, payload: data.toString() });
211
+ });
212
+ } else if (message.type === 'file-frame') {
213
+ const socket = fileLocals.get(message.session);
214
+ if (socket?.readyState === WebSocket.OPEN) {
215
+ if (socket.bufferedAmount > 1024 * 1024) { socket.close(1013, 'File receiver too slow'); return; }
216
+ socket.send(message.payload);
217
+ }
218
+ } else if (message.type === 'file-close') fileLocals.get(message.session)?.terminate();
219
+ else throw new Error('Invalid file receiver frame');
220
+ } catch (error) { events.emit('diagnostic', error.message); receiver.close(1008, 'Invalid file frame'); }
221
+ });
222
+ }
223
+ const ready = new Promise((resolve, reject) => {
224
+ let registered = false;
225
+ const timer = setTimeout(() => { reject(new Error('Relay registration timeout')); remote.terminate(); }, 5000);
226
+ remote.on('open', () => send(remote, { type: 'host', key: hostKey, hostName, lanCandidates, p2pProbe: p2pProbe ? 1 : 0, p2pProbeRepeat: true,
227
+ ...(identity ? { deviceId: identity.id, deviceSecret: identity.secret } : {}) }));
228
+ remote.on('error', error => { reject(error); events.emit('diagnostic', error.message); });
229
+ remote.on('close', (code, reason) => {
230
+ clearTimeout(timer); clearTimeout(heartbeat); dropLocals();
231
+ const error = lastClose = Object.assign(new Error(`Relay closed (${code}): ${reason.toString()}`), { code });
232
+ if (!registered) reject(error);
233
+ if (operation) { clearTimeout(operation.timer); operation.reject(error); operation = undefined; }
234
+ events.emit('disconnected', { code, reason: reason.toString() });
235
+ });
236
+ remote.on('message', (bytes, binary) => {
237
+ try {
238
+ const message = decode(bytes, binary);
239
+ if (message.type === 'registered' && !registered) {
240
+ registered = true; clearTimeout(timer); startFileReceiver(); startTurnReceiver();
241
+ turnPulseTimer = setInterval(() => {
242
+ verifyActiveTurns();
243
+ }, 10000);
244
+ resolve(message); return;
245
+ }
246
+ if (!registered) throw new Error('Relay not registered');
247
+ if (['pair-code', 'revoked', 'status', 'invite-create', 'invite-list', 'invite-revoke', 'devices-list', 'device-revoke'].includes(message.type)) {
248
+ if (operation && message.commandId === operation.id && message.type === operation.responseType) {
249
+ clearTimeout(operation.timer); const pending = operation; operation = undefined;
250
+ message.error ? pending.reject(new Error(message.error)) : pending.resolve(message);
251
+ }
252
+ return;
253
+ }
254
+ if (message.type === 'authorization-changed' && Array.isArray(message.clients) && Number.isSafeInteger(message.revision)) {
255
+ events.emit('authorization-changed', { clients: message.clients, revision: message.revision }); return;
256
+ }
257
+ if (message.type === 'attach') {
258
+ events.emit('attached', { session: message.session, clientId: message.clientId });
259
+ dropLocal(message.session);
260
+ const current = { socket: undefined, ready: false, timer: undefined, p2pProbe: p2pProbe && message.p2pProbe === 1,
261
+ bridgeOutbound: Promise.resolve(), healthRequests: new Set(), usageRequests: new Map(), detached: false, detachedTimer: undefined };
262
+ locals.set(message.session, current);
263
+ let connection;
264
+ try {
265
+ connection = bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken };
266
+ loopbackUrl(connection.url, '/api/socket');
267
+ } catch {
268
+ send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
269
+ events.emit('diagnostic', 'Local desktop service connection not ready'); return;
270
+ }
271
+ const socket = current.socket = new WebSocket(connection.url, { maxPayload: 64 * 1024 * 1024, perMessageDeflate: false });
272
+ current.forwardToBridge = payload => {
273
+ if (locals.get(message.session) !== current || !current.ready) throw new Error('Local desktop service connection not ready');
274
+ const request = decode(payload, false);
275
+ if (!['request', 'subscribe', 'unsubscribe', 'response-ack', 'cancel'].includes(request.type)) throw new Error('Unsupported LAN message');
276
+ if (request.type === 'request' && request.method === 'GET' && request.path === '/api/health' && Number.isSafeInteger(request.id)) {
277
+ current.healthRequests.add(request.id);
278
+ }
279
+ if (request.type === 'cancel') current.healthRequests.delete(request.id);
280
+ if (request.type === 'request' && request.method === 'POST' && Number.isSafeInteger(request.id)) {
281
+ const path = (request.path || '').split('?')[0];
282
+ const existing = /^\/api\/threads\/([^/?]+)\/(?:messages|queue)$/.exec(path);
283
+ const created = /^\/api\/projects\/[^/?]+\/threads$/.test(path);
284
+ if (existing || created) {
285
+ const threadId = existing ? decodeURIComponent(existing[1]) : null;
286
+ const messageKey = typeof request.body?.clientUserMessageId === 'string' && request.body.clientUserMessageId.length <= 64
287
+ ? request.body.clientUserMessageId : `${message.session}:${request.id}`;
288
+ current.usageRequests.set(request.id, { threadId, messageKey, created });
289
+ send(remote, { type: 'usage-arm', session: message.session, requestId: request.id, threadId });
290
+ }
291
+ }
292
+ send(socket, request);
293
+ };
294
+ current.timer = setTimeout(() => socket.terminate(), 5000);
295
+ socket.on('open', () => send(socket, { type: 'auth', token: connection.token }));
296
+ socket.on('error', error => events.emit('diagnostic', `Local desktop service: ${error.message}`));
297
+ socket.on('close', () => {
298
+ clearTimeout(current.detachedTimer);
299
+ if (locals.get(message.session)?.socket !== socket) return;
300
+ send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); events.emit('detached', { session: message.session });
301
+ });
302
+ socket.on('message', (data, isBinary) => {
303
+ if (locals.get(message.session)?.socket !== socket && !current.detached) return;
304
+ try {
305
+ const response = decode(data, isBinary);
306
+ if (!current.ready) {
307
+ if (response.type !== 'ready') throw new Error('Local desktop service authentication failed');
308
+ clearTimeout(current.timer); current.ready = true; send(remote, { type: 'ready', session: message.session });
309
+ } else {
310
+ const usage = response.type === 'response' ? current.usageRequests.get(response.id) : null;
311
+ if (usage) {
312
+ current.usageRequests.delete(response.id);
313
+ const threadId = usage.threadId || response.body?.threadId;
314
+ if (response.status >= 200 && response.status < 300 && typeof threadId === 'string') {
315
+ const turnId = usage.created ? response.body?.turnId : response.body?.usage?.turnId;
316
+ const turnKind = usage.created ? 'start' : response.body?.usage?.kind;
317
+ send(remote, { type: 'usage-message', session: message.session, requestId: response.id,
318
+ threadId, messageKey: usage.messageKey, turnId, turnKind,
319
+ completed: completedTurns.has(`${threadId}:${turnId}`), observerReady: turnObserverReady,
320
+ active: activeTurns.get(threadId) === turnId });
321
+ if (['start', 'steer'].includes(turnKind) && typeof turnId === 'string' &&
322
+ !completedTurns.has(`${threadId}:${turnId}`)) {
323
+ trackedTurns.set(threadId, turnId); saveTrackedTurns();
324
+ if (turnObserverReady && !activeTurns.has(threadId)) queueTurnResync(threadId, turnId);
325
+ }
326
+ } else send(remote, { type: 'usage-cancel', session: message.session, requestId: response.id });
327
+ }
328
+ if (current.detached) {
329
+ if (!current.usageRequests.size) socket.terminate();
330
+ return;
331
+ }
332
+ const payload = data.toString();
333
+ // Native health probes use the relay socket. Keep their replies on that
334
+ // route, outside the RTC queue shared by chat and file downloads.
335
+ if (response.type === 'response' && current.healthRequests.delete(response.id)) {
336
+ send(remote, { type: 'data', session: message.session, payload });
337
+ return;
338
+ }
339
+ current.bridgeOutbound = current.bridgeOutbound.then(async () => {
340
+ const probe = await current.probe;
341
+ if (locals.get(message.session) !== current || !current.ready) return;
342
+ if (!probe?.send?.(payload)) send(remote, { type: 'data', session: message.session, payload });
343
+ });
344
+ void current.bridgeOutbound.catch(error => { events.emit('diagnostic', error.message); socket.terminate(); });
345
+ }
346
+ } catch (error) { events.emit('diagnostic', error.message); socket.terminate(); }
347
+ });
348
+ } else if (message.type === 'p2p-probe' && locals.has(message.session)) {
349
+ const current = locals.get(message.session);
350
+ if (!current.ready || !current.p2pProbe || !validProbeSignal(message)) throw new Error('Invalid probe signal');
351
+ if (message.purpose === 'diagnostic') {
352
+ if (message.kind === 'start' && current.diagnosticProbeId !== message.probeId) {
353
+ void current.diagnosticProbe?.then(probe => probe.close());
354
+ current.diagnosticProbeId = message.probeId;
355
+ const probeId = message.probeId;
356
+ current.diagnosticProbe = createP2pProbe({ iceServers,
357
+ signal: signal => {
358
+ if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
359
+ send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
360
+ version: 1, probeId, ...signal });
361
+ }
362
+ },
363
+ report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic })
364
+ }).catch(() => {
365
+ if (locals.get(message.session) === current && current.diagnosticProbeId === probeId) {
366
+ send(remote, { type: 'p2p-probe', purpose: 'diagnostic', session: message.session,
367
+ version: 1, probeId, kind: 'close' });
368
+ }
369
+ return { receive() {}, close() {} };
370
+ });
371
+ } else if (current.diagnosticProbe && current.diagnosticProbeId === message.probeId) {
372
+ void current.diagnosticProbe.then(probe => probe.receive(message));
373
+ }
374
+ return;
375
+ }
376
+ if (message.purpose && message.purpose !== 'file') throw new Error('Invalid probe purpose');
377
+ const file = message.purpose === 'file';
378
+ const probeField = file ? 'fileProbe' : 'probe', idField = file ? 'fileProbeId' : 'probeId';
379
+ const signalsField = file ? 'fileRouteSignals' : 'routeSignals', helperField = file ? 'fileRouteHelper' : 'routeHelper';
380
+ if (message.kind === 'start' && current[idField] !== message.probeId) {
381
+ void current[probeField]?.then(probe => probe.close());
382
+ current[idField] = message.probeId;
383
+ current[signalsField] = Promise.resolve();
384
+ const probeId = message.probeId;
385
+ current[helperField] = createWindowsRouteHelper({ enabled: p2pWindowsRouteHelper, token: p2pWindowsRouteHelperToken, iceServers,
386
+ report: (event, details) => events.emit('diagnostic', `${event};count=${details.count}`) });
387
+ current[probeField] = current[helperField].prepare().then(() => createP2pProbe({ iceServers, purpose: file ? 'file' : '',
388
+ signal: signal => {
389
+ if (locals.get(message.session) === current && current[idField] === probeId) send(remote, { type: 'p2p-probe', session: message.session, ...(file ? { purpose: 'file' } : {}),
390
+ version: 1, probeId, ...signal });
391
+ },
392
+ onData: file ? undefined : payload => current.forwardToBridge(payload),
393
+ onFileChannel: file ? channel => attachRtcFile(channel, bridgeConnection?.() || { url: bridgeUrl, token: bridgeToken },
394
+ () => locals.get(message.session) === current && current.ready && current[idField] === probeId && remote.readyState === WebSocket.OPEN) : undefined,
395
+ report: (state, diagnostic) => events.emit('p2p-probe', { state, diagnostic, purpose: file ? 'file' : 'chat' })
396
+ })).catch(() => {
397
+ events.emit('diagnostic', p2pWindowsRouteHelper ? 'Windows RTC 路由辅助不可用' : 'WebRTC probe initialization failed');
398
+ if (locals.get(message.session) === current && current[idField] === probeId) send(remote, {
399
+ type: 'p2p-probe', session: message.session, version: 1, probeId, ...(file ? { purpose: 'file' } : {}), kind: 'close' });
400
+ return { receive() {}, close() {} };
401
+ });
402
+ } else if (current[probeField] && current[idField] === message.probeId) {
403
+ const pending = current[probeField];
404
+ current[signalsField] ||= Promise.resolve();
405
+ current[signalsField] = current[signalsField].then(async () => {
406
+ await current[helperField]?.include(message);
407
+ const probe = await pending;
408
+ if (locals.get(message.session) === current && current[idField] === message.probeId) probe.receive(message);
409
+ });
410
+ void current[signalsField]
411
+ .catch(() => { events.emit('diagnostic', 'WebRTC probe negotiation failed'); void pending.then(probe => probe.close()); });
412
+ }
413
+ } else if (message.type === 'detach' && locals.has(message.session)) { dropLocal(message.session); events.emit('detached', { session: message.session }); }
414
+ else if (message.type === 'data' && locals.has(message.session)) {
415
+ const current = locals.get(message.session);
416
+ if (!current.ready) { send(remote, { type: 'detach', session: message.session }); dropLocal(message.session); return; }
417
+ current.forwardToBridge(message.payload);
418
+ }
419
+ } catch (error) { events.emit('diagnostic', error.message); remote.close(1008, 'Invalid relay message'); }
420
+ });
421
+ });
422
+ return { ready, events,
423
+ assertConnected() { if (remote.readyState !== WebSocket.OPEN) throw lastClose || new Error('Relay disconnected during registration'); },
424
+ reportDeviceConnection(info) { return send(remote, { ...info, type: 'device-connected' }); },
425
+ command(type, details = {}) {
426
+ if (operation) return Promise.reject(new Error('已有设备管理操作正在执行'));
427
+ return new Promise((resolve, reject) => {
428
+ const timer = setTimeout(() => { operation = undefined; reject(new Error('设备管理请求超时,结果未确认;未自动重发')); }, 7000);
429
+ const id = randomUUID();
430
+ operation = { resolve, reject, timer, id, responseType: type === 'revoke' ? 'revoked' : type };
431
+ if (!send(remote, { type, commandId: id, ...details })) { clearTimeout(timer); operation = undefined; reject(new Error('中继未连接')); }
432
+ });
433
+ },
434
+ updateLanCandidates(candidates) {
435
+ if (!send(remote, { type: 'lan-candidates', candidates })) throw new Error('中继未连接');
436
+ },
437
+ close() { closing = true; clearTimeout(heartbeat); dropLocals(); remote.terminate(); }
438
+ };
439
+ }