From c55af60342ffb8c804a7a29008d5c10be2b1942e Mon Sep 17 00:00:00 2001
From: noufalrahim <noufalrahim6784@gmail.com>
Date: Sat, 14 Dec 2024 21:11:35 +0530
Subject: [PATCH] refactor: modified error message

---
 src/Utils/request/request.ts               | 10 +++++++++-
 src/Utils/request/useQuery.ts              |  6 +++++-
 src/components/Facility/DischargeModal.tsx | 15 +++++++++++----
 3 files changed, 25 insertions(+), 6 deletions(-)

diff --git a/src/Utils/request/request.ts b/src/Utils/request/request.ts
index bd1cabc523c..8033efcc417 100644
--- a/src/Utils/request/request.ts
+++ b/src/Utils/request/request.ts
@@ -18,6 +18,10 @@ export default async function request<TData, TBody>(
     silent,
     signal,
   }: Options<TData, TBody> = {},
+  errorObj?: {
+    code: string;
+    detail: string;
+  },
 ): Promise<RequestResult<TData>> {
   const url = `${careConfig.apiUrl}${makeUrl(path, query, pathParams)}`;
 
@@ -43,7 +47,11 @@ export default async function request<TData, TBody>(
     result = {
       res,
       data: res.ok ? data : undefined,
-      error: res.ok ? undefined : (data as Record<string, unknown>),
+      error: res.ok
+        ? undefined
+        : errorObj
+          ? errorObj
+          : (data as Record<string, unknown>),
     };
 
     onResponse?.(result);
diff --git a/src/Utils/request/useQuery.ts b/src/Utils/request/useQuery.ts
index a0d337a1aa8..524ef67ca42 100644
--- a/src/Utils/request/useQuery.ts
+++ b/src/Utils/request/useQuery.ts
@@ -17,6 +17,10 @@ export interface QueryOptions<TData> extends RequestOptions<TData> {
 export default function useTanStackQueryInstead<TData>(
   route: QueryRoute<TData>,
   options?: QueryOptions<TData>,
+  errorObj?: {
+    code: string;
+    detail: string;
+  },
 ) {
   const overridesRef = useRef<QueryOptions<TData>>();
 
@@ -35,7 +39,7 @@ export default function useTanStackQueryInstead<TData>(
         ? mergeRequestOptions(options || {}, overridesRef.current)
         : options;
 
-      return await request(route, { ...resolvedOptions, signal });
+      return await request(route, { ...resolvedOptions, signal }, errorObj);
     },
     enabled: options?.prefetch ?? true,
     refetchOnWindowFocus: false,
diff --git a/src/components/Facility/DischargeModal.tsx b/src/components/Facility/DischargeModal.tsx
index be71997aa43..081fff75640 100644
--- a/src/components/Facility/DischargeModal.tsx
+++ b/src/components/Facility/DischargeModal.tsx
@@ -92,10 +92,17 @@ const DischargeModal = ({
     setFacility(referred_to);
   }, [referred_to]);
 
-  const initialDiagnoses = useTanStackQueryInstead(routes.getConsultation, {
-    pathParams: { id: consultationData.id ?? "" },
-    prefetch: !!consultationData.id,
-  }).data?.diagnoses;
+  const initialDiagnoses = useTanStackQueryInstead(
+    routes.getConsultation,
+    {
+      pathParams: { id: consultationData.id ?? "" },
+      prefetch: !!consultationData.id,
+    },
+    {
+      code: "",
+      detail: t("consultation_missing_warning"),
+    },
+  ).data?.diagnoses;
 
   const discharge_reason =
     new_discharge_reason ?? preDischargeForm.new_discharge_reason;