fix: allow users cancel backend download

extensions/llamacpp-extension/src/backend.ts

@@ -319,7 +319,10 @@ export async function downloadBackend(
     events.emit('onFileDownloadSuccess', { modelId: taskId, downloadType })
   } catch (error) {
     // Fallback: if GitHub fails, retry once with CDN
-    if (source === 'github') {
+    if (
+      source === 'github' &&
+      error?.toString() !== 'Error: Download cancelled'
+    ) {
       console.warn(`GitHub download failed, falling back to CDN:`, error)
       return await downloadBackend(backend, version, 'cdn')
     }

web-app/src/containers/DownloadManegement.tsx

@@ -1,88 +1,88 @@
 import {
   Popover,
   PopoverContent,
   PopoverTrigger,
 } from '@/components/ui/popover'
 import { Progress } from '@/components/ui/progress'
 import { useDownloadStore } from '@/hooks/useDownloadStore'
 import { useLeftPanel } from '@/hooks/useLeftPanel'
 import { useAppUpdater } from '@/hooks/useAppUpdater'
 import { useServiceHub } from '@/hooks/useServiceHub'
 import { DownloadEvent, DownloadState, events, AppEvent } from '@janhq/core'
 import { IconDownload, IconX } from '@tabler/icons-react'
 import { useCallback, useEffect, useMemo, useState } from 'react'
 import { toast } from 'sonner'
 import { useTranslation } from '@/i18n/react-i18next-compat'
 
 export function DownloadManagement() {
   const { t } = useTranslation()
   const { open: isLeftPanelOpen } = useLeftPanel()
   const [isPopoverOpen, setIsPopoverOpen] = useState(false)
   const serviceHub = useServiceHub()
   const {
     downloads,
     updateProgress,
     localDownloadingModels,
     removeDownload,
     removeLocalDownloadingModel,
   } = useDownloadStore()
   const { updateState } = useAppUpdater()
 
   const [appUpdateState, setAppUpdateState] = useState({
     isDownloading: false,
     downloadProgress: 0,
     downloadedBytes: 0,
     totalBytes: 0,
   })
 
   useEffect(() => {
     setAppUpdateState({
       isDownloading: updateState.isDownloading,
       downloadProgress: updateState.downloadProgress,
       downloadedBytes: updateState.downloadedBytes,
       totalBytes: updateState.totalBytes,
     })
   }, [updateState])
 
   const onAppUpdateDownloadUpdate = useCallback(
     (data: {
       progress?: number
       downloadedBytes?: number
       totalBytes?: number
     }) => {
       setAppUpdateState((prev) => ({
         ...prev,
         isDownloading: true,
         downloadProgress: data.progress || 0,
         downloadedBytes: data.downloadedBytes || 0,
         totalBytes: data.totalBytes || 0,
       }))
     },
     []
   )
 
   const onAppUpdateDownloadSuccess = useCallback(() => {
     setAppUpdateState((prev) => ({
       ...prev,
       isDownloading: false,
       downloadProgress: 1,
     }))
     toast.success(t('common:toast.appUpdateDownloaded.title'), {
       description: t('common:toast.appUpdateDownloaded.description'),
     })
   }, [t])
 
   const onAppUpdateDownloadError = useCallback(() => {
     setAppUpdateState((prev) => ({
       ...prev,
       isDownloading: false,
     }))
     toast.error(t('common:toast.appUpdateDownloadFailed.title'), {
       description: t('common:toast.appUpdateDownloadFailed.description'),
     })
   }, [t])
 
   const downloadProcesses = useMemo(() => {
     // Get downloads with progress data
     const downloadsWithProgress = Object.values(downloads).map((download) => ({
       id: download.name,
@@ -400,23 +400,33 @@
                           className="text-main-view-fg/70 cursor-pointer"
                           title="Cancel download"
                           onClick={() => {
-                            serviceHub
-                              .models()
-                              .abortDownload(download.name)
-                              .then(() => {
-                                toast.info(
-                                  t('common:toast.downloadCancelled.title'),
-                                  {
-                                    id: 'cancel-download',
-                                    description: t(
-                                      'common:toast.downloadCancelled.description'
-                                    ),
-                                  }
+                            // TODO: Consolidate cancellation logic
+                            if (download.id.startsWith('llamacpp')) {
+                              const downloadManager =
+                                window.core.extensionManager.getByName(
+                                  '@janhq/download-extension'
                                 )
-                                if (downloadProcesses.length === 0) {
-                                  setIsPopoverOpen(false)
-                                }
-                              })
+
+                              downloadManager.cancelDownload(download.id)
+                            } else {
+                              serviceHub
+                                .models()
+                                .abortDownload(download.name)
+                                .then(() => {
+                                  toast.info(
+                                    t('common:toast.downloadCancelled.title'),
+                                    {
+                                      id: 'cancel-download',
+                                      description: t(
+                                        'common:toast.downloadCancelled.description'
+                                      ),
+                                    }
+                                  )
+                                  if (downloadProcesses.length === 0) {
+                                    setIsPopoverOpen(false)
+                                  }
+                                })
+                            }
                           }}
                         />
                       </div>