@mastra/hono 1.4.8-alpha.0 → 1.4.8-alpha.2

package/dist/index.js

@@ -943,12 +943,9 @@ var MastraServer = class extends MastraServer$1 {
     );
   }
   async registerCustomApiRoutes() {
-    const routes = this.customApiRoutes ?? this.mastra.getServer()?.apiRoutes;
-    if (!routes || routes.length === 0) return;
+    if (!await this.buildCustomRouteHandler()) return;
+    const routes = this.customApiRoutes ?? this.mastra.getServer()?.apiRoutes ?? [];
     for (const route of routes) {
-      const handler = "handler" in route && route.handler ? route.handler : "createHandler" in route ? await route.createHandler({ mastra: this.mastra }) : void 0;
-      if (!handler) continue;
-      const middlewares = [];
       const serverRoute = {
         method: route.method,
         path: route.path,
@@ -957,7 +954,7 @@ var MastraServer = class extends MastraServer$1 {
         },
         requiresAuth: route.requiresAuth
       };
-      middlewares.push(async (c, next) => {
+      const routeHandler = async (c) => {
         const authError = await this.checkRouteAuth(serverRoute, {
           path: c.req.path,
           method: c.req.method,
@@ -984,14 +981,24 @@ var MastraServer = class extends MastraServer$1 {
             }
           }
         }
-        return next();
-      });
-      if (route.middleware) {
-        middlewares.push(...Array.isArray(route.middleware) ? route.middleware : [route.middleware]);
-      }
-      const allHandlers = [...middlewares, handler];
+        const reqHeaders = {};
+        c.req.raw.headers.forEach((v, k) => {
+          reqHeaders[k] = v;
+        });
+        const response = await this.handleCustomRouteRequest(
+          c.req.url,
+          c.req.method,
+          reqHeaders,
+          c.req.raw.body,
+          c.get("requestContext")
+        );
+        if (!response) {
+          return c.json({ error: "Not Found" }, 404);
+        }
+        return response;
+      };
       const method = route.method.toLowerCase();
-      this.app[method](route.path, allHandlers[0], ...allHandlers.slice(1));
+      this.app[method](route.path, routeHandler);
     }
   }
   registerContextMiddleware() {