aboutsummaryrefslogtreecommitdiffstats
path: root/src/electron/ipc-api/download.js
blob: 36eca3c9fe8ec5c4fe877aa68e01ea217c8a1159 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import { ipcMain, dialog } from 'electron';
import { download } from 'electron-dl';
import mime from 'mime-types';
import fs from 'fs-extra';

const debug = require('debug')('Ferdi:ipcApi:download');

function decodeBase64Image(dataString) {
  const matches = dataString.match(/^data:([A-Za-z-+/]+);base64,(.+)$/);

  if (matches.length !== 3) {
    return new Error('Invalid input string');
  }

  return Buffer.from(matches[2], 'base64');
}

export default (params) => {
  ipcMain.on('download-file', async (event, { url, content, fileOptions = {} }) => {
    try {
      if (!content) {
        const dl = await download(params.mainWindow, url, {
          saveAs: true,
        });
        debug('File saved to', dl.savePath);
      } else {
        const extension = mime.extension(fileOptions.mime);
        const filename = `${fileOptions.name}.${extension}`;

        dialog.showSaveDialog(params.mainWindow, {
          defaultPath: filename,
        }, (name) => {
          const binaryImage = decodeBase64Image(content);
          fs.writeFileSync(name, binaryImage, 'binary');

          debug('File blob saved to', name);
        });
      }
    } catch (e) {
      console.error(e);
    }
  });
};