package network

// Copyright (c) Microsoft and contributors.  All rights reserved.

//

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

// http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

//

// See the License for the specific language governing permissions and

// limitations under the License.

//

// Code generated by Microsoft (R) AutoRest Code Generator.

// Changes may cause incorrect behavior and will be lost if the code is regenerated.

import (

	"context"

	"github.com/Azure/go-autorest/autorest"

	"github.com/Azure/go-autorest/autorest/azure"

	"github.com/Azure/go-autorest/autorest/validation"

	"github.com/Azure/go-autorest/tracing"

	"net/http"

)

// UsagesClient is the network Client

type UsagesClient struct {

	BaseClient

}

// NewUsagesClient creates an instance of the UsagesClient client.

func NewUsagesClient(subscriptionID string) UsagesClient {

	return NewUsagesClientWithBaseURI(DefaultBaseURI, subscriptionID)

}

// NewUsagesClientWithBaseURI creates an instance of the UsagesClient client using a custom endpoint.  Use this when

// interacting with an Azure cloud that uses a non-standard base URI (sovereign clouds, Azure stack).

func NewUsagesClientWithBaseURI(baseURI string, subscriptionID string) UsagesClient {

	return UsagesClient{NewWithBaseURI(baseURI, subscriptionID)}

}

// List list network usages for a subscription.

// Parameters:

// location - the location where resource usage is queried.

func (client UsagesClient) List(ctx context.Context, location string) (result UsagesListResultPage, err error) {

	if tracing.IsEnabled() {

		ctx = tracing.StartSpan(ctx, fqdn+"/UsagesClient.List")

		defer func() {

			sc := -1

			if result.ulr.Response.Response != nil {

				sc = result.ulr.Response.Response.StatusCode

			}

			tracing.EndSpan(ctx, sc, err)

		}()

	}

	if err := validation.Validate([]validation.Validation{

		{TargetValue: location,

			Constraints: []validation.Constraint{{Target: "location", Name: validation.Pattern, Rule: `^[-\w\._ ]+$`, Chain: nil}}}}); err != nil {

		return result, validation.NewError("network.UsagesClient", "List", err.Error())

	}

	result.fn = client.listNextResults

	req, err := client.ListPreparer(ctx, location)

	if err != nil {

		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", nil, "Failure preparing request")

		return

	}

	resp, err := client.ListSender(req)

	if err != nil {

		result.ulr.Response = autorest.Response{Response: resp}

		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", resp, "Failure sending request")

		return

	}

	result.ulr, err = client.ListResponder(resp)

	if err != nil {

		err = autorest.NewErrorWithError(err, "network.UsagesClient", "List", resp, "Failure responding to request")

	}

	return

}

// ListPreparer prepares the List request.

func (client UsagesClient) ListPreparer(ctx context.Context, location string) (*http.Request, error) {

	pathParameters := map[string]interface{}{

		"location":       autorest.Encode("path", location),

		"subscriptionId": autorest.Encode("path", client.SubscriptionID),

	}

	const APIVersion = "2019-09-01"

	queryParameters := map[string]interface{}{

		"api-version": APIVersion,

	}

	preparer := autorest.CreatePreparer(

		autorest.AsGet(),

		autorest.WithBaseURL(client.BaseURI),

		autorest.WithPathParameters("/subscriptions/{subscriptionId}/providers/Microsoft.Network/locations/{location}/usages", pathParameters),

		autorest.WithQueryParameters(queryParameters))

	return preparer.Prepare((&http.Request{}).WithContext(ctx))

}

// ListSender sends the List request. The method will close the

// http.Response Body if it receives an error.

func (client UsagesClient) ListSender(req *http.Request) (*http.Response, error) {

	return client.Send(req, azure.DoRetryWithRegistration(client.Client))

}

// ListResponder handles the response to the List request. The method always

// closes the http.Response Body.

func (client UsagesClient) ListResponder(resp *http.Response) (result UsagesListResult, err error) {

	err = autorest.Respond(

		resp,

		client.ByInspecting(),

		azure.WithErrorUnlessStatusCode(http.StatusOK),

		autorest.ByUnmarshallingJSON(&result),

		autorest.ByClosing())

	result.Response = autorest.Response{Response: resp}

	return

}

// listNextResults retrieves the next set of results, if any.

func (client UsagesClient) listNextResults(ctx context.Context, lastResults UsagesListResult) (result UsagesListResult, err error) {

	req, err := lastResults.usagesListResultPreparer(ctx)

	if err != nil {

		return result, autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", nil, "Failure preparing next results request")

	}

	if req == nil {

		return

	}

	resp, err := client.ListSender(req)

	if err != nil {

		result.Response = autorest.Response{Response: resp}

		return result, autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", resp, "Failure sending next results request")

	}

	result, err = client.ListResponder(resp)

	if err != nil {

		err = autorest.NewErrorWithError(err, "network.UsagesClient", "listNextResults", resp, "Failure responding to next results request")

	}

	return

}

// ListComplete enumerates all values, automatically crossing page boundaries as required.

func (client UsagesClient) ListComplete(ctx context.Context, location string) (result UsagesListResultIterator, err error) {

	if tracing.IsEnabled() {

		ctx = tracing.StartSpan(ctx, fqdn+"/UsagesClient.List")

		defer func() {

			sc := -1

			if result.Response().Response.Response != nil {

				sc = result.page.Response().Response.Response.StatusCode

			}

			tracing.EndSpan(ctx, sc, err)

		}()

	}

	result.page, err = client.List(ctx, location)

	return

}