add middlewares for mutation builder

Author: KurtGokhan

src/builder/HttpMutationBuilder.ts

@@ -3,11 +3,13 @@ import { WithOptional } from '../types/utils';
 import { HttpQueryBuilder } from './HttpQueryBuilder';
 import { MutationBuilder, MutationBuilderConfig } from './MutationBuilder';
 import { QueryBuilder } from './QueryBuilder';
+import { MiddlewareFn } from './middlewares';
 import {
   HttpBaseHeaders,
   HttpBaseParams,
   HttpBaseSearch,
   HttpBuilderTypeTemplate,
+  SetAllTypes,
   SetDataType,
   SetErrorType,
 } from './types';
@@ -75,6 +77,11 @@ export class HttpMutationBuilder<
     vars?: TVars,
     reset?: TReset,
   ) => HttpMutationBuilder<AppendVarsType<T, Partial<TVars>, TReset>>;
+
+  declare withMiddleware: <TVars = T['vars'], TData = T['data'], TError = T['error']>(
+    middleware: MiddlewareFn<TVars, TData, TError, T>,
+  ) => HttpMutationBuilder<SetAllTypes<T, TData, TError, TVars, true>>;
+
   declare asQueryBuilder: () => HttpQueryBuilder<T>;
   protected override QueryBuilderConstructor: typeof QueryBuilder = HttpQueryBuilder as typeof QueryBuilder;
 }

src/builder/HttpQueryBuilder.ts

@@ -67,9 +67,11 @@ export class HttpQueryBuilder<T extends HttpBuilderTypeTemplate = HttpBuilderTyp
     vars?: TVars,
     reset?: TReset,
   ) => HttpQueryBuilder<AppendVarsType<T, Partial<TVars>, TReset>>;
-  declare asMutationBuilder: () => HttpMutationBuilder<T>;
+
   declare withMiddleware: <TVars = T['vars'], TData = T['data'], TError = T['error']>(
     middleware: MiddlewareFn<TVars, TData, TError, T>,
   ) => HttpQueryBuilder<SetAllTypes<T, TData, TError, TVars, true>>;
+
+  declare asMutationBuilder: () => HttpMutationBuilder<T>;
   protected override MutationBuilderConstructor: typeof MutationBuilder = HttpMutationBuilder as typeof MutationBuilder;
 }

src/builder/MutationBuilder.ts

@@ -15,7 +15,8 @@ import {
 import { mergeTagOptions } from '../tags/mergeTagOptions';
 import { QueryInvalidatesMetadata } from '../tags/types';
 import { QueryBuilder } from './QueryBuilder';
-import { BuilderMergeVarsFn, BuilderQueryFn, SetDataType, SetErrorType } from './types';
+import { MiddlewareFn, applyMiddleware } from './middlewares';
+import { BuilderMergeVarsFn, BuilderQueryFn, SetAllTypes, SetDataType, SetErrorType } from './types';
 import { AppendVarsType, BuilderTypeTemplate } from './types';
 import { areKeysEqual, mergeMutationOptions, mergeVars } from './utils';
 
@@ -180,6 +181,14 @@ export class MutationBuilder<T extends BuilderTypeTemplate = BuilderTypeTemplate
     return new ctor<T>(newConfig) as this;
   }
 
+  withMiddleware<TVars = T['vars'], TData = T['data'], TError = T['error']>(
+    middleware: MiddlewareFn<TVars, TData, TError, T>,
+  ): MutationBuilder<SetAllTypes<T, TData, TError, TVars, true>> {
+    const newBuilder = this as unknown as MutationBuilder<SetAllTypes<T, TData, TError, TVars, true>>;
+
+    return newBuilder.withConfig({ queryFn: applyMiddleware(this.config.queryFn, middleware) });
+  }
+
   freeze(): MutationBuilderFrozen<T> {
     return this;
   }