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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
import { ipcMain, dialog, BrowserWindow } from 'electron';
import { download } from 'electron-dl';
import mime from 'mime-types';
import { writeFileSync } from 'fs-extra';
import { PathLike } from 'fs';
const debug = require('../../preload-safe-debug')('Ferdium:ipcApi:download');
function decodeBase64Image(dataString: string) {
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: { mainWindow: BrowserWindow }) => {
ipcMain.on(
'download-file',
async (_event, { url, content, fileOptions = {} }) => {
const win = BrowserWindow.getFocusedWindow();
try {
if (content) {
const extension = mime.extension(fileOptions.mime);
const filename = `${fileOptions.name}.${extension}`;
try {
const saveDialog = await dialog.showSaveDialog(params.mainWindow, {
defaultPath: filename,
});
if (saveDialog.canceled) return;
const binaryImage = decodeBase64Image(content);
writeFileSync(
saveDialog.filePath as PathLike,
binaryImage as unknown as string,
'binary',
);
debug('File blob saved to', saveDialog.filePath);
} catch (error) {
console.error(error);
}
} else {
const dl = await download(win!, url, {
saveAs: true,
});
debug('File saved to', dl.savePath);
}
} catch (error) {
console.error(error);
}
},
);
};
|