From e3370e9474f977378490d118f20c937bd2a9c40b Mon Sep 17 00:00:00 2001
From: Ankit Singh <2501.ankit@gmail.com>
Date: Mon, 28 Oct 2024 16:07:14 +0530
Subject: [PATCH] add: turboscale build_artifact url

---
 bin/commands/runs.js          | 4 ++--
 bin/helpers/buildArtifacts.js | 6 ++++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/bin/commands/runs.js b/bin/commands/runs.js
index 57a6d0f9..baa04b12 100644
--- a/bin/commands/runs.js
+++ b/bin/commands/runs.js
@@ -365,10 +365,10 @@ module.exports = function run(args, rawArgs) {
                     await new Promise(resolve => setTimeout(resolve, 5000));
 
                     // download build artifacts
-                    if (exitCode != Constants.BUILD_FAILED_EXIT_CODE && !turboScaleSession) {
+                    if (exitCode != Constants.BUILD_FAILED_EXIT_CODE) {
                       if (utils.nonEmptyArray(bsConfig.run_settings.downloads)) {
                         logger.debug("Downloading build artifacts");
-                        await downloadBuildArtifacts(bsConfig, data.build_id, args, rawArgs, buildReportData);
+                        await downloadBuildArtifacts(bsConfig, data.build_id, args, rawArgs, buildReportData, turboScaleSession);
                       }
 
                       // Generate custom report!
diff --git a/bin/helpers/buildArtifacts.js b/bin/helpers/buildArtifacts.js
index 844ad3b6..cd064fa7 100644
--- a/bin/helpers/buildArtifacts.js
+++ b/bin/helpers/buildArtifacts.js
@@ -213,13 +213,15 @@ const sendUpdatesToBstack = async (bsConfig, buildId, args, options, rawArgs, bu
   });
 }
 
-exports.downloadBuildArtifacts = async (bsConfig, buildId, args, rawArgs, buildReportData = null) => {
+exports.downloadBuildArtifacts = async (bsConfig, buildId, args, rawArgs, buildReportData = null, isTurboScaleSession = false) => {
   return new Promise ( async (resolve, reject) => {
     BUILD_ARTIFACTS_FAIL_COUNT = 0;
     BUILD_ARTIFACTS_TOTAL_COUNT = 0;
 
     let options = {
-      url: `${config.buildUrl}${buildId}/build_artifacts`,
+      url: isTurboScaleSession
+        ? `${config.turboScaleBuildsUrl}/${buildId}/build_artifacts`
+        : `${config.buildUrl}${buildId}/build_artifacts`,
       auth: {
         username: bsConfig.auth.username,
         password: bsConfig.auth.access_key,