@linkshell/gateway 0.3.2 → 0.3.4

package/src/sessions.ts

@@ -33,12 +33,16 @@ export interface HostDevice {
 }
 
 const OUTPUT_BUFFER_CAPACITY = 200;
+const OUTPUT_BUFFER_MAX_PAYLOAD_BYTES = Number(
+  process.env.OUTPUT_BUFFER_MAX_PAYLOAD_BYTES ?? 64 * 1024,
+);
 const HOST_RECONNECT_WINDOW = 60_000;
 const CLEANUP_INTERVAL = 30_000;
 
 export class DeviceManager {
   private devices = new Map<string, HostDevice>();
   private cleanupTimer: ReturnType<typeof setInterval>;
+  private droppedClients = 0;
 
   constructor() {
     this.cleanupTimer = setInterval(() => this.cleanup(), CLEANUP_INTERVAL);
@@ -106,7 +110,7 @@ export class DeviceManager {
   removeClient(hostDeviceId: string, deviceId: string): void {
     const device = this.devices.get(hostDeviceId);
     if (!device) return;
-    device.clients.delete(deviceId);
+    if (device.clients.delete(deviceId)) this.droppedClients++;
     if (device.controllerId === deviceId) {
       const next = device.clients.keys().next();
       device.controllerId = next.done ? undefined : next.value;
@@ -126,6 +130,7 @@ export class DeviceManager {
       device.clients.delete(deviceId);
       closed++;
     }
+    this.droppedClients += closed;
     if (device.controllerId && !device.clients.has(device.controllerId)) {
       const next = device.clients.keys().next();
       device.controllerId = next.done ? undefined : next.value;
@@ -154,6 +159,14 @@ export class DeviceManager {
   bufferOutput(hostDeviceId: string, envelope: Envelope): void {
     const device = this.devices.get(hostDeviceId);
     if (!device) return;
+    const payload = envelope.payload as { data?: unknown } | undefined;
+    if (
+      typeof payload?.data === "string" &&
+      Buffer.byteLength(payload.data, "utf8") > OUTPUT_BUFFER_MAX_PAYLOAD_BYTES
+    ) {
+      device.lastActivity = Date.now();
+      return;
+    }
     const terminalId = envelope.terminalId ?? "default";
     let buffer = device.outputBuffers.get(terminalId);
     if (!buffer) {
@@ -257,6 +270,29 @@ export class DeviceManager {
     };
   }
 
+  getStats() {
+    let clientCount = 0;
+    let bufferedTerminalFrames = 0;
+    let hostAbsentDevices = 0;
+    let terminalCount = 0;
+    for (const device of this.devices.values()) {
+      clientCount += device.clients.size;
+      terminalCount += device.outputBuffers.size;
+      bufferedTerminalFrames += [...device.outputBuffers.values()].reduce((sum, buf) => sum + buf.length, 0);
+      if (!device.host || device.host.socket.readyState !== device.host.socket.OPEN) hostAbsentDevices++;
+    }
+    return {
+      devices: this.devices.size,
+      activeDevices: this.listActive().length,
+      clients: clientCount,
+      droppedClients: this.droppedClients,
+      hostAbsentDevices,
+      terminalsWithReplay: terminalCount,
+      bufferedTerminalFrames,
+      bufferedAgentFrames: 0,
+    };
+  }
+
   setMetadata(
     hostDeviceId: string,
     _provider?: string,