From 05ea73bffe2c5f1c03f71b616729b9054abae63c Mon Sep 17 00:00:00 2001
From: Alex Yong <alexjyong@gmail.com>
Date: Thu, 11 Apr 2024 20:08:14 +0000
Subject: [PATCH] adding in updates

---
 frontend/src/components/prompts/UploadFiles.vue |  3 ++-
 frontend/src/stores/upload.ts                   | 14 ++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/frontend/src/components/prompts/UploadFiles.vue b/frontend/src/components/prompts/UploadFiles.vue
index 288d3fc9..be27bd07 100644
--- a/frontend/src/components/prompts/UploadFiles.vue
+++ b/frontend/src/components/prompts/UploadFiles.vue
@@ -11,7 +11,7 @@
           <div class="upload-speed">{{ uploadSpeed.toFixed(2) }} MB/s</div>
           <div class="upload-eta">{{ formattedETA }} remaining</div>
           <div class="upload-percentage"> {{ getProgress }} % Completed </div>
-          <div class="upload-fraction"> / {{ getTotalSize }} </div>
+          <div class="upload-fraction"> {{ getTotalProgressBytes }} / {{ getTotalSize }} </div>
         </div>
         <button
           class="action"
@@ -75,6 +75,7 @@ export default {
       "uploadSpeed",
       "getETA",
       "getProgress",
+      "getTotalProgressBytes",
       "getTotalSize"
     ]),
     ...mapWritableState(useFileStore, ["reload"]),
diff --git a/frontend/src/stores/upload.ts b/frontend/src/stores/upload.ts
index 0c1e1cd8..fde357e3 100644
--- a/frontend/src/stores/upload.ts
+++ b/frontend/src/stores/upload.ts
@@ -47,6 +47,20 @@ export const useUploadStore = defineStore("upload", {
       const sum = state.progress.reduce((acc, val) => +acc + +val) as number;
       return Math.ceil((sum / totalSize) * 100);
     },
+    getTotalProgressBytes: (state) => {
+     if (state.progress.length === 0 || state.sizes.length === 0) {
+        return '0 Bytes';
+      }
+    
+      const totalSize = state.sizes.reduce((a, b) => a + b, 0);
+
+      // TODO: this looks ugly but it works with ts now
+      const sum = state.progress.reduce((acc, val) => +acc + +val) as number;
+      const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'];
+      const i = Math.floor(Math.log(sum) / Math.log(1024));
+      return parseFloat((sum / Math.pow(1024, i)).toFixed(2)) + ' ' + sizes[i];
+      
+    },
     getTotalSize: (state) => {
       if (state.sizes.length === 0) {
         return '0 Bytes';