Handle redirects. Koenkk#114

nukusinji · May 16, 2022 · 0cdb428 · 0cdb428
1 parent 19e73f7
commit 0cdb428
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/scripts/add.js b/scripts/add.js
@@ -41,14 +41,20 @@ const main = async () => {
             const lib = url.toLowerCase().startsWith("https") ? require('https') : require('http');
             const file = fs.createWriteStream(path);
 
-            return new Promise((resolve) => {
+            return new Promise((resolve, reject) => {
                 const request = lib.get(url, function(response) {
-                  response.pipe(file);
-                  file.on('finish', function() {
-                    file.close(function() {
-                        resolve();
-                    });
-                  });
+                    if (response.statusCode >= 200 && response.statusCode < 300) {
+                        response.pipe(file);
+                        file.on('finish', function() {
+                          file.close(function() {
+                              resolve();
+                          });
+                        });
+                    } else if (response.headers.location) {
+                        resolve(downloadFile(response.headers.location, path));
+                    } else {
+                        reject(new Error(response.statusCode + ' ' + response.statusMessage));
+                    }
                 });
             });
         }