proxy-rotator-js 1.0.3 → 1.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "proxy-rotator-js",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "type": "module",
   "description": "Proxy Rotator",
   "main": "index.js",

package/src/ProxyRotator.js

@@ -18,12 +18,12 @@ class ProxyRotator {
             proxies.forEach( p => this._add(p) );
         }
         // handle options
-        let { returnType, revive_timer, shuffle, protocol, assume_aliveness, check_on_next } = options;
+        let { proxyType, revive_timer, shuffle, protocol, assume_aliveness, check_on_next } = options;
         // how long to wait before reviving a dead proxy
         // default: 30 minutes
         this.revive_timer = revive_timer ?? 1000 * 60 * 30;
-        // return type of proxy
-        this.returnType = returnType ?? 'string'; // or 'object'
+        // return type of proxy, either 'string' or 'object'
+        this.proxyType = _handleProxyTypeInput(proxyType);
         // assume a a protocol for all proxies
         this.protocol = protocol ?? null;
         // shuffle the proxies before adding them to the queue
@@ -113,7 +113,7 @@ class ProxyRotator {
         return this.setDead(proxy);
     }
 
-    next(returnType='string'){
+    next(proxyType=null){
         // resurect a proxy from the graveyard
         if(this.check_on_next) _resurection();
         // if there are no proxies in the pool
@@ -123,11 +123,11 @@ class ProxyRotator {
         // add to back
         this.pool.enqueue(proxy);
         // if returnType is 'string'
-        if(this.returnType === 'string')
-            // return proxy as string
-            return proxy.proxy;
-        else if(this.returnType === 'object')
-            return proxy.toObject();
+        if(proxyType === null) proxyType = this.proxyType;
+        else proxyType = this._handleProxyTypeInput(proxyType);
+        // return proxy
+        if(proxyType === 'string') return proxy.proxy;
+        else if(this.returnType === 'object') return proxy.toObject();
     }
 
     /* Randomize array in-place using Durstenfeld shuffle algorithm */
@@ -141,6 +141,17 @@ class ProxyRotator {
         return array;
     }
 
+    _handleProxyTypeInput(proxyType){
+        proxyType = proxyType.toLowerCase();
+        if(proxyType === 'str') proxyType = 'string';
+        if(proxyType === 'obj') proxyType = 'object';
+        if(proxyType !== 'string' && proxyType !== 'object'){
+            console.error('proxyType must be either "string" or "object"');
+            proxyType = 'string';
+        }
+        return proxyType;
+    }
+    
     _parseFile(filename) {
         // read file
         let str = fs.readFileSync(filename, 'utf8');