From cbd634144e7434d3e97cba7324407d958f928e26 Mon Sep 17 00:00:00 2001
From: Hrishikesh Vaipurkar <hrishikesh.vaipurkar@scandichotels.com>
Date: Tue, 2 Jul 2024 14:20:42 +0200
Subject: [PATCH] feat: WEB-210 Updated types as review comments

---
 .../MyPages/Blocks/Stays/Soonest/index.tsx    |  2 +-
 server/routers/user/query.ts                  | 31 +++++++++++++------
 2 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/components/MyPages/Blocks/Stays/Soonest/index.tsx b/components/MyPages/Blocks/Stays/Soonest/index.tsx
index 950a72726..a3d724d6f 100644
--- a/components/MyPages/Blocks/Stays/Soonest/index.tsx
+++ b/components/MyPages/Blocks/Stays/Soonest/index.tsx
@@ -26,7 +26,7 @@ export default async function SoonestStays({
       <SectionHeader title={title} subtitle={subtitle} link={link} />
       {response.data.length ? (
         <Grids.Stackable>
-          {response.data.map((stay: any) => (
+          {response.data.map((stay) => (
             <StayCard
               key={stay.attributes.confirmationNumber}
               lang={lang}
diff --git a/server/routers/user/query.ts b/server/routers/user/query.ts
index db30c5a06..13323917a 100644
--- a/server/routers/user/query.ts
+++ b/server/routers/user/query.ts
@@ -23,6 +23,7 @@ import {
   getMembershipCardsSchema,
   getStaysSchema,
   getUserSchema,
+  Stay,
 } from "./output"
 import { benefits, extendedUser, nextLevelPerks } from "./temp"
 
@@ -65,19 +66,23 @@ function fakingRequest<T>(payload: T): Promise<T> {
   })
 }
 
-const updateStaysBookingUrl = async (verifiedData: any, ctx: any) => {
+const updateStaysBookingUrl = async (
+  data: Stay[],
+  token: string,
+  lang: string
+) => {
   // Tenporary API call needed till we have user name in ctx session data
   const apiResponse = await api.get(api.endpoints.v1.profile, {
     cache: "no-store",
     headers: {
-      Authorization: `Bearer ${ctx.session.token.access_token}`,
+      Authorization: `Bearer ${token}`,
     },
   })
 
   // Temporary domain and lang support for current web
   let localeDomain = env.PUBLIC_URL
   let fullBookingUrl = localeDomain + "/hotelreservation/my-booking"
-  switch (ctx.lang) {
+  switch (lang) {
     case "sv":
       localeDomain = localeDomain?.replace(".com", ".se")
       fullBookingUrl = localeDomain + "/hotelreservation/din-bokning"
@@ -105,7 +110,7 @@ const updateStaysBookingUrl = async (verifiedData: any, ctx: any) => {
   if (apiResponse.ok) {
     const apiJson = await apiResponse.json()
     if (apiJson.data?.attributes) {
-      verifiedData.data.data.forEach((stay: any) => {
+      data.map((stay: Stay) => {
         const originalString =
           stay.attributes.confirmationNumber.toString() +
           "," +
@@ -116,7 +121,7 @@ const updateStaysBookingUrl = async (verifiedData: any, ctx: any) => {
       })
     }
   }
-  return verifiedData
+  return data
 }
 
 export const userQueryRouter = router({
@@ -269,10 +274,14 @@ export const userQueryRouter = router({
             ? verifiedData.data.links.offset
             : undefined
 
-        const updatedData = await updateStaysBookingUrl(verifiedData, ctx)
+        const updatedData = await updateStaysBookingUrl(
+          verifiedData.data.data,
+          ctx.session.token.access_token,
+          ctx.lang
+        )
 
         return {
-          data: updatedData.data.data,
+          data: updatedData,
           nextCursor,
         }
       }),
@@ -331,10 +340,14 @@ export const userQueryRouter = router({
             ? verifiedData.data.links.offset
             : undefined
 
-        const updatedData = await updateStaysBookingUrl(verifiedData, ctx)
+        const updatedData = await updateStaysBookingUrl(
+          verifiedData.data.data,
+          ctx.session.token.access_token,
+          ctx.lang
+        )
 
         return {
-          data: updatedData.data.data,
+          data: updatedData,
           nextCursor,
         }
       }),