Рефакторинг

This commit is contained in:
Book Pauk
2022-10-15 18:59:13 +07:00
parent f519bf3f67
commit 0894a38978
2 changed files with 6 additions and 10 deletions

View File

@@ -9,10 +9,10 @@ class FileDownloader {
this.limitDownloadSize = limitDownloadSize;
}
async load(url, callback, abort) {
async load(url, opts, callback, abort) {
let errMes = '';
const options = {
let options = {
headers: {
'user-agent': userAgent,
timeout: 300*1000,
@@ -22,6 +22,8 @@ class FileDownloader {
}),
responseType: 'stream',
};
if (opts)
options = Object.assign({}, opts, options);
try {
const res = await axios.get(url, options);

View File

@@ -64,17 +64,11 @@ class RemoteLib {
const response = await await this.wsRequest({action: 'get-book-link', bookPath, downFileName});
const link = response.link;
const buf = await this.down.load(`${this.remoteHost}${link}`);
const buf = await this.down.load(`${this.remoteHost}${link}`, {decompress: false});
const tmpFile = `${this.config.tempDir}/${utils.randomHexString(30)}`;
const tmpFile2 = `${this.config.tempDir}/${utils.randomHexString(30)}`;
const publicPath = `${this.config.publicDir}${link}`;
await fs.writeFile(tmpFile, buf);
await utils.gzipFile(tmpFile, tmpFile2, 4);
await fs.remove(tmpFile);
await fs.move(tmpFile2, publicPath, {overwrite: true});
await fs.writeFile(publicPath, buf);
return path.basename(link);
} catch (e) {