From f6c40d21cb099a9164bb380c607d5cc752fc3549 Mon Sep 17 00:00:00 2001
From: Srijan Tripathi <72181144+srijantrpth@users.noreply.github.com>
Date: Mon, 16 Dec 2024 12:44:00 +0000
Subject: [PATCH] Added Custom Headers Logic to Query

---
 src/Utils/request/query.ts | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/Utils/request/query.ts b/src/Utils/request/query.ts
index 3431f625728..c198de5fea5 100644
--- a/src/Utils/request/query.ts
+++ b/src/Utils/request/query.ts
@@ -5,15 +5,26 @@ import { getResponseBody } from "@/Utils/request/request";
 import { QueryOptions, Route } from "@/Utils/request/types";
 import { makeHeaders, makeUrl } from "@/Utils/request/utils";
 
+// Extend the QueryOptions interface to include customHeaders
+export interface ExtendedQueryOptions<TBody> extends QueryOptions<TBody> {
+  customHeaders?: Record<string, string>;
+}
+
 async function queryRequest<TData, TBody>(
   { path, method, noAuth }: Route<TData, TBody>,
-  options?: QueryOptions<TBody>,
+  options?: ExtendedQueryOptions<TBody>,
 ): Promise<TData> {
   const url = `${careConfig.apiUrl}${makeUrl(path, options?.queryParams, options?.pathParams)}`;
 
+  // Merge customHeaders with default headers
+  const headers = {
+    ...makeHeaders(noAuth ?? false),
+    ...(options?.customHeaders || {}),
+  };
+
   const fetchOptions: RequestInit = {
     method,
-    headers: makeHeaders(noAuth ?? false),
+    headers,
     signal: options?.signal,
   };
 
@@ -48,7 +59,7 @@ async function queryRequest<TData, TBody>(
  */
 export default function query<TData, TBody>(
   route: Route<TData, TBody>,
-  options?: QueryOptions<TBody>,
+  options?: ExtendedQueryOptions<TBody>,
 ) {
   return ({ signal }: { signal: AbortSignal }) => {
     return queryRequest(route, { ...options, signal });