@bytespell/amux 0.0.11 → 0.0.13

package/src/ws-adapter.ts

@@ -0,0 +1,245 @@
+import { WebSocket, WebSocketServer } from 'ws';
+import type { AgentSession, AgentSessionEvents } from './session.js';
+
+/**
+ * WebSocket message from client
+ */
+interface WsClientMessage {
+  type: string;
+  [key: string]: unknown;
+}
+
+/**
+ * Options for the WebSocket adapter
+ */
+export interface WsAdapterOptions {
+  /**
+   * Send history on connect (default: true)
+   */
+  sendHistoryOnConnect?: boolean;
+}
+
+/**
+ * Create a WebSocket adapter for an AgentSession.
+ *
+ * Bridges session events to WebSocket clients and handles incoming messages.
+ *
+ * @example
+ * ```typescript
+ * import { AgentSession } from 'amux';
+ * import { createWsAdapter } from 'amux/ws';
+ * import { WebSocketServer } from 'ws';
+ *
+ * const session = new AgentSession({ ... });
+ * const wss = new WebSocketServer({ server, path: '/ws' });
+ *
+ * createWsAdapter(session, wss);
+ *
+ * await session.spawnAgent();
+ * ```
+ */
+export function createWsAdapter(
+  session: AgentSession,
+  wss: WebSocketServer,
+  options: WsAdapterOptions = {}
+): {
+  /** Get connected client count */
+  clientCount: () => number;
+  /** Broadcast a custom message to all clients */
+  broadcast: (message: unknown) => void;
+  /** Close all connections */
+  close: () => void;
+} {
+  const clients = new Set<WebSocket>();
+  const sendHistoryOnConnect = options.sendHistoryOnConnect ?? true;
+
+  // Helper to broadcast to all clients
+  function broadcast(message: unknown): void {
+    const data = JSON.stringify(message);
+    for (const client of clients) {
+      if (client.readyState === WebSocket.OPEN) {
+        client.send(data);
+      }
+    }
+  }
+
+  // Wire up session events to broadcast
+  const eventHandlers: { [K in keyof AgentSessionEvents]?: (data: AgentSessionEvents[K]) => void } = {
+    ready: (data) => broadcast({ type: 'ready', ...data }),
+    connecting: () => broadcast({ type: 'connecting' }),
+    update: (data) => broadcast({ type: 'session_update', update: data }),
+    turn_start: () => broadcast({ type: 'turn_start' }),
+    turn_end: () => broadcast({ type: 'turn_end' }),
+    permission_request: (data) => broadcast({ type: 'permission_request', ...data }),
+    prompt_complete: (data) => broadcast({ type: 'prompt_complete', ...data }),
+    session_created: (data) => broadcast({ type: 'session_created', ...data }),
+    session_switched: (data) => broadcast({ type: 'session_switched', ...data }),
+    history_replay: (data) => broadcast({ type: 'history_replay', ...data }),
+    error: (data) => broadcast({ type: 'error', ...data }),
+    agent_exit: (data) => broadcast({ type: 'agent_exit', ...data }),
+  };
+
+  // Register all event handlers
+  for (const [event, handler] of Object.entries(eventHandlers)) {
+    session.on(event as keyof AgentSessionEvents, handler as (data: unknown) => void);
+  }
+
+  // Handle WebSocket connections
+  wss.on('connection', (ws: WebSocket) => {
+    console.log('[amux-ws] Client connected');
+    clients.add(ws);
+
+    // Send current state to new client
+    if (session.isConnected) {
+      ws.send(JSON.stringify({
+        type: 'ready',
+        cwd: session.cwd,
+        sessionId: session.sessionId,
+        capabilities: session.agentCapabilities,
+        agent: session.getAgentInfo(),
+        availableAgents: session.getAvailableAgents(),
+      }));
+
+      // Send history to hydrate the chat UI
+      if (sendHistoryOnConnect) {
+        const history = session.loadHistory();
+        if (history.length > 0) {
+          ws.send(JSON.stringify({
+            type: 'history_replay',
+            previousSessionId: session.sessionId,
+            events: history,
+            eventCount: history.length,
+          }));
+        }
+      }
+    } else {
+      ws.send(JSON.stringify({ type: 'connecting' }));
+    }
+
+    // Handle incoming messages
+    ws.on('message', async (data: Buffer) => {
+      try {
+        const msg = JSON.parse(data.toString()) as WsClientMessage;
+        await handleMessage(ws, msg, session);
+      } catch (err) {
+        console.error('[amux-ws] Invalid message:', err);
+        ws.send(JSON.stringify({ type: 'error', message: 'Invalid message format' }));
+      }
+    });
+
+    ws.on('close', () => {
+      console.log('[amux-ws] Client disconnected');
+      clients.delete(ws);
+    });
+
+    ws.on('error', (err) => {
+      console.error('[amux-ws] WebSocket error:', err);
+      clients.delete(ws);
+    });
+  });
+
+  return {
+    clientCount: () => clients.size,
+    broadcast,
+    close: () => {
+      for (const client of clients) {
+        client.close();
+      }
+      clients.clear();
+    },
+  };
+}
+
+/**
+ * Handle incoming WebSocket message
+ */
+async function handleMessage(
+  ws: WebSocket,
+  msg: WsClientMessage,
+  session: AgentSession
+): Promise<void> {
+  switch (msg.type) {
+    case 'prompt':
+      if (!session.isConnected || !session.sessionId) {
+        ws.send(JSON.stringify({ type: 'error', message: 'Agent not ready' }));
+        return;
+      }
+      try {
+        await session.prompt(msg.message as string);
+      } catch {
+        // Error already emitted by session
+      }
+      break;
+
+    case 'cancel':
+      await session.cancel();
+      break;
+
+    case 'permission_response':
+      session.respondToPermission(msg.requestId as string, msg.optionId as string);
+      break;
+
+    case 'change_cwd':
+      try {
+        await session.changeCwd(msg.path as string);
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    case 'new_session':
+      try {
+        await session.newSession();
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    case 'set_mode':
+      try {
+        await session.setMode(msg.modeId as string);
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    case 'set_model':
+      try {
+        await session.setModel(msg.modelId as string);
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    case 'change_agent':
+      try {
+        await session.changeAgent(msg.agentType as string);
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    case 'get_history': {
+      const history = session.loadHistory();
+      ws.send(JSON.stringify({ type: 'history', events: history, sessionId: session.sessionId }));
+      break;
+    }
+
+    case 'list_sessions': {
+      const sessions = session.listSessions();
+      ws.send(JSON.stringify({ type: 'sessions', sessions }));
+      break;
+    }
+
+    case 'switch_session':
+      try {
+        await session.switchSession(msg.sessionId as string);
+      } catch (err) {
+        ws.send(JSON.stringify({ type: 'error', message: (err as Error).message }));
+      }
+      break;
+
+    default:
+      ws.send(JSON.stringify({ type: 'error', message: `Unknown message type: ${msg.type}` }));
+  }
+}

package/tsconfig.json

@@ -0,0 +1,22 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "lib": ["ES2022"],
+    "outDir": "dist",
+    "rootDir": "src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "noUncheckedIndexedAccess": true,
+    "noUnusedLocals": true,
+    "noUnusedParameters": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "src/**/*.test.ts"]
+}

package/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    include: ['src/**/*.test.ts'],
+  },
+});

