Merge pull request #206 from DavidVujic/retry_download

Retry download if failed

Author: DavidVujic

scripts/helper.js

@@ -11,6 +11,22 @@ function exec(action) {
     return res.stdout;
 }
 
+async function retry(func, ...args) {
+    let res;
+
+    try {
+        res = await func(...args);
+    } catch (e) {
+        shell.echo(e.message);
+        shell.echo(`Retrying ${func.name}`);
+
+        res = await func(...args);
+    }
+
+    return res;
+}
+
 module.exports = {
     exec,
+    retry,
 };

scripts/prepublish.js

@@ -4,7 +4,7 @@ const shell = require('shelljs');
 const decompress = require('decompress');
 const decompressTargz = require('decompress-targz');
 const env = require('./env.js');
-const { exec } = require('./helper.js');
+const { exec, retry } = require('./helper.js');
 
 function writeFile(url, destination, resolve, reject) {
     const file = fs.createWriteStream(destination);
@@ -14,12 +14,12 @@ function writeFile(url, destination, resolve, reject) {
     });
 
     req.on('error', (e) => {
-        file.unlink(destination);
+        shell.rm(destination);
         reject(e);
     });
 
     file.on('error', () => {
-        file.unlink(destination);
+        shell.rm(destination);
         reject(new Error('File error'));
     });
 
@@ -115,7 +115,7 @@ if (env.isVerbose) {
 
 shell.cd(env.workFolder);
 
-download(env.downloadUrl, env.downloadedFileName)
+retry(download, env.downloadUrl, env.downloadedFileName)
     .then(() => {
         validateFile(env.downloadedFileName);