Рефакторинг

server/core/FileDownloader.js

@@ -6,15 +6,23 @@ class FileDownloader {
 
     async load(url, callback) {
         const maxDownloadSize = 10*1024*1024;
+        let errMes = '';
+
+        const response = await got(url, {method: 'HEAD'});
 
         let estSize = 100000;
-        const request = got(url).on('downloadProgress', progress => {
+        if (response.headers['content-length']) {
+            estSize = response.headers['content-length'];
+        }
+
+        const request = got(url, {encoding: null}).on('downloadProgress', progress => {
             if (progress.transferred > maxDownloadSize) {
+                errMes = 'file too big';
                 request.cancel();
             }
             const prog = Math.round(progress.transferred/estSize*100);
             if (callback)
-                callback(prog);
+                callback((prog > 100 ? 100 : prog));
             if (prog > 100)
                 estSize *= 1.5;
         });
@@ -23,10 +31,8 @@ class FileDownloader {
         try {
             return (await request).body;
         } catch (error) {
-            if (request.isCanceled) {
-                throw new Error('file too big')
-            }
-            throw error;
+            errMes = (errMes ? errMes : error.message);
+            throw new Error(errMes);
         }
     }
 }