Добавил ограничение на максимальный размер скачиваемого файла

This commit is contained in:
Book Pauk
2019-01-11 22:56:15 +07:00
parent 75bed20f0c
commit c46edd4cad

View File

@@ -15,6 +15,8 @@ class ReaderWorker {
}
async loadBook(wState, url) {
const maxDownloadSize = 10*1024*1024;
let errMes = '';
try {
wState.set({state: 'download', step: 1, totalSteps: 3, url});
@@ -22,13 +24,16 @@ class ReaderWorker {
const d = download(url);
d.on('downloadProgress', progress => {
wState.set({progress: Math.round(progress.percent*100)});
})
if (progress.transferred > maxDownloadSize) {
errMes = 'file too big';
d.destroy();
}
});
await pipeline(d, fs.createWriteStream(`${this.tempDownloadDir}/${tempFilename}`));
wState.finish({step: 3, file: tempFilename});
} catch (e) {
wState.set({state: 'error', error: e.message});
wState.set({state: 'error', error: (errMes ? errMes : e.message)});
}
}