Рефакторинг

This commit is contained in:
Book Pauk
2019-01-13 15:43:54 +07:00
parent d040dd0718
commit 329b828f1f

View File

@@ -6,15 +6,23 @@ class FileDownloader {
async load(url, callback) { async load(url, callback) {
const maxDownloadSize = 10*1024*1024; const maxDownloadSize = 10*1024*1024;
let errMes = '';
const response = await got(url, {method: 'HEAD'});
let estSize = 100000; let estSize = 100000;
const request = got(url).on('downloadProgress', progress => { if (response.headers['content-length']) {
estSize = response.headers['content-length'];
}
const request = got(url, {encoding: null}).on('downloadProgress', progress => {
if (progress.transferred > maxDownloadSize) { if (progress.transferred > maxDownloadSize) {
errMes = 'file too big';
request.cancel(); request.cancel();
} }
const prog = Math.round(progress.transferred/estSize*100); const prog = Math.round(progress.transferred/estSize*100);
if (callback) if (callback)
callback(prog); callback((prog > 100 ? 100 : prog));
if (prog > 100) if (prog > 100)
estSize *= 1.5; estSize *= 1.5;
}); });
@@ -23,10 +31,8 @@ class FileDownloader {
try { try {
return (await request).body; return (await request).body;
} catch (error) { } catch (error) {
if (request.isCanceled) { errMes = (errMes ? errMes : error.message);
throw new Error('file too big') throw new Error(errMes);
}
throw error;
} }
} }
} }