@agentmeshhq/agent 0.1.0

package/src/core/websocket.ts

@@ -0,0 +1,94 @@
+import WebSocket from "ws";
+
+export interface WebSocketConfig {
+  url: string;
+  token: string;
+  onMessage: (data: WebSocketEvent) => void;
+  onConnect?: () => void;
+  onDisconnect?: () => void;
+  onError?: (error: Error) => void;
+}
+
+export interface WebSocketEvent {
+  type: string;
+  [key: string]: unknown;
+}
+
+export class AgentWebSocket {
+  private ws: WebSocket | null = null;
+  private config: WebSocketConfig;
+  private reconnectAttempts = 0;
+  private maxReconnectAttempts = 10;
+  private reconnectDelay = 1000;
+  private shouldReconnect = true;
+
+  constructor(config: WebSocketConfig) {
+    this.config = config;
+  }
+
+  connect(): void {
+    if (this.ws?.readyState === WebSocket.OPEN) {
+      return;
+    }
+
+    const wsUrl = `${this.config.url}?token=${this.config.token}`;
+
+    this.ws = new WebSocket(wsUrl);
+
+    this.ws.on("open", () => {
+      this.reconnectAttempts = 0;
+      this.config.onConnect?.();
+    });
+
+    this.ws.on("message", (data) => {
+      try {
+        const event = JSON.parse(data.toString()) as WebSocketEvent;
+        this.config.onMessage(event);
+      } catch (error) {
+        console.error("Failed to parse WebSocket message:", error);
+      }
+    });
+
+    this.ws.on("close", () => {
+      this.config.onDisconnect?.();
+      this.scheduleReconnect();
+    });
+
+    this.ws.on("error", (error) => {
+      this.config.onError?.(error);
+    });
+  }
+
+  private scheduleReconnect(): void {
+    if (!this.shouldReconnect) {
+      return;
+    }
+
+    if (this.reconnectAttempts >= this.maxReconnectAttempts) {
+      console.error("Max reconnect attempts reached");
+      return;
+    }
+
+    this.reconnectAttempts++;
+    const delay = this.reconnectDelay * Math.pow(2, this.reconnectAttempts - 1);
+
+    setTimeout(() => {
+      console.log(
+        `Reconnecting... (attempt ${this.reconnectAttempts}/${this.maxReconnectAttempts})`
+      );
+      this.connect();
+    }, delay);
+  }
+
+  disconnect(): void {
+    this.shouldReconnect = false;
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
+    }
+  }
+
+  isConnected(): boolean {
+    return this.ws?.readyState === WebSocket.OPEN;
+  }
+}

package/src/index.ts

@@ -0,0 +1,9 @@
+// Re-export core modules for programmatic usage
+export { AgentDaemon } from "./core/daemon.js";
+export { AgentWebSocket } from "./core/websocket.js";
+export { Heartbeat } from "./core/heartbeat.js";
+export * from "./core/tmux.js";
+export * from "./core/registry.js";
+export * from "./core/injector.js";
+export * from "./config/schema.js";
+export * from "./config/loader.js";

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "../../tsconfig.base.json",
+  "compilerOptions": {
+    "outDir": "dist",
+    "rootDir": "src"
+  },
+  "include": ["src/**/*"]
+}