feat: allow chaining use/error

Wroud · Nov 26, 2024 · 41992e1 · 41992e1
1 parent f4f1061
commit 41992e1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/@wroud/flow-middleware/src/FlowMiddleware.ts b/packages/@wroud/flow-middleware/src/FlowMiddleware.ts
@@ -24,16 +24,18 @@ export class FlowMiddleware<Data = Record<string, any>>
    * Registers a middleware globally.
    * @param {Middleware<Data>} middleware - The middleware function to register.
    */
-  public use(...middleware: IMiddleware<Data>[]): void {
+  public use(...middleware: IMiddleware<Data>[]): this {
     this.middlewares.push(...middleware);
+    return this;
   }
 
   /**
    * Registers an error-handling middleware globally.
    * @param {ErrorMiddleware<Data>} errorMiddleware - The error middleware function to register.
    */
-  public error(...errorMiddleware: IErrorMiddleware<Data>[]): void {
+  public error(...errorMiddleware: IErrorMiddleware<Data>[]): this {
     this.errorMiddlewares.push(...errorMiddleware);
+    return this;
   }
 
   /**

diff --git a/packages/@wroud/flow-middleware/src/interfaces/IFlowMiddleware.ts b/packages/@wroud/flow-middleware/src/interfaces/IFlowMiddleware.ts
@@ -11,13 +11,13 @@ export interface IFlowMiddleware<Data = Record<string, any>> {
    * Registers a middleware globally.
    * @param {Middleware<Data>} middleware - The middleware function to register.
    */
-  use(...middleware: IMiddleware<Data>[]): void;
+  use(...middleware: IMiddleware<Data>[]): this;
 
   /**
    * Registers an error-handling middleware globally.
    * @param {ErrorMiddleware<Data>} errorMiddleware - The error middleware function to register.
    */
-  error(...errorMiddleware: IErrorMiddleware<Data>[]): void;
+  error(...errorMiddleware: IErrorMiddleware<Data>[]): this;
 
   /**
    * Creates a new MiddlewareRequest instance.