From 21ede77fceaa0a185a700b943fa1a9578b12952d Mon Sep 17 00:00:00 2001
From: Nikola Milosavljevic <nikolam@microsoft.com>
Date: Tue, 1 Oct 2024 13:37:03 -0700
Subject: [PATCH] Obtain token if not provided

---
 eng/Get-DropVersions.ps1 | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/eng/Get-DropVersions.ps1 b/eng/Get-DropVersions.ps1
index c515b097ba..24bd2d7727 100644
--- a/eng/Get-DropVersions.ps1
+++ b/eng/Get-DropVersions.ps1
@@ -237,13 +237,16 @@ if ($UseInternalBuild) {
     }
 
     if ($BuildId) {
-        if ($InternalArtifactsAccessToken) {
-            $internalBaseUrl = GetInternalBaseUrl
-        }
-        else {
-            Write-Error "'InternalArtifactsAccessToken' parameter is required for obtaining internal base-url, when specifying 'BuildId' option"
-            exit 1
+        if (!$InternalArtifactsAccessToken) {
+            $InternalArtifactsAccessToken = az account get-access-token --query accessToken --output tsv
+            if ($LASTEXITCODE -ne 0) {
+                Write-Error "Failed to obtain access token using Azure CLI"
+                Write-Error "Please provide 'InternalArtifactsAccessToken' parameter when using 'BuildId' option"
+                exit 1
+            }
         }
+
+        $internalBaseUrl = GetInternalBaseUrl
     }
 
     $queryString = "$BlobStorageSasQueryString"