@azure/arm-baremetalinfrastructure 1.0.0-beta.1 → 1.0.0-beta.2

package/src/operations/azureBareMetalInstances.ts

@@ -13,6 +13,12 @@ import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { BareMetalInfrastructureClient } from "../bareMetalInfrastructureClient";
+import {
+  SimplePollerLike,
+  OperationState,
+  createHttpPoller
+} from "@azure/core-lro";
+import { createLroSpec } from "../lroImpl";
 import {
   AzureBareMetalInstance,
   AzureBareMetalInstancesListBySubscriptionNextOptionalParams,
@@ -21,6 +27,12 @@ import {
   AzureBareMetalInstancesListByResourceGroupNextOptionalParams,
   AzureBareMetalInstancesListByResourceGroupOptionalParams,
   AzureBareMetalInstancesListByResourceGroupResponse,
+  AzureBareMetalInstancesStartOptionalParams,
+  AzureBareMetalInstancesStartResponse,
+  AzureBareMetalInstancesRestartOptionalParams,
+  AzureBareMetalInstancesRestartResponse,
+  AzureBareMetalInstancesShutdownOptionalParams,
+  AzureBareMetalInstancesShutdownResponse,
   AzureBareMetalInstancesGetOptionalParams,
   AzureBareMetalInstancesGetResponse,
   Tags,
@@ -44,8 +56,8 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Gets a list of AzureBareMetal instances in the specified subscription. The operations returns
-   * various properties of each Azure BareMetal instance.
+   * Returns a list of Azure Bare Metal Instances in the specified subscription. The operations returns
+   * various properties of each Azure Bare Metal Instance.
    * @param options The options parameters.
    */
   public listBySubscription(
@@ -99,8 +111,8 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Gets a list of AzureBareMetal instances in the specified subscription and resource group. The
-   * operations returns various properties of each Azure BareMetal instance.
+   * Gets a list of Azure Bare Metal Instances in the specified subscription and resource group. The
+   * operations returns various properties of each Azure Bare Metal Instance.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param options The options parameters.
    */
@@ -169,8 +181,287 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Gets a list of AzureBareMetal instances in the specified subscription. The operations returns
-   * various properties of each Azure BareMetal instance.
+   * The operation to start an Azure Bare Metal instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginStart(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesStartOptionalParams
+  ): Promise<
+    SimplePollerLike<
+      OperationState<AzureBareMetalInstancesStartResponse>,
+      AzureBareMetalInstancesStartResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<AzureBareMetalInstancesStartResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, azureBareMetalInstanceName, options },
+      spec: startOperationSpec
+    });
+    const poller = await createHttpPoller<
+      AzureBareMetalInstancesStartResponse,
+      OperationState<AzureBareMetalInstancesStartResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location"
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * The operation to start an Azure Bare Metal instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginStartAndWait(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesStartOptionalParams
+  ): Promise<AzureBareMetalInstancesStartResponse> {
+    const poller = await this.beginStart(
+      resourceGroupName,
+      azureBareMetalInstanceName,
+      options
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * The operation to restart an Azure Bare Metal Instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginRestart(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesRestartOptionalParams
+  ): Promise<
+    SimplePollerLike<
+      OperationState<AzureBareMetalInstancesRestartResponse>,
+      AzureBareMetalInstancesRestartResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<AzureBareMetalInstancesRestartResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, azureBareMetalInstanceName, options },
+      spec: restartOperationSpec
+    });
+    const poller = await createHttpPoller<
+      AzureBareMetalInstancesRestartResponse,
+      OperationState<AzureBareMetalInstancesRestartResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location"
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * The operation to restart an Azure Bare Metal Instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginRestartAndWait(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesRestartOptionalParams
+  ): Promise<AzureBareMetalInstancesRestartResponse> {
+    const poller = await this.beginRestart(
+      resourceGroupName,
+      azureBareMetalInstanceName,
+      options
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * The operation to shutdown an Azure Bare Metal Instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginShutdown(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesShutdownOptionalParams
+  ): Promise<
+    SimplePollerLike<
+      OperationState<AzureBareMetalInstancesShutdownResponse>,
+      AzureBareMetalInstancesShutdownResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<AzureBareMetalInstancesShutdownResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperationFn = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, azureBareMetalInstanceName, options },
+      spec: shutdownOperationSpec
+    });
+    const poller = await createHttpPoller<
+      AzureBareMetalInstancesShutdownResponse,
+      OperationState<AzureBareMetalInstancesShutdownResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      resourceLocationConfig: "location"
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * The operation to shutdown an Azure Bare Metal Instance
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
+   * @param options The options parameters.
+   */
+  async beginShutdownAndWait(
+    resourceGroupName: string,
+    azureBareMetalInstanceName: string,
+    options?: AzureBareMetalInstancesShutdownOptionalParams
+  ): Promise<AzureBareMetalInstancesShutdownResponse> {
+    const poller = await this.beginShutdown(
+      resourceGroupName,
+      azureBareMetalInstanceName,
+      options
+    );
+    return poller.pollUntilDone();
+  }
+
+  /**
+   * Returns a list of Azure Bare Metal Instances in the specified subscription. The operations returns
+   * various properties of each Azure Bare Metal Instance.
    * @param options The options parameters.
    */
   private _listBySubscription(
@@ -183,8 +474,8 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Gets a list of AzureBareMetal instances in the specified subscription and resource group. The
-   * operations returns various properties of each Azure BareMetal instance.
+   * Gets a list of Azure Bare Metal Instances in the specified subscription and resource group. The
+   * operations returns various properties of each Azure Bare Metal Instance.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param options The options parameters.
    */
@@ -199,9 +490,10 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Gets an Azure BareMetal instance for the specified subscription, resource group, and instance name.
+   * Gets an Azure Bare Metal Instance for the specified subscription, resource group, and instance name.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param azureBareMetalInstanceName Name of the Azure BareMetal on Azure instance.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
    * @param options The options parameters.
    */
   get(
@@ -216,10 +508,11 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
   }
 
   /**
-   * Patches the Tags field of a Azure BareMetal instance for the specified subscription, resource group,
-   * and instance name.
+   * Patches the Tags field of a Azure Bare Metal Instance for the specified subscription, resource
+   * group, and instance name.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param azureBareMetalInstanceName Name of the Azure BareMetal on Azure instance.
+   * @param azureBareMetalInstanceName Name of the Azure Bare Metal Instance, also known as the
+   *                                   ResourceName.
    * @param tagsParameter Request body that only contains the new Tags field
    * @param options The options parameters.
    */
@@ -270,6 +563,101 @@ export class AzureBareMetalInstancesImpl implements AzureBareMetalInstances {
 // Operation Specifications
 const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
 
+const startOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.BareMetalInfrastructure/bareMetalInstances/{azureBareMetalInstanceName}/start",
+  httpMethod: "POST",
+  responses: {
+    200: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    201: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    202: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    204: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse
+    }
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.azureBareMetalInstanceName
+  ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const restartOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.BareMetalInfrastructure/bareMetalInstances/{azureBareMetalInstanceName}/restart",
+  httpMethod: "POST",
+  responses: {
+    200: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    201: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    202: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    204: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse
+    }
+  },
+  requestBody: Parameters.forceParameter,
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.azureBareMetalInstanceName
+  ],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer
+};
+const shutdownOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.BareMetalInfrastructure/bareMetalInstances/{azureBareMetalInstanceName}/shutdown",
+  httpMethod: "POST",
+  responses: {
+    200: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    201: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    202: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    204: {
+      bodyMapper: Mappers.OperationStatus
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponse
+    }
+  },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.azureBareMetalInstanceName
+  ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
 const listBySubscriptionOperationSpec: coreClient.OperationSpec = {
   path:
     "/subscriptions/{subscriptionId}/providers/Microsoft.BareMetalInfrastructure/bareMetalInstances",