feat: add cloud endpoint support

Azure · Nov 18, 2024 · c5eae87 · c5eae87
1 parent bd76686
commit c5eae87
Show file tree

Hide file tree

Showing 4 changed files with 68 additions and 4 deletions.
diff --git a/action.yml b/action.yml
@@ -3,8 +3,8 @@ description: "GitHub Action for deploying to Azure"
 author: "Microsoft"
 
 branding:
-  icon: 'upload-cloud'
-  color: 'blue'
+  icon: "upload-cloud"
+  color: "blue"
 
 inputs:
   type:
@@ -54,7 +54,6 @@ inputs:
     description: "Specifies the parameters to use."
     required: false
 
-
   what-if-exclude-change-types:
     description: "Specifies the change types to exclude from the 'What If' operation."
     required: false
@@ -93,6 +92,10 @@ inputs:
     description: "Specifies output names to mask values for."
     required: false
 
+  cloud:
+    description: "Specifies the cloud to use. Choose from 'AzureCloud', 'AzureChinaCloud', 'AzureGermanCloud', 'AzureUSGovernment'."
+    required: false
+
 runs:
   using: node20
   main: dist/index.js
diff --git a/dist/index.js b/dist/index.js
@@ -59791,8 +59791,34 @@ const arm_resources_1 = __nccwpck_require__(6271);
 const arm_resourcesdeploymentstacks_1 = __nccwpck_require__(9821);
 const identity_1 = __nccwpck_require__(3983);
 const core_1 = __nccwpck_require__(7484);
+const input_1 = __nccwpck_require__(5397);
 const userAgentPrefix = "gh-azure-bicep-deploy";
 const dummySubscriptionId = "00000000-0000-0000-0000-000000000000";
+function getEndpoint() {
+    const cloud = (0, input_1.getOptionalEnumInput)("cloud", [
+        "AzureCloud",
+        "AzureChinaCloud",
+        "AzureUSGovernment",
+        "AzureGermanCloud",
+    ]);
+    // Set default endpoint to AzureCloud
+    let endpoint = "AzureCloud";
+    switch (cloud) {
+        case "AzureCloud":
+            endpoint = "https://management.azure.com";
+            break;
+        case "AzureChinaCloud":
+            endpoint = "https://management.chinacloudapi.cn";
+            break;
+        case "AzureGermanCloud":
+            endpoint = "https://management.microsoftazure.de";
+            break;
+        case "AzureUSGovernment":
+            endpoint = "https://management.usgovcloudapi.net";
+            break;
+    }
+    return endpoint;
+}
 function createDeploymentClient(subscriptionId, tenantId) {
     const credentials = new identity_1.DefaultAzureCredential({ tenantId });
     return new arm_resources_1.ResourceManagementClient(credentials, 
@@ -59804,6 +59830,7 @@ function createDeploymentClient(subscriptionId, tenantId) {
         additionalPolicies: [debugLoggingPolicy],
         // Use a recent API version to take advantage of error improvements
         apiVersion: "2024-03-01",
+        endpoint: getEndpoint(),
     });
 }
 function createStacksClient(subscriptionId, tenantId) {
@@ -59815,6 +59842,7 @@ function createStacksClient(subscriptionId, tenantId) {
             userAgentPrefix: userAgentPrefix,
         },
         additionalPolicies: [debugLoggingPolicy],
+        endpoint: getEndpoint(),
     });
 }
 // Log request + response bodies to GitHub Actions debug output if enabled

diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/helpers/azure.ts b/src/helpers/azure.ts
@@ -6,9 +6,40 @@ import { DefaultAzureCredential } from "@azure/identity";
 import { AdditionalPolicyConfig } from "@azure/core-client";
 import { debug, isDebug } from "@actions/core";
 
+import { getOptionalEnumInput } from "./input";
+
 const userAgentPrefix = "gh-azure-bicep-deploy";
 const dummySubscriptionId = "00000000-0000-0000-0000-000000000000";
 
+function getEndpoint(): string {
+  const cloud = getOptionalEnumInput("cloud", [
+    "AzureCloud",
+    "AzureChinaCloud",
+    "AzureUSGovernment",
+    "AzureGermanCloud",
+  ]);
+
+  // Set default endpoint to AzureCloud
+  let endpoint = "AzureCloud";
+
+  switch (cloud) {
+    case "AzureCloud":
+      endpoint = "https://management.azure.com";
+      break;
+    case "AzureChinaCloud":
+      endpoint = "https://management.chinacloudapi.cn";
+      break;
+    case "AzureGermanCloud":
+      endpoint = "https://management.microsoftazure.de";
+      break;
+    case "AzureUSGovernment":
+      endpoint = "https://management.usgovcloudapi.net";
+      break;
+  }
+
+  return endpoint;
+}
+
 export function createDeploymentClient(
   subscriptionId?: string,
   tenantId?: string,
@@ -26,6 +57,7 @@ export function createDeploymentClient(
       additionalPolicies: [debugLoggingPolicy],
       // Use a recent API version to take advantage of error improvements
       apiVersion: "2024-03-01",
+      endpoint: getEndpoint(),
     },
   );
 }
@@ -45,6 +77,7 @@ export function createStacksClient(
         userAgentPrefix: userAgentPrefix,
       },
       additionalPolicies: [debugLoggingPolicy],
+      endpoint: getEndpoint(),
     },
   );
 }