@beastmode-develeap/beastmode 0.1.132 → 0.1.134

package/dist/index.js

@@ -3616,6 +3616,9 @@ import { readFileSync as readFileSync11, writeFileSync as writeFileSync11, exist
 import { join as join12 } from "path";
 import http from "http";
 import { WebSocketServer, WebSocket as WsWebSocket } from "ws";
+function getRecentTokenUsage() {
+  return recentTokenUsage.slice();
+}
 function boardGet(path, boardUrl) {
   return new Promise((resolve20) => {
     const url = new URL(path, boardUrl);
@@ -4544,10 +4547,12 @@ Respond concisely. Continue the conversation naturally.`;
   }
   return { sessions, startSession, stopSession, handleUpgrade, shutdown };
 }
+var recentTokenUsage;
 var init_chat_handler = __esm({
   "src/cli/ui/chat-handler.ts"() {
     "use strict";
     init_inception();
+    recentTokenUsage = [];
   }
 });
 
@@ -6312,6 +6317,45 @@ import { createHmac } from "crypto";
 import { readFileSync as readFileSync13, existsSync as existsSync16 } from "fs";
 import { join as join14, dirname as dirname6 } from "path";
 import { fileURLToPath as fileURLToPath2 } from "url";
+import { WebSocketServer as WebSocketServer2, WebSocket as WsClient } from "ws";
+function proxyBoardWebSocket(req, socket, head, boardWsUrl) {
+  _boardWsProxy.handleUpgrade(req, socket, head, (clientWs) => {
+    const upstream = new WsClient(boardWsUrl);
+    let closed = false;
+    const cleanup = (code, reason) => {
+      if (closed) return;
+      closed = true;
+      try {
+        clientWs.close(code, reason);
+      } catch {
+      }
+      try {
+        upstream.close(code, reason);
+      } catch {
+      }
+    };
+    upstream.on("open", () => {
+      clientWs.on("message", (data) => {
+        try {
+          upstream.send(data);
+        } catch {
+          cleanup(1011, "relay failed");
+        }
+      });
+      upstream.on("message", (data) => {
+        try {
+          clientWs.send(data);
+        } catch {
+          cleanup(1011, "relay failed");
+        }
+      });
+    });
+    upstream.on("error", () => cleanup(1011, "board WS upstream error"));
+    upstream.on("close", (code, reason) => cleanup(code || 1e3, reason?.toString() || "upstream closed"));
+    clientWs.on("error", () => cleanup(1011, "client WS error"));
+    clientWs.on("close", (code, reason) => cleanup(code || 1e3, reason?.toString() || "client closed"));
+  });
+}
 function resolveStaticDir() {
   const devPath = join14(__dirname, "static");
   if (existsSync16(join14(devPath, "index.html"))) {
@@ -6387,6 +6431,7 @@ function isAuthExempt(method, url) {
   if (url.startsWith("/static/")) return true;
   if (url === "/health" && method === "GET") return true;
   if (url === "/api/build-stamp" && method === "GET") return true;
+  if (url === "/api/token-usage/recent" && method === "GET") return true;
   return false;
 }
 function setSessionCookie(res) {
@@ -6519,6 +6564,10 @@ async function startServer(options = {}) {
       sendJson(res, 200, { status: "ok", uptime: process.uptime() });
       return;
     }
+    if (method === "GET" && url === "/api/token-usage/recent") {
+      sendJson(res, 200, getRecentTokenUsage());
+      return;
+    }
     if (method === "GET" && url === "/api/build-stamp") {
       let stamp = "unknown";
       let commit_sha = null;
@@ -6692,17 +6741,30 @@ async function startServer(options = {}) {
     }
     sendJson(res, 404, { error: "Not found" });
   });
-  if (chatManager) {
-    server.on("upgrade", (req, socket, head) => {
-      const url = (req.url || "").split("?")[0];
-      if (url === "/ws/chat") {
-        chatManager.handleUpgrade(req, socket, head);
-      } else {
-        socket.write("HTTP/1.1 404 Not Found\r\n\r\n");
+  server.on("upgrade", (req, socket, head) => {
+    const url = (req.url || "").split("?")[0];
+    if (url === "/ws/chat") {
+      if (!chatManager) {
+        socket.write("HTTP/1.1 503 Service Unavailable\r\n\r\n");
         socket.end();
+        return;
       }
-    });
-  }
+      if (options.factoryPath && chatManager.sessions.size === 0) {
+        chatManager.startSession(options.factoryPath);
+      }
+      chatManager.handleUpgrade(req, socket, head);
+      return;
+    }
+    if (url === "/ws") {
+      const factoryDir = options.factoryPath || process.cwd();
+      const boardUrl = getBoardUrl2(factoryDir);
+      const wsBoardUrl = boardUrl.replace(/^http/, "ws").replace(/\/$/, "") + "/ws";
+      proxyBoardWebSocket(req, socket, head, wsBoardUrl);
+      return;
+    }
+    socket.write("HTTP/1.1 404 Not Found\r\n\r\n");
+    socket.end();
+  });
   async function shutdown() {
     clearTimeout(inactivityTimer);
     if (chatManager) {
@@ -6733,13 +6795,14 @@ async function startServer(options = {}) {
     });
   });
 }
-var __filename, __dirname, INACTIVITY_TIMEOUT_MS, UI_PASSWORD, AUTH_COOKIE_NAME, AUTH_COOKIE_MAX_AGE, LOGIN_PAGE;
+var _boardWsProxy, __filename, __dirname, INACTIVITY_TIMEOUT_MS, UI_PASSWORD, AUTH_COOKIE_NAME, AUTH_COOKIE_MAX_AGE, LOGIN_PAGE;
 var init_server = __esm({
   "src/cli/ui/server.ts"() {
     "use strict";
     init_api_routes();
     init_board_api_routes();
     init_chat_handler();
+    _boardWsProxy = new WebSocketServer2({ noServer: true });
     __filename = fileURLToPath2(import.meta.url);
     __dirname = dirname6(__filename);
     INACTIVITY_TIMEOUT_MS = 30 * 60 * 1e3;