update

add proxy test

cleanup

version bump

2 files changed, 23 insertions, 0 deletions

diff --git a/lib/login.js b/lib/login.js
index d7f3fdf..01ecd6d 100644
--- a/lib/login.js
+++ b/lib/login.js
@@ -193,5 +193,20 @@ module.exports = {
 
             callback(new Error(res.statusCode));
         });
+    },
+
+    checkProxy: function checkProxy(target, callback) {
+        request(target, function get(err, response) {
+            if (err !== null) { return callback(err); }
+            if (response.statusCode === 200) {
+                var stack = '';
+                if (response.headers.server) { stack = 'Proxy Server: '.bold + response.headers.server.cyan + ' '; }
+                if (response.headers['x-powered-by']) { stack += 'Powered-by: '.bold + response.headers['x-powered-by'].cyan + ' '; }
+                stack.trim();
+                if (stack === '') { stack = target; }
+                return callback(null, stack);
+            }
+            callback(new Error('Server Response: '.red + response.statusCode));
+        });
     }
 };
diff --git a/lib/videos.js b/lib/videos.js
index a71d4e3..bf8973e 100644
--- a/lib/videos.js
+++ b/lib/videos.js
@@ -325,5 +325,13 @@ module.exports = {
 
         });
 
+    },
+
+    checkProxyDownload: function checkProxyDownload(argv) {
+        setOptions(argv);
+        var video = 'https://youtu.be/nm20j_x9Ol8', opt = { all: undefined, cancel: undefined };
+        if (typeof argv.test === 'string') { video = argv.test; }
+        opt[video] = 0;
+        handleList([video], opt);
     }
 };