add Local.startSync

Author: 07souravkunda

lib/Local.js

@@ -26,6 +26,51 @@ function Local(){
   this.errorRegex = /\*\*\* Error: [^\r\n]*/i;
   this.doneRegex = /Press Ctrl-C to exit/i;
 
+  this.startSync = function(options) {
+    this.userArgs = [];
+    var that = this;
+    this.addArgs(options);
+
+    if(typeof options['onlyCommand'] !== 'undefined')
+      return;
+    
+    const binaryPath = this.getBinaryPathSync();
+    that.binaryPath = binaryPath;
+    childProcess.exec('echo "" > ' + that.logfile);
+    that.opcode = 'start';
+    if(!this.binaryPath){
+      return new LocalError("Couldn't find binary file");
+    }
+    try{
+      const obj = childProcess.spawnSync(that.binaryPath, that.getBinaryArgs());
+      var data = {};
+      if(obj.stdout.length > 0)
+        data = JSON.parse(obj.stdout);
+      else if(obj.stderr.length > 0)
+        data = JSON.parse(obj.stderr);
+      else
+        return new LocalError('No output received');
+      if(data['state'] != 'connected'){
+        return new LocalError(data['message']['message']);
+      } else {
+        that.pid = data['pid'];
+        that.isProcessRunning = true;
+        return;
+      }
+    }catch(error){
+      console.error('Error while trying to execute binary', error);
+      if(that.retriesLeft > 0) {
+        console.log('Retrying Binary Download. Retries Left', that.retriesLeft);
+        that.retriesLeft -= 1;
+        fs.unlinkSync(that.binaryPath);
+        delete(that.binaryPath);
+        return that.startSync(options);
+      } else {
+        return new LocalError(error.toString());
+      }
+    }
+  }
+
   this.start = function(options, callback){
     this.userArgs = [];
     var that = this;
@@ -191,6 +236,21 @@ function Local(){
     }
   };
 
+  this.getBinaryPathSync = function(){
+    if(typeof(this.binaryPath) == 'undefined'){
+      this.binary = new LocalBinary();
+      var conf = {};
+      if(this.proxyHost && this.proxyPort){
+        conf.proxyHost = this.proxyHost;
+        conf.proxyPort = this.proxyPort;
+      }
+      return this.binary.binaryPathSync(conf);
+    } else {
+      console.log('BINARY PATH IS DEFINED');
+      return this.binaryPath;
+    }
+  }
+
   this.getBinaryPath = function(callback){
     if(typeof(this.binaryPath) == 'undefined'){
       this.binary = new LocalBinary();

lib/LocalBinary.js

@@ -39,6 +39,22 @@ function LocalBinary(){
 
   this.httpPath = this.getDownloadPath();
 
+  this.retryBinaryDownloadSync = function(conf, destParentDir, retries, binaryPath){
+    var that = this;
+    if(retries > 0) {
+      console.log('Retrying Download. Retries left', retries);
+      try {
+        const stats = fs.statSync(binaryPath)
+        fs.unlinkSync(binaryPath);
+      } catch(er) {}
+
+      that.downloadSync(conf, destParentDir, retries - 1)
+
+    } else {
+      console.error('Number of retries to download exceeded.');
+    }
+  }
+
   this.retryBinaryDownload = function(conf, destParentDir, callback, retries, binaryPath) {
     var that = this;
     if(retries > 0) {
@@ -54,6 +70,56 @@ function LocalBinary(){
     }
   };
 
+  this.downloadSync = function(conf, destParentDir, retries) {
+    console.log('downloading in sync')
+    var that = this;
+    if(!this.checkPath(destParentDir))
+      fs.mkdirSync(destParentDir);
+    
+    var destBinaryName = (this.windows) ? 'BrowserStackLocal.exe' : 'BrowserStackLocal';
+    var binaryPath = path.join(destParentDir, destBinaryName);
+
+    var options = url.parse(this.httpPath);
+    if(conf.proxyHost && conf.proxyPort) {
+      options.agent = new HttpsProxyAgent({
+        host: conf.proxyHost,
+        port: conf.proxyPort
+      });
+    }
+
+    const cmd = `curl`;
+    const opts = ['-o', binaryPath, options.href];
+
+    if(conf.proxyHost && conf.proxyPort) {
+      opts.push('--proxy', `${conf.proxyHost}:${conf.proxyPort}`);
+    }
+
+    try{
+      const obj = childProcess.spawnSync(cmd, opts);
+      let output;
+      if(obj.stdout.length > 0) {
+        output = Buffer.from(JSON.parse(JSON.stringify(obj.stderr)).data).toString();
+      } else if(obj.stderr.length > 0) {
+        output = Buffer.from(JSON.parse(JSON.stringify(obj.stderr)).data).toString();
+      }
+      if(output.includes('failed')){
+        console.log('failed to download');
+        that.retryBinaryDownloadSync(conf, destParentDir, retries, binaryPath);
+      }else{
+        if(fs.existsSync(binaryPath)){
+          fs.chmodSync(binaryPath, '0755')
+          return binaryPath;
+        }else{
+          console.log('failed to download');
+          that.retryBinaryDownloadSync(conf, destParentDir, retries, binaryPath);
+        }
+      }
+    } catch(err) {
+      console.log('Download failed with error', err);
+      that.retryBinaryDownloadSync(conf, destParentDir, retries, binaryPath);
+    }
+  }
+
   this.download = function(conf, destParentDir, callback, retries){
     var that = this;
     if(!this.checkPath(destParentDir))
@@ -92,6 +158,17 @@ function LocalBinary(){
     });
   };
 
+  this.binaryPathSync = function(conf){
+    var destParentDir = this.getAvailableDirs();
+    var destBinaryName = (this.windows) ? 'BrowserStackLocal.exe' : 'BrowserStackLocal';
+    var binaryPath = path.join(destParentDir, destBinaryName);
+    if(this.checkPath(binaryPath, fs.X_OK)){
+      return binaryPath;
+    } else {
+      return this.downloadSync(conf, destParentDir, 5);
+    }
+  }
+
   this.binaryPath = function(conf, callback){
     var destParentDir = this.getAvailableDirs();
     var destBinaryName = (this.windows) ? 'BrowserStackLocal.exe' : 'BrowserStackLocal';