fix: add missing option when calling use function and next middleware context

Author: minosss

packages/api/src/client/client.ts

@@ -150,10 +150,11 @@ export class ApiClient<
     return handler;
   }
 
-  use<NC extends object>(middleware: Middleware<Ctx, NC>) {
-    return new ApiClient<Req, Ctx & NC>({
+  use<NC extends SomeObject>(middleware: Middleware<Ctx, NC>) {
+    return new ApiClient<Req, Prettify<Ctx & NC>>({
       action: this.#action,
-      middlewares: [...this.middlewares, middleware],
+      handleError: this.handleError,
+      middlewares: [...this.middlewares, middleware] as Middleware<any>[],
     });
   }
 }

packages/api/src/compose.ts

@@ -1,16 +1,19 @@
-import type { AnyAsyncFn, ApiResponse, Options } from './types.js';
+import type { AnyAsyncFn, ApiResponse, Options, SomeObject } from './types.js';
 
 export type MiddlewareOptions<Ctx> = Options & {
   ctx: Ctx;
   execute: AnyAsyncFn;
   res: ApiResponse;
 };
 
-export type Middleware<Ctx, _NextCtx = any> = (
+export type Middleware<Ctx, NextCtx extends SomeObject = SomeObject> = (
   opts: MiddlewareOptions<Ctx> & {
-    next: <NC extends object>(opts?: { ctx?: NC }) => Promise<any>;
+    next: <NC extends SomeObject>(opts?: {
+      ctx?: NC;
+    }) => Promise<MiddlewareOptions<Ctx & NC>>;
   },
-) => Promise<any>;
+  // biome-ignore lint/suspicious/noConfusingVoidType: <explanation>
+) => Promise<void | MiddlewareOptions<Ctx & NextCtx>>;
 
 export function compose(middlewares: Middleware<any>[]) {
   return function composed(

packages/api/src/next/action.ts

@@ -13,6 +13,7 @@ import type {
   Prettify,
   Options,
   ApiRequest,
+  SomeObject,
 } from '../types.js';
 import { validate } from '../validate.js';
 import { noop } from './noop.js';
@@ -323,9 +324,10 @@ export class NextAction<
     return handler;
   }
 
-  use<NC extends object>(middleware: Middleware<Ctx, NC>) {
-    return new NextAction<Req, Ctx & NC>({
-      middlewares: [...this.middlewares, middleware],
+  use<NC extends SomeObject>(middleware: Middleware<Ctx, NC>) {
+    return new NextAction<Req, Prettify<Ctx & NC>>({
+      handleError: this.handleError,
+      middlewares: [...this.middlewares, middleware] as Middleware<any>[],
     });
   }
 }

packages/api/src/next/route.ts

@@ -8,6 +8,7 @@ import type {
   InferInput,
   InferOutput,
   Prettify,
+  SomeObject,
   Transform,
 } from '../types.js';
 import type { Middleware } from '../compose.js';
@@ -206,9 +207,10 @@ export class NextRoute<
     return handler;
   }
 
-  use<NC extends object>(middleware: Middleware<Ctx, NC>) {
-    return new NextRoute<Req, Ctx & NC>({
-      middlewares: [...this.middlewares, middleware],
+  use<NC extends SomeObject>(middleware: Middleware<Ctx, NC>) {
+    return new NextRoute<Req, Prettify<Ctx & NC>>({
+      handleError: this.handleError,
+      middlewares: [...this.middlewares, middleware] as Middleware<any>[],
     });
   }
 }