Throw error on incorrect download sizes

This commit is contained in:
2019-08-26 17:57:18 -05:00
parent b36878b80c
commit 6521f37471

View File

@@ -182,32 +182,43 @@ module.exports.downloadFile = (url, destination, progressCallback, completeCallb
.get(url, {
responseType: 'stream',
})
.then((response) => {
const stream = fs.createWriteStream(destination);
.then(response => {
const total = parseInt(response.headers['content-length'], 10);
let downloaded = 0;
response.data.on('data', (chunk) => {
stream.write(Buffer.from(chunk));
downloaded += chunk.length;
if (progressCallback) {
progressCallback((downloaded / total * 100.0).toFixed(2));
}
});
if (total === 0) {
throw Error('No data available for download');
} else {
const stream = fs.createWriteStream(destination);
response.data.on('end', () => {
stream.end(() => {
completeCallback();
let downloaded = 0;
response.data.on('data', (chunk) => {
stream.write(Buffer.from(chunk));
downloaded += chunk.length;
if (progressCallback) {
progressCallback((downloaded / total * 100.0).toFixed(2));
}
});
});
response.data.on('error', (e) => {
stream.end(() => {
completeCallback(e);
response.data.on('end', () => {
stream.end(() => {
if (downloaded === 0) {
throw Error('Received 0 bytes');
} else if (downloaded !== total) {
throw Error('Received incorrect number of bytes');
} else {
completeCallback();
}
});
});
});
response.data.on('error', error => {
stream.end(() => {
completeCallback(error);
});
});
}
})
.catch((e)=> {
completeCallback(e);
.catch(error => {
completeCallback(error);
});
};