@azure/arm-hardwaresecuritymodules 2.0.0-beta.1 → 2.0.0-beta.2

package/src/azureHSMResourceProvider.ts

@@ -8,28 +8,38 @@
 
 import * as coreClient from "@azure/core-client";
 import * as coreRestPipeline from "@azure/core-rest-pipeline";
+import {
+  PipelineRequest,
+  PipelineResponse,
+  SendRequest,
+} from "@azure/core-rest-pipeline";
 import * as coreAuth from "@azure/core-auth";
 import {
   CloudHsmClustersImpl,
   CloudHsmClusterPrivateLinkResourcesImpl,
   CloudHsmClusterPrivateEndpointConnectionsImpl,
   PrivateEndpointConnectionsImpl,
+  CloudHsmClusterBackupStatusImpl,
+  CloudHsmClusterRestoreStatusImpl,
+  DedicatedHsmOperationsImpl,
   OperationsImpl,
-  DedicatedHsmOperationsImpl
 } from "./operations";
 import {
   CloudHsmClusters,
   CloudHsmClusterPrivateLinkResources,
   CloudHsmClusterPrivateEndpointConnections,
   PrivateEndpointConnections,
+  CloudHsmClusterBackupStatus,
+  CloudHsmClusterRestoreStatus,
+  DedicatedHsmOperations,
   Operations,
-  DedicatedHsmOperations
 } from "./operationsInterfaces";
 import { AzureHSMResourceProviderOptionalParams } from "./models";
 
 export class AzureHSMResourceProvider extends coreClient.ServiceClient {
   $host: string;
   subscriptionId: string;
+  apiVersion: string;
 
   /**
    * Initializes a new instance of the AzureHSMResourceProvider class.
@@ -40,7 +50,7 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
   constructor(
     credentials: coreAuth.TokenCredential,
     subscriptionId: string,
-    options?: AzureHSMResourceProviderOptionalParams
+    options?: AzureHSMResourceProviderOptionalParams,
   ) {
     if (credentials === undefined) {
       throw new Error("'credentials' cannot be null");
@@ -55,10 +65,10 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
     }
     const defaults: AzureHSMResourceProviderOptionalParams = {
       requestContentType: "application/json; charset=utf-8",
-      credential: credentials
+      credential: credentials,
     };
 
-    const packageDetails = `azsdk-js-arm-hardwaresecuritymodules/2.0.0-beta.1`;
+    const packageDetails = `azsdk-js-arm-hardwaresecuritymodules/2.0.0-beta.2`;
     const userAgentPrefix =
       options.userAgentOptions && options.userAgentOptions.userAgentPrefix
         ? `${options.userAgentOptions.userAgentPrefix} ${packageDetails}`
@@ -68,20 +78,21 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
       ...defaults,
       ...options,
       userAgentOptions: {
-        userAgentPrefix
+        userAgentPrefix,
       },
       endpoint:
-        options.endpoint ?? options.baseUri ?? "https://management.azure.com"
+        options.endpoint ?? options.baseUri ?? "https://management.azure.com",
     };
     super(optionsWithDefaults);
 
     let bearerTokenAuthenticationPolicyFound: boolean = false;
     if (options?.pipeline && options.pipeline.getOrderedPolicies().length > 0) {
-      const pipelinePolicies: coreRestPipeline.PipelinePolicy[] = options.pipeline.getOrderedPolicies();
+      const pipelinePolicies: coreRestPipeline.PipelinePolicy[] =
+        options.pipeline.getOrderedPolicies();
       bearerTokenAuthenticationPolicyFound = pipelinePolicies.some(
         (pipelinePolicy) =>
           pipelinePolicy.name ===
-          coreRestPipeline.bearerTokenAuthenticationPolicyName
+          coreRestPipeline.bearerTokenAuthenticationPolicyName,
       );
     }
     if (
@@ -91,7 +102,7 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
       !bearerTokenAuthenticationPolicyFound
     ) {
       this.pipeline.removePolicy({
-        name: coreRestPipeline.bearerTokenAuthenticationPolicyName
+        name: coreRestPipeline.bearerTokenAuthenticationPolicyName,
       });
       this.pipeline.addPolicy(
         coreRestPipeline.bearerTokenAuthenticationPolicy({
@@ -101,9 +112,9 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
             `${optionsWithDefaults.endpoint}/.default`,
           challengeCallbacks: {
             authorizeRequestOnChallenge:
-              coreClient.authorizeRequestOnClaimChallenge
-          }
-        })
+              coreClient.authorizeRequestOnClaimChallenge,
+          },
+        }),
       );
     }
     // Parameter assignments
@@ -111,22 +122,58 @@ export class AzureHSMResourceProvider extends coreClient.ServiceClient {
 
     // Assigning values to Constant parameters
     this.$host = options.$host || "https://management.azure.com";
+    this.apiVersion = options.apiVersion || "2024-06-30-preview";
     this.cloudHsmClusters = new CloudHsmClustersImpl(this);
-    this.cloudHsmClusterPrivateLinkResources = new CloudHsmClusterPrivateLinkResourcesImpl(
-      this
+    this.cloudHsmClusterPrivateLinkResources =
+      new CloudHsmClusterPrivateLinkResourcesImpl(this);
+    this.cloudHsmClusterPrivateEndpointConnections =
+      new CloudHsmClusterPrivateEndpointConnectionsImpl(this);
+    this.privateEndpointConnections = new PrivateEndpointConnectionsImpl(this);
+    this.cloudHsmClusterBackupStatus = new CloudHsmClusterBackupStatusImpl(
+      this,
     );
-    this.cloudHsmClusterPrivateEndpointConnections = new CloudHsmClusterPrivateEndpointConnectionsImpl(
-      this
+    this.cloudHsmClusterRestoreStatus = new CloudHsmClusterRestoreStatusImpl(
+      this,
     );
-    this.privateEndpointConnections = new PrivateEndpointConnectionsImpl(this);
-    this.operations = new OperationsImpl(this);
     this.dedicatedHsmOperations = new DedicatedHsmOperationsImpl(this);
+    this.operations = new OperationsImpl(this);
+    this.addCustomApiVersionPolicy(options.apiVersion);
+  }
+
+  /** A function that adds a policy that sets the api-version (or equivalent) to reflect the library version. */
+  private addCustomApiVersionPolicy(apiVersion?: string) {
+    if (!apiVersion) {
+      return;
+    }
+    const apiVersionPolicy = {
+      name: "CustomApiVersionPolicy",
+      async sendRequest(
+        request: PipelineRequest,
+        next: SendRequest,
+      ): Promise<PipelineResponse> {
+        const param = request.url.split("?");
+        if (param.length > 1) {
+          const newParams = param[1].split("&").map((item) => {
+            if (item.indexOf("api-version") > -1) {
+              return "api-version=" + apiVersion;
+            } else {
+              return item;
+            }
+          });
+          request.url = param[0] + "?" + newParams.join("&");
+        }
+        return next(request);
+      },
+    };
+    this.pipeline.addPolicy(apiVersionPolicy);
   }
 
   cloudHsmClusters: CloudHsmClusters;
   cloudHsmClusterPrivateLinkResources: CloudHsmClusterPrivateLinkResources;
   cloudHsmClusterPrivateEndpointConnections: CloudHsmClusterPrivateEndpointConnections;
   privateEndpointConnections: PrivateEndpointConnections;
-  operations: Operations;
+  cloudHsmClusterBackupStatus: CloudHsmClusterBackupStatus;
+  cloudHsmClusterRestoreStatus: CloudHsmClusterRestoreStatus;
   dedicatedHsmOperations: DedicatedHsmOperations;
+  operations: Operations;
 }

package/src/lroImpl.ts

@@ -7,7 +7,7 @@
  */
 
 // Copyright (c) Microsoft Corporation.
-// Licensed under the MIT license.
+// Licensed under the MIT License.
 
 import { AbortSignalLike } from "@azure/abort-controller";
 import { LongRunningOperation, LroResponse } from "@azure/core-lro";
@@ -28,15 +28,15 @@ export function createLroSpec<T>(inputs: {
     sendInitialRequest: () => sendOperationFn(args, spec),
     sendPollRequest: (
       path: string,
-      options?: { abortSignal?: AbortSignalLike }
+      options?: { abortSignal?: AbortSignalLike },
     ) => {
       const { requestBody, ...restSpec } = spec;
       return sendOperationFn(args, {
         ...restSpec,
         httpMethod: "GET",
         path,
-        abortSignal: options?.abortSignal
+        abortSignal: options?.abortSignal,
       });
-    }
+    },
   };
 }