zz_generated_iotdevicevulnerability_client.go

//go:build go1.16
// +build go1.16

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.
// Code generated by Microsoft (R) AutoRest Code Generator.
// Changes may cause incorrect behavior and will be lost if the code is regenerated.

package armiotsecurity

import (
	"context"
	"errors"
	"fmt"
	"github.com/Azure/azure-sdk-for-go/sdk/azcore"
	"github.com/Azure/azure-sdk-for-go/sdk/azcore/arm"
	armruntime "github.com/Azure/azure-sdk-for-go/sdk/azcore/arm/runtime"
	"github.com/Azure/azure-sdk-for-go/sdk/azcore/policy"
	"github.com/Azure/azure-sdk-for-go/sdk/azcore/runtime"
	"net/http"
	"net/url"
	"strings"
)

// IotDeviceVulnerabilityClient contains the methods for the IotDeviceVulnerability group.
// Don't use this type directly, use NewIotDeviceVulnerabilityClient() instead.
type IotDeviceVulnerabilityClient struct {
	ep                  string
	pl                  runtime.Pipeline
	subscriptionID      string
	iotDefenderLocation string
}

// NewIotDeviceVulnerabilityClient creates a new instance of IotDeviceVulnerabilityClient with the specified values.
func NewIotDeviceVulnerabilityClient(subscriptionID string, iotDefenderLocation string, credential azcore.TokenCredential, options *arm.ClientOptions) *IotDeviceVulnerabilityClient {
	cp := arm.ClientOptions{}
	if options != nil {
		cp = *options
	}
	if len(cp.Host) == 0 {
		cp.Host = arm.AzurePublicCloud
	}
	return &IotDeviceVulnerabilityClient{subscriptionID: subscriptionID, iotDefenderLocation: iotDefenderLocation, ep: string(cp.Host), pl: armruntime.NewPipeline(module, version, credential, &cp)}
}

// Get - Get IoT Device Vulnerability
// If the operation fails it returns the *ErrorResponse error type.
func (client *IotDeviceVulnerabilityClient) Get(ctx context.Context, deviceGroupName string, vulnerabilityID string, options *IotDeviceVulnerabilityGetOptions) (IotDeviceVulnerabilityGetResponse, error) {
	req, err := client.getCreateRequest(ctx, deviceGroupName, vulnerabilityID, options)
	if err != nil {
		return IotDeviceVulnerabilityGetResponse{}, err
	}
	resp, err := client.pl.Do(req)
	if err != nil {
		return IotDeviceVulnerabilityGetResponse{}, err
	}
	if !runtime.HasStatusCode(resp, http.StatusOK) {
		return IotDeviceVulnerabilityGetResponse{}, client.getHandleError(resp)
	}
	return client.getHandleResponse(resp)
}

