feat: Add Download Manager (pause, stop, delete) (#1339)

1 files changed, 80 insertions, 1 deletions

diff --git a/src/models/Service.ts b/src/models/Service.ts
index 265b3e13c..b1f0bc271 100644
--- a/src/models/Service.ts
+++ b/src/models/Service.ts
@@ -1,9 +1,10 @@
-import { join } from 'node:path';
+import { join, basename } from 'node:path';
 import { autorun, action, computed, makeObservable, observable } from 'mobx';
 import { ipcRenderer } from 'electron';
 import { webContents } from '@electron/remote';
 import ElectronWebView from 'react-electron-web-view';
 
+import { v4 as uuidV4 } from 'uuid';
 import { todosStore } from '../features/todos';
 import { isValidExternalURL, normalizedUrl } from '../helpers/url-helpers';
 import UserAgent from './UserAgent';
@@ -523,6 +524,84 @@ export default class Service {
     });
 
     if (webviewWebContents) {
+      webviewWebContents.session.on('will-download', (event, item) => {
+        event.preventDefault();
+
+        const downloadId = uuidV4();
+
+        window['ferdium'].actions.app.addDownload({
+          id: downloadId,
+          serviceId: this.id,
+          filename: item.getFilename(),
+          url: item.getURL(),
+          savePath: item.getSavePath(),
+        });
+
+        item.addListener('updated', (event, state) => {
+          if (state === 'interrupted') {
+            debug('Download is interrupted but can be resumed');
+          } else if (state === 'progressing') {
+            if (item.isPaused()) {
+              debug('Download is paused');
+            } else {
+              debug(`Received bytes: ${item.getReceivedBytes()}`);
+            }
+          }
+          window['ferdium'].actions.app.updateDownload({
+            id: downloadId,
+            serviceId: this.id,
+            filename: basename(item.getSavePath()),
+            url: item.getURL(),
+            savePath: item.getSavePath(),
+            receivedBytes: item.getReceivedBytes(),
+            totalBytes: item.getTotalBytes(),
+            state,
+          });
+          debug('download updated', event, state);
+        });
+        item.addListener('done', (event, state) => {
+          debug('download done', event, state);
+          if (state === 'completed') {
+            debug('Download successfully');
+          } else {
+            if (state === 'cancelled' && item.getSavePath() === '') {
+              window['ferdium'].actions.app.removeDownload(downloadId);
+              debug('Download is cancelled');
+            }
+            debug(`Download failed: ${state}`);
+          }
+
+          window['ferdium'].actions.app.endedDownload({
+            id: downloadId,
+            serviceId: this.id,
+            receivedBytes: item.getReceivedBytes(),
+            totalBytes: item.getTotalBytes(),
+            state,
+          });
+        });
+
+        ipcRenderer.on('toggle-pause-download', (_, data) => {
+          debug('toggle-pause-download', item.isPaused(), item.getState());
+          if (data.downloadId === downloadId || data.downloadId === undefined) {
+            if (item.isPaused()) {
+              item.resume();
+            } else {
+              item.pause();
+            }
+          }
+          debug('toggle-pause-download', item.isPaused(), item.getState());
+          window['ferdium'].actions.app.updateDownload({
+            id: downloadId,
+            paused: item.isPaused(),
+          });
+        });
+
+        ipcRenderer.on('stop-download', (_, data) => {
+          if (data === undefined || downloadId === data.downloadId) {
+            item.cancel();
+          }
+        });
+      });
       webviewWebContents.on('login', (event, _, authInfo, callback) => {
         // const authCallback = callback;
         debug('browser login event', authInfo);