add queries synchronization feature

packages/normy-react-query/src/index.ts

@@ -1,4 +1,9 @@
-import { QueryClient, QueryKey } from '@tanstack/react-query';
+import {
+  QueryClient,
+  QueryKey,
+  SetDataOptions,
+  Updater,
+} from '@tanstack/react-query';
 import { createNormalizer, NormalizerConfig, Data } from '@normy/core';
 
 const shouldBeNormalized = (
@@ -12,6 +17,23 @@ const shouldBeNormalized = (
   return localNormalize;
 };
 
+const updateQueriesFromQueryData = (
+  originalQueryKey: string,
+  queryData: Data,
+  normalizer: ReturnType<typeof createNormalizer>,
+  queryClient: QueryClient,
+) => {
+  const queriesToUpdate = normalizer
+    .getQueriesToUpdate(queryData)
+    .filter(query => query.queryKey !== originalQueryKey);
+  queriesToUpdate.forEach(query => {
+    queryClient.setQueryData(
+      JSON.parse(query.queryKey) as QueryKey,
+      () => query.data,
+    );
+  });
+};
+
 const updateQueriesFromMutationData = (
   mutationData: Data,
   normalizer: ReturnType<typeof createNormalizer>,
@@ -29,7 +51,9 @@ const updateQueriesFromMutationData = (
 
 export const createQueryNormalizer = (
   queryClient: QueryClient,
-  normalizerConfig: NormalizerConfig & { normalize?: boolean } = {},
+  normalizerConfig: NormalizerConfig & {
+    normalize?: boolean;
+  } = {},
 ) => {
   const normalize = normalizerConfig.normalize ?? true;
   const normalizer = createNormalizer(normalizerConfig);
@@ -46,10 +70,14 @@ export const createQueryNormalizer = (
         event.query.meta?.normalize as boolean | undefined,
       )
     ) {
-      normalizer.setQuery(
-        JSON.stringify(event.query.queryKey),
-        event.action.data as Data,
-      );
+      const queryKey = JSON.stringify(event.query.queryKey);
+      const data = event.action.data as Data;
+
+      normalizer.setQuery(queryKey, data);
+
+      if (!event.action.manual) {
+        updateQueriesFromQueryData(queryKey, data, normalizer, queryClient);
+      }
     }
   });
 
@@ -95,12 +123,31 @@ export const createQueryNormalizer = (
       }
     });
 
+  const setQueryData = <TQueryFnData>(
+    queryKey: QueryKey,
+    updater: Updater<TQueryFnData | undefined, TQueryFnData | undefined>,
+    options?: SetDataOptions,
+  ): TQueryFnData | undefined => {
+    const data = queryClient.setQueryData(queryKey, updater, options);
+    const queryKeyString = JSON.stringify(queryKey);
+
+    updateQueriesFromQueryData(
+      queryKeyString,
+      data as Data,
+      normalizer,
+      queryClient,
+    );
+
+    return data;
+  };
+
   return {
     getNormalizedData: normalizer.getNormalizedData,
     clear: () => {
       unsubscribeQueryCache();
       unsubscribeMutationCache();
       normalizer.clearNormalizedData();
     },
+    setQueryData,
   };
 };