// getCreateRequest creates the Get request.
func (client *IotDeviceVulnerabilityClient) getCreateRequest(ctx context.Context, deviceGroupName string, vulnerabilityID string, options *IotDeviceVulnerabilityGetOptions) (*policy.Request, error) {
	urlPath := "/subscriptions/{subscriptionId}/providers/Microsoft.IoTSecurity/locations/{iotDefenderLocation}/deviceGroups/{deviceGroupName}/vulnerabilities/{vulnerabilityId}"
	if client.subscriptionID == "" {
		return nil, errors.New("parameter client.subscriptionID cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{subscriptionId}", url.PathEscape(client.subscriptionID))
	if client.iotDefenderLocation == "" {
		return nil, errors.New("parameter client.iotDefenderLocation cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{iotDefenderLocation}", url.PathEscape(client.iotDefenderLocation))
	if deviceGroupName == "" {
		return nil, errors.New("parameter deviceGroupName cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{deviceGroupName}", url.PathEscape(deviceGroupName))
	if vulnerabilityID == "" {
		return nil, errors.New("parameter vulnerabilityID cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{vulnerabilityId}", url.PathEscape(vulnerabilityID))
	req, err := runtime.NewRequest(ctx, http.MethodGet, runtime.JoinPaths(client.ep, urlPath))
	if err != nil {
		return nil, err
	}
	reqQP := req.Raw().URL.Query()
	reqQP.Set("api-version", "2021-07-01-preview")
	req.Raw().URL.RawQuery = reqQP.Encode()
	req.Raw().Header.Set("Accept", "application/json")
	return req, nil
}

// getHandleResponse handles the Get response.
func (client *IotDeviceVulnerabilityClient) getHandleResponse(resp *http.Response) (IotDeviceVulnerabilityGetResponse, error) {
	result := IotDeviceVulnerabilityGetResponse{RawResponse: resp}
	if err := runtime.UnmarshalAsJSON(resp, &result.DeviceVulnerabilityModel); err != nil {
		return IotDeviceVulnerabilityGetResponse{}, runtime.NewResponseError(err, resp)
	}
	return result, nil
}

// getHandleError handles the Get error response.
func (client *IotDeviceVulnerabilityClient) getHandleError(resp *http.Response) error {
	body, err := runtime.Payload(resp)
	if err != nil {
		return runtime.NewResponseError(err, resp)
	}
	errType := ErrorResponse{raw: string(body)}
	if err := runtime.UnmarshalAsJSON(resp, &errType); err != nil {
		return runtime.NewResponseError(fmt.Errorf("%s\n%s", string(body), err), resp)
	}
	return runtime.NewResponseError(&errType, resp)
}

// List - List IoT Device Vulnerabilities
// If the operation fails it returns the *ErrorResponse error type.
func (client *IotDeviceVulnerabilityClient) List(deviceGroupName string, options *IotDeviceVulnerabilityListOptions) *IotDeviceVulnerabilityListPager {
	return &IotDeviceVulnerabilityListPager{
		client: client,
		requester: func(ctx context.Context) (*policy.Request, error) {
			return client.listCreateRequest(ctx, deviceGroupName, options)
		},
		advancer: func(ctx context.Context, resp IotDeviceVulnerabilityListResponse) (*policy.Request, error) {
			return runtime.NewRequest(ctx, http.MethodGet, *resp.DeviceVulnerabilityListModel.NextLink)
		},
	}
}

// listCreateRequest creates the List request.
func (client *IotDeviceVulnerabilityClient) listCreateRequest(ctx context.Context, deviceGroupName string, options *IotDeviceVulnerabilityListOptions) (*policy.Request, error) {
	urlPath := "/subscriptions/{subscriptionId}/providers/Microsoft.IoTSecurity/locations/{iotDefenderLocation}/deviceGroups/{deviceGroupName}/vulnerabilities"
	if client.subscriptionID == "" {
		return nil, errors.New("parameter client.subscriptionID cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{subscriptionId}", url.PathEscape(client.subscriptionID))
	if client.iotDefenderLocation == "" {
		return nil, errors.New("parameter client.iotDefenderLocation cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{iotDefenderLocation}", url.PathEscape(client.iotDefenderLocation))
	if deviceGroupName == "" {
		return nil, errors.New("parameter deviceGroupName cannot be empty")
	}
	urlPath = strings.ReplaceAll(urlPath, "{deviceGroupName}", url.PathEscape(deviceGroupName))
	req, err := runtime.NewRequest(ctx, http.MethodGet, runtime.JoinPaths(client.ep, urlPath))
	if err != nil {
		return nil, err
	}
	reqQP := req.Raw().URL.Query()
	reqQP.Set("api-version", "2021-07-01-preview")
	if options != nil && options.DeviceID != nil {
		reqQP.Set("deviceId", *options.DeviceID)
	}
	if options != nil && options.SkipToken != nil {
		reqQP.Set("$skipToken", *options.SkipToken)
	}
	req.Raw().URL.RawQuery = reqQP.Encode()
	req.Raw().Header.Set("Accept", "application/json")
	return req, nil
}

// listHandleResponse handles the List response.
func (client *IotDeviceVulnerabilityClient) listHandleResponse(resp *http.Response) (IotDeviceVulnerabilityListResponse, error) {
	result := IotDeviceVulnerabilityListResponse{RawResponse: resp}
	if err := runtime.UnmarshalAsJSON(resp, &result.DeviceVulnerabilityListModel); err != nil {
		return IotDeviceVulnerabilityListResponse{}, runtime.NewResponseError(err, resp)
	}
	return result, nil
}

// listHandleError handles the List error response.
func (client *IotDeviceVulnerabilityClient) listHandleError(resp *http.Response) error {
	body, err := runtime.Payload(resp)
	if err != nil {
		return runtime.NewResponseError(err, resp)
	}
	errType := ErrorResponse{raw: string(body)}
	if err := runtime.UnmarshalAsJSON(resp, &errType); err != nil {
		return runtime.NewResponseError(fmt.Errorf("%s\n%s", string(body), err), resp)
	}
	return runtime.NewResponseError(&errType, resp)
}