2020117-agent 0.1.13 → 0.1.15

package/dist/agent.js

@@ -121,6 +121,8 @@ const state = {
     swarmNode: null,
     processor: null,
     skill: loadSkill(),
+    p2pSessionsCompleted: 0,
+    p2pTotalEarnedSats: 0,
 };
 // --- Capacity management ---
 function acquireSlot() {
@@ -182,7 +184,11 @@ async function setupPlatform(label) {
         models,
         skill: state.skill,
     });
-    state.stopHeartbeat = startHeartbeatLoop(() => getAvailableCapacity());
+    state.stopHeartbeat = startHeartbeatLoop(() => getAvailableCapacity(), () => ({
+        sessions: state.p2pSessionsCompleted,
+        earned_sats: state.p2pTotalEarnedSats,
+        active: activeSessions.size > 0,
+    }));
 }
 // --- 3. Async Inbox Poller ---
 function startInboxPoller(label) {
@@ -685,6 +691,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,14 +765,22 @@ 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`);
+    // Update P2P lifetime counters
+    state.p2pSessionsCompleted++;
+    state.p2pTotalEarnedSats += session.totalEarned;
     // Batch claim tokens
     batchClaim(session.tokens, session.sessionId, label);
     activeSessions.delete(session.sessionId);

package/dist/api.d.ts

@@ -25,9 +25,14 @@ export declare function registerService(opts: {
     models?: string[];
     skill?: Record<string, unknown> | null;
 }): Promise<unknown | null>;
-export declare function sendHeartbeat(capacity?: number): Promise<boolean>;
+export interface P2PStats {
+    sessions: number;
+    earned_sats: number;
+    active: boolean;
+}
+export declare function sendHeartbeat(capacity?: number, p2pStats?: P2PStats): Promise<boolean>;
 export declare function getOnlineProviders(kind: number): Promise<OnlineAgent[]>;
-export declare function startHeartbeatLoop(capacityOrFn?: number | (() => number)): () => void;
+export declare function startHeartbeatLoop(capacityOrFn?: number | (() => number), p2pStatsFn?: () => P2PStats): () => void;
 export interface InboxJob {
     id: string;
     kind: number;

package/dist/api.js

@@ -161,13 +161,15 @@ export async function registerService(opts) {
         return null;
     }
 }
-export async function sendHeartbeat(capacity) {
+export async function sendHeartbeat(capacity, p2pStats) {
     if (!API_KEY)
         return false;
     try {
         const body = {};
         if (capacity !== undefined)
             body.capacity = capacity;
+        if (p2pStats)
+            body.p2p_stats = p2pStats;
         const resp = await fetch(`${BASE_URL}/api/heartbeat`, {
             method: 'POST',
             headers: {
@@ -206,17 +208,17 @@ export async function getOnlineProviders(kind) {
     }
 }
 const HEARTBEAT_INTERVAL = 5 * 60 * 1000; // 5 minutes
-export function startHeartbeatLoop(capacityOrFn) {
+export function startHeartbeatLoop(capacityOrFn, p2pStatsFn) {
     const getCapacity = typeof capacityOrFn === 'function'
         ? capacityOrFn
         : () => capacityOrFn;
     // Send first heartbeat immediately
-    sendHeartbeat(getCapacity()).then((ok) => {
+    sendHeartbeat(getCapacity(), p2pStatsFn?.()).then((ok) => {
         if (ok)
             console.log('[api] Heartbeat sent');
     });
     const timer = setInterval(() => {
-        sendHeartbeat(getCapacity()).then((ok) => {
+        sendHeartbeat(getCapacity(), p2pStatsFn?.()).then((ok) => {
             if (ok)
                 console.log('[api] Heartbeat sent');
         });

package/package.json

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