From a404fb043da2573bf04385863b2d34b1f918b8e1 Mon Sep 17 00:00:00 2001
From: M A E R Y O <pu8046@gmail.com>
Date: Mon, 28 Aug 2023 06:59:49 +0900
Subject: [PATCH] feat: implement abort upload functionality (#2673)

---
 frontend/src/api/tus.js                       | 22 ++++++++++++++++++
 .../src/components/prompts/UploadFiles.vue    | 23 +++++++++++++++++--
 frontend/src/i18n/en.json                     |  3 +++
 frontend/src/store/getters.js                 |  6 ++---
 frontend/src/store/mutations.js               |  7 ++++++
 5 files changed, 55 insertions(+), 6 deletions(-)

diff --git a/frontend/src/api/tus.js b/frontend/src/api/tus.js
index bda4f40c..f2e613e2 100644
--- a/frontend/src/api/tus.js
+++ b/frontend/src/api/tus.js
@@ -6,6 +6,7 @@ import { fetchURL } from "./utils";
 
 const RETRY_BASE_DELAY = 1000;
 const RETRY_MAX_DELAY = 20000;
+const CURRENT_UPLOAD_LIST = {};
 
 export async function upload(
   filePath,
@@ -34,6 +35,7 @@ export async function upload(
         "X-Auth": store.state.jwt,
       },
       onError: function (error) {
+        delete CURRENT_UPLOAD_LIST[filePath];
         reject("Upload failed: " + error);
       },
       onProgress: function (bytesUploaded) {
@@ -44,10 +46,12 @@ export async function upload(
         }
       },
       onSuccess: function () {
+        delete CURRENT_UPLOAD_LIST[filePath];
         resolve();
       },
     });
     upload.start();
+    CURRENT_UPLOAD_LIST[filePath] = upload;
   });
 }
 
@@ -88,3 +92,21 @@ export async function useTus(content) {
 function isTusSupported() {
   return tus.isSupported === true;
 }
+
+export function abortUpload(filePath) {
+  const upload = CURRENT_UPLOAD_LIST[filePath];
+  if (upload) {
+    upload.abort();
+    delete CURRENT_UPLOAD_LIST[filePath];
+  }
+}
+
+export function abortAllUploads() {
+  for (let filePath in CURRENT_UPLOAD_LIST) {
+    const upload = CURRENT_UPLOAD_LIST[filePath];
+    if (upload) {
+      upload.abort();
+      delete CURRENT_UPLOAD_LIST[filePath];
+    }
+  }
+}
\ No newline at end of file
diff --git a/frontend/src/components/prompts/UploadFiles.vue b/frontend/src/components/prompts/UploadFiles.vue
index 37ea3a36..b9336c9a 100644
--- a/frontend/src/components/prompts/UploadFiles.vue
+++ b/frontend/src/components/prompts/UploadFiles.vue
@@ -7,7 +7,14 @@
     <div class="card floating">
       <div class="card-title">
         <h2>{{ $t("prompts.uploadFiles", { files: filesInUploadCount }) }}</h2>
-
+        <button
+          class="action"
+          @click="abortAll"
+          aria-label="Abort upload"
+          title="Abort upload"
+        >
+        <i class="material-icons">{{ "cancel" }}</i>
+        </button>
         <button
           class="action"
           @click="toggle"
@@ -42,7 +49,9 @@
 </template>
 
 <script>
-import { mapGetters } from "vuex";
+import { mapGetters, mapMutations } from "vuex";
+import { abortAllUploads } from "@/api/tus";
+import buttons from "@/utils/buttons";
 
 export default {
   name: "uploadFiles",
@@ -53,11 +62,21 @@ export default {
   },
   computed: {
     ...mapGetters(["filesInUpload", "filesInUploadCount"]),
+    ...mapMutations(['resetUpload']),
   },
   methods: {
     toggle: function () {
       this.open = !this.open;
     },
+    abortAll() {
+      if (confirm(this.$t('upload.abortUpload'))) {
+        abortAllUploads();
+        buttons.done('upload');
+        this.open = false;
+        this.$store.commit('resetUpload');
+        this.$store.commit("setReload", true);
+      }
+    },
   },
 };
 </script>
diff --git a/frontend/src/i18n/en.json b/frontend/src/i18n/en.json
index a348b488..d9ba48cc 100644
--- a/frontend/src/i18n/en.json
+++ b/frontend/src/i18n/en.json
@@ -45,6 +45,9 @@
     "downloadFolder": "Download Folder",
     "downloadSelected": "Download Selected"
   },
+  "upload": {
+    "abortUpload": "Are you sure you want to abort?"
+  },
   "errors": {
     "forbidden": "You don't have permissions to access this.",
     "internal": "Something really went wrong.",
diff --git a/frontend/src/store/getters.js b/frontend/src/store/getters.js
index ccad1749..5ceff381 100644
--- a/frontend/src/store/getters.js
+++ b/frontend/src/store/getters.js
@@ -4,7 +4,7 @@ const getters = {
   isListing: (state, getters) => getters.isFiles && state.req.isDir,
   selectedCount: (state) => state.selected.length,
   progress: (state) => {
-    if (state.upload.progress.length == 0) {
+    if (state.upload.progress.length === 0) {
       return 0;
     }
 
@@ -14,9 +14,7 @@ const getters = {
     return Math.ceil((sum / totalSize) * 100);
   },
   filesInUploadCount: (state) => {
-    let total =
-      Object.keys(state.upload.uploads).length + state.upload.queue.length;
-    return total;
+    return Object.keys(state.upload.uploads).length + state.upload.queue.length;
   },
   filesInUpload: (state) => {
     let files = [];
diff --git a/frontend/src/store/mutations.js b/frontend/src/store/mutations.js
index bc8cece7..81947a7d 100644
--- a/frontend/src/store/mutations.js
+++ b/frontend/src/store/mutations.js
@@ -97,6 +97,13 @@ const mutations = {
     state.clipboard.key = "";
     state.clipboard.items = [];
   },
+  resetUpload(state) {
+    state.upload.uploads = {};
+    state.upload.queue = [];
+    state.upload.progress = [];
+    state.upload.sizes = [];
+    state.upload.id = 0;
+  },
 };
 
 export default mutations;