From 58cf7e816b3ffb5c008a3c4d8eeb72bba33de82b Mon Sep 17 00:00:00 2001
From: Jack Bailey <81699395+JackBailey@users.noreply.github.com>
Date: Fri, 24 Nov 2023 11:39:22 +0000
Subject: [PATCH] Add total price indicator to labels and locations

---
 frontend/lib/api/types/data-contracts.ts |  2 +
 frontend/pages/label/[id].vue            | 15 +++++-
 frontend/pages/location/[id].vue         | 66 ++++++++++++++++++++++--
 3 files changed, 78 insertions(+), 5 deletions(-)

diff --git a/frontend/lib/api/types/data-contracts.ts b/frontend/lib/api/types/data-contracts.ts
index e5457bb..2990549 100644
--- a/frontend/lib/api/types/data-contracts.ts
+++ b/frontend/lib/api/types/data-contracts.ts
@@ -193,6 +193,7 @@ export interface LabelOut {
   id: string;
   name: string;
   updatedAt: Date | string;
+  totalPrice: number | undefined;
 }
 
 export interface LabelSummary {
@@ -217,6 +218,7 @@ export interface LocationOut {
   name: string;
   parent: LocationSummary;
   updatedAt: Date | string;
+  totalPrice: number | undefined;
 }
 
 export interface LocationOutCount {
diff --git a/frontend/pages/label/[id].vue b/frontend/pages/label/[id].vue
index f190782..5c7e5d9 100644
--- a/frontend/pages/label/[id].vue
+++ b/frontend/pages/label/[id].vue
@@ -1,4 +1,6 @@
 <script setup lang="ts">
+  import Currency from "~~/components/global/Currency.vue";
+
   definePageMeta({
     middleware: ["auth"],
   });
@@ -84,6 +86,8 @@
       return [];
     }
 
+    label.value.totalPrice = resp.data.items.map(item => Number(item.purchasePrice)).reduce((a, b) => a + b, 0);
+
     return resp.data.items;
   });
 </script>
@@ -111,8 +115,17 @@
               </div>
             </div>
             <div>
-              <h1 class="text-2xl pb-1">
+              <h1 class="text-2xl pb-1 flex items-center gap-3">
                 {{ label ? label.name : "" }}
+
+                <div
+                  v-if="label && label.totalPrice"
+                  class="text-xs bg-secondary text-secondary-content rounded-full px-2 py-1"
+                >
+                  <div>
+                    <Currency :amount="label.totalPrice" />
+                  </div>
+                </div>
               </h1>
               <div class="flex gap-1 flex-wrap text-xs">
                 <div>
diff --git a/frontend/pages/location/[id].vue b/frontend/pages/location/[id].vue
index 05c6e5f..7728f5b 100644
--- a/frontend/pages/location/[id].vue
+++ b/frontend/pages/location/[id].vue
@@ -1,6 +1,7 @@
 <script setup lang="ts">
-  import { LocationSummary, LocationUpdate } from "~~/lib/api/types/data-contracts";
+  import { LocationSummary, LocationUpdate, TreeItem } from "~~/lib/api/types/data-contracts";
   import { useLocationStore } from "~~/stores/locations";
+  import Currency from "~~/components/global/Currency.vue";
 
   definePageMeta({
     middleware: ["auth"],
@@ -89,8 +90,47 @@
       return [];
     }
 
+    const locationTreeResp = await api.locations.getTree();
+
+    if (locationTreeResp.error) {
+      toast.error("Failed to load items");
+      return [];
+    }
+
+    const fullLocationTree: TreeItem[] = locationTreeResp.data;
+
+    function flattenLocation(location: TreeItem): string[] {
+      return location.children.reduce((acc, child) => [...acc, ...flattenLocation(child)], [location.id]);
+    }
+
+    function findObjectById(locations: TreeItem[], targetId: string): TreeItem | undefined {
+      if (!locations || !Array.isArray(locations)) {
+        return undefined;
+      }
+
+      const target = locations.find(location => location.id === targetId);
+      if (target) return target;
+
+      for (let i = 0; i < locations.length; i++) {
+        const result = findObjectById(locations[i].children, targetId);
+        if (result) return result;
+      }
+
+      return undefined;
+    }
+
+    const locationTree: TreeItem | undefined = findObjectById(fullLocationTree, location.value.id);
+
+    if (!locationTree) {
+      toast.error("Failed to get tree for current location");
+
+      return [];
+    }
+
+    const allChildLocations = flattenLocation(locationTree);
+
     const resp = await api.items.getAll({
-      locations: [location.value.id],
+      locations: allChildLocations,
     });
 
     if (resp.error) {
@@ -98,7 +138,16 @@
       return [];
     }
 
-    return resp.data.items;
+    const locationItems = resp.data.items.filter(item => {
+      if (item.location && location.value) return item?.location.id === location?.value.id;
+      return false;
+    });
+
+    const totalPrice = resp.data.items.map(item => Number(item.purchasePrice)).reduce((a, b) => a + b, 0);
+
+    location.value.totalPrice = totalPrice;
+
+    return locationItems;
   });
 </script>
 
@@ -135,8 +184,17 @@
                   <li>{{ location.name }}</li>
                 </ul>
               </div>
-              <h1 class="text-2xl pb-1">
+              <h1 class="text-2xl pb-1 flex items-center gap-3">
                 {{ location ? location.name : "" }}
+
+                <div
+                  v-if="location && location.totalPrice"
+                  class="text-xs bg-secondary text-secondary-content rounded-full px-2 py-1"
+                >
+                  <div>
+                    <Currency :amount="location.totalPrice" />
+                  </div>
+                </div>
               </h1>
               <div class="flex gap-1 flex-wrap text-xs">
                 <div>