2020117-agent 0.1.13 → 0.1.14

package/dist/agent.js

@@ -685,6 +685,37 @@ async function startSwarmListener(label) {
                 handleStop(job, msg.id, label);
         }
     });
+    // Handle customer disconnect — claim any earned tokens immediately
+    node.on('peer-leave', (peerId) => {
+        const tag = peerId.slice(0, 8);
+        // Find and end all sessions for this peer
+        for (const [sessionId, session] of activeSessions) {
+            if (session.peerId === peerId) {
+                console.log(`[${label}] Peer ${tag} disconnected — ending session ${sessionId} (${session.totalEarned} sats earned)`);
+                endSession(node, session, label);
+            }
+        }
+        // Clean up any P2P streaming jobs for this peer
+        for (const [jobId, job] of p2pJobs) {
+            if (job.socket?.remotePublicKey?.toString('hex') === peerId) {
+                console.log(`[${label}] Peer ${tag} disconnected — claiming P2P job ${jobId} tokens`);
+                job.stopped = true;
+                batchClaim(job.tokens, jobId, label);
+                p2pJobs.delete(jobId);
+                releaseSlot();
+            }
+        }
+        // Clean up backend WebSockets for this peer
+        for (const [wsId, entry] of backendWebSockets) {
+            if (entry.peerId === peerId) {
+                try {
+                    entry.ws.close(1001, 'Peer disconnected');
+                }
+                catch { }
+                backendWebSockets.delete(wsId);
+            }
+        }
+    });
 }
 async function runP2PGeneration(node, job, msg, label) {
     const source = SUB_KIND
@@ -728,13 +759,18 @@ function endSession(node, session, label) {
             backendWebSockets.delete(wsId);
         }
     }
-    node.send(session.socket, {
-        type: 'session_end',
-        id: session.sessionId,
-        session_id: session.sessionId,
-        total_sats: session.totalEarned,
-        duration_s: durationS,
-    });
+    try {
+        node.send(session.socket, {
+            type: 'session_end',
+            id: session.sessionId,
+            session_id: session.sessionId,
+            total_sats: session.totalEarned,
+            duration_s: durationS,
+        });
+    }
+    catch {
+        // Socket may already be closed (peer disconnect)
+    }
     console.log(`[${label}] Session ${session.sessionId} ended: ${session.totalEarned} sats, ${durationS}s`);
     // Batch claim tokens
     batchClaim(session.tokens, session.sessionId, label);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "2020117-agent",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "2020117 agent runtime — API polling + Hyperswarm P2P + Cashu streaming payments",
   "type": "module",
   "bin": {