feat: add gzip support, send UA containing version

Author: shirish87

lib/LocalBinary.js

@@ -4,9 +4,13 @@ var https = require('https'),
   path = require('path'),
   os = require('os'),
   childProcess = require('child_process'),
+  zlib = require('zlib'),
   HttpsProxyAgent = require('https-proxy-agent'),
+  version = require('../package.json').version,
   LocalError = require('./LocalError');
 
+const packageName = 'browserstack-local-nodejs';
+
 function LocalBinary(){
   this.hostOS = process.platform;
   this.is64bits = process.arch == 'x64';
@@ -81,7 +85,9 @@ function LocalBinary(){
     }
 
     try{
-      const obj = childProcess.spawnSync(cmd, opts);
+      const userAgent = [packageName, version].join('/');
+      const env = Object.assign({ 'USER_AGENT': userAgent }, process.env);
+      const obj = childProcess.spawnSync(cmd, opts, { env: env });
       let output;
       if(obj.stdout.length > 0) {
         if(fs.existsSync(binaryPath)){
@@ -122,12 +128,23 @@ function LocalBinary(){
       try {
         options.ca = fs.readFileSync(conf.useCaCertificate);
       } catch(err) {
-        console.log("failed to read cert file", err)
+        console.log('failed to read cert file', err);
       }
     }
 
+    options.headers = Object.assign({}, options.headers, {
+      'accept-encoding': 'gzip, *',
+      'user-agent': [packageName, version].join('/'),
+    });
+
     https.get(options, function (response) {
-      response.pipe(fileStream);
+      const contentEncoding = response.headers['content-encoding'];
+      if (typeof contentEncoding === 'string' && contentEncoding.match(/gzip/i)) {
+        response.pipe(zlib.createGunzip()).pipe(fileStream);
+      } else {
+        response.pipe(fileStream);
+      }
+
       response.on('error', function(err) {
         console.error('Got Error in binary download response', err);
         that.retryBinaryDownload(conf, destParentDir, callback, retries, binaryPath);

lib/download.js

@@ -1,7 +1,8 @@
 const https = require('https'),
   fs = require('fs'),
   HttpsProxyAgent = require('https-proxy-agent'),
-  url = require('url');
+  url = require('url'),
+  zlib = require('zlib');
 
 const binaryPath = process.argv[2], httpPath = process.argv[3], proxyHost = process.argv[4], proxyPort = process.argv[5], useCaCertificate = process.argv[6];
 
@@ -17,13 +18,24 @@ if(proxyHost && proxyPort) {
     try {
       options.ca = fs.readFileSync(useCaCertificate);
     } catch(err) {
-      console.log("failed to read cert file", err)
+      console.log('failed to read cert file', err);
     }
   }
 }
 
+options.headers = Object.assign({}, options.headers, {
+  'accept-encoding': 'gzip, *',
+  'user-agent': process.env.USER_AGENT,
+});
+
 https.get(options, function (response) {
-  response.pipe(fileStream);
+  const contentEncoding = response.headers['content-encoding'];
+  if (typeof contentEncoding === 'string' && contentEncoding.match(/gzip/i)) {
+    response.pipe(zlib.createGunzip()).pipe(fileStream);
+  } else {
+    response.pipe(fileStream);
+  }
+
   response.on('error', function(err) {
     console.error('Got Error in binary download response', err);
   });