From 52b96e69a5923add137a19e71cdab99507859e48 Mon Sep 17 00:00:00 2001
From: Faisal Amir <urmauur@gmail.com>
Date: Mon, 13 Nov 2023 23:02:54 +0700
Subject: [PATCH] Remove file from application data when user cancel download a
 model

---
 electron/main.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/electron/main.ts b/electron/main.ts
index cecc90f42..86d9d0b34 100644
--- a/electron/main.ts
+++ b/electron/main.ts
@@ -396,7 +396,20 @@ function handleIPCs() {
   ipcMain.handle("abortDownload", async (_event, fileName) => {
     const rq = networkRequests[fileName];
     networkRequests[fileName] = undefined;
+    const userDataPath = app.getPath("userData");
+    const fullPath = join(userDataPath, fileName);
     rq?.abort();
+    let result = "NULL";
+    unlink(fullPath, function (err) {
+      if (err && err.code == "ENOENT") {
+        result = `File not exist: ${err}`;
+      } else if (err) {
+        result = `File delete error: ${err}`;
+      } else {
+        result = "File deleted successfully";
+      }
+      console.log(`Delete file ${fileName} from ${fullPath} result: ${result}`);
+    });
   });
 
   /**