From e9336e9a6727e06db01c6e15fb3ff4e98e6e8235 Mon Sep 17 00:00:00 2001
From: Fredrik Burmester <fredrik.burmester@gmail.com>
Date: Sun, 5 Jan 2025 11:56:46 +0100
Subject: [PATCH] feat: new useQuery specifically for react navigation to
 invalidate !enabled queries on screen re-mount

---
 utils/useReactNavigationQuery.ts | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 utils/useReactNavigationQuery.ts

diff --git a/utils/useReactNavigationQuery.ts b/utils/useReactNavigationQuery.ts
new file mode 100644
index 00000000..a0c5b307
--- /dev/null
+++ b/utils/useReactNavigationQuery.ts
@@ -0,0 +1,32 @@
+import { useFocusEffect } from "@react-navigation/core";
+import {
+  QueryKey,
+  useQuery,
+  UseQueryOptions,
+  UseQueryResult,
+} from "@tanstack/react-query";
+import { useCallback } from "react";
+
+export function useReactNavigationQuery<
+  TQueryFnData = unknown,
+  TError = unknown,
+  TData = TQueryFnData,
+  TQueryKey extends QueryKey = QueryKey
+>(
+  options: UseQueryOptions<TQueryFnData, TError, TData, TQueryKey>
+): UseQueryResult<TData, TError> {
+  const useQueryReturn = useQuery(options);
+
+  useFocusEffect(
+    useCallback(() => {
+      if (
+        ((options.refetchOnWindowFocus && useQueryReturn.isStale) ||
+          options.refetchOnWindowFocus === "always") &&
+        options.enabled !== false
+      )
+        useQueryReturn.refetch();
+    }, [options.enabled, options.refetchOnWindowFocus])
+  );
+
+  return useQueryReturn;
+}