@mastra/deployer 1.22.0-alpha.0 → 1.22.0-alpha.2

package/dist/server/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAGlD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAGhE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAY5B,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAoBnD,KAAK,QAAQ,GAAG,YAAY,CAAC;AAE7B,KAAK,SAAS,GAAG,aAAa,GAAG;IAC/B,OAAO,EAAE,GAAG,CAAC;QAAE,UAAU,EAAE,+BAA+B,CAAA;KAAE,CAAC,CAAC;CAC/D,CAAC;AAEF,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,wCAmB/D;AAED,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,mBAER;cAegC,QAAQ;eAAa,SAAS;2CAgYhE;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,mBAAmC,mDA0DlG"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAGlD,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAGhE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAY5B,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAoBnD,KAAK,QAAQ,GAAG,YAAY,CAAC;AAE7B,KAAK,SAAS,GAAG,aAAa,GAAG;IAC/B,OAAO,EAAE,GAAG,CAAC;QAAE,UAAU,EAAE,+BAA+B,CAAA;KAAE,CAAC,CAAC;CAC/D,CAAC;AAEF,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,wCAmB/D;AAED,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,mBAER;cAegC,QAAQ;eAAa,SAAS;2CAgZhE;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,mBAAmC,mDA+DlG"}

package/dist/server/index.js

@@ -15,6 +15,7 @@ import { Tool } from '@mastra/core/tools';
 import { MastraServer as MastraServer$1, normalizeQueryParams, redactStreamChunk } from '@mastra/server/server-adapter';
 import util from 'util';
 import { Buffer as Buffer$1 } from 'buffer';
+import { ViewerRegistry, handleInputMessage } from '@mastra/server/browser-stream';
 import { InMemoryTaskStore } from '@mastra/server/a2a/store';
 import { Hono } from 'hono';
 import { compress } from 'hono/compress';
@@ -3089,6 +3090,83 @@ var initializer2 = (inst, issues) => {
   });
 };
 var ZodError = /* @__PURE__ */ $constructor("ZodError", initializer2);
+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 };
+}
 var _hasPermissionPromise;
 function loadHasPermission() {
   if (!_hasPermissionPromise) {
@@ -3994,6 +4072,12 @@ async function createHonoServer(mastra, options = {
       app.use(m.path, m.handler);
     }
   }
+  const browserStreamSetup = await setupBrowserStream(app, {
+    getToolset: (agentId) => {
+      const agent = mastra.getAgentById(agentId);
+      return agent?.browser;
+    }
+  });
   if (server?.cors === false) {
     app.use("*", timeout(server?.timeout ?? 3 * 60 * 1e3));
   } else {
@@ -4212,10 +4296,12 @@ async function createHonoServer(mastra, options = {
       })
     );
   }
+  app.injectWebSocket = browserStreamSetup?.injectWebSocket;
   return app;
 }
 async function createNodeServer(mastra, options = { tools: {} }) {
   const app = await createHonoServer(mastra, options);
+  const injectWebSocket = app.injectWebSocket;
   const serverOptions = mastra.getServer();
   const apiPrefix = serverOptions?.apiPrefix ?? "/api";
   const key = serverOptions?.https?.key ?? (process.env.MASTRA_HTTPS_KEY ? Buffer.from(process.env.MASTRA_HTTPS_KEY, "base64") : void 0);
@@ -4258,6 +4344,7 @@ async function createNodeServer(mastra, options = { tools: {} }) {
       }
     }
   );
+  injectWebSocket?.(server);
   await mastra.startEventEngine();
   return server;
 }