Obtain token if not provided

dotnet · Oct 1, 2024 · 21ede77 · 21ede77
1 parent a930c7b
commit 21ede77
Showing 1 changed file with 9 additions and 6 deletions.
diff --git 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"