proxy-rotator-js 1.0.2 → 1.0.4

package/package.json

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

package/src/ProxyRotator.js

@@ -18,10 +18,12 @@ class ProxyRotator {
             proxies.forEach( p => this._add(p) );
         }
         // handle options
-        let { 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, 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
@@ -111,7 +113,7 @@ class ProxyRotator {
         return this.setDead(proxy);
     }
 
-    next(){
+    next(proxyType=null){
         // resurect a proxy from the graveyard
         if(this.check_on_next) _resurection();
         // if there are no proxies in the pool
@@ -120,8 +122,12 @@ class ProxyRotator {
         let proxy = this.pool.dequeue();
         // add to back
         this.pool.enqueue(proxy);
-        // return 
-        return proxy
+        // if returnType is 'string'
+        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 */
@@ -135,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');