@azure/arm-subscriptions 3.1.2 → 5.0.0

package/src/operations/subscriptionPolicy.ts

@@ -0,0 +1,207 @@
+/*
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
+ */
+
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { SubscriptionPolicy } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
+import * as Parameters from "../models/parameters";
+import { SubscriptionClient } from "../subscriptionClient";
+import {
+  GetTenantPolicyResponse,
+  SubscriptionPolicyListPolicyForTenantNextOptionalParams,
+  SubscriptionPolicyListPolicyForTenantOptionalParams,
+  PutTenantPolicyRequestProperties,
+  SubscriptionPolicyAddUpdatePolicyForTenantOptionalParams,
+  SubscriptionPolicyAddUpdatePolicyForTenantResponse,
+  SubscriptionPolicyGetPolicyForTenantOptionalParams,
+  SubscriptionPolicyGetPolicyForTenantResponse,
+  SubscriptionPolicyListPolicyForTenantResponse,
+  SubscriptionPolicyListPolicyForTenantNextResponse
+} from "../models";
+
+/// <reference lib="esnext.asynciterable" />
+/** Class containing SubscriptionPolicy operations. */
+export class SubscriptionPolicyImpl implements SubscriptionPolicy {
+  private readonly client: SubscriptionClient;
+
+  /**
+   * Initialize a new instance of the class SubscriptionPolicy class.
+   * @param client Reference to the service client
+   */
+  constructor(client: SubscriptionClient) {
+    this.client = client;
+  }
+
+  /**
+   * Get the subscription tenant policy for the user's tenant.
+   * @param options The options parameters.
+   */
+  public listPolicyForTenant(
+    options?: SubscriptionPolicyListPolicyForTenantOptionalParams
+  ): PagedAsyncIterableIterator<GetTenantPolicyResponse> {
+    const iter = this.listPolicyForTenantPagingAll(options);
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listPolicyForTenantPagingPage(options);
+      }
+    };
+  }
+
+  private async *listPolicyForTenantPagingPage(
+    options?: SubscriptionPolicyListPolicyForTenantOptionalParams
+  ): AsyncIterableIterator<GetTenantPolicyResponse[]> {
+    let result = await this._listPolicyForTenant(options);
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listPolicyForTenantNext(continuationToken, options);
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listPolicyForTenantPagingAll(
+    options?: SubscriptionPolicyListPolicyForTenantOptionalParams
+  ): AsyncIterableIterator<GetTenantPolicyResponse> {
+    for await (const page of this.listPolicyForTenantPagingPage(options)) {
+      yield* page;
+    }
+  }
+
+  /**
+   * Create or Update Subscription tenant policy for user's tenant.
+   * @param body Put tenant policy request properties.
+   * @param options The options parameters.
+   */
+  addUpdatePolicyForTenant(
+    body: PutTenantPolicyRequestProperties,
+    options?: SubscriptionPolicyAddUpdatePolicyForTenantOptionalParams
+  ): Promise<SubscriptionPolicyAddUpdatePolicyForTenantResponse> {
+    return this.client.sendOperationRequest(
+      { body, options },
+      addUpdatePolicyForTenantOperationSpec
+    );
+  }
+
+  /**
+   * Get the subscription tenant policy for the user's tenant.
+   * @param options The options parameters.
+   */
+  getPolicyForTenant(
+    options?: SubscriptionPolicyGetPolicyForTenantOptionalParams
+  ): Promise<SubscriptionPolicyGetPolicyForTenantResponse> {
+    return this.client.sendOperationRequest(
+      { options },
+      getPolicyForTenantOperationSpec
+    );
+  }
+
+  /**
+   * Get the subscription tenant policy for the user's tenant.
+   * @param options The options parameters.
+   */
+  private _listPolicyForTenant(
+    options?: SubscriptionPolicyListPolicyForTenantOptionalParams
+  ): Promise<SubscriptionPolicyListPolicyForTenantResponse> {
+    return this.client.sendOperationRequest(
+      { options },
+      listPolicyForTenantOperationSpec
+    );
+  }
+
+  /**
+   * ListPolicyForTenantNext
+   * @param nextLink The nextLink from the previous successful call to the ListPolicyForTenant method.
+   * @param options The options parameters.
+   */
+  private _listPolicyForTenantNext(
+    nextLink: string,
+    options?: SubscriptionPolicyListPolicyForTenantNextOptionalParams
+  ): Promise<SubscriptionPolicyListPolicyForTenantNextResponse> {
+    return this.client.sendOperationRequest(
+      { nextLink, options },
+      listPolicyForTenantNextOperationSpec
+    );
+  }
+}
+// Operation Specifications
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const addUpdatePolicyForTenantOperationSpec: coreClient.OperationSpec = {
+  path: "/providers/Microsoft.Subscription/policies/default",
+  httpMethod: "PUT",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GetTenantPolicyResponse
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponseBody
+    }
+  },
+  requestBody: Parameters.body3,
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [Parameters.$host],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer
+};
+const getPolicyForTenantOperationSpec: coreClient.OperationSpec = {
+  path: "/providers/Microsoft.Subscription/policies/default",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GetTenantPolicyResponse
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponseBody
+    }
+  },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [Parameters.$host],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const listPolicyForTenantOperationSpec: coreClient.OperationSpec = {
+  path: "/providers/Microsoft.Subscription/policies",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GetTenantPolicyListResponse
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponseBody
+    }
+  },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [Parameters.$host],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const listPolicyForTenantNextOperationSpec: coreClient.OperationSpec = {
+  path: "{nextLink}",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.GetTenantPolicyListResponse
+    },
+    default: {
+      bodyMapper: Mappers.ErrorResponseBody
+    }
+  },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [Parameters.$host, Parameters.nextLink],
+  headerParameters: [Parameters.accept],
+  serializer
+};

