fix: cancel download does not work

Author: louis-jan

core/src/core.ts

@@ -47,6 +47,14 @@ const downloadFile: (url: string, fileName: string) => Promise<any> = (
 const deleteFile: (path: string) => Promise<any> = (path) =>
   window.coreAPI?.deleteFile(path) ?? window.electronAPI?.deleteFile(path);
 
+/**
+ * Aborts the download of a specific file.
+ * @param {string} fileName - The name of the file whose download is to be aborted.
+ * @returns {Promise<any>} A promise that resolves when the download has been aborted.
+ */
+const abortDownload: (fileName: string) => Promise<any> = (fileName) =>
+  window.coreAPI?.abortDownload(fileName);
+
 /**
  * Retrieves the path to the app data directory using the `coreAPI` object.
  * If the `coreAPI` object is not available, the function returns `undefined`.
@@ -79,6 +87,7 @@ export const core = {
   invokePluginFunc,
   executeOnMain,
   downloadFile,
+  abortDownload,
   deleteFile,
   appDataPath,
   getUserSpace,
@@ -91,6 +100,7 @@ export {
   invokePluginFunc,
   executeOnMain,
   downloadFile,
+  abortDownload,
   deleteFile,
   appDataPath,
   getUserSpace,

core/src/index.ts

@@ -8,7 +8,13 @@ export { core, deleteFile, invokePluginFunc } from "./core";
  * Core module exports.
  * @module
  */
-export { downloadFile, executeOnMain, appDataPath, getUserSpace } from "./core";
+export {
+  downloadFile,
+  executeOnMain,
+  appDataPath,
+  getUserSpace,
+  abortDownload,
+} from "./core";
 
 /**
  * Events module exports.

core/src/plugins/model.ts

@@ -16,6 +16,13 @@ export abstract class ModelPlugin extends JanPlugin {
    */
   abstract downloadModel(model: Model): Promise<void>;
 
+  /**
+   * Cancels the download of a specific model.
+   * @param {string} modelId - The ID of the model to cancel the download for.
+   * @returns {Promise<void>} A promise that resolves when the download has been cancelled.
+   */
+  abstract cancelModelDownload(modelId: string): Promise<void>;
+
   /**
    * Deletes a model.
    * @param filePath - The file path of the model to delete.

plugins/model-plugin/src/index.ts

@@ -1,4 +1,4 @@
-import { PluginType, fs, downloadFile } from '@janhq/core'
+import { PluginType, fs, downloadFile, abortDownload } from '@janhq/core'
 import { ModelPlugin } from '@janhq/core/lib/plugins'
 import { Model, ModelCatalog } from '@janhq/core/lib/types'
 import { parseToModel } from './helpers/modelParser'
@@ -50,6 +50,15 @@ export default class JanModelPlugin implements ModelPlugin {
     downloadFile(model.downloadLink, path)
   }
 
+  /**
+   * Cancels the download of a specific machine learning model.
+   * @param {string} modelId - The ID of the model whose download is to be cancelled.
+   * @returns {Promise<void>} A promise that resolves when the download has been cancelled.
+   */
+  async cancelModelDownload(modelId: string): Promise<void> {
+    return abortDownload(join(JanModelPlugin._homeDir, modelId, modelId))
+  }
+
   /**
    * Deletes a machine learning model.
    * @param filePath - The path to the model file to delete.

web/containers/ModalCancelDownload/index.tsx

@@ -1,5 +1,7 @@
 import { useMemo } from 'react'
 
+import { PluginType } from '@janhq/core'
+import { ModelPlugin } from '@janhq/core/lib/plugins'
 import { ModelVersion } from '@janhq/core/lib/types'
 
 import {
@@ -19,6 +21,8 @@ import { useDownloadState } from '@/hooks/useDownloadState'
 
 import { formatDownloadPercentage } from '@/utils/converter'
 
+import { pluginManager } from '@/plugin'
+
 type Props = {
   suitableModel: ModelVersion
   isFromList?: boolean
@@ -67,9 +71,9 @@ export default function ModalCancelDownload({
                 themes="danger"
                 onClick={() => {
                   if (downloadState?.fileName)
-                    window.coreAPI?.abortDownload(
-                      `models/${downloadState?.fileName}`
-                    )
+                    pluginManager
+                      .get<ModelPlugin>(PluginType.Model)
+                      ?.cancelModelDownload(downloadState.fileName)
                 }}
               >
                 Yes