55 lines
1.6 KiB
JavaScript
55 lines
1.6 KiB
JavaScript
// Copyright (c) 2015-2016 Yuya Ochiai
|
|
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
import fs from 'fs';
|
|
import path from 'path';
|
|
import zlib from 'zlib';
|
|
|
|
import electron from 'electron';
|
|
const {app, dialog} = electron;
|
|
|
|
export default function downloadURL(browserWindow, URL, callback) {
|
|
const {net} = electron;
|
|
const request = net.request(URL);
|
|
request.setHeader('Accept-Encoding', 'gzip,deflate');
|
|
request.on('response', (response) => {
|
|
const file = getAttachmentName(response.headers);
|
|
const dialogOptions = {
|
|
defaultPath: path.join(app.getPath('downloads'), file),
|
|
};
|
|
dialog.showSaveDialog(browserWindow, dialogOptions, (filename) => {
|
|
if (filename) {
|
|
saveResponseBody(response, filename, callback);
|
|
}
|
|
});
|
|
}).on('error', callback);
|
|
request.end();
|
|
}
|
|
|
|
function getAttachmentName(headers) {
|
|
if (headers['content-disposition']) {
|
|
const contentDisposition = headers['content-disposition'][0];
|
|
const matched = contentDisposition.match(/filename="(.*)"/);
|
|
if (matched) {
|
|
return path.basename(matched[1]);
|
|
}
|
|
}
|
|
return '';
|
|
}
|
|
|
|
function saveResponseBody(response, filename, callback) {
|
|
const output = fs.createWriteStream(filename);
|
|
output.on('close', callback);
|
|
switch (response.headers['content-encoding']) {
|
|
case 'gzip':
|
|
response.pipe(zlib.createGunzip()).pipe(output).on('error', callback);
|
|
break;
|
|
case 'deflate':
|
|
response.pipe(zlib.createInflate()).pipe(output).on('error', callback);
|
|
break;
|
|
default:
|
|
response.pipe(output).on('error', callback);
|
|
break;
|
|
}
|
|
}
|