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-12 00:33:58 +07:00
parent 6d9fa4cf9d
commit 15177d3fed
2 changed files with 26 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
server/core/ReaderWorker.js
View File

@@ -1,5 +1,11 @@
const workerState = require('./workerState');
const utils = require('./utils');
const fs = require('fs-extra');
const util = require('util');
const stream = require('stream');
const pipeline = util.promisify(stream.pipeline);
const download = require('download');
class ReaderWorker {
constructor(config) {
@@ -9,10 +15,26 @@ class ReaderWorker {
}
async loadBook(url, wState) {
const loader = require('./readerLoader');
loader(url, this.config, (state) => {
wState.set(state)
});
const maxDownloadSize = 10*1024*1024;
let errMes = '';
try {
wState.set({state: 'download', step: 1, totalSteps: 3, url});
const tempFilename = utils.randomHexString(30);
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.config.tempDownloadDir}/${tempFilename}`));
wState.finish({step: 3, file: tempFilename});
} catch (e) {
wState.set({state: 'error', error: (errMes ? errMes : e.message)});
}
}
loadBookUrl(url) {

32
server/core/readerLoader.js
View File

@@ -1,32 +0,0 @@
const utils = require('./utils');
const fs = require('fs-extra');
const util = require('util');
const stream = require('stream');
const pipeline = util.promisify(stream.pipeline);
const download = require('download');
async function main(url, config, setState) {
const maxDownloadSize = 10*1024*1024;
let errMes = '';
try {
setState({state: 'download', step: 1, totalSteps: 3, url});
const tempFilename = utils.randomHexString(30);
const d = download(url);
d.on('downloadProgress', progress => {
setState({progress: Math.round(progress.percent*100)});
if (progress.transferred > maxDownloadSize) {
errMes = 'file too big';
d.destroy();
}
});
await pipeline(d, fs.createWriteStream(`${config.tempDownloadDir}/${tempFilename}`));
setState({state: 'finish', step: 3, file: tempFilename});
} catch (e) {
setState({state: 'error', error: (errMes ? errMes : e.message)});
}
}
module.exports = main;