@agentxjs/network 0.2.0

package/dist/index.js

@@ -0,0 +1,192 @@
+import {
+  WebSocketClient,
+  __publicField,
+  createWebSocketClient
+} from "./chunk-63P5VUHB.js";
+
+// src/WebSocketServer.ts
+import { createLogger } from "@agentxjs/common";
+var logger = createLogger("network/WebSocketServer");
+var WebSocketConnection = class {
+  constructor(ws, options) {
+    __publicField(this, "id");
+    __publicField(this, "ws");
+    __publicField(this, "messageHandlers", /* @__PURE__ */ new Set());
+    __publicField(this, "closeHandlers", /* @__PURE__ */ new Set());
+    __publicField(this, "errorHandlers", /* @__PURE__ */ new Set());
+    __publicField(this, "heartbeatInterval");
+    __publicField(this, "isAlive", true);
+    this.ws = ws;
+    this.id = `conn_${Date.now()}_${Math.random().toString(36).substring(2, 8)}`;
+    if (options.heartbeat !== false) {
+      const interval = options.heartbeatInterval || 3e4;
+      ws.on("pong", () => {
+        this.isAlive = true;
+        logger.debug("Heartbeat pong received", { id: this.id });
+      });
+      this.heartbeatInterval = setInterval(() => {
+        if (!this.isAlive) {
+          logger.warn("Client heartbeat timeout, terminating connection", { id: this.id });
+          clearInterval(this.heartbeatInterval);
+          ws.terminate();
+          return;
+        }
+        this.isAlive = false;
+        ws.ping();
+        logger.debug("Heartbeat ping sent", { id: this.id });
+      }, interval);
+    }
+    ws.on("message", (data) => {
+      const message = data.toString();
+      for (const handler of this.messageHandlers) {
+        handler(message);
+      }
+    });
+    ws.on("close", () => {
+      if (this.heartbeatInterval) {
+        clearInterval(this.heartbeatInterval);
+      }
+      for (const handler of this.closeHandlers) {
+        handler();
+      }
+    });
+    ws.on("error", (err) => {
+      if (this.heartbeatInterval) {
+        clearInterval(this.heartbeatInterval);
+      }
+      for (const handler of this.errorHandlers) {
+        handler(err);
+      }
+    });
+  }
+  send(message) {
+    if (this.ws.readyState === 1) {
+      this.ws.send(message);
+    }
+  }
+  onMessage(handler) {
+    this.messageHandlers.add(handler);
+    return () => {
+      this.messageHandlers.delete(handler);
+    };
+  }
+  onClose(handler) {
+    this.closeHandlers.add(handler);
+    return () => {
+      this.closeHandlers.delete(handler);
+    };
+  }
+  onError(handler) {
+    this.errorHandlers.add(handler);
+    return () => {
+      this.errorHandlers.delete(handler);
+    };
+  }
+  close() {
+    if (this.heartbeatInterval) {
+      clearInterval(this.heartbeatInterval);
+    }
+    this.ws.close();
+  }
+};
+var WebSocketServer = class {
+  constructor(options = {}) {
+    __publicField(this, "wss", null);
+    __publicField(this, "connections", /* @__PURE__ */ new Set());
+    __publicField(this, "connectionHandlers", /* @__PURE__ */ new Set());
+    __publicField(this, "options");
+    __publicField(this, "attachedToServer", false);
+    this.options = options;
+  }
+  async listen(port, host = "0.0.0.0") {
+    if (this.wss) {
+      throw new Error("Server already listening");
+    }
+    if (this.attachedToServer) {
+      throw new Error(
+        "Cannot listen when attached to existing server. The server should call listen() instead."
+      );
+    }
+    const { WebSocketServer: WSS } = await import("ws");
+    this.wss = new WSS({ port, host });
+    this.wss.on("connection", (ws) => {
+      this.handleConnection(ws);
+    });
+    logger.info("WebSocket server listening", { port, host });
+  }
+  attach(server, path = "/ws") {
+    if (this.wss) {
+      throw new Error("Server already initialized");
+    }
+    import("ws").then(({ WebSocketServer: WSS }) => {
+      this.wss = new WSS({ noServer: true });
+      server.on("upgrade", (request, socket, head) => {
+        const url = new URL(request.url || "", `http://${request.headers.host}`);
+        if (url.pathname === path) {
+          this.wss.handleUpgrade(request, socket, head, (ws) => {
+            this.wss.emit("connection", ws, request);
+          });
+        } else {
+          socket.destroy();
+        }
+      });
+      this.wss.on("connection", (ws) => {
+        this.handleConnection(ws);
+      });
+      this.attachedToServer = true;
+      logger.info("WebSocket attached to existing HTTP server", { path });
+    });
+  }
+  handleConnection(ws) {
+    const connection = new WebSocketConnection(ws, this.options);
+    this.connections.add(connection);
+    logger.info("Client connected", {
+      connectionId: connection.id,
+      totalConnections: this.connections.size
+    });
+    connection.onClose(() => {
+      this.connections.delete(connection);
+      logger.info("Client disconnected", {
+        connectionId: connection.id,
+        totalConnections: this.connections.size
+      });
+    });
+    for (const handler of this.connectionHandlers) {
+      handler(connection);
+    }
+  }
+  onConnection(handler) {
+    this.connectionHandlers.add(handler);
+    return () => {
+      this.connectionHandlers.delete(handler);
+    };
+  }
+  broadcast(message) {
+    for (const connection of this.connections) {
+      connection.send(message);
+    }
+  }
+  async close() {
+    if (!this.wss) return;
+    for (const connection of this.connections) {
+      connection.close();
+    }
+    this.connections.clear();
+    if (!this.attachedToServer) {
+      await new Promise((resolve) => {
+        this.wss.close(() => resolve());
+      });
+    }
+    this.wss = null;
+  }
+  async dispose() {
+    await this.close();
+    this.connectionHandlers.clear();
+  }
+};
+export {
+  WebSocketClient,
+  WebSocketServer,
+  createWebSocketClient
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/WebSocketServer.ts"],"sourcesContent":["/**\n * WebSocket Server implementation of ChannelServer\n */\n\nimport type { WebSocket as WS, WebSocketServer as WSS } from \"ws\";\nimport type {\n  ChannelServer,\n  ChannelConnection,\n  ChannelServerOptions,\n  MinimalHTTPServer,\n  Unsubscribe,\n} from \"@agentxjs/types/network\";\nimport { createLogger } from \"@agentxjs/common\";\n\nconst logger = createLogger(\"network/WebSocketServer\");\n\n/**\n * WebSocket connection implementation\n */\nclass WebSocketConnection implements ChannelConnection {\n  public readonly id: string;\n  private ws: WS;\n  private messageHandlers = new Set<(message: string) => void>();\n  private closeHandlers = new Set<() => void>();\n  private errorHandlers = new Set<(error: Error) => void>();\n  private heartbeatInterval?: ReturnType<typeof setInterval>;\n  private isAlive = true;\n\n  constructor(ws: WS, options: ChannelServerOptions) {\n    this.ws = ws;\n    this.id = `conn_${Date.now()}_${Math.random().toString(36).substring(2, 8)}`;\n\n    // Setup heartbeat if enabled\n    if (options.heartbeat !== false) {\n      const interval = options.heartbeatInterval || 30000;\n\n      ws.on(\"pong\", () => {\n        this.isAlive = true;\n        logger.debug(\"Heartbeat pong received\", { id: this.id });\n      });\n\n      this.heartbeatInterval = setInterval(() => {\n        if (!this.isAlive) {\n          logger.warn(\"Client heartbeat timeout, terminating connection\", { id: this.id });\n          clearInterval(this.heartbeatInterval);\n          ws.terminate();\n          return;\n        }\n        this.isAlive = false;\n        ws.ping();\n        logger.debug(\"Heartbeat ping sent\", { id: this.id });\n      }, interval);\n    }\n\n    // Setup message handler\n    ws.on(\"message\", (data: Buffer) => {\n      const message = data.toString();\n      for (const handler of this.messageHandlers) {\n        handler(message);\n      }\n    });\n\n    // Setup close handler\n    ws.on(\"close\", () => {\n      if (this.heartbeatInterval) {\n        clearInterval(this.heartbeatInterval);\n      }\n      for (const handler of this.closeHandlers) {\n        handler();\n      }\n    });\n\n    // Setup error handler\n    ws.on(\"error\", (err: Error) => {\n      if (this.heartbeatInterval) {\n        clearInterval(this.heartbeatInterval);\n      }\n      for (const handler of this.errorHandlers) {\n        handler(err);\n      }\n    });\n  }\n\n  send(message: string): void {\n    if (this.ws.readyState === 1) {\n      // WebSocket.OPEN\n      this.ws.send(message);\n    }\n  }\n\n  onMessage(handler: (message: string) => void): Unsubscribe {\n    this.messageHandlers.add(handler);\n    return () => {\n      this.messageHandlers.delete(handler);\n    };\n  }\n\n  onClose(handler: () => void): Unsubscribe {\n    this.closeHandlers.add(handler);\n    return () => {\n      this.closeHandlers.delete(handler);\n    };\n  }\n\n  onError(handler: (error: Error) => void): Unsubscribe {\n    this.errorHandlers.add(handler);\n    return () => {\n      this.errorHandlers.delete(handler);\n    };\n  }\n\n  close(): void {\n    if (this.heartbeatInterval) {\n      clearInterval(this.heartbeatInterval);\n    }\n    this.ws.close();\n  }\n}\n\n/**\n * WebSocket Server\n */\nexport class WebSocketServer implements ChannelServer {\n  private wss: WSS | null = null;\n  private connections = new Set<WebSocketConnection>();\n  private connectionHandlers = new Set<(connection: ChannelConnection) => void>();\n  private options: ChannelServerOptions;\n  private attachedToServer = false;\n\n  constructor(options: ChannelServerOptions = {}) {\n    this.options = options;\n  }\n\n  async listen(port: number, host: string = \"0.0.0.0\"): Promise<void> {\n    if (this.wss) {\n      throw new Error(\"Server already listening\");\n    }\n    if (this.attachedToServer) {\n      throw new Error(\n        \"Cannot listen when attached to existing server. The server should call listen() instead.\"\n      );\n    }\n\n    const { WebSocketServer: WSS } = await import(\"ws\");\n    this.wss = new WSS({ port, host });\n\n    this.wss.on(\"connection\", (ws: WS) => {\n      this.handleConnection(ws);\n    });\n\n    logger.info(\"WebSocket server listening\", { port, host });\n  }\n\n  attach(server: MinimalHTTPServer, path: string = \"/ws\"): void {\n    if (this.wss) {\n      throw new Error(\"Server already initialized\");\n    }\n\n    import(\"ws\").then(({ WebSocketServer: WSS }) => {\n      this.wss = new WSS({ noServer: true });\n\n      // Handle WebSocket upgrade on the HTTP server\n      server.on(\"upgrade\", (request, socket, head) => {\n        const url = new URL(request.url || \"\", `http://${request.headers.host}`);\n        if (url.pathname === path) {\n          this.wss!.handleUpgrade(request as any, socket as any, head as any, (ws: WS) => {\n            this.wss!.emit(\"connection\", ws, request);\n          });\n        } else {\n          (socket as any).destroy();\n        }\n      });\n\n      this.wss.on(\"connection\", (ws: WS) => {\n        this.handleConnection(ws);\n      });\n\n      this.attachedToServer = true;\n      logger.info(\"WebSocket attached to existing HTTP server\", { path });\n    });\n  }\n\n  private handleConnection(ws: WS): void {\n    const connection = new WebSocketConnection(ws, this.options);\n    this.connections.add(connection);\n\n    logger.info(\"Client connected\", {\n      connectionId: connection.id,\n      totalConnections: this.connections.size,\n    });\n\n    connection.onClose(() => {\n      this.connections.delete(connection);\n      logger.info(\"Client disconnected\", {\n        connectionId: connection.id,\n        totalConnections: this.connections.size,\n      });\n    });\n\n    // Notify handlers\n    for (const handler of this.connectionHandlers) {\n      handler(connection);\n    }\n  }\n\n  onConnection(handler: (connection: ChannelConnection) => void): Unsubscribe {\n    this.connectionHandlers.add(handler);\n    return () => {\n      this.connectionHandlers.delete(handler);\n    };\n  }\n\n  broadcast(message: string): void {\n    for (const connection of this.connections) {\n      connection.send(message);\n    }\n  }\n\n  async close(): Promise<void> {\n    if (!this.wss) return;\n\n    for (const connection of this.connections) {\n      connection.close();\n    }\n    this.connections.clear();\n\n    // Don't close the server if attached to existing HTTP server\n    if (!this.attachedToServer) {\n      await new Promise<void>((resolve) => {\n        this.wss!.close(() => resolve());\n      });\n    }\n    this.wss = null;\n  }\n\n  async dispose(): Promise<void> {\n    await this.close();\n    this.connectionHandlers.clear();\n  }\n}\n"],"mappings":";;;;;;;AAYA,SAAS,oBAAoB;AAE7B,IAAM,SAAS,aAAa,yBAAyB;AAKrD,IAAM,sBAAN,MAAuD;AAAA,EASrD,YAAY,IAAQ,SAA+B;AARnD,wBAAgB;AAChB,wBAAQ;AACR,wBAAQ,mBAAkB,oBAAI,IAA+B;AAC7D,wBAAQ,iBAAgB,oBAAI,IAAgB;AAC5C,wBAAQ,iBAAgB,oBAAI,IAA4B;AACxD,wBAAQ;AACR,wBAAQ,WAAU;AAGhB,SAAK,KAAK;AACV,SAAK,KAAK,QAAQ,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,UAAU,GAAG,CAAC,CAAC;AAG1E,QAAI,QAAQ,cAAc,OAAO;AAC/B,YAAM,WAAW,QAAQ,qBAAqB;AAE9C,SAAG,GAAG,QAAQ,MAAM;AAClB,aAAK,UAAU;AACf,eAAO,MAAM,2BAA2B,EAAE,IAAI,KAAK,GAAG,CAAC;AAAA,MACzD,CAAC;AAED,WAAK,oBAAoB,YAAY,MAAM;AACzC,YAAI,CAAC,KAAK,SAAS;AACjB,iBAAO,KAAK,oDAAoD,EAAE,IAAI,KAAK,GAAG,CAAC;AAC/E,wBAAc,KAAK,iBAAiB;AACpC,aAAG,UAAU;AACb;AAAA,QACF;AACA,aAAK,UAAU;AACf,WAAG,KAAK;AACR,eAAO,MAAM,uBAAuB,EAAE,IAAI,KAAK,GAAG,CAAC;AAAA,MACrD,GAAG,QAAQ;AAAA,IACb;AAGA,OAAG,GAAG,WAAW,CAAC,SAAiB;AACjC,YAAM,UAAU,KAAK,SAAS;AAC9B,iBAAW,WAAW,KAAK,iBAAiB;AAC1C,gBAAQ,OAAO;AAAA,MACjB;AAAA,IACF,CAAC;AAGD,OAAG,GAAG,SAAS,MAAM;AACnB,UAAI,KAAK,mBAAmB;AAC1B,sBAAc,KAAK,iBAAiB;AAAA,MACtC;AACA,iBAAW,WAAW,KAAK,eAAe;AACxC,gBAAQ;AAAA,MACV;AAAA,IACF,CAAC;AAGD,OAAG,GAAG,SAAS,CAAC,QAAe;AAC7B,UAAI,KAAK,mBAAmB;AAC1B,sBAAc,KAAK,iBAAiB;AAAA,MACtC;AACA,iBAAW,WAAW,KAAK,eAAe;AACxC,gBAAQ,GAAG;AAAA,MACb;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,KAAK,SAAuB;AAC1B,QAAI,KAAK,GAAG,eAAe,GAAG;AAE5B,WAAK,GAAG,KAAK,OAAO;AAAA,IACtB;AAAA,EACF;AAAA,EAEA,UAAU,SAAiD;AACzD,SAAK,gBAAgB,IAAI,OAAO;AAChC,WAAO,MAAM;AACX,WAAK,gBAAgB,OAAO,OAAO;AAAA,IACrC;AAAA,EACF;AAAA,EAEA,QAAQ,SAAkC;AACxC,SAAK,cAAc,IAAI,OAAO;AAC9B,WAAO,MAAM;AACX,WAAK,cAAc,OAAO,OAAO;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,QAAQ,SAA8C;AACpD,SAAK,cAAc,IAAI,OAAO;AAC9B,WAAO,MAAM;AACX,WAAK,cAAc,OAAO,OAAO;AAAA,IACnC;AAAA,EACF;AAAA,EAEA,QAAc;AACZ,QAAI,KAAK,mBAAmB;AAC1B,oBAAc,KAAK,iBAAiB;AAAA,IACtC;AACA,SAAK,GAAG,MAAM;AAAA,EAChB;AACF;AAKO,IAAM,kBAAN,MAA+C;AAAA,EAOpD,YAAY,UAAgC,CAAC,GAAG;AANhD,wBAAQ,OAAkB;AAC1B,wBAAQ,eAAc,oBAAI,IAAyB;AACnD,wBAAQ,sBAAqB,oBAAI,IAA6C;AAC9E,wBAAQ;AACR,wBAAQ,oBAAmB;AAGzB,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,MAAM,OAAO,MAAc,OAAe,WAA0B;AAClE,QAAI,KAAK,KAAK;AACZ,YAAM,IAAI,MAAM,0BAA0B;AAAA,IAC5C;AACA,QAAI,KAAK,kBAAkB;AACzB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,EAAE,iBAAiB,IAAI,IAAI,MAAM,OAAO,IAAI;AAClD,SAAK,MAAM,IAAI,IAAI,EAAE,MAAM,KAAK,CAAC;AAEjC,SAAK,IAAI,GAAG,cAAc,CAAC,OAAW;AACpC,WAAK,iBAAiB,EAAE;AAAA,IAC1B,CAAC;AAED,WAAO,KAAK,8BAA8B,EAAE,MAAM,KAAK,CAAC;AAAA,EAC1D;AAAA,EAEA,OAAO,QAA2B,OAAe,OAAa;AAC5D,QAAI,KAAK,KAAK;AACZ,YAAM,IAAI,MAAM,4BAA4B;AAAA,IAC9C;AAEA,WAAO,IAAI,EAAE,KAAK,CAAC,EAAE,iBAAiB,IAAI,MAAM;AAC9C,WAAK,MAAM,IAAI,IAAI,EAAE,UAAU,KAAK,CAAC;AAGrC,aAAO,GAAG,WAAW,CAAC,SAAS,QAAQ,SAAS;AAC9C,cAAM,MAAM,IAAI,IAAI,QAAQ,OAAO,IAAI,UAAU,QAAQ,QAAQ,IAAI,EAAE;AACvE,YAAI,IAAI,aAAa,MAAM;AACzB,eAAK,IAAK,cAAc,SAAgB,QAAe,MAAa,CAAC,OAAW;AAC9E,iBAAK,IAAK,KAAK,cAAc,IAAI,OAAO;AAAA,UAC1C,CAAC;AAAA,QACH,OAAO;AACL,UAAC,OAAe,QAAQ;AAAA,QAC1B;AAAA,MACF,CAAC;AAED,WAAK,IAAI,GAAG,cAAc,CAAC,OAAW;AACpC,aAAK,iBAAiB,EAAE;AAAA,MAC1B,CAAC;AAED,WAAK,mBAAmB;AACxB,aAAO,KAAK,8CAA8C,EAAE,KAAK,CAAC;AAAA,IACpE,CAAC;AAAA,EACH;AAAA,EAEQ,iBAAiB,IAAc;AACrC,UAAM,aAAa,IAAI,oBAAoB,IAAI,KAAK,OAAO;AAC3D,SAAK,YAAY,IAAI,UAAU;AAE/B,WAAO,KAAK,oBAAoB;AAAA,MAC9B,cAAc,WAAW;AAAA,MACzB,kBAAkB,KAAK,YAAY;AAAA,IACrC,CAAC;AAED,eAAW,QAAQ,MAAM;AACvB,WAAK,YAAY,OAAO,UAAU;AAClC,aAAO,KAAK,uBAAuB;AAAA,QACjC,cAAc,WAAW;AAAA,QACzB,kBAAkB,KAAK,YAAY;AAAA,MACrC,CAAC;AAAA,IACH,CAAC;AAGD,eAAW,WAAW,KAAK,oBAAoB;AAC7C,cAAQ,UAAU;AAAA,IACpB;AAAA,EACF;AAAA,EAEA,aAAa,SAA+D;AAC1E,SAAK,mBAAmB,IAAI,OAAO;AACnC,WAAO,MAAM;AACX,WAAK,mBAAmB,OAAO,OAAO;AAAA,IACxC;AAAA,EACF;AAAA,EAEA,UAAU,SAAuB;AAC/B,eAAW,cAAc,KAAK,aAAa;AACzC,iBAAW,KAAK,OAAO;AAAA,IACzB;AAAA,EACF;AAAA,EAEA,MAAM,QAAuB;AAC3B,QAAI,CAAC,KAAK,IAAK;AAEf,eAAW,cAAc,KAAK,aAAa;AACzC,iBAAW,MAAM;AAAA,IACnB;AACA,SAAK,YAAY,MAAM;AAGvB,QAAI,CAAC,KAAK,kBAAkB;AAC1B,YAAM,IAAI,QAAc,CAAC,YAAY;AACnC,aAAK,IAAK,MAAM,MAAM,QAAQ,CAAC;AAAA,MACjC,CAAC;AAAA,IACH;AACA,SAAK,MAAM;AAAA,EACb;AAAA,EAEA,MAAM,UAAyB;AAC7B,UAAM,KAAK,MAAM;AACjB,SAAK,mBAAmB,MAAM;AAAA,EAChC;AACF;","names":[]}

package/package.json

@@ -0,0 +1,60 @@
+{
+  "name": "@agentxjs/network",
+  "version": "0.2.0",
+  "description": "Network layer abstraction for AgentX platform - WebSocket client/server with heartbeat and auto-reconnect",
+  "keywords": [
+    "agentx",
+    "network",
+    "websocket",
+    "channel",
+    "internal"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/deepractice/Agent",
+    "directory": "packages/network"
+  },
+  "license": "MIT",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    },
+    "./browser": {
+      "types": "./dist/browser.d.ts",
+      "import": "./dist/browser.js",
+      "require": "./dist/browser.cjs"
+    }
+  },
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "lint": "eslint .",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@agentxjs/common": "workspace:*",
+    "@agentxjs/types": "workspace:*",
+    "reconnecting-websocket": "^4.4.0",
+    "ws": "^8.18.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.19.1",
+    "@types/ws": "^8.5.13",
+    "tsup": "^8.5.0",
+    "typescript": "^5.9.3"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}