feat: added invalidation map

src/assets/useMutationEffects.ts

@@ -1,6 +1,6 @@
 import { useCallback } from "react";
 
-import { OpenApiQueryConfig } from "@povio/openapi-codegen-cli";
+import { OpenApiQueryConfig } from "../lib/config/queryConfig.context";
 import { QueryKey, useQueryClient } from "@tanstack/react-query";
 
 import { QueryModule } from "./queryModules";
@@ -10,6 +10,7 @@ export interface MutationEffectsOptions {
   invalidateModules?: QueryModule[];
   invalidateKeys?: QueryKey[];
   preferUpdate?: boolean;
+  context?: any;
 }
 
 export interface UseMutationEffectsProps {
@@ -22,7 +23,13 @@ export function useMutationEffects({ currentModule }: UseMutationEffectsProps) {
 
   const runMutationEffects = useCallback(
     async <TData>(data: TData, options: MutationEffectsOptions = {}, updateKeys?: QueryKey[]) => {
-      const { invalidateCurrentModule = true, invalidateModules, invalidateKeys, preferUpdate } = options;
+      const {
+        invalidateCurrentModule = true,
+        invalidateModules,
+        invalidateKeys,
+        preferUpdate,
+        context,
+      } = options;
       const shouldUpdate = preferUpdate || (preferUpdate === undefined && config.preferUpdate);
       const shouldInvalidateCurrentModule =
         invalidateCurrentModule || (invalidateCurrentModule === undefined && config.invalidateCurrentModule);
@@ -40,15 +47,19 @@ export function useMutationEffects({ currentModule }: UseMutationEffectsProps) {
           const isCurrentModule = shouldInvalidateCurrentModule && queryKey[0] === currentModule;
           const isInvalidateModule = !!invalidateModules && invalidateModules.some((module) => queryKey[0] === module);
           const isInvalidateKey = !!invalidateKeys && invalidateKeys.some((key) => isQueryKeyEqual(queryKey, key));
-          return isCurrentModule || isInvalidateModule || isInvalidateKey;
+
+          const map = config.invalidationMap?.(context)?.[currentModule];
+          const isMappedKey = !!map && map.some((key) => isQueryKeyEqual(queryKey, key));
+
+          return isCurrentModule || isInvalidateModule || isInvalidateKey || isMappedKey;
         },
       });
 
       if (shouldUpdate && updateKeys) {
         updateKeys.map((queryKey) => queryClient.setQueryData(queryKey, data));
       }
     },
-    [queryClient, currentModule, config.preferUpdate],
+    [queryClient, currentModule, config.preferUpdate, config.invalidationMap],
   );
 
   return { runMutationEffects };

src/lib/config/queryConfig.context.tsx

@@ -1,18 +1,28 @@
 import { createContext, use, useMemo } from "react";
 import { PropsWithChildren } from "react";
+import type { QueryKey } from "@tanstack/react-query";
 
 export namespace OpenApiQueryConfig {
-  interface Type {
+  export interface Type {
     preferUpdate?: boolean;
     invalidateCurrentModule?: boolean;
+    invalidationMap?: (context?: any) => Record<string, QueryKey[]>;
   }
 
   const Context = createContext<Type>({});
 
   type ProviderProps = Type;
 
-  export const Provider = ({ preferUpdate, invalidateCurrentModule, children }: PropsWithChildren<ProviderProps>) => {
-    const value = useMemo(() => ({ preferUpdate, invalidateCurrentModule }), [preferUpdate, invalidateCurrentModule]);
+  export const Provider = ({
+    preferUpdate,
+    invalidateCurrentModule,
+    invalidationMap,
+    children,
+  }: PropsWithChildren<ProviderProps>) => {
+    const value = useMemo(
+      () => ({ preferUpdate, invalidateCurrentModule, invalidationMap }),
+      [preferUpdate, invalidateCurrentModule, invalidationMap],
+    );
 
     return <Context.Provider value={value}>{children}</Context.Provider>;
   };