From f70ddd2f6abfa1d41f3707c3c42a2b1b4dcc691d Mon Sep 17 00:00:00 2001
From: AdityaJ2305 <adityajindal2305@gmail.com>
Date: Thu, 26 Dec 2024 19:24:51 +0530
Subject: [PATCH] moved to usequery

---
 public/locale/en.json                      |  1 +
 src/components/Facility/DischargeModal.tsx | 34 ++++++++++------------
 2 files changed, 17 insertions(+), 18 deletions(-)

diff --git a/public/locale/en.json b/public/locale/en.json
index 758ef1965db..83e7d4f8357 100644
--- a/public/locale/en.json
+++ b/public/locale/en.json
@@ -48,6 +48,7 @@
   "DOMESTIC_HEALTHCARE_SUPPORT__FAMILY_MEMBER": "Family member",
   "DOMESTIC_HEALTHCARE_SUPPORT__NO_SUPPORT": "No support",
   "DOMESTIC_HEALTHCARE_SUPPORT__PAID_CAREGIVER": "Paid caregiver",
+  "ERROR_FETCHING_CONSULTATION": "Error while retrieving consultation data",
   "GENDER__1": "Male",
   "GENDER__2": "Female",
   "GENDER__3": "Non-binary",
diff --git a/src/components/Facility/DischargeModal.tsx b/src/components/Facility/DischargeModal.tsx
index 1183b2d1909..f614765ca53 100644
--- a/src/components/Facility/DischargeModal.tsx
+++ b/src/components/Facility/DischargeModal.tsx
@@ -1,3 +1,4 @@
+import { useQuery } from "@tanstack/react-query";
 import { useEffect, useState } from "react";
 import { useTranslation } from "react-i18next";
 
@@ -20,7 +21,6 @@ import TextAreaFormField from "@/components/Form/FormFields/TextAreaFormField";
 import TextFormField from "@/components/Form/FormFields/TextFormField";
 import PrescriptionBuilder from "@/components/Medicine/PrescriptionBuilder";
 
-import useAppHistory from "@/hooks/useAppHistory";
 import useConfirmedAction from "@/hooks/useConfirmedAction";
 
 import { DISCHARGE_REASONS } from "@/common/constants";
@@ -29,8 +29,8 @@ import { PLUGIN_Component } from "@/PluginEngine";
 import * as Notification from "@/Utils/Notifications";
 import dayjs from "@/Utils/dayjs";
 import routes from "@/Utils/request/api";
+import query from "@/Utils/request/query";
 import request from "@/Utils/request/request";
-import useTanStackQueryInstead from "@/Utils/request/useQuery";
 
 interface PreDischargeFormInterface {
   new_discharge_reason: number | null;
@@ -93,21 +93,22 @@ const DischargeModal = ({
     setFacility(referred_to);
   }, [referred_to]);
 
-  const { goBack } = useAppHistory();
-
-  const { data } = useTanStackQueryInstead(routes.getConsultation, {
-    pathParams: { id: consultationData.id ?? "" },
-    prefetch: !!consultationData.id,
-    silent: true,
-    onResponse: ({ error }) => {
-      if (error) {
-        goBack();
-        Notification.Error({ msg: "Error in fetching consultation data" });
-        return;
-      }
-    },
+  const { data, isLoading, error } = useQuery({
+    queryKey: [routes.getConsultation.path, consultationData.id],
+    queryFn: query(routes.getConsultation, {
+      pathParams: { id: consultationData.id },
+      silent: true,
+    }),
+    enabled: consultationData.id !== undefined,
   });
 
+  if (isLoading) return <Loading />;
+
+  if (error) {
+    Notification.Error({ msg: t("ERROR_FETCHING_CONSULTATION") });
+    return;
+  }
+
   const initialDiagnoses = data?.diagnoses ?? [];
 
   const discharge_reason =
@@ -190,9 +191,6 @@ const DischargeModal = ({
         ? "death_datetime"
         : "discharge_date"
     ];
-  if (initialDiagnoses == null) {
-    return <Loading />;
-  }
 
   return (
     <>