add user-agent while calling api

browserstack · May 12, 2020 · 67dd517 · 67dd517
1 parent 8d9d4d3
commit 67dd517
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 11 deletions.
diff --git a/bin/commands/info.js b/bin/commands/info.js
@@ -20,12 +20,15 @@ function buildInfo(args) {
 
     let options = {
       url: config.buildUrl + buildId,
-      method: 'GET',
+      method: "GET",
       auth: {
         user: bsConfig.auth.username,
-        password: bsConfig.auth.access_key
-      }
-    }
+        password: bsConfig.auth.access_key,
+      },
+      headers: {
+        "User-Agent": util.getUserAgent,
+      },
+    };
 
     request(options, function (err, resp, body) {
       let message = null;

diff --git a/bin/commands/stop.js b/bin/commands/stop.js
@@ -20,12 +20,15 @@ function buildStop(args) {
 
     let options = {
       url: config.buildStopUrl + buildId,
-      method: 'POST',
+      method: "POST",
       auth: {
         user: bsConfig.auth.username,
-        password: bsConfig.auth.access_key
-      }
-    }
+        password: bsConfig.auth.access_key,
+      },
+      headers: {
+        "User-Agent": util.getUserAgent(),
+      },
+    };
 
     request(options, function (err, resp, body) {
       let message = null;

diff --git a/bin/helpers/build.js b/bin/helpers/build.js
@@ -3,7 +3,8 @@ const request = require('request');
 
 const config = require('./config'),
   capabilityHelper = require("../helpers/capabilityHelper"),
-  Constants = require('../helpers/constants');
+  Constants = require('../helpers/constants'),
+  util =require('../helpers/util');
 
 const createBuild = (bsConfig, zip) => {
   return new Promise(function (resolve, reject) {
@@ -15,7 +16,8 @@ const createBuild = (bsConfig, zip) => {
           password: bsConfig.auth.access_key
         },
         headers: {
-          'Content-Type': 'application/json'
+          'Content-Type': 'application/json',
+          "User-Agent": util.getUserAgent(),
         },
         body: data
       }

diff --git a/bin/helpers/util.js b/bin/helpers/util.js
@@ -67,3 +67,7 @@ exports.setUsageReportingFlag = (bsConfig, disableUsageReporting) => {
     process.env.DISABLE_USAGE_REPORTING = disableUsageReporting;
   }
 }
+
+exports.getUserAgent = () => {
+  return `BStack-Cypress-CLI`;
+}
diff --git a/bin/helpers/zipUpload.js b/bin/helpers/zipUpload.js
@@ -3,7 +3,8 @@ const config = require("./config"),
   request = require("request"),
   fs = require("fs"),
   logger = require("./logger").winstonLogger,
-  Constants = require("./constants");
+  Constants = require("./constants"),
+  util = require("./util");
 
 const uploadCypressZip = (bsConfig, filePath) => {
   return new Promise(function (resolve, reject) {
@@ -17,6 +18,9 @@ const uploadCypressZip = (bsConfig, filePath) => {
         file: fs.createReadStream(filePath),
         filetype: 'zip',
         filename: 'tests'
+      },
+      headers: {
+        "User-Agent": util.getUserAgent,
       }
     }