@yeaft/webchat-agent 1.0.445 → 1.0.447

package/local-runtime/server/ws-agent.js

@@ -5,7 +5,7 @@ import { verifyAgent } from './auth.js';
 
 import { encodeKey } from './encryption.js';
 import { agents, pendingAgentConnections } from './context.js';
-import { userDb } from './database.js';
+import { agentInventoryDb, userDb } from './database.js';
 import {
   parseMessage, broadcastAgentList, clearAgentDirCache
 } from './ws-utils.js';
@@ -52,6 +52,35 @@ function pruneAgentConnectionGenerations() {
   }
 }
 
+function persistAgentInventory(agentId, agent) {
+  if (!agentInventoryDb?.upsert || !agent?.name) return;
+  try {
+    agentInventoryDb.upsert({
+      ...agent,
+      id: agentId,
+      lastSeenAt: agent.lastSeenAt,
+      lastConnectedAt: agent.lastConnectedAt,
+    });
+  } catch (error) {
+    // The inventory is an admin read model. Never reject a live Agent because
+    // this best-effort durability write failed.
+    console.error(`[AgentInventory] Failed to persist ${agentId}:`, error.message);
+  }
+}
+
+function persistAgentHeartbeat(agentId, agent, now = Date.now()) {
+  if (!agentInventoryDb?.touch || !agent) return;
+  const lastPersistedAt = Number(agent.inventoryLastPersistedAt) || 0;
+  if (lastPersistedAt && now - lastPersistedAt < 10_000) return;
+  agent.inventoryLastPersistedAt = now;
+  try {
+    const touched = agentInventoryDb.touch(agentId, now);
+    if (touched === false) persistAgentInventory(agentId, agent);
+  } catch (error) {
+    console.error(`[AgentInventory] Failed to persist heartbeat for ${agentId}:`, error.message);
+  }
+}
+
 export function handleAgentConnection(ws, url) {
   const clientAgentId = url.searchParams.get('id') || randomUUID();
   const agentName = url.searchParams.get('name') || `Agent-${clientAgentId.slice(0, 8)}`;
@@ -183,7 +212,9 @@ export function handleAgentConnection(ws, url) {
         ws.close(1008, 'Account disabled');
         return;
       }
-      markAgentHeartbeatSeen(agent);
+      const now = Date.now();
+      markAgentHeartbeatSeen(agent, now);
+      persistAgentHeartbeat(resolvedAgentId, agent, now);
       const msg = await parseMessage(data, agent.sessionKey);
       if (msg) {
         console.log(`[Agent] Received message from ${resolvedAgentId}: ${msg.type}`);
@@ -235,6 +266,9 @@ export function handleAgentConnection(ws, url) {
 function handleAgentDisconnect(agentId, agentName, ws) {
   const agent = agents.get(agentId);
   if (!agent || agent.ws !== ws) return;
+  // Capture the last in-memory heartbeat before removing the live socket. The
+  // durable row remains an offline historical record; it never grants access.
+  persistAgentInventory(agentId, agent);
   // Phase 4: 清理目录缓存
   clearAgentDirCache(agentId);
   clearWorkbenchCorrelationsForAgent(agentId);
@@ -280,6 +314,7 @@ function completeAgentRegistration(ws, agentId, agentName, workDir, sessionKey,
     sessionKey,
     isAlive: true,
     lastSeenAt: Date.now(),
+    lastConnectedAt: Date.now(),
     capabilities: effectiveCapabilities,
     // The fallback list supports old UI features but cannot prove that an
     // Agent made an explicit capability claim.
@@ -305,6 +340,7 @@ function completeAgentRegistration(ws, agentId, agentName, workDir, sessionKey,
     }
   }, 30000);
   agents.get(agentId)._syncTimeout = syncTimeout;
+  persistAgentInventory(agentId, agents.get(agentId));
 
   if (existingAgent?.ws && existingAgent.ws !== ws) {
     clearBrowserRuntimeForAgent(agentId);
@@ -315,7 +351,9 @@ function completeAgentRegistration(ws, agentId, agentName, workDir, sessionKey,
   ws.on('pong', () => {
     const agent = agents.get(agentId);
     if (agent?.ws === ws) {
-      markAgentHeartbeatSeen(agent);
+      const now = Date.now();
+      markAgentHeartbeatSeen(agent, now);
+      persistAgentHeartbeat(agentId, agent, now);
       if (agent.pingSentAt) {
         agent.latency = Date.now() - agent.pingSentAt;
         agent.pingSentAt = null;

package/local-runtime/server/ws-client.js

@@ -5,7 +5,7 @@ import { generateSkipAuthSession } from './auth.js';
 import { authenticateRequest } from './auth/request-auth.js';
 import { encodeKey } from './encryption.js';
 import { userDb } from './database.js';
-import { agents, clearYeaftDebugRequestsForClient, webClients, isHeartbeatMessageType, trackRequest } from './context.js';
+import { agents, clearYeaftDebugRequestsForClient, webClients } from './context.js';
 import {
   applyClientHello,
   BROWSER_RUNTIME_PROTOCOL,
@@ -136,12 +136,6 @@ export function handleWebConnection(ws, url, req = {}) {
     const client = webClients.get(clientId);
     const msg = await parseMessage(data, client?.sessionKey);
     if (!msg) return;
-    // Stats tracking: exclude heartbeat/control frames. User-turn bytes are
-    // accounted at the send handlers, not here, so pings and dashboard polling
-    // don't inflate traffic.
-    if (!isHeartbeatMessageType(msg.type)) {
-      trackRequest(client?.userId, data.length || 0, msg.type);
-    }
     if (msg.perfTraceId) {
       recordPerfTraceEvent({
         traceId: msg.perfTraceId,

package/local-runtime/version.json

@@ -1 +1 @@
-{"version":"1.0.445"}
+{"version":"1.0.447"}