// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// Code generated by Microsoft (R) AutoRest Code Generator.

package com.azure.analytics.purview.administration.implementation;

import com.azure.core.annotation.ExpectedResponses;
import com.azure.core.annotation.Get;
import com.azure.core.annotation.HeaderParam;
import com.azure.core.annotation.Host;
import com.azure.core.annotation.HostParam;
import com.azure.core.annotation.PathParam;
import com.azure.core.annotation.QueryParam;
import com.azure.core.annotation.ReturnType;
import com.azure.core.annotation.ServiceInterface;
import com.azure.core.annotation.ServiceMethod;
import com.azure.core.annotation.UnexpectedResponseExceptionType;
import com.azure.core.exception.ClientAuthenticationException;
import com.azure.core.exception.HttpResponseException;
import com.azure.core.exception.ResourceModifiedException;
import com.azure.core.exception.ResourceNotFoundException;
import com.azure.core.http.rest.PagedFlux;
import com.azure.core.http.rest.PagedIterable;
import com.azure.core.http.rest.PagedResponse;
import com.azure.core.http.rest.PagedResponseBase;
import com.azure.core.http.rest.RequestOptions;
import com.azure.core.http.rest.Response;
import com.azure.core.http.rest.RestProxy;
import com.azure.core.util.BinaryData;
import com.azure.core.util.Context;
import com.azure.core.util.FluxUtil;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import reactor.core.publisher.Mono;

/** An instance of this class provides access to all the operations defined in MetadataRoles. */
public final class MetadataRolesImpl {
    /** The proxy service used to perform REST calls. */
    private final MetadataRolesService service;

    /** The service client containing this operation class. */
    private final PurviewMetadataClientImpl client;

    /**
     * Initializes an instance of MetadataRolesImpl.
     *
     * @param client the instance of the service client containing this operation class.
     */
    MetadataRolesImpl(PurviewMetadataClientImpl client) {
        this.service =
                RestProxy.create(MetadataRolesService.class, client.getHttpPipeline(), client.getSerializerAdapter());
        this.client = client;
    }

    /**
     * The interface defining all the services for PurviewMetadataClientMetadataRoles to be used by the proxy service to
     * perform REST calls.
     */
    @Host("{Endpoint}/policyStore")
    @ServiceInterface(name = "PurviewMetadataClien")
    public interface MetadataRolesService {
        @Get("/metadataRoles")
        @ExpectedResponses({200})
        @UnexpectedResponseExceptionType(
                value = ClientAuthenticationException.class,
                code = {401})
        @UnexpectedResponseExceptionType(
                value = ResourceNotFoundException.class,
                code = {404})
        @UnexpectedResponseExceptionType(
                value = ResourceModifiedException.class,
                code = {409})
        @UnexpectedResponseExceptionType(HttpResponseException.class)
        Mono<Response<BinaryData>> list(
                @HostParam("Endpoint") String endpoint,
                @QueryParam("api-version") String apiVersion,
                @HeaderParam("Accept") String accept,
                RequestOptions requestOptions,
                Context context);

        @Get("/metadataRoles")
        @ExpectedResponses({200})
        @UnexpectedResponseExceptionType(
                value = ClientAuthenticationException.class,
                code = {401})
        @UnexpectedResponseExceptionType(
                value = ResourceNotFoundException.class,
                code = {404})
        @UnexpectedResponseExceptionType(
                value = ResourceModifiedException.class,
                code = {409})
        @UnexpectedResponseExceptionType(HttpResponseException.class)
        Response<BinaryData> listSync(
                @HostParam("Endpoint") String endpoint,
                @QueryParam("api-version") String apiVersion,
                @HeaderParam("Accept") String accept,
                RequestOptions requestOptions,
                Context context);

        @Get("{nextLink}")
        @ExpectedResponses({200})
        @UnexpectedResponseExceptionType(
                value = ClientAuthenticationException.class,
                code = {401})
        @UnexpectedResponseExceptionType(
                value = ResourceNotFoundException.class,
                code = {404})
        @UnexpectedResponseExceptionType(
                value = ResourceModifiedException.class,
                code = {409})
        @UnexpectedResponseExceptionType(HttpResponseException.class)
        Mono<Response<BinaryData>> listNext(
                @PathParam(value = "nextLink", encoded = true) String nextLink,
                @HostParam("Endpoint") String endpoint,
                @HeaderParam("Accept") String accept,
                RequestOptions requestOptions,
                Context context);

        @Get("{nextLink}")
        @ExpectedResponses({200})
        @UnexpectedResponseExceptionType(
                value = ClientAuthenticationException.class,
                code = {401})
        @UnexpectedResponseExceptionType(
                value = ResourceNotFoundException.class,
                code = {404})
        @UnexpectedResponseExceptionType(
                value = ResourceModifiedException.class,
                code = {409})
        @UnexpectedResponseExceptionType(HttpResponseException.class)
        Response<BinaryData> listNextSync(
                @PathParam(value = "nextLink", encoded = true) String nextLink,
                @HostParam("Endpoint") String endpoint,
                @HeaderParam("Accept") String accept,
                RequestOptions requestOptions,
                Context context);
    }

    /**
     * Lists roles for Purview Account.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles along with {@link PagedResponse} on successful completion of {@link Mono}.
     */
    @ServiceMethod(returns = ReturnType.SINGLE)
    private Mono<PagedResponse<BinaryData>> listSinglePageAsync(RequestOptions requestOptions) {
        final String accept = "application/json";
        return FluxUtil.withContext(
                        context ->
                                service.list(
                                        this.client.getEndpoint(),
                                        this.client.getServiceVersion().getVersion(),
                                        accept,
                                        requestOptions,
                                        context))
                .map(
                        res ->
                                new PagedResponseBase<>(
                                        res.getRequest(),
                                        res.getStatusCode(),
                                        res.getHeaders(),
                                        getValues(res.getValue(), "values"),
                                        getNextLink(res.getValue(), "nextLink"),
                                        null));
    }

    /**
     * Lists roles for Purview Account.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles as paginated response with {@link PagedFlux}.
     */
    @ServiceMethod(returns = ReturnType.COLLECTION)
    public PagedFlux<BinaryData> listAsync(RequestOptions requestOptions) {
        RequestOptions requestOptionsForNextPage = new RequestOptions();
        requestOptionsForNextPage.setContext(
                requestOptions != null && requestOptions.getContext() != null
                        ? requestOptions.getContext()
                        : Context.NONE);
        return new PagedFlux<>(
                () -> listSinglePageAsync(requestOptions),
                nextLink -> listNextSinglePageAsync(nextLink, requestOptionsForNextPage));
    }

    /**
     * Lists roles for Purview Account.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles along with {@link PagedResponse}.
     */
    @ServiceMethod(returns = ReturnType.SINGLE)
    private PagedResponse<BinaryData> listSinglePage(RequestOptions requestOptions) {
        final String accept = "application/json";
        Response<BinaryData> res =
                service.listSync(
                        this.client.getEndpoint(),
                        this.client.getServiceVersion().getVersion(),
                        accept,
                        requestOptions,
                        Context.NONE);
        return new PagedResponseBase<>(
                res.getRequest(),
                res.getStatusCode(),
                res.getHeaders(),
                getValues(res.getValue(), "values"),
                getNextLink(res.getValue(), "nextLink"),
                null);
    }

    /**
     * Lists roles for Purview Account.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles as paginated response with {@link PagedIterable}.
     */
    @ServiceMethod(returns = ReturnType.COLLECTION)
    public PagedIterable<BinaryData> list(RequestOptions requestOptions) {
        RequestOptions requestOptionsForNextPage = new RequestOptions();
        requestOptionsForNextPage.setContext(
                requestOptions != null && requestOptions.getContext() != null
                        ? requestOptions.getContext()
                        : Context.NONE);
        return new PagedIterable<>(
                () -> listSinglePage(requestOptions),
                nextLink -> listNextSinglePage(nextLink, requestOptionsForNextPage));
    }

    /**
     * Get the next page of items.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param nextLink The URL to get the next list of items
     *     <p>The nextLink parameter.
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles along with {@link PagedResponse} on successful completion of {@link Mono}.
     */
    @ServiceMethod(returns = ReturnType.SINGLE)
    private Mono<PagedResponse<BinaryData>> listNextSinglePageAsync(String nextLink, RequestOptions requestOptions) {
        final String accept = "application/json";
        return FluxUtil.withContext(
                        context ->
                                service.listNext(nextLink, this.client.getEndpoint(), accept, requestOptions, context))
                .map(
                        res ->
                                new PagedResponseBase<>(
                                        res.getRequest(),
                                        res.getStatusCode(),
                                        res.getHeaders(),
                                        getValues(res.getValue(), "values"),
                                        getNextLink(res.getValue(), "nextLink"),
                                        null));
    }

    /**
     * Get the next page of items.
     *
     * <p><strong>Response Body Schema</strong>
     *
     * <pre>{@code
     * {
     *     id: String (Optional)
     *     name: String (Optional)
     *     type: String (Optional)
     *     properties (Optional): {
     *         provisioningState: String (Optional)
     *         roleType: String (Optional)
     *         friendlyName: String (Optional)
     *         description: String (Optional)
     *         cnfCondition (Optional): [
     *              (Optional)[
     *                  (Optional){
     *                     attributeName: String (Optional)
     *                     attributeValueIncludes: String (Optional)
     *                     attributeValueIncludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                     attributeValueExcludes: String (Optional)
     *                     attributeValueExcludedIn (Optional): [
     *                         String (Optional)
     *                     ]
     *                 }
     *             ]
     *         ]
     *         dnfCondition (Optional): [
     *              (Optional)[
     *                 (recursive schema, see above)
     *             ]
     *         ]
     *         version: Long (Optional)
     *     }
     * }
     * }</pre>
     *
     * @param nextLink The URL to get the next list of items
     *     <p>The nextLink parameter.
     * @param requestOptions The options to configure the HTTP request before HTTP client sends it.
     * @throws HttpResponseException thrown if the request is rejected by server.
     * @throws ClientAuthenticationException thrown if the request is rejected by server on status code 401.
     * @throws ResourceNotFoundException thrown if the request is rejected by server on status code 404.
     * @throws ResourceModifiedException thrown if the request is rejected by server on status code 409.
     * @return list of Metadata roles along with {@link PagedResponse}.
     */
    @ServiceMethod(returns = ReturnType.SINGLE)
    private PagedResponse<BinaryData> listNextSinglePage(String nextLink, RequestOptions requestOptions) {
        final String accept = "application/json";
        Response<BinaryData> res =
                service.listNextSync(nextLink, this.client.getEndpoint(), accept, requestOptions, Context.NONE);
        return new PagedResponseBase<>(
                res.getRequest(),
                res.getStatusCode(),
                res.getHeaders(),
                getValues(res.getValue(), "values"),
                getNextLink(res.getValue(), "nextLink"),
                null);
    }

    private List<BinaryData> getValues(BinaryData binaryData, String path) {
        try {
            Map<?, ?> obj = binaryData.toObject(Map.class);
            List<?> values = (List<?>) obj.get(path);
            return values.stream().map(BinaryData::fromObject).collect(Collectors.toList());
        } catch (RuntimeException e) {
            return null;
        }
    }

    private String getNextLink(BinaryData binaryData, String path) {
        try {
            Map<?, ?> obj = binaryData.toObject(Map.class);
            return (String) obj.get(path);
        } catch (RuntimeException e) {
            return null;
        }
    }
}
