@mastra/hono 1.3.5-alpha.0 → 1.4.0-alpha.2

package/dist/index.js

@@ -2,6 +2,7 @@ import { MastraServer as MastraServer$1, normalizeQueryParams, redactStreamChunk
 import { toReqRes, toFetchResponse } from 'fetch-to-node';
 import { RequestContext } from '@mastra/core/request-context';
 import { coreAuthMiddleware } from '@mastra/server/auth';
+import { ViewerRegistry, handleInputMessage } from '@mastra/server/browser-stream';
 
 // src/index.ts
 
@@ -458,6 +459,83 @@ function createAuthMiddleware({ mastra, requiresAuth = true }) {
     return c.json(result.body, result.status);
   };
 }
+async function setupBrowserStream(app, config2) {
+  let createNodeWebSocket;
+  try {
+    const mod = "@hono/node-ws";
+    const honoNodeWs = await import(
+      /* @vite-ignore */
+      /* webpackIgnore: true */
+      mod
+    );
+    createNodeWebSocket = honoNodeWs.createNodeWebSocket;
+  } catch {
+    return null;
+  }
+  const { injectWebSocket, upgradeWebSocket } = createNodeWebSocket({ app });
+  const registry2 = new ViewerRegistry();
+  app.get(
+    "/browser/:agentId/stream",
+    upgradeWebSocket((c) => {
+      const agentId = c.req.param("agentId");
+      const threadId = c.req.query("threadId");
+      const viewerKey = threadId ? `${agentId}:${threadId}` : agentId;
+      return {
+        onOpen(_event, ws) {
+          ws.send(JSON.stringify({ status: "connected" }));
+          void registry2.addViewer(viewerKey, ws, config2.getToolset, agentId, threadId);
+        },
+        onMessage(event, _ws) {
+          const data = typeof event.data === "string" ? event.data : null;
+          if (data) {
+            handleInputMessage(data, config2.getToolset, agentId, threadId);
+          }
+        },
+        onClose(_event, ws) {
+          void registry2.removeViewer(viewerKey, ws);
+        },
+        onError(event, ws) {
+          console.error("[BrowserStream] WebSocket error:", event);
+          void registry2.removeViewer(viewerKey, ws);
+        }
+      };
+    })
+  );
+  app.post("/api/agents/:agentId/browser/close", async (c) => {
+    const agentId = c.req.param("agentId");
+    if (!agentId) {
+      return c.json({ error: "Agent ID is required" }, 400);
+    }
+    const toolset = config2.getToolset(agentId);
+    if (!toolset) {
+      return c.json({ error: "No browser session for this agent" }, 404);
+    }
+    try {
+      let threadId;
+      try {
+        const body = await c.req.json();
+        threadId = body?.threadId;
+      } catch {
+      }
+      const scope = toolset.getScope();
+      const viewerKey = threadId ? `${agentId}:${threadId}` : agentId;
+      if (scope === "thread" && threadId) {
+        await registry2.closeBrowserSession(viewerKey);
+        if ("closeThreadSession" in toolset && typeof toolset.closeThreadSession === "function") {
+          await toolset.closeThreadSession(threadId);
+        }
+      } else {
+        await registry2.closeBrowserSession(viewerKey);
+        await toolset.close();
+      }
+      return c.json({ success: true });
+    } catch (error) {
+      console.error(`[BrowserStream] Error closing browser for ${agentId}:`, error);
+      return c.json({ error: "Failed to close browser" }, 500);
+    }
+  });
+  return { injectWebSocket, registry: registry2 };
+}
 
 // src/index.ts
 var _hasPermissionPromise;
@@ -961,6 +1039,6 @@ var MastraServer = class extends MastraServer$1 {
   }
 };
 
-export { MastraServer, createAuthMiddleware };
+export { MastraServer, createAuthMiddleware, setupBrowserStream };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map