@mastra/koa 1.5.7-alpha.5 → 1.5.7-alpha.8

package/dist/index.js

@@ -4,6 +4,9 @@ import { MastraServer as MastraServer$1, checkRouteFGA, redactStreamChunk, norma
 import { RequestContext } from '@mastra/core/request-context';
 
 // src/index.ts
+
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/core.js
+var _a;
 // @__NO_SIDE_EFFECTS__
 function $constructor(name, initializer3, params) {
   function init(inst, def) {
@@ -56,15 +59,16 @@ function $constructor(name, initializer3, params) {
   Object.defineProperty(_, "name", { value: name });
   return _;
 }
+(_a = globalThis).__zod_globalConfig ?? (_a.__zod_globalConfig = {});
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/util.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/util.js
 function jsonStringifyReplacer(_, value) {
   if (typeof value === "bigint")
     return value.toString();
   return value;
 }
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/errors.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/core/errors.js
 var initializer = (inst, def) => {
   inst.name = "$ZodError";
   Object.defineProperty(inst, "_zod", {
@@ -82,7 +86,7 @@ var initializer = (inst, def) => {
   });
 };
 var $ZodError = $constructor("$ZodError", initializer);
-function flattenError(error, mapper = (issue2) => issue2.message) {
+function flattenError(error, mapper = (issue) => issue.message) {
   const fieldErrors = {};
   const formErrors = [];
   for (const sub of error.issues) {
@@ -95,32 +99,35 @@ function flattenError(error, mapper = (issue2) => issue2.message) {
   }
   return { formErrors, fieldErrors };
 }
-function formatError(error, mapper = (issue2) => issue2.message) {
+function formatError(error, mapper = (issue) => issue.message) {
   const fieldErrors = { _errors: [] };
-  const processError = (error2) => {
-    for (const issue2 of error2.issues) {
-      if (issue2.code === "invalid_union" && issue2.errors.length) {
-        issue2.errors.map((issues) => processError({ issues }));
-      } else if (issue2.code === "invalid_key") {
-        processError({ issues: issue2.issues });
-      } else if (issue2.code === "invalid_element") {
-        processError({ issues: issue2.issues });
-      } else if (issue2.path.length === 0) {
-        fieldErrors._errors.push(mapper(issue2));
+  const processError = (error2, path = []) => {
+    for (const issue of error2.issues) {
+      if (issue.code === "invalid_union" && issue.errors.length) {
+        issue.errors.map((issues) => processError({ issues }, [...path, ...issue.path]));
+      } else if (issue.code === "invalid_key") {
+        processError({ issues: issue.issues }, [...path, ...issue.path]);
+      } else if (issue.code === "invalid_element") {
+        processError({ issues: issue.issues }, [...path, ...issue.path]);
       } else {
-        let curr = fieldErrors;
-        let i = 0;
-        while (i < issue2.path.length) {
-          const el = issue2.path[i];
-          const terminal = i === issue2.path.length - 1;
-          if (!terminal) {
-            curr[el] = curr[el] || { _errors: [] };
-          } else {
-            curr[el] = curr[el] || { _errors: [] };
-            curr[el]._errors.push(mapper(issue2));
+        const fullpath = [...path, ...issue.path];
+        if (fullpath.length === 0) {
+          fieldErrors._errors.push(mapper(issue));
+        } else {
+          let curr = fieldErrors;
+          let i = 0;
+          while (i < fullpath.length) {
+            const el = fullpath[i];
+            const terminal = i === fullpath.length - 1;
+            if (!terminal) {
+              curr[el] = curr[el] || { _errors: [] };
+            } else {
+              curr[el] = curr[el] || { _errors: [] };
+              curr[el]._errors.push(mapper(issue));
+            }
+            curr = curr[el];
+            i++;
           }
-          curr = curr[el];
-          i++;
         }
       }
     }
@@ -129,54 +136,7 @@ function formatError(error, mapper = (issue2) => issue2.message) {
   return fieldErrors;
 }
 
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/core/registries.js
-var _a;
-var $ZodRegistry = class {
-  constructor() {
-    this._map = /* @__PURE__ */ new WeakMap();
-    this._idmap = /* @__PURE__ */ new Map();
-  }
-  add(schema, ..._meta) {
-    const meta = _meta[0];
-    this._map.set(schema, meta);
-    if (meta && typeof meta === "object" && "id" in meta) {
-      this._idmap.set(meta.id, schema);
-    }
-    return this;
-  }
-  clear() {
-    this._map = /* @__PURE__ */ new WeakMap();
-    this._idmap = /* @__PURE__ */ new Map();
-    return this;
-  }
-  remove(schema) {
-    const meta = this._map.get(schema);
-    if (meta && typeof meta === "object" && "id" in meta) {
-      this._idmap.delete(meta.id);
-    }
-    this._map.delete(schema);
-    return this;
-  }
-  get(schema) {
-    const p = schema._zod.parent;
-    if (p) {
-      const pm = { ...this.get(p) ?? {} };
-      delete pm.id;
-      const f = { ...pm, ...this._map.get(schema) };
-      return Object.keys(f).length ? f : void 0;
-    }
-    return this._map.get(schema);
-  }
-  has(schema) {
-    return this._map.has(schema);
-  }
-};
-function registry() {
-  return new $ZodRegistry();
-}
-(_a = globalThis).__zod_globalRegistry ?? (_a.__zod_globalRegistry = registry());
-
-// ../../node_modules/.pnpm/zod@4.3.6/node_modules/zod/v4/classic/errors.js
+// ../../node_modules/.pnpm/zod@4.4.3/node_modules/zod/v4/classic/errors.js
 var initializer2 = (inst, issues) => {
   $ZodError.init(inst, issues);
   inst.name = "ZodError";
@@ -190,8 +150,8 @@ var initializer2 = (inst, issues) => {
       // enumerable: false,
     },
     addIssue: {
-      value: (issue2) => {
-        inst.issues.push(issue2);
+      value: (issue) => {
+        inst.issues.push(issue);
         inst.message = JSON.stringify(inst.issues, jsonStringifyReplacer, 2);
       }
       // enumerable: false,
@@ -211,7 +171,7 @@ var initializer2 = (inst, issues) => {
     }
   });
 };
-var ZodError = $constructor("ZodError", initializer2);
+var ZodError = /* @__PURE__ */ $constructor("ZodError", initializer2);
 function toWebRequest(ctx) {
   const protocol = ctx.protocol || "http";
   const host = ctx.host || "localhost";
@@ -674,6 +634,10 @@ var MastraServer = class extends MastraServer$1 {
         const { done, value } = await reader.read();
         if (done) break;
         if (value) {
+          if (streamFormat === "sse" && typeof value === "string" && value.startsWith(":")) {
+            ctx.res.write(value);
+            continue;
+          }
           const shouldRedact = this.streamOptions?.redact ?? true;
           const outputValue = shouldRedact ? redactStreamChunk(value) : value;
           if (streamFormat === "sse") {
@@ -913,77 +877,94 @@ var MastraServer = class extends MastraServer$1 {
       );
       const shouldRunCustomRouteAuth = isProtectedCustomRoute(path, method, server.customRouteAuthConfig);
       const shouldRunCustomRouteFGA = !!matchedRoute?.route.fga;
-      if (shouldRunCustomRouteAuth || shouldRunCustomRouteFGA) {
-        const serverRoute = {
-          method: matchedRoute?.route.method ?? method,
-          path: matchedRoute?.route.path ?? path,
-          responseType: "json",
-          handler: async () => {
-          },
-          requiresAuth: matchedRoute?.route.requiresAuth,
-          requiresPermission: matchedRoute?.route.requiresPermission,
-          fga: matchedRoute?.route.fga
-        };
-        if (shouldRunCustomRouteAuth) {
-          const authError = await server.checkRouteAuth(serverRoute, {
-            path,
-            method,
-            getHeader: (name) => ctx.headers[name.toLowerCase()],
-            getQuery: (name) => ctx.query[name],
-            requestContext: ctx.state.requestContext,
-            request: toWebRequest2(ctx),
-            buildAuthorizeContext: () => toWebRequest2(ctx)
-          });
-          if (authError) {
-            if (authError.headers) {
-              for (const [key, value] of Object.entries(authError.headers)) {
-                ctx.set(key, value);
+      const customRouteAbortController = new AbortController();
+      const abortCustomRoute = () => {
+        customRouteAbortController.abort();
+      };
+      const abortCustomRouteIfOpen = () => {
+        if (!ctx.res.writableEnded) {
+          abortCustomRoute();
+        }
+      };
+      ctx.res.once("close", abortCustomRouteIfOpen);
+      ctx.res.once("error", abortCustomRouteIfOpen);
+      try {
+        if (shouldRunCustomRouteAuth || shouldRunCustomRouteFGA) {
+          const serverRoute = {
+            method: matchedRoute?.route.method ?? method,
+            path: matchedRoute?.route.path ?? path,
+            responseType: "json",
+            handler: async () => {
+            },
+            requiresAuth: matchedRoute?.route.requiresAuth,
+            requiresPermission: matchedRoute?.route.requiresPermission,
+            fga: matchedRoute?.route.fga
+          };
+          if (shouldRunCustomRouteAuth) {
+            const authError = await server.checkRouteAuth(serverRoute, {
+              path,
+              method,
+              getHeader: (name) => ctx.headers[name.toLowerCase()],
+              getQuery: (name) => ctx.query[name],
+              requestContext: ctx.state.requestContext,
+              request: toWebRequest2(ctx),
+              buildAuthorizeContext: () => toWebRequest2(ctx)
+            });
+            if (authError) {
+              if (authError.headers) {
+                for (const [key, value] of Object.entries(authError.headers)) {
+                  ctx.set(key, value);
+                }
               }
-            }
-            if (authError.error) {
-              ctx.status = authError.status;
-              ctx.body = { error: authError.error };
-              return;
-            }
-          }
-          const authConfig = server.mastra.getServer()?.auth;
-          if (authConfig) {
-            const hasPermission = await loadHasPermission();
-            if (hasPermission) {
-              const userPermissions = ctx.state.requestContext.get("mastra__userPermissions");
-              const permissionError = server.checkRoutePermission(serverRoute, userPermissions, hasPermission);
-              if (permissionError) {
-                ctx.status = permissionError.status;
-                ctx.body = {
-                  error: permissionError.error,
-                  message: permissionError.message
-                };
+              if (authError.error) {
+                ctx.status = authError.status;
+                ctx.body = { error: authError.error };
                 return;
               }
             }
+            const authConfig = server.mastra.getServer()?.auth;
+            if (authConfig) {
+              const hasPermission = await loadHasPermission();
+              if (hasPermission) {
+                const userPermissions = ctx.state.requestContext.get("mastra__userPermissions");
+                const permissionError = server.checkRoutePermission(serverRoute, userPermissions, hasPermission);
+                if (permissionError) {
+                  ctx.status = permissionError.status;
+                  ctx.body = {
+                    error: permissionError.error,
+                    message: permissionError.message
+                  };
+                  return;
+                }
+              }
+            }
+          }
+          const fgaError = await checkRouteFGA(server.mastra, serverRoute, ctx.state.requestContext, {
+            ...matchedRoute?.params ?? {},
+            ...ctx.query,
+            ...typeof ctx.request.body === "object" && ctx.request.body !== null ? ctx.request.body : {}
+          });
+          if (fgaError) {
+            ctx.status = fgaError.status;
+            ctx.body = { error: fgaError.error, message: fgaError.message };
+            return;
           }
         }
-        const fgaError = await checkRouteFGA(server.mastra, serverRoute, ctx.state.requestContext, {
-          ...matchedRoute?.params ?? {},
-          ...ctx.query,
-          ...typeof ctx.request.body === "object" && ctx.request.body !== null ? ctx.request.body : {}
-        });
-        if (fgaError) {
-          ctx.status = fgaError.status;
-          ctx.body = { error: fgaError.error, message: fgaError.message };
-          return;
-        }
+        const response = await server.handleCustomRouteRequest(
+          `${ctx.protocol}://${ctx.host}${ctx.originalUrl || ctx.url}`,
+          ctx.method,
+          ctx.headers,
+          ctx.request.body,
+          ctx.state.requestContext,
+          customRouteAbortController.signal
+        );
+        if (!response) return next();
+        ctx.respond = false;
+        await server.writeCustomRouteResponse(response, ctx.res, customRouteAbortController.signal);
+      } finally {
+        ctx.res.off("close", abortCustomRouteIfOpen);
+        ctx.res.off("error", abortCustomRouteIfOpen);
       }
-      const response = await server.handleCustomRouteRequest(
-        `${ctx.protocol}://${ctx.host}${ctx.originalUrl || ctx.url}`,
-        ctx.method,
-        ctx.headers,
-        ctx.request.body,
-        ctx.state.requestContext
-      );
-      if (!response) return next();
-      ctx.respond = false;
-      await server.writeCustomRouteResponse(response, ctx.res);
     });
   }
   registerContextMiddleware() {
@@ -1004,14 +985,14 @@ var MastraServer = class extends MastraServer$1 {
       const method = ctx.method;
       const path = ctx.path;
       await next();
-      const duration2 = Date.now() - start;
+      const duration = Date.now() - start;
       const status = ctx.status;
       const level = server.httpLoggingConfig?.level || "info";
       const logData = {
         method,
         path,
         status,
-        duration: `${duration2}ms`
+        duration: `${duration}ms`
       };
       if (server.httpLoggingConfig?.includeQueryParams) {
         logData.query = ctx.query;
@@ -1027,7 +1008,7 @@ var MastraServer = class extends MastraServer$1 {
         });
         logData.headers = headers;
       }
-      server.logger[level](`${method} ${path} ${status} ${duration2}ms`, logData);
+      server.logger[level](`${method} ${path} ${status} ${duration}ms`, logData);
     });
   }
 };