From d2bd0c7c7e559bf9742ed2ebb4944e3913970caa Mon Sep 17 00:00:00 2001
From: Marian Moravcik <marian.moravcik@emea.ironmountain.com>
Date: Thu, 12 Jan 2023 12:18:25 +0100
Subject: [PATCH] Fix task progress notification

---
 frontend/src/modules/task/index.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/frontend/src/modules/task/index.ts b/frontend/src/modules/task/index.ts
index a1439148c..7e2687c57 100644
--- a/frontend/src/modules/task/index.ts
+++ b/frontend/src/modules/task/index.ts
@@ -10,10 +10,12 @@ import { notification } from "./notification";
 class TaskDispatcher {
   private running: boolean;
   private tasks: Record<string, Task.Callable[]> = {};
+  private progress: Record<string, Site.Progress> = {};
 
   constructor() {
     this.running = false;
     this.tasks = {};
+    this.progress = {};
 
     window.addEventListener("beforeunload", this.onBeforeUnload.bind(this));
   }
@@ -108,9 +110,10 @@ class TaskDispatcher {
       // TODO: FIX ME!
       item.value += 1;
 
-      if (item.value >= item.count) {
+      if (item.value >= item.count && this.progress[item.id] !== undefined) {
         updateNotification(notification.progress.end(item.id, item.header));
-      } else if (item.value > 1) {
+        delete this.progress[item.id];
+      } else if (item.value > 1 && this.progress[item.id] !== undefined) {
         updateNotification(
           notification.progress.update(
             item.id,
@@ -122,6 +125,7 @@ class TaskDispatcher {
         );
       } else {
         showNotification(notification.progress.pending(item.id, item.header));
+        this.progress[item.id] = item;
       }
     });
   }