deadcxap/liberama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Book Pauk
2019-01-11 22:56:15 +07:00
parent 75bed20f0c
commit c46edd4cad
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server/core/ReaderWorker.js
View File

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