From 89de6c834fe3aa4946d1d8d6717b186c55d06f1c Mon Sep 17 00:00:00 2001
From: Thor Galle <thorgalle@gmail.com>
Date: Thu, 14 Mar 2024 17:31:33 +0200
Subject: [PATCH] fix: make garden REST API call compatible with local dev
 environment

---
 src/lib/api/garden.ts | 101 +++++++++++++++++++++++-------------------
 1 file changed, 55 insertions(+), 46 deletions(-)

diff --git a/src/lib/api/garden.ts b/src/lib/api/garden.ts
index e6b1cdb0..726120c6 100644
--- a/src/lib/api/garden.ts
+++ b/src/lib/api/garden.ts
@@ -124,9 +124,13 @@ export const getAllListedGardens = async () => {
 
   let appCheckTokenResponse;
   try {
-    appCheckTokenResponse = await getToken(appCheck(), /* forceRefresh= */ false);
+    // Use AppCheck if it is initialized (not on localhost development, for example)
+    if (typeof import.meta.env.VITE_FIREBASE_APP_CHECK_PUBLIC_KEY !== 'undefined') {
+      appCheckTokenResponse = await getToken(appCheck(), /* forceRefresh= */ false);
+    }
   } catch (err) {
     // Handle any errors if the token was not retrieved.
+    console.error('Error fetching app check token:', err);
     return;
   }
 
@@ -135,55 +139,60 @@ export const getAllListedGardens = async () => {
   do {
     iteration++;
 
+    const url = `${
+      // Change the REST API base URL depending on the environment
+      import.meta.env.VITE_FIREBASE_PROJECT_ID === 'demo-test'
+        ? 'http://127.0.0.1:8080/v1/projects/'
+        : 'https://firestore.googleapis.com/v1/projects/'
+    }${import.meta.env.VITE_FIREBASE_PROJECT_ID}/databases/(default)/documents:runQuery`;
     // Query the chunk of gardens using the REST api
-    const gardensChunkResponse = (await fetch(
-      `https://firestore.googleapis.com/v1/projects/${
-        import.meta.env.VITE_FIREBASE_PROJECT_ID
-      }/databases/(default)/documents:runQuery`,
-      {
-        headers: {
-          'X-Firebase-AppCheck': appCheckTokenResponse.token
-        },
-        method: 'POST',
-        body: JSON.stringify({
-          structuredQuery: {
-            from: [
-              {
-                collectionId: 'campsites',
-                allDescendants: false
+    const gardensChunkResponse = (await fetch(url, {
+      ...(appCheckTokenResponse
+        ? {
+            headers: {
+              'X-Firebase-AppCheck': appCheckTokenResponse.token
+            }
+          }
+        : {}),
+      method: 'POST',
+      body: JSON.stringify({
+        structuredQuery: {
+          from: [
+            {
+              collectionId: 'campsites',
+              allDescendants: false
+            }
+          ],
+          where: {
+            fieldFilter: {
+              field: {
+                fieldPath: 'listed'
+              },
+              op: 'EQUAL',
+              value: {
+                booleanValue: true
               }
-            ],
-            where: {
-              fieldFilter: {
-                field: {
-                  fieldPath: 'listed'
-                },
-                op: 'EQUAL',
-                value: {
-                  booleanValue: true
+            }
+          },
+          limit: CHUNK_SIZE,
+          // https://stackoverflow.com/a/71812269/4973029
+          orderBy: [
+            {
+              direction: 'ASCENDING',
+              field: { fieldPath: '__name__' }
+            }
+          ],
+          ...(startAfterDocRef
+            ? {
+                startAt: {
+                  before: false,
+                  values: [{ referenceValue: startAfterDocRef }]
                 }
               }
-            },
-            limit: CHUNK_SIZE,
-            // https://stackoverflow.com/a/71812269/4973029
-            orderBy: [
-              {
-                direction: 'ASCENDING',
-                field: { fieldPath: '__name__' }
-              }
-            ],
-            ...(startAfterDocRef
-              ? {
-                  startAt: {
-                    before: false,
-                    values: [{ referenceValue: startAfterDocRef }]
-                  }
-                }
-              : {})
-          }
-        })
-      }
-    ).then((r) => r.json())) as RESTGardenDoc[];
+            : {})
+        }
+      })
+    }).then((r) => r.json())) as RESTGardenDoc[];
 
     // Query the chunk of gardens
     if (gardensChunkResponse.length === CHUNK_SIZE) {