Merge pull request #13 from tekliner/master

Make REST support proxy_server env too

Author: jbenguira

API/REST/Screenshot.js

@@ -10,17 +10,18 @@ if (maxConcurrency < 1){
 }
 
 exports.handler = async (event, context, callback) => {
-    
+
     var sharedmem = context.sharedmem;
     var beginPipeline = process.hrtime();
-    
+    var proxy_server = process.env.PROXY_SERVER;
+
     if ( hardcodedAPIKey != "" && event.queryStringParameters.apiKey != hardcodedAPIKey ){
         callback(null, {
             status: 400,
             content: "Invalid API Key"
         });
         return;
-    }   
+    }
 
     while ( sharedmem.getInteger("nbPuppeteerProcess") >= maxConcurrency ){
         await tools.sleep(20);
@@ -44,13 +45,13 @@ exports.handler = async (event, context, callback) => {
 
     //var screenshotResult = await tools.screnshotForUrl(url, isFullPage, resX, resY, outFormat);
     var screenshotResult = null;
-    
-    try{    
-        screenshotResult = await tools.screnshotForUrlTab(url, isFullPage, resX, resY, outFormat, waitTime);
+
+    try{
+        screenshotResult = await tools.screnshotForUrlTab(url, isFullPage, resX, resY, outFormat, waitTime, proxy_server);
     }
     catch(ex){
         //do nothing
-        
+
     }
 
     sharedmem.incInteger("nbPuppeteerProcess", -1);
@@ -64,11 +65,11 @@ exports.handler = async (event, context, callback) => {
             data: "",
             mimeType: ""
         }
-    }   
+    }
 
     callback(null, {
             status: 200,
-            content: screenshotResult.data, 
+            content: screenshotResult.data,
             details: screenshotResult.details,
             headers:{
                 "execTime": durationMS.toFixed(2) + "ms",