From 0ef12d4a29d24d6057a65fa8a2ebddd91266bd1d Mon Sep 17 00:00:00 2001
From: Pontus Dreij <pontus.dreij@scandichotels.com>
Date: Mon, 25 Nov 2024 15:38:08 +0100
Subject: [PATCH] fix(SW-1014): Fixed issue where google didn't found city.

---
 .../select-hotel/@modal/(.)map/page.tsx       |  5 ++-
 .../(standard)/select-hotel/page.tsx          |  5 +++
 components/Maps/StaticMap/index.tsx           |  5 ++-
 lib/trpc/memoizedRequests/index.ts            |  5 ++-
 server/routers/hotels/input.ts                |  3 ++
 server/routers/hotels/query.ts                | 35 +++++++++++++++----
 types/components/maps/staticMap.ts            |  1 +
 7 files changed, 50 insertions(+), 9 deletions(-)

diff --git a/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/@modal/(.)map/page.tsx b/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/@modal/(.)map/page.tsx
index cdf236184..ea19ffda1 100644
--- a/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/@modal/(.)map/page.tsx
+++ b/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/@modal/(.)map/page.tsx
@@ -58,7 +58,10 @@ export default async function SelectHotelMapPage({
 
   const hotelPins = getHotelPins(hotels)
   const filterList = getFiltersFromHotels(hotels)
-  const cityCoordinates = await getCityCoordinates({ city: city.name })
+  const cityCoordinates = await getCityCoordinates({
+    city: city.name,
+    hotel: { address: hotels[0].hotelData.address.streetAddress },
+  })
 
   return (
     <MapModal>
diff --git a/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/page.tsx b/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/page.tsx
index 49893c54d..3954be688 100644
--- a/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/page.tsx
+++ b/app/[lang]/(live)/(public)/hotelreservation/(standard)/select-hotel/page.tsx
@@ -51,8 +51,12 @@ export default async function SelectHotelPage({
     (location) =>
       location.name.toLowerCase() === searchParams.city.toLowerCase()
   )
+
   if (!city) return notFound()
 
+  const isCityWithCountry = (city: any): city is { country: string } =>
+    "country" in city
+
   const intl = await getIntl()
   const selectHotelParams = new URLSearchParams(searchParams)
   const selectHotelParamsObject =
@@ -124,6 +128,7 @@ export default async function SelectHotelPage({
               <div className={styles.mapContainer}>
                 <StaticMap
                   city={searchParams.city}
+                  country={isCityWithCountry(city) ? city.country : undefined}
                   width={340}
                   height={180}
                   zoomLevel={11}
diff --git a/components/Maps/StaticMap/index.tsx b/components/Maps/StaticMap/index.tsx
index 52f7dd7b3..7c80aa01f 100644
--- a/components/Maps/StaticMap/index.tsx
+++ b/components/Maps/StaticMap/index.tsx
@@ -7,6 +7,7 @@ import { StaticMapProps } from "@/types/components/maps/staticMap"
 
 export default function StaticMap({
   city,
+  country,
   coordinates,
   width,
   height,
@@ -18,7 +19,9 @@ export default function StaticMap({
   const key = env.GOOGLE_STATIC_MAP_KEY
   const secret = env.GOOGLE_STATIC_MAP_SIGNATURE_SECRET
   const baseUrl = "https://maps.googleapis.com/maps/api/staticmap"
-  const center = coordinates ? `${coordinates.lat},${coordinates.lng}` : city
+  const center = coordinates
+    ? `${coordinates.lat},${coordinates.lng}`
+    : `${city}, ${country}`
 
   if (!center) {
     return null
diff --git a/lib/trpc/memoizedRequests/index.ts b/lib/trpc/memoizedRequests/index.ts
index 75e58765f..1380517e3 100644
--- a/lib/trpc/memoizedRequests/index.ts
+++ b/lib/trpc/memoizedRequests/index.ts
@@ -160,7 +160,10 @@ export const getBookingConfirmation = cache(
 )
 
 export const getCityCoordinates = cache(
-  async function getMemoizedCityCoordinates(input: { city: string }) {
+  async function getMemoizedCityCoordinates(input: {
+    city: string
+    hotel: { address: string }
+  }) {
     return serverClient().hotel.map.city(input)
   }
 )
diff --git a/server/routers/hotels/input.ts b/server/routers/hotels/input.ts
index 7fe4c02fd..a4e47caf3 100644
--- a/server/routers/hotels/input.ts
+++ b/server/routers/hotels/input.ts
@@ -76,4 +76,7 @@ export const getRoomPackagesInputSchema = z.object({
 })
 export const getCityCoordinatesInputSchema = z.object({
   city: z.string(),
+  hotel: z.object({
+    address: z.string(),
+  }),
 })
diff --git a/server/routers/hotels/query.ts b/server/routers/hotels/query.ts
index 900b31a89..910044af8 100644
--- a/server/routers/hotels/query.ts
+++ b/server/routers/hotels/query.ts
@@ -1087,14 +1087,37 @@ export const hotelQueryRouter = router({
       .input(getCityCoordinatesInputSchema)
       .query(async function ({ input }) {
         const apiKey = process.env.GOOGLE_STATIC_MAP_KEY
-        const { city } = input
-        const url = `https://maps.googleapis.com/maps/api/geocode/json?address=${encodeURIComponent(city)}&key=${apiKey}`
+        const { city, hotel } = input
 
-        const response = await fetch(url)
-        const data = await response.json()
-        const { lat, lng } = data.results[0].geometry.location
+        async function fetchCoordinates(address: string) {
+          const url = `https://maps.googleapis.com/maps/api/geocode/json?address=${encodeURIComponent(address)}&key=${apiKey}`
+          const response = await fetch(url)
+          const data = await response.json()
 
-        return { lat, lng }
+          if (data.status !== "OK") {
+            console.error(`Geocode error: ${data.status}`)
+            return null
+          }
+
+          const location = data.results[0]?.geometry?.location
+          if (!location) {
+            console.error("No location found in geocode response")
+            return null
+          }
+
+          return location
+        }
+
+        let location = await fetchCoordinates(city)
+        if (!location) {
+          location = await fetchCoordinates(`${city}, ${hotel.address}`)
+        }
+
+        if (!location) {
+          throw new Error("Unable to fetch coordinates")
+        }
+
+        return location
       }),
   }),
 })
diff --git a/types/components/maps/staticMap.ts b/types/components/maps/staticMap.ts
index 8519f4101..4694eb8bc 100644
--- a/types/components/maps/staticMap.ts
+++ b/types/components/maps/staticMap.ts
@@ -2,6 +2,7 @@ import type { Coordinates } from "./coordinates"
 
 export type StaticMapProps = {
   city?: string
+  country?: string
   coordinates?: Coordinates
   width: number
   height: number