feat: add cloud endpoint support

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

dist/index.js

@@ -59791,8 +59791,29 @@ 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", [
+        "AzureChinaCloud",
+        "AzureUSGovernment",
+        "AzureGermanCloud",
+    ]);
+    let endpoint;
+    switch (cloud) {
+        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 +59825,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 +59837,7 @@ function createStacksClient(subscriptionId, tenantId) {
             userAgentPrefix: userAgentPrefix,
         },
         additionalPolicies: [debugLoggingPolicy],
+        endpoint: getEndpoint(),
     });
 }
 // Log request + response bodies to GitHub Actions debug output if enabled

src/helpers/azure.ts

@@ -6,9 +6,34 @@ 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 | undefined {
+  const cloud = getOptionalEnumInput("cloud", [
+    "AzureChinaCloud",
+    "AzureUSGovernment",
+    "AzureGermanCloud",
+  ]);
+  let endpoint: string | undefined;
+
+  switch (cloud) {
+    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 +51,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 +71,7 @@ export function createStacksClient(
         userAgentPrefix: userAgentPrefix,
       },
       additionalPolicies: [debugLoggingPolicy],
+      endpoint: getEndpoint(),
     },
   );
 }