From 20657bb3da91b9769c19278c833786043786497d Mon Sep 17 00:00:00 2001
From: NamH <NamNh0122@gmail.com>
Date: Mon, 15 Apr 2024 21:52:22 +0700
Subject: [PATCH] fix: cannot cancel model which is progress at 0 percent
 (#2727)

Signed-off-by: James <james@jan.ai>
Co-authored-by: James <james@jan.ai>
---
 core/src/node/api/processors/download.ts | 13 ++++++++-----
 core/src/node/helper/download.ts         |  4 ++++
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/core/src/node/api/processors/download.ts b/core/src/node/api/processors/download.ts
index 98464dd52..d26e46d59 100644
--- a/core/src/node/api/processors/download.ts
+++ b/core/src/node/api/processors/download.ts
@@ -66,6 +66,7 @@ export class Downloader implements Processor {
       localPath: normalizedPath,
     }
     DownloadManager.instance.downloadProgressMap[modelId] = initialDownloadState
+    DownloadManager.instance.downloadInfo[normalizedPath] = initialDownloadState
 
     if (downloadRequest.downloadType === 'extension') {
       observer?.(DownloadEvent.onFileDownloadUpdate, initialDownloadState)
@@ -118,12 +119,14 @@ export class Downloader implements Processor {
     if (rq) {
       DownloadManager.instance.networkRequests[fileName] = undefined
       rq?.abort()
-    } else {
-      observer?.(DownloadEvent.onFileDownloadError, {
-        fileName,
-        error: 'aborted',
-      })
     }
+
+    const downloadInfo = DownloadManager.instance.downloadInfo[fileName]
+    observer?.(DownloadEvent.onFileDownloadError, {
+      ...downloadInfo,
+      fileName,
+      error: 'aborted',
+    })
   }
 
   resumeDownload(observer: any, fileName: any) {
diff --git a/core/src/node/helper/download.ts b/core/src/node/helper/download.ts
index b9fb88bb5..b7560d100 100644
--- a/core/src/node/helper/download.ts
+++ b/core/src/node/helper/download.ts
@@ -8,8 +8,12 @@ export class DownloadManager {
 
   public static instance: DownloadManager = new DownloadManager()
 
+  // store the download information with key is model id
   public downloadProgressMap: Record<string, DownloadState> = {}
 
+  // store the download infomation with key is normalized file path
+  public downloadInfo: Record<string, DownloadState> = {}
+
   constructor() {
     if (DownloadManager.instance) {
       return DownloadManager.instance