From 2e740e56a77a152a38f425ce4a9d895e082bdbcd Mon Sep 17 00:00:00 2001
From: Jason Hills <jason.hills@mongodb.com>
Date: Tue, 11 Nov 2025 15:42:53 -0500
Subject: [PATCH] SERVER-113676 Update profiling data script to use environment
 variable (#43783)

GitOrigin-RevId: ce80b48f068e5388bf38591bbad2bf3c55b590ac
---
 buildscripts/profile_data_pr.py               | 20 ++++++++++++++++---
 .../tasks/compile_tasks.yml                   |  5 +++--
 2 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/buildscripts/profile_data_pr.py b/buildscripts/profile_data_pr.py
index 0f70d800d88..644f20f700b 100755
--- a/buildscripts/profile_data_pr.py
+++ b/buildscripts/profile_data_pr.py
@@ -190,8 +190,22 @@ if __name__ == "__main__":
     parser.add_argument("gcc_pgo_url", help="URL that gcc pgo data was uploaded to.")
     parser.add_argument("target_branch", help="The branch you want to create a PR into.")
     parser.add_argument("new_branch", help="The new branch to create a PR from.")
-    parser.add_argument("app_id", help="App ID used for authentication.")
-    parser.add_argument("private_key", help="Key to use for authentication.")
+    parser.add_argument(
+        "--app_id", help="App ID used for authentication.", default=os.getenv("MONGO_PR_BOT_APP_ID")
+    )
+    parser.add_argument(
+        "--private_key",
+        help="Key to use for authentication.",
+        default=os.getenv("MONGO_PR_BOT_PRIVATE_KEY"),
+    )
     args = parser.parse_args()
-    repo = get_mongo_repository(args.app_id, args.private_key)
+    if not args.app_id or not args.private_key:
+        parser.error(
+            "Must define --app-id or env MONGO_PR_BOT_APP_ID and --private-key or env MONGO_PR_BOT_PRIVATE_KEY."
+        )
+    # Replace spaces with newline, if applicable
+    private_key = (
+        args.private_key[:31] + args.private_key[31:-29].replace(" ", "\n") + args.private_key[-29:]
+    )
+    repo = get_mongo_repository(args.app_id, private_key)
     create_profile_data_pr(repo, args, args.target_branch, args.new_branch)
diff --git a/etc/evergreen_yml_components/tasks/compile_tasks.yml b/etc/evergreen_yml_components/tasks/compile_tasks.yml
index 6efca0cd692..58dd9a2fbce 100644
--- a/etc/evergreen_yml_components/tasks/compile_tasks.yml
+++ b/etc/evergreen_yml_components/tasks/compile_tasks.yml
@@ -1302,6 +1302,9 @@ tasks:
       - command: subprocess.exec
         params:
           binary: bash
+          include_expansions_in_env:
+            - MONGO_PR_BOT_APP_ID
+            - MONGO_PR_BOT_PRIVATE_KEY
           args:
             - "src/evergreen/run_python_script.sh"
             - "buildscripts/profile_data_pr.py"
@@ -1310,8 +1313,6 @@ tasks:
             - "https://mdb-build-public.s3.us-east-1.amazonaws.com/profiling_data/pgo/mongod_${revision}_${build_arch}_gcc_lto_pgo_${version}.tgz"
             - "${branch_name}"
             - "SERVER-110427-${revision}"
-            - "${MONGO_PR_BOT_APP_ID}"
-            - "${MONGO_PR_BOT_PRIVATE_KEY}"
   - name: publish_crypt_packages
     run_on: rhel8.7-small
     tags: