claude-simple 0.1.0 → 0.2.1

package/client.js

@@ -1,5 +1,6 @@
 #!/usr/bin/env node
-import { createApp, init } from './src/app.js';
+import { createApp, init, hasBusyAgent } from './src/app.js';
+import { startAutoUpdate } from './src/auto-update.js';
 import { WebSocket } from 'ws';
 import os from 'os';
 import crypto from 'crypto';
@@ -18,6 +19,10 @@ const { server } = createApp();
 await new Promise(resolve => server.listen(LOCAL_PORT, '127.0.0.1', resolve));
 console.log(`[client] Local app listening on 127.0.0.1:${LOCAL_PORT}`);
 
+// Self-update from npm (default every 12h). Disable with AUTO_UPDATE=0.
+// Requires global install + systemd Restart= (or equivalent) for full apply.
+startAutoUpdate({ role: 'edge-client', isBusy: hasBusyAgent });
+
 // ─── Active WS tunnels: tunnelId → local WebSocket ───────────────────────────
 const tunnels = new Map();
 
@@ -35,6 +40,7 @@ function connect() {
   console.log(`[client] Connecting to gateway ${GATEWAY_URL} ...`);
   controlWs = new WebSocket(GATEWAY_URL, {
     headers: { 'x-machine-token': MACHINE_TOKEN },
+    perMessageDeflate: true,
   });
 
   controlWs.on('open', () => {
@@ -60,12 +66,35 @@ function connect() {
           headers: { ...msg.headers, host: `127.0.0.1:${LOCAL_PORT}` },
           body: msg.body || undefined,
         });
+        const headers = Object.fromEntries(res.headers.entries());
+        const contentType = res.headers.get('content-type') || '';
+
+        // SSE (and any other body-less-until-done response) is forwarded as
+        // it arrives instead of buffered in full, so remote/hub viewers get
+        // real-time streaming instead of one lump delivered at the end.
+        if (contentType.includes('text/event-stream') && res.body) {
+          sendControl({ type: 'http-res-start', reqId: msg.reqId, status: res.status, headers });
+          const reader = res.body.getReader();
+          const decoder = new TextDecoder();
+          for (;;) {
+            const { done, value } = await reader.read();
+            if (done) break;
+            sendControl({ type: 'http-chunk', reqId: msg.reqId, data: decoder.decode(value, { stream: true }) });
+          }
+          // Flush a possibly incomplete UTF-8 code point held by TextDecoder
+          // before terminating the stream (SSE payloads often contain Chinese).
+          const tail = decoder.decode();
+          if (tail) sendControl({ type: 'http-chunk', reqId: msg.reqId, data: tail });
+          sendControl({ type: 'http-end', reqId: msg.reqId });
+          return;
+        }
+
         const body = await res.text();
         sendControl({
           type: 'http-res',
           reqId: msg.reqId,
           status: res.status,
-          headers: Object.fromEntries(res.headers.entries()),
+          headers,
           body,
         });
       } catch (e) {