package/src/operations/subscriptions.ts

@@ -1,227 +1,237 @@
 /*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for
- * license information.
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
  *
  * Code generated by Microsoft (R) AutoRest Code Generator.
- * Changes may cause incorrect behavior and will be lost if the code is
- * regenerated.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 
-import * as msRest from "@azure/ms-rest-js";
-import * as Models from "../models";
-import * as Mappers from "../models/subscriptionsMappers";
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { Subscriptions } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
-import { SubscriptionClientContext } from "../subscriptionClientContext";
+import { SubscriptionClient } from "../subscriptionClient";
+import {
+  Location,
+  SubscriptionsListLocationsOptionalParams,
+  Subscription,
+  SubscriptionsListNextOptionalParams,
+  SubscriptionsListOptionalParams,
+  SubscriptionsListLocationsResponse,
+  SubscriptionsGetOptionalParams,
+  SubscriptionsGetResponse,
+  SubscriptionsListResponse,
+  SubscriptionsListNextResponse
+} from "../models";
 
-/** Class representing a Subscriptions. */
-export class Subscriptions {
-  private readonly client: SubscriptionClientContext;
+/// <reference lib="esnext.asynciterable" />
+/** Class containing Subscriptions operations. */
+export class SubscriptionsImpl implements Subscriptions {
+  private readonly client: SubscriptionClient;
 
   /**
-   * Create a Subscriptions.
-   * @param {SubscriptionClientContext} client Reference to the service client.
+   * Initialize a new instance of the class Subscriptions class.
+   * @param client Reference to the service client
    */
-  constructor(client: SubscriptionClientContext) {
+  constructor(client: SubscriptionClient) {
     this.client = client;
   }
 
   /**
-   * This operation provides all the locations that are available for resource providers; however,
-   * each resource provider may support a subset of this list.
-   * @summary Gets all available geo-locations.
+   * This operation provides all the locations that are available for resource providers; however, each
+   * resource provider may support a subset of this list.
    * @param subscriptionId The ID of the target subscription.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.SubscriptionsListLocationsResponse>
+   * @param options The options parameters.
    */
-  listLocations(subscriptionId: string, options?: msRest.RequestOptionsBase): Promise<Models.SubscriptionsListLocationsResponse>;
+  public listLocations(
+    subscriptionId: string,
+    options?: SubscriptionsListLocationsOptionalParams
+  ): PagedAsyncIterableIterator<Location> {
+    const iter = this.listLocationsPagingAll(subscriptionId, options);
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listLocationsPagingPage(subscriptionId, options);
+      }
+    };
+  }
+
+  private async *listLocationsPagingPage(
+    subscriptionId: string,
+    options?: SubscriptionsListLocationsOptionalParams
+  ): AsyncIterableIterator<Location[]> {
+    let result = await this._listLocations(subscriptionId, options);
+    yield result.value || [];
+  }
+
+  private async *listLocationsPagingAll(
+    subscriptionId: string,
+    options?: SubscriptionsListLocationsOptionalParams
+  ): AsyncIterableIterator<Location> {
+    for await (const page of this.listLocationsPagingPage(
+      subscriptionId,
+      options
+    )) {
+      yield* page;
+    }
+  }
+
   /**
-   * @param subscriptionId The ID of the target subscription.
-   * @param callback The callback
+   * Gets all subscriptions for a tenant.
+   * @param options The options parameters.
    */
-  listLocations(subscriptionId: string, callback: msRest.ServiceCallback<Models.LocationListResult>): void;
+  public list(
+    options?: SubscriptionsListOptionalParams
+  ): PagedAsyncIterableIterator<Subscription> {
+    const iter = this.listPagingAll(options);
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listPagingPage(options);
+      }
+    };
+  }
+
+  private async *listPagingPage(
+    options?: SubscriptionsListOptionalParams
+  ): AsyncIterableIterator<Subscription[]> {
+    let result = await this._list(options);
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listNext(continuationToken, options);
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listPagingAll(
+    options?: SubscriptionsListOptionalParams
+  ): AsyncIterableIterator<Subscription> {
+    for await (const page of this.listPagingPage(options)) {
+      yield* page;
+    }
+  }
+
   /**
+   * This operation provides all the locations that are available for resource providers; however, each
+   * resource provider may support a subset of this list.
    * @param subscriptionId The ID of the target subscription.
-   * @param options The optional parameters
-   * @param callback The callback
+   * @param options The options parameters.
    */
-  listLocations(subscriptionId: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.LocationListResult>): void;
-  listLocations(subscriptionId: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.LocationListResult>, callback?: msRest.ServiceCallback<Models.LocationListResult>): Promise<Models.SubscriptionsListLocationsResponse> {
+  private _listLocations(
+    subscriptionId: string,
+    options?: SubscriptionsListLocationsOptionalParams
+  ): Promise<SubscriptionsListLocationsResponse> {
     return this.client.sendOperationRequest(
-      {
-        subscriptionId,
-        options
-      },
-      listLocationsOperationSpec,
-      callback) as Promise<Models.SubscriptionsListLocationsResponse>;
+      { subscriptionId, options },
+      listLocationsOperationSpec
+    );
   }
 
   /**
    * Gets details about a specified subscription.
    * @param subscriptionId The ID of the target subscription.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.SubscriptionsGetResponse>
-   */
-  get(subscriptionId: string, options?: msRest.RequestOptionsBase): Promise<Models.SubscriptionsGetResponse>;
-  /**
-   * @param subscriptionId The ID of the target subscription.
-   * @param callback The callback
-   */
-  get(subscriptionId: string, callback: msRest.ServiceCallback<Models.Subscription>): void;
-  /**
-   * @param subscriptionId The ID of the target subscription.
-   * @param options The optional parameters
-   * @param callback The callback
+   * @param options The options parameters.
    */
-  get(subscriptionId: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.Subscription>): void;
-  get(subscriptionId: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.Subscription>, callback?: msRest.ServiceCallback<Models.Subscription>): Promise<Models.SubscriptionsGetResponse> {
+  get(
+    subscriptionId: string,
+    options?: SubscriptionsGetOptionalParams
+  ): Promise<SubscriptionsGetResponse> {
     return this.client.sendOperationRequest(
-      {
-        subscriptionId,
-        options
-      },
-      getOperationSpec,
-      callback) as Promise<Models.SubscriptionsGetResponse>;
+      { subscriptionId, options },
+      getOperationSpec
+    );
   }
 
   /**
    * Gets all subscriptions for a tenant.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.SubscriptionsListResponse>
-   */
-  list(options?: msRest.RequestOptionsBase): Promise<Models.SubscriptionsListResponse>;
-  /**
-   * @param callback The callback
-   */
-  list(callback: msRest.ServiceCallback<Models.SubscriptionListResult>): void;
-  /**
-   * @param options The optional parameters
-   * @param callback The callback
+   * @param options The options parameters.
    */
-  list(options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.SubscriptionListResult>): void;
-  list(options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.SubscriptionListResult>, callback?: msRest.ServiceCallback<Models.SubscriptionListResult>): Promise<Models.SubscriptionsListResponse> {
-    return this.client.sendOperationRequest(
-      {
-        options
-      },
-      listOperationSpec,
-      callback) as Promise<Models.SubscriptionsListResponse>;
+  private _list(
+    options?: SubscriptionsListOptionalParams
+  ): Promise<SubscriptionsListResponse> {
+    return this.client.sendOperationRequest({ options }, listOperationSpec);
   }
 
   /**
-   * Gets all subscriptions for a tenant.
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.SubscriptionsListNextResponse>
-   */
-  listNext(nextPageLink: string, options?: msRest.RequestOptionsBase): Promise<Models.SubscriptionsListNextResponse>;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param callback The callback
-   */
-  listNext(nextPageLink: string, callback: msRest.ServiceCallback<Models.SubscriptionListResult>): void;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param options The optional parameters
-   * @param callback The callback
+   * ListNext
+   * @param nextLink The nextLink from the previous successful call to the List method.
+   * @param options The options parameters.
    */
-  listNext(nextPageLink: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.SubscriptionListResult>): void;
-  listNext(nextPageLink: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.SubscriptionListResult>, callback?: msRest.ServiceCallback<Models.SubscriptionListResult>): Promise<Models.SubscriptionsListNextResponse> {
+  private _listNext(
+    nextLink: string,
+    options?: SubscriptionsListNextOptionalParams
+  ): Promise<SubscriptionsListNextResponse> {
     return this.client.sendOperationRequest(
-      {
-        nextPageLink,
-        options
-      },
-      listNextOperationSpec,
-      callback) as Promise<Models.SubscriptionsListNextResponse>;
+      { nextLink, options },
+      listNextOperationSpec
+    );
   }
 }
-
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const listLocationsOperationSpec: msRest.OperationSpec = {
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const listLocationsOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/locations",
   httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/locations",
-  urlParameters: [
-    Parameters.subscriptionId
-  ],
-  queryParameters: [
-    Parameters.apiVersion0
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {
       bodyMapper: Mappers.LocationListResult
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [Parameters.$host, Parameters.subscriptionId],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const getOperationSpec: msRest.OperationSpec = {
+const getOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}",
   httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}",
-  urlParameters: [
-    Parameters.subscriptionId
-  ],
-  queryParameters: [
-    Parameters.apiVersion0
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {
       bodyMapper: Mappers.Subscription
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [Parameters.$host, Parameters.subscriptionId],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const listOperationSpec: msRest.OperationSpec = {
+const listOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions",
   httpMethod: "GET",
-  path: "subscriptions",
-  queryParameters: [
-    Parameters.apiVersion0
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {
       bodyMapper: Mappers.SubscriptionListResult
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [Parameters.$host],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const listNextOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  baseUrl: "https://management.azure.com",
+const listNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
-  urlParameters: [
-    Parameters.nextPageLink
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.SubscriptionListResult
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [Parameters.$host, Parameters.nextLink],
+  headerParameters: [Parameters.accept],
   serializer
 };