package/dist/chunk-5IPYOXBE.js

@@ -1,32 +0,0 @@
-// src/lib/logger.ts
-var verbose = false;
-function setVerbose(v) {
-  verbose = v;
-}
-function isVerbose() {
-  return verbose;
-}
-function debug(tag, message) {
-  if (verbose) {
-    console.log(`[${tag}] ${message}`);
-  }
-}
-function info(tag, message) {
-  console.log(`[${tag}] ${message}`);
-}
-function warn(tag, message) {
-  console.warn(`[${tag}] ${message}`);
-}
-function error(tag, message) {
-  console.error(`[${tag}] ${message}`);
-}
-
-export {
-  setVerbose,
-  isVerbose,
-  debug,
-  info,
-  warn,
-  error
-};
-//# sourceMappingURL=chunk-5IPYOXBE.js.map

package/dist/chunk-5IPYOXBE.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/lib/logger.ts"],"sourcesContent":["/**\n * Simple logger with verbose mode toggle.\n * In production, debug logs are suppressed unless verbose mode is enabled.\n */\n\nlet verbose = false;\n\nexport function setVerbose(v: boolean) {\n  verbose = v;\n}\n\nexport function isVerbose(): boolean {\n  return verbose;\n}\n\n/**\n * Log a debug message. Only shown when verbose mode is enabled.\n */\nexport function debug(tag: string, message: string) {\n  if (verbose) {\n    console.log(`[${tag}] ${message}`);\n  }\n}\n\n/**\n * Log an info message. Always shown.\n */\nexport function info(tag: string, message: string) {\n  console.log(`[${tag}] ${message}`);\n}\n\n/**\n * Log a warning. Always shown.\n */\nexport function warn(tag: string, message: string) {\n  console.warn(`[${tag}] ${message}`);\n}\n\n/**\n * Log an error. Always shown.\n */\nexport function error(tag: string, message: string) {\n  console.error(`[${tag}] ${message}`);\n}\n"],"mappings":";AAKA,IAAI,UAAU;AAEP,SAAS,WAAW,GAAY;AACrC,YAAU;AACZ;AAEO,SAAS,YAAqB;AACnC,SAAO;AACT;AAKO,SAAS,MAAM,KAAa,SAAiB;AAClD,MAAI,SAAS;AACX,YAAQ,IAAI,IAAI,GAAG,KAAK,OAAO,EAAE;AAAA,EACnC;AACF;AAKO,SAAS,KAAK,KAAa,SAAiB;AACjD,UAAQ,IAAI,IAAI,GAAG,KAAK,OAAO,EAAE;AACnC;AAKO,SAAS,KAAK,KAAa,SAAiB;AACjD,UAAQ,KAAK,IAAI,GAAG,KAAK,OAAO,EAAE;AACpC;AAKO,SAAS,MAAM,KAAa,SAAiB;AAClD,UAAQ,MAAM,IAAI,GAAG,KAAK,OAAO,EAAE;AACrC;","names":[]}

package/dist/chunk-C73RKCTS.js

@@ -1,36 +0,0 @@
-// src/lib/mentions.ts
-import path from "path";
-function parseMessageToContentBlocks(message, workingDir) {
-  const blocks = [];
-  const mentionRegex = /@(\.{0,2}[\w\/\.\-]+)/g;
-  let lastIndex = 0;
-  for (const match of message.matchAll(mentionRegex)) {
-    if (match.index > lastIndex) {
-      const text = message.slice(lastIndex, match.index);
-      if (text.trim()) {
-        blocks.push({ type: "text", text });
-      }
-    }
-    const relativePath = match[1];
-    const absolutePath = path.resolve(workingDir, relativePath);
-    blocks.push({
-      type: "resource_link",
-      uri: `file://${absolutePath}`,
-      name: relativePath
-      // Required per ACP spec - use the path the user typed
-    });
-    lastIndex = match.index + match[0].length;
-  }
-  if (lastIndex < message.length) {
-    const text = message.slice(lastIndex);
-    if (text.trim()) {
-      blocks.push({ type: "text", text });
-    }
-  }
-  return blocks.length > 0 ? blocks : [{ type: "text", text: message }];
-}
-
-export {
-  parseMessageToContentBlocks
-};
-//# sourceMappingURL=chunk-C73RKCTS.js.map

package/dist/chunk-C73RKCTS.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/lib/mentions.ts"],"sourcesContent":["import type { ContentBlock } from '../types.js';\nimport path from 'path';\n\n/**\n * Parse message text with @mentions into ContentBlock array.\n * @mentions become resource_link blocks, other text becomes text blocks.\n *\n * Examples:\n * - \"hello world\" → [{ type: 'text', text: 'hello world' }]\n * - \"@src/foo.ts\" → [{ type: 'resource_link', uri: 'file://...', name: 'src/foo.ts' }]\n * - \"check @src/foo.ts for bugs\" → text + resource_link + text\n */\nexport function parseMessageToContentBlocks(\n  message: string,\n  workingDir: string\n): ContentBlock[] {\n  const blocks: ContentBlock[] = [];\n  // Match @path/to/file.ts, @./local.ts, @../parent/file.ts\n  // Allows leading ./ or ../ followed by path characters\n  const mentionRegex = /@(\\.{0,2}[\\w\\/\\.\\-]+)/g;\n\n  let lastIndex = 0;\n  for (const match of message.matchAll(mentionRegex)) {\n    // Add text before mention\n    if (match.index! > lastIndex) {\n      const text = message.slice(lastIndex, match.index);\n      if (text.trim()) {\n        blocks.push({ type: 'text', text });\n      }\n    }\n\n    // Add resource_link for mention\n    const relativePath = match[1];\n    const absolutePath = path.resolve(workingDir, relativePath);\n    blocks.push({\n      type: 'resource_link',\n      uri: `file://${absolutePath}`,\n      name: relativePath, // Required per ACP spec - use the path the user typed\n    });\n\n    lastIndex = match.index! + match[0].length;\n  }\n\n  // Add remaining text\n  if (lastIndex < message.length) {\n    const text = message.slice(lastIndex);\n    if (text.trim()) {\n      blocks.push({ type: 'text', text });\n    }\n  }\n\n  // If no mentions found, return single text block\n  return blocks.length > 0 ? blocks : [{ type: 'text', text: message }];\n}\n"],"mappings":";AACA,OAAO,UAAU;AAWV,SAAS,4BACd,SACA,YACgB;AAChB,QAAM,SAAyB,CAAC;AAGhC,QAAM,eAAe;AAErB,MAAI,YAAY;AAChB,aAAW,SAAS,QAAQ,SAAS,YAAY,GAAG;AAElD,QAAI,MAAM,QAAS,WAAW;AAC5B,YAAM,OAAO,QAAQ,MAAM,WAAW,MAAM,KAAK;AACjD,UAAI,KAAK,KAAK,GAAG;AACf,eAAO,KAAK,EAAE,MAAM,QAAQ,KAAK,CAAC;AAAA,MACpC;AAAA,IACF;AAGA,UAAM,eAAe,MAAM,CAAC;AAC5B,UAAM,eAAe,KAAK,QAAQ,YAAY,YAAY;AAC1D,WAAO,KAAK;AAAA,MACV,MAAM;AAAA,MACN,KAAK,UAAU,YAAY;AAAA,MAC3B,MAAM;AAAA;AAAA,IACR,CAAC;AAED,gBAAY,MAAM,QAAS,MAAM,CAAC,EAAE;AAAA,EACtC;AAGA,MAAI,YAAY,QAAQ,QAAQ;AAC9B,UAAM,OAAO,QAAQ,MAAM,SAAS;AACpC,QAAI,KAAK,KAAK,GAAG;AACf,aAAO,KAAK,EAAE,MAAM,QAAQ,KAAK,CAAC;AAAA,IACpC;AAAA,EACF;AAGA,SAAO,OAAO,SAAS,IAAI,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,CAAC;AACtE;","names":[]}