From b5b194f55ad55d9eb46ca6d8ff70cff101937e88 Mon Sep 17 00:00:00 2001
From: Nikhil Agarwal <nikhil.ag@browserstack.com>
Date: Wed, 20 Mar 2024 12:55:25 +0530
Subject: [PATCH 1/2] add instrumentation for turboscale cypress sessions

---
 bin/helpers/usageReporting.js | 55 ++++++++++++++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 1 deletion(-)

diff --git a/bin/helpers/usageReporting.js b/bin/helpers/usageReporting.js
index ed5e42ba..b8b7726e 100644
--- a/bin/helpers/usageReporting.js
+++ b/bin/helpers/usageReporting.js
@@ -10,6 +10,7 @@ const config = require('./config'),
   utils = require('./utils');
 
 const { AUTH_REGEX, REDACTED_AUTH, REDACTED, CLI_ARGS_REGEX, RAW_ARGS_REGEX } = require("./constants");
+const { isTurboScaleSession } = require('../helpers/atsHelper');
 
 function get_version(package_name) {
   try {
@@ -197,10 +198,58 @@ function redactKeys(str, regex, redact) {
   return str.replace(regex, redact);
 }
 
+function sendTurboscaleErrorLogs(args) {
+  let bsConfig = JSON.parse(JSON.stringify(args.bstack_config));
+  let data = utils.isUndefined(args.data) ? {} : args.data;
+  const turboscaleErrorPayload = {
+    kind: 'hst-cypress-cli-error',
+    userId: data.user_id,
+    error: args.message
+  }
+
+  const options = {
+    headers: {
+      'User-Agent': utils.getUserAgent()
+    },
+    method: "POST",
+    auth: {
+      username: bsConfig.auth.username,
+      password: bsConfig.auth.access_key,
+    },
+    url: `${config.turboScaleAPIUrl}/send-instrumentation`,
+    body: turboscaleErrorPayload,
+    json: true,
+    maxAttempts: 10, // (default) try 3 times
+    retryDelay: 2000, // (default) wait for 2s before trying again
+    retrySrategy: request.RetryStrategies.HTTPOrNetworkError, // (default) retry on 5xx or network errors
+  };
+
+  fileLogger.info(`Sending ${JSON.stringify(turboscaleErrorPayload)} to ${config.turboScaleAPIUrl}/send-instrumentation`);
+  request(options, function (error, res, body) {
+    if (error) {
+      //write err response to file
+      fileLogger.error(JSON.stringify(error));
+      return;
+    }
+    // write response file
+    let response = {
+      attempts: res.attempts,
+      statusCode: res.statusCode,
+      body: body
+    };
+    fileLogger.info(`${JSON.stringify(response)}`);
+  });
+}
+
 function send(args) {
+  let bsConfig = JSON.parse(JSON.stringify(args.bstack_config));
+
+  if (isTurboScaleSession(bsConfig) && args.message_type === 'error') {
+    sendTurboscaleErrorLogs(args);
+  }
+
   if (isUsageReportingEnabled() === "true") return;
 
-  let bsConfig = JSON.parse(JSON.stringify(args.bstack_config));
   let runSettings = "";
   let sanitizedbsConfig = "";
   let cli_details = cli_version_and_path(bsConfig);
@@ -258,6 +307,10 @@ function send(args) {
     },
   };
 
+  if (isTurboScaleSession(bsConfig)) {
+    payload.event_type = 'hst_cypress_cli_stats';
+  }
+
   const options = {
     headers: {
       "Content-Type": "text/json",

From 49d1a7ec7c9a33fbdb7fc59d20ca6367108be062 Mon Sep 17 00:00:00 2001
From: Nikhil Agarwal <nikhil.ag@browserstack.com>
Date: Wed, 20 Mar 2024 13:13:45 +0530
Subject: [PATCH 2/2] update payload

---
 bin/helpers/usageReporting.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bin/helpers/usageReporting.js b/bin/helpers/usageReporting.js
index b8b7726e..5a7a250e 100644
--- a/bin/helpers/usageReporting.js
+++ b/bin/helpers/usageReporting.js
@@ -203,7 +203,7 @@ function sendTurboscaleErrorLogs(args) {
   let data = utils.isUndefined(args.data) ? {} : args.data;
   const turboscaleErrorPayload = {
     kind: 'hst-cypress-cli-error',
-    userId: data.user_id,
+    data: data,
     error: args.message
   }