feat: refactor version api (#2534)

* feat: create useGetVersionQuery function

* [autofix.ci] apply automated fixes

* fix: extend useQuery type to accpet functions with no parameters

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: anovazzi1 <otavio2204@gmail.com>

src/frontend/src/App.tsx

@@ -14,6 +14,7 @@ import {
 } from "./constants/constants";
 import { AuthContext } from "./contexts/authContext";
 import { autoLogin, getGlobalVariables, getHealth } from "./controllers/API";
+import { useGetVersionQuery } from "./controllers/API/queries/version";
 import { setupAxiosDefaults } from "./controllers/API/utils";
 import useTrackLastVisitedPath from "./hooks/use-track-last-visited-path";
 import Router from "./routes";
@@ -27,22 +28,19 @@ export default function App() {
   const queryClient = new QueryClient();
 
   useTrackLastVisitedPath();
-
   const [fetchError, setFetchError] = useState(false);
   const isLoading = useFlowsManagerStore((state) => state.isLoading);
-
   const { isAuthenticated, login, setUserData, setAutoLogin, getUser } =
     useContext(AuthContext);
   const setLoading = useAlertStore((state) => state.setLoading);
-  const refreshVersion = useDarkStore((state) => state.refreshVersion);
   const refreshStars = useDarkStore((state) => state.refreshStars);
   const navigate = useNavigate();
   const dark = useDarkStore((state) => state.dark);
 
   const isLoadingFolders = useFolderStore((state) => state.isLoadingFolders);
+  useGetVersionQuery(undefined, "updateState");
 
   const [isLoadingHealth, setIsLoadingHealth] = useState(false);
-
   useEffect(() => {
     if (!dark) {
       document.getElementById("body")!.classList.remove("dark");
@@ -85,10 +83,9 @@ export default function App() {
     */
     return () => abortController.abort();
   }, []);
-
   const fetchAllData = async () => {
     setTimeout(async () => {
-      await Promise.all([refreshStars(), refreshVersion(), fetchData()]);
+      await Promise.all([refreshStars(), fetchData()]);
     }, 1000);
   };
 
@@ -154,7 +151,6 @@ export default function App() {
   return (
     //need parent component with width and height
     <div className="flex h-full flex-col">
-      <QueryClientProvider client={queryClient}>
       <ErrorBoundary
         onReset={() => {
           // any reset function
@@ -185,7 +181,6 @@ export default function App() {
           </Case>
         </>
       </ErrorBoundary>
-      </QueryClientProvider>
       <div></div>
       <div className="app-div">
         <AlertDisplayArea />

src/frontend/src/contexts/index.tsx

@@ -1,3 +1,4 @@
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
 import { ReactNode } from "react";
 import { BrowserRouter } from "react-router-dom";
 import { ReactFlowProvider } from "reactflow";
@@ -6,10 +7,12 @@ import { ApiInterceptor } from "../controllers/API/api";
 import { AuthProvider } from "./authContext";
 
 export default function ContextWrapper({ children }: { children: ReactNode }) {
+  const queryClient = new QueryClient();
   //element to wrap all context
   return (
     <>
       <BrowserRouter>
+        <QueryClientProvider client={queryClient}>
           <AuthProvider>
             <TooltipProvider>
               <ReactFlowProvider>
@@ -18,6 +21,7 @@ export default function ContextWrapper({ children }: { children: ReactNode }) {
               </ReactFlowProvider>
             </TooltipProvider>
           </AuthProvider>
+        </QueryClientProvider>
       </BrowserRouter>
     </>
   );

src/frontend/src/controllers/API/helpers/constants.ts

@@ -3,6 +3,7 @@ import { BASE_URL_API } from "../../../constants/constants";
 export const URLs = {
   TRANSACTIONS: `monitor/transactions`,
   API_KEY: `api_key`,
+  VERSION: `version`,
 } as const;
 
 export function getURL(key: keyof typeof URLs, params: any = {}) {

src/frontend/src/controllers/API/queries/version/index.ts

@@ -0,0 +1 @@
+export * from "./use-get-version";

src/frontend/src/controllers/API/queries/version/use-get-version.ts

@@ -0,0 +1,41 @@
+import { useDarkStore } from "@/stores/darkStore";
+import { useQueryFunctionType } from "@/types/api";
+import { api } from "../../api";
+import { getURL } from "../../helpers/constants";
+import { UseRequestProcessor } from "../../services/request-processor";
+
+interface versionQueryResponse {
+  version: string;
+  package: string;
+}
+
+export const useGetVersionQuery: useQueryFunctionType<
+  undefined,
+  versionQueryResponse
+> = (_, onFetch) => {
+  const { query } = UseRequestProcessor();
+
+  const responseFn = (data: versionQueryResponse) => {
+    if (!onFetch) return data;
+    if (typeof onFetch === "function") return onFetch(data);
+    const refreshVersion = useDarkStore.getState().refreshVersion;
+    switch (onFetch) {
+      case "updateState": {
+        return refreshVersion(data.version);
+      }
+      default:
+        return data;
+    }
+  };
+
+  const getVersionFn = async () => {
+    return await api.get<versionQueryResponse>(`${getURL("VERSION")}`);
+  };
+
+  const queryResult = query(["useGetVersionQuery"], async () => {
+    const { data } = await getVersionFn();
+    return responseFn(data);
+  });
+
+  return queryResult;
+};

src/frontend/src/stores/darkStore.ts

@@ -12,10 +12,8 @@ export const useDarkStore = create<DarkStoreType>((set, get) => ({
     set(() => ({ dark: dark }));
     window.localStorage.setItem("isDark", dark.toString());
   },
-  refreshVersion: () => {
+  refreshVersion: (v) => {
-    getVersion().then((data) => {
+    set(() => ({ version: v }));
-      set(() => ({ version: data.version }));
-    });
   },
   refreshStars: () => {
     if (import.meta.env.CI) {

src/frontend/src/types/api/index.ts

@@ -233,10 +233,9 @@ export type ResponseErrorDetailAPI = {
   response: { data: { detail: string } };
 };
 
-export type useQueryFunctionType<T, R> = (
+export type useQueryFunctionType<T = undefined, R = any> = T extends undefined
-  props: T,
+  ? (props?: T, onFetch?: ((data: R) => void) | string) => UseQueryResult<R>
-  onFetch?: ((data: R) => void) | string,
+  : (props: T, onFetch?: ((data: R) => void) | string) => UseQueryResult<R>;
-) => UseQueryResult<R>;
 
 export type QueryFunctionType = (
   queryKey: UndefinedInitialDataOptions["queryKey"],

src/frontend/src/types/zustand/dark/index.ts

@@ -3,6 +3,6 @@ export type DarkStoreType = {
   stars: number;
   version: string;
   setDark: (dark: boolean) => void;
-  refreshVersion: () => void;
+  refreshVersion: (v: string) => void;
   refreshStars: () => void;
 };