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

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 (185) 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 +62 -58
  7. package/dist/bin/self-relay-admin.mjs +8 -8
  8. package/dist/bin/self-relay-demo.mjs +33 -33
  9. package/dist/bin/self-relay-server.mjs +66 -66
  10. package/dist/build-platform.json +1 -1
  11. package/dist/get-lan-networks.ps1 +40 -40
  12. package/dist/get-shared-service-url.ps1 +6 -6
  13. package/dist/shared/app-server-events.cjs +422 -412
  14. package/dist/shared/code-browser.cjs +159 -159
  15. package/dist/shared/file-changes.cjs +31 -31
  16. package/dist/shared/file-citations.cjs +117 -117
  17. package/dist/shared/file-mentions.cjs +92 -92
  18. package/dist/shared/message-delivery.cjs +54 -54
  19. package/dist/shared/p2p-data-channel.cjs +72 -72
  20. package/dist/shared/p2p-diagnostics.cjs +28 -28
  21. package/dist/shared/p2p-file-channel.cjs +74 -74
  22. package/dist/shared/plan-pricing.cjs +32 -32
  23. package/dist/shared/question-attention.cjs +40 -0
  24. package/dist/shared/server-requests.cjs +219 -219
  25. package/dist/shared/timeline-sync.cjs +173 -173
  26. package/dist/skills/code-relax/SKILL.md +22 -22
  27. package/dist/src/archived-threads.mjs +22 -0
  28. package/dist/src/background-process.mjs +18 -18
  29. package/dist/src/bounded-process-log.mjs +63 -63
  30. package/dist/src/celsius-client.mjs +107 -107
  31. package/dist/src/code-browser.mjs +100 -100
  32. package/dist/src/desktop-launcher.mjs +46 -46
  33. package/dist/src/dots-approval-timeline.mjs +99 -99
  34. package/dist/src/dots-avatar.mjs +42 -42
  35. package/dist/src/dots-chat-projection.mjs +50 -50
  36. package/dist/src/dots-chat-server.mjs +461 -461
  37. package/dist/src/dots-message-images.mjs +319 -319
  38. package/dist/src/dots-profile.mjs +77 -77
  39. package/dist/src/dots-room-activity.mjs +94 -94
  40. package/dist/src/dots-room-client.mjs +135 -135
  41. package/dist/src/dots-room-projection.mjs +94 -94
  42. package/dist/src/dots-schedules.mjs +66 -66
  43. package/dist/src/dots-summary-tasks.mjs +32 -32
  44. package/dist/src/dots-task-catalog.mjs +430 -430
  45. package/dist/src/dots-task-files.mjs +52 -52
  46. package/dist/src/execution-health.mjs +26 -26
  47. package/dist/src/file-transfer-range.mjs +22 -22
  48. package/dist/src/file-upload-session.mjs +130 -130
  49. package/dist/src/hosted-history-app-server.mjs +103 -103
  50. package/dist/src/hosted-history-cache.mjs +113 -113
  51. package/dist/src/lan-socket-server.mjs +265 -263
  52. package/dist/src/managed-app-server.mjs +244 -244
  53. package/dist/src/message-images.mjs +9 -9
  54. package/dist/src/message-upload-files.mjs +74 -74
  55. package/dist/src/onboarding.mjs +12 -12
  56. package/dist/src/plan-pricing.mjs +47 -47
  57. package/dist/src/platform/README.md +12 -12
  58. package/dist/src/platform/codex-protocol.mjs +46 -46
  59. package/dist/src/platform/codex-runtime.mjs +3 -3
  60. package/dist/src/platform/credential-store.mjs +3 -3
  61. package/dist/src/platform/desktop-host.mjs +3 -3
  62. package/dist/src/platform/desktop-project-registration.mjs +3 -3
  63. package/dist/src/platform/desktop-project-store.mjs +208 -149
  64. package/dist/src/platform/file-opener.mjs +9 -0
  65. package/dist/src/platform/linux/README.md +3 -3
  66. package/dist/src/platform/linux/codex-runtime.mjs +56 -0
  67. package/dist/src/platform/linux/credential-store.mjs +28 -0
  68. package/dist/src/platform/linux/desktop-host.mjs +83 -0
  69. package/dist/src/platform/linux/desktop-installation.mjs +36 -0
  70. package/dist/src/platform/linux/desktop-project-registration.mjs +11 -0
  71. package/dist/src/platform/linux/network-info.mjs +11 -0
  72. package/dist/src/platform/linux/process.mjs +42 -0
  73. package/dist/src/platform/linux/project-locations.mjs +17 -0
  74. package/dist/src/platform/linux/resident-startup.mjs +52 -0
  75. package/dist/src/platform/linux/retained-runtime.mjs +76 -0
  76. package/dist/src/platform/linux/service-host.mjs +18 -0
  77. package/dist/src/platform/linux/system.mjs +49 -0
  78. package/dist/src/platform/linux/user-environment.mjs +7 -0
  79. package/dist/src/platform/macos/README.md +3 -3
  80. package/dist/src/platform/macos/codex-runtime.mjs +125 -125
  81. package/dist/src/platform/macos/credential-store.mjs +10 -10
  82. package/dist/src/platform/macos/desktop-host.mjs +82 -82
  83. package/dist/src/platform/macos/desktop-project-registration.mjs +11 -10
  84. package/dist/src/platform/macos/network-info.mjs +8 -28
  85. package/dist/src/platform/macos/project-locations.mjs +8 -8
  86. package/dist/src/platform/macos/resident-startup.mjs +84 -84
  87. package/dist/src/platform/macos/retained-runtime.mjs +127 -127
  88. package/dist/src/platform/macos/service-host.mjs +18 -18
  89. package/dist/src/platform/macos/system.mjs +29 -29
  90. package/dist/src/platform/macos/user-environment.mjs +7 -7
  91. package/dist/src/platform/native-package.mjs +58 -46
  92. package/dist/src/platform/network-info.mjs +3 -3
  93. package/dist/src/platform/path-identity.mjs +5 -5
  94. package/dist/src/platform/posix/network-candidates.mjs +23 -0
  95. package/dist/src/platform/posix/runtime-paths.mjs +25 -25
  96. package/dist/src/platform/project-locations.mjs +3 -3
  97. package/dist/src/platform/runtime-paths.mjs +8 -8
  98. package/dist/src/platform/select.mjs +5 -5
  99. package/dist/src/platform/service-host.mjs +3 -3
  100. package/dist/src/platform/user-environment.mjs +3 -3
  101. package/dist/src/platform/windows/IsolatedProcess.cs +79 -79
  102. package/dist/src/platform/windows/activate-desktop.ps1 +48 -48
  103. package/dist/src/platform/windows/background-process.mjs +30 -30
  104. package/dist/src/platform/windows/codex-runtime.mjs +215 -215
  105. package/dist/src/platform/windows/credential-store.mjs +27 -27
  106. package/dist/src/platform/windows/credential-store.ps1 +49 -49
  107. package/dist/src/platform/windows/desktop-helper.mjs +179 -179
  108. package/dist/src/platform/windows/desktop-host.mjs +58 -58
  109. package/dist/src/platform/windows/desktop-info.ps1 +14 -14
  110. package/dist/src/platform/windows/desktop-project-registration.mjs +4 -4
  111. package/dist/src/platform/windows/desktop-runtime.ps1 +10 -10
  112. package/dist/src/platform/windows/desktop-shortcuts.ps1 +21 -21
  113. package/dist/src/platform/windows/dots-profile.mjs +49 -49
  114. package/dist/src/platform/windows/launch-worker.mjs +41 -41
  115. package/dist/src/platform/windows/network-info.mjs +34 -34
  116. package/dist/src/platform/windows/project-locations.mjs +16 -16
  117. package/dist/src/platform/windows/resident-start.js +20 -20
  118. package/dist/src/platform/windows/resident-start.ps1 +7 -7
  119. package/dist/src/platform/windows/resident-startup.ps1 +83 -83
  120. package/dist/src/platform/windows/resolve-state-root.ps1 +14 -14
  121. package/dist/src/platform/windows/runtime-paths.mjs +20 -20
  122. package/dist/src/platform/windows/rustdesk-video-probe.mjs +167 -167
  123. package/dist/src/platform/windows/service-host.mjs +31 -31
  124. package/dist/src/platform/windows/start-hidden-console.ps1 +49 -49
  125. package/dist/src/platform/windows/stop-desktop.ps1 +22 -22
  126. package/dist/src/platform/windows/user-environment.mjs +50 -50
  127. package/dist/src/platform/windows/user-environment.ps1 +41 -41
  128. package/dist/src/project-archive.mjs +30 -0
  129. package/dist/src/project-create.mjs +121 -119
  130. package/dist/src/project-management.mjs +59 -69
  131. package/dist/src/project-remove.mjs +55 -0
  132. package/dist/src/project-update.mjs +56 -0
  133. package/dist/src/runtime-discovery.mjs +47 -47
  134. package/dist/src/self-relay/account-cleanup.mjs +30 -30
  135. package/dist/src/self-relay/account-connections.mjs +52 -52
  136. package/dist/src/self-relay/admin-cloud.mjs +101 -101
  137. package/dist/src/self-relay/admin-page.mjs +80 -80
  138. package/dist/src/self-relay/admin-server.mjs +74 -74
  139. package/dist/src/self-relay/admin-state.mjs +127 -127
  140. package/dist/src/self-relay/admin-style.mjs +14 -14
  141. package/dist/src/self-relay/android-releases.mjs +61 -61
  142. package/dist/src/self-relay/client.mjs +61 -61
  143. package/dist/src/self-relay/connector.mjs +439 -438
  144. package/dist/src/self-relay/debug-allowlist.mjs +33 -33
  145. package/dist/src/self-relay/demo.mjs +37 -37
  146. package/dist/src/self-relay/device-info.mjs +27 -0
  147. package/dist/src/self-relay/device-store.mjs +92 -84
  148. package/dist/src/self-relay/diagnostics.mjs +54 -54
  149. package/dist/src/self-relay/email-accounts.mjs +183 -183
  150. package/dist/src/self-relay/file-wire.mjs +13 -13
  151. package/dist/src/self-relay/lan-authorizations.mjs +71 -42
  152. package/dist/src/self-relay/lifecycle.mjs +264 -250
  153. package/dist/src/self-relay/long-pairing.mjs +40 -40
  154. package/dist/src/self-relay/mysql-state.mjs +118 -105
  155. package/dist/src/self-relay/p2p-probe.mjs +159 -159
  156. package/dist/src/self-relay/privacy-page.mjs +26 -26
  157. package/dist/src/self-relay/public-releases.mjs +34 -0
  158. package/dist/src/self-relay/rtc-file-bridge.mjs +41 -41
  159. package/dist/src/self-relay/server.mjs +1006 -932
  160. package/dist/src/self-relay/usage-daily.mjs +55 -55
  161. package/dist/src/self-relay/usage-meter.mjs +141 -141
  162. package/dist/src/self-relay/windows-route-helper.mjs +75 -75
  163. package/dist/src/self-relay/wire.mjs +36 -36
  164. package/dist/src/server.mjs +1861 -1778
  165. package/dist/src/service-control.mjs +82 -82
  166. package/dist/src/service-doctor.mjs +110 -110
  167. package/dist/src/service-health.mjs +33 -33
  168. package/dist/src/service-install.mjs +91 -91
  169. package/dist/src/service-lifecycle.mjs +138 -138
  170. package/dist/src/service-restart.mjs +81 -81
  171. package/dist/src/shared-app-server.mjs +615 -611
  172. package/dist/src/shared-catalog.mjs +132 -132
  173. package/dist/src/shared-recovery.mjs +146 -146
  174. package/dist/src/shared-thread-stream.mjs +164 -164
  175. package/dist/src/skill-install.mjs +51 -51
  176. package/dist/src/thread-title-generation.mjs +150 -150
  177. package/dist/src/turn-model-history.mjs +47 -47
  178. package/dist/src/usage-reset.mjs +17 -17
  179. package/dist/src/workstation-downloads.mjs +120 -120
  180. package/dist/tools/rtc-route-helper.ps1 +99 -99
  181. package/dist/tools/start-rtc-route-helper.ps1 +15 -15
  182. package/dist/web/vendor/markdown-it.LICENSE.txt +22 -22
  183. package/dist/web/vendor/markdown-it.umd.min.js +13 -13
  184. package/package.json +7 -6
  185. 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
+ }