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

package com.azure.resourcemanager.machinelearningservices;

import com.azure.core.credential.TokenCredential;
import com.azure.core.http.HttpClient;
import com.azure.core.http.HttpPipeline;
import com.azure.core.http.HttpPipelineBuilder;
import com.azure.core.http.policy.AddDatePolicy;
import com.azure.core.http.policy.BearerTokenAuthenticationPolicy;
import com.azure.core.http.policy.HttpLogOptions;
import com.azure.core.http.policy.HttpLoggingPolicy;
import com.azure.core.http.policy.HttpPipelinePolicy;
import com.azure.core.http.policy.HttpPolicyProviders;
import com.azure.core.http.policy.RequestIdPolicy;
import com.azure.core.http.policy.RetryPolicy;
import com.azure.core.http.policy.UserAgentPolicy;
import com.azure.core.management.profile.AzureProfile;
import com.azure.core.util.Configuration;
import com.azure.core.util.logging.ClientLogger;
import com.azure.resourcemanager.machinelearningservices.fluent.AzureMachineLearningWorkspaces;
import com.azure.resourcemanager.machinelearningservices.implementation.AzureMachineLearningWorkspacesBuilder;
import com.azure.resourcemanager.machinelearningservices.implementation.MachineLearningComputesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.MachineLearningServicesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.NotebooksImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.OperationsImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.PrivateEndpointConnectionsImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.PrivateLinkResourcesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.QuotasImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.StorageAccountsImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.UsagesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.VirtualMachineSizesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.WorkspaceConnectionsImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.WorkspaceFeaturesImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.WorkspaceOperationsImpl;
import com.azure.resourcemanager.machinelearningservices.implementation.WorkspacesImpl;
import com.azure.resourcemanager.machinelearningservices.models.MachineLearningComputes;
import com.azure.resourcemanager.machinelearningservices.models.MachineLearningServices;
import com.azure.resourcemanager.machinelearningservices.models.Notebooks;
import com.azure.resourcemanager.machinelearningservices.models.Operations;
import com.azure.resourcemanager.machinelearningservices.models.PrivateEndpointConnections;
import com.azure.resourcemanager.machinelearningservices.models.PrivateLinkResources;
import com.azure.resourcemanager.machinelearningservices.models.Quotas;
import com.azure.resourcemanager.machinelearningservices.models.StorageAccounts;
import com.azure.resourcemanager.machinelearningservices.models.Usages;
import com.azure.resourcemanager.machinelearningservices.models.VirtualMachineSizes;
import com.azure.resourcemanager.machinelearningservices.models.WorkspaceConnections;
import com.azure.resourcemanager.machinelearningservices.models.WorkspaceFeatures;
import com.azure.resourcemanager.machinelearningservices.models.WorkspaceOperations;
import com.azure.resourcemanager.machinelearningservices.models.Workspaces;
import java.time.Duration;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
 * Entry point to MachineLearningServicesManager. These APIs allow end users to operate on Azure Machine Learning
 * Workspace resources.
 */
public final class MachineLearningServicesManager {
    private Operations operations;

    private Workspaces workspaces;

    private WorkspaceFeatures workspaceFeatures;

    private Usages usages;

    private VirtualMachineSizes virtualMachineSizes;

    private Quotas quotas;

    private MachineLearningComputes machineLearningComputes;

    private WorkspaceOperations workspaceOperations;

    private PrivateEndpointConnections privateEndpointConnections;

    private PrivateLinkResources privateLinkResources;

    private MachineLearningServices machineLearningServices;

    private Notebooks notebooks;

    private StorageAccounts storageAccounts;

    private WorkspaceConnections workspaceConnections;

    private final AzureMachineLearningWorkspaces clientObject;

    private MachineLearningServicesManager(
        HttpPipeline httpPipeline, AzureProfile profile, Duration defaultPollInterval) {
        Objects.requireNonNull(httpPipeline, "'httpPipeline' cannot be null.");
        Objects.requireNonNull(profile, "'profile' cannot be null.");
        this.clientObject =
            new AzureMachineLearningWorkspacesBuilder()
                .pipeline(httpPipeline)
                .endpoint(profile.getEnvironment().getResourceManagerEndpoint())
                .subscriptionId(profile.getSubscriptionId())
                .defaultPollInterval(defaultPollInterval)
                .buildClient();
    }

    /**
     * Creates an instance of MachineLearningServices service API entry point.
     *
     * @param credential the credential to use.
     * @param profile the Azure profile for client.
     * @return the MachineLearningServices service API instance.
     */
    public static MachineLearningServicesManager authenticate(TokenCredential credential, AzureProfile profile) {
        Objects.requireNonNull(credential, "'credential' cannot be null.");
        Objects.requireNonNull(profile, "'profile' cannot be null.");
        return configure().authenticate(credential, profile);
    }

    /**
     * Gets a Configurable instance that can be used to create MachineLearningServicesManager with optional
     * configuration.
     *
     * @return the Configurable instance allowing configurations.
     */
    public static Configurable configure() {
        return new MachineLearningServicesManager.Configurable();
    }

    /** The Configurable allowing configurations to be set. */
    public static final class Configurable {
        private final ClientLogger logger = new ClientLogger(Configurable.class);

        private HttpClient httpClient;
        private HttpLogOptions httpLogOptions;
        private final List<HttpPipelinePolicy> policies = new ArrayList<>();
        private RetryPolicy retryPolicy;
        private Duration defaultPollInterval;

        private Configurable() {
        }

        /**
         * Sets the http client.
         *
         * @param httpClient the HTTP client.
         * @return the configurable object itself.
         */
        public Configurable withHttpClient(HttpClient httpClient) {
            this.httpClient = Objects.requireNonNull(httpClient, "'httpClient' cannot be null.");
            return this;
        }

        /**
         * Sets the logging options to the HTTP pipeline.
         *
         * @param httpLogOptions the HTTP log options.
         * @return the configurable object itself.
         */
        public Configurable withLogOptions(HttpLogOptions httpLogOptions) {
            this.httpLogOptions = Objects.requireNonNull(httpLogOptions, "'httpLogOptions' cannot be null.");
            return this;
        }

        /**
         * Adds the pipeline policy to the HTTP pipeline.
         *
         * @param policy the HTTP pipeline policy.
         * @return the configurable object itself.
         */
        public Configurable withPolicy(HttpPipelinePolicy policy) {
            this.policies.add(Objects.requireNonNull(policy, "'policy' cannot be null."));
            return this;
        }

        /**
         * Sets the retry policy to the HTTP pipeline.
         *
         * @param retryPolicy the HTTP pipeline retry policy.
         * @return the configurable object itself.
         */
        public Configurable withRetryPolicy(RetryPolicy retryPolicy) {
            this.retryPolicy = Objects.requireNonNull(retryPolicy, "'retryPolicy' cannot be null.");
            return this;
        }

        /**
         * Sets the default poll interval, used when service does not provide "Retry-After" header.
         *
         * @param defaultPollInterval the default poll interval.
         * @return the configurable object itself.
         */
        public Configurable withDefaultPollInterval(Duration defaultPollInterval) {
            this.defaultPollInterval = Objects.requireNonNull(defaultPollInterval, "'retryPolicy' cannot be null.");
            if (this.defaultPollInterval.isNegative()) {
                throw logger.logExceptionAsError(new IllegalArgumentException("'httpPipeline' cannot be negative"));
            }
            return this;
        }

        /**
         * Creates an instance of MachineLearningServices service API entry point.
         *
         * @param credential the credential to use.
         * @param profile the Azure profile for client.
         * @return the MachineLearningServices service API instance.
         */
        public MachineLearningServicesManager authenticate(TokenCredential credential, AzureProfile profile) {
            Objects.requireNonNull(credential, "'credential' cannot be null.");
            Objects.requireNonNull(profile, "'profile' cannot be null.");

            StringBuilder userAgentBuilder = new StringBuilder();
            userAgentBuilder
                .append("azsdk-java")
                .append("-")
                .append("com.azure.resourcemanager.machinelearningservices")
                .append("/")
                .append("1.0.0-beta.1");
            if (!Configuration.getGlobalConfiguration().get("AZURE_TELEMETRY_DISABLED", false)) {
                userAgentBuilder
                    .append(" (")
                    .append(Configuration.getGlobalConfiguration().get("java.version"))
                    .append("; ")
                    .append(Configuration.getGlobalConfiguration().get("os.name"))
                    .append("; ")
                    .append(Configuration.getGlobalConfiguration().get("os.version"))
                    .append("; auto-generated)");
            } else {
                userAgentBuilder.append(" (auto-generated)");
            }

            if (retryPolicy == null) {
                retryPolicy = new RetryPolicy("Retry-After", ChronoUnit.SECONDS);
            }
            List<HttpPipelinePolicy> policies = new ArrayList<>();
            policies.add(new UserAgentPolicy(userAgentBuilder.toString()));
            policies.add(new RequestIdPolicy());
            HttpPolicyProviders.addBeforeRetryPolicies(policies);
            policies.add(retryPolicy);
            policies.add(new AddDatePolicy());
            policies
                .add(
                    new BearerTokenAuthenticationPolicy(
                        credential, profile.getEnvironment().getManagementEndpoint() + "/.default"));
            policies.addAll(this.policies);
            HttpPolicyProviders.addAfterRetryPolicies(policies);
            policies.add(new HttpLoggingPolicy(httpLogOptions));
            HttpPipeline httpPipeline =
                new HttpPipelineBuilder()
                    .httpClient(httpClient)
                    .policies(policies.toArray(new HttpPipelinePolicy[0]))
                    .build();
            return new MachineLearningServicesManager(httpPipeline, profile, defaultPollInterval);
        }
    }

    /** @return Resource collection API of Operations. */
    public Operations operations() {
        if (this.operations == null) {
            this.operations = new OperationsImpl(clientObject.getOperations(), this);
        }
        return operations;
    }

    /** @return Resource collection API of Workspaces. */
    public Workspaces workspaces() {
        if (this.workspaces == null) {
            this.workspaces = new WorkspacesImpl(clientObject.getWorkspaces(), this);
        }
        return workspaces;
    }

    /** @return Resource collection API of WorkspaceFeatures. */
    public WorkspaceFeatures workspaceFeatures() {
        if (this.workspaceFeatures == null) {
            this.workspaceFeatures = new WorkspaceFeaturesImpl(clientObject.getWorkspaceFeatures(), this);
        }
        return workspaceFeatures;
    }

    /** @return Resource collection API of Usages. */
    public Usages usages() {
        if (this.usages == null) {
            this.usages = new UsagesImpl(clientObject.getUsages(), this);
        }
        return usages;
    }

    /** @return Resource collection API of VirtualMachineSizes. */
    public VirtualMachineSizes virtualMachineSizes() {
        if (this.virtualMachineSizes == null) {
            this.virtualMachineSizes = new VirtualMachineSizesImpl(clientObject.getVirtualMachineSizes(), this);
        }
        return virtualMachineSizes;
    }

    /** @return Resource collection API of Quotas. */
    public Quotas quotas() {
        if (this.quotas == null) {
            this.quotas = new QuotasImpl(clientObject.getQuotas(), this);
        }
        return quotas;
    }

    /** @return Resource collection API of MachineLearningComputes. */
    public MachineLearningComputes machineLearningComputes() {
        if (this.machineLearningComputes == null) {
            this.machineLearningComputes =
                new MachineLearningComputesImpl(clientObject.getMachineLearningComputes(), this);
        }
        return machineLearningComputes;
    }

    /** @return Resource collection API of WorkspaceOperations. */
    public WorkspaceOperations workspaceOperations() {
        if (this.workspaceOperations == null) {
            this.workspaceOperations = new WorkspaceOperationsImpl(clientObject.getWorkspaceOperations(), this);
        }
        return workspaceOperations;
    }

    /** @return Resource collection API of PrivateEndpointConnections. */
    public PrivateEndpointConnections privateEndpointConnections() {
        if (this.privateEndpointConnections == null) {
            this.privateEndpointConnections =
                new PrivateEndpointConnectionsImpl(clientObject.getPrivateEndpointConnections(), this);
        }
        return privateEndpointConnections;
    }

    /** @return Resource collection API of PrivateLinkResources. */
    public PrivateLinkResources privateLinkResources() {
        if (this.privateLinkResources == null) {
            this.privateLinkResources = new PrivateLinkResourcesImpl(clientObject.getPrivateLinkResources(), this);
        }
        return privateLinkResources;
    }

    /** @return Resource collection API of MachineLearningServices. */
    public MachineLearningServices machineLearningServices() {
        if (this.machineLearningServices == null) {
            this.machineLearningServices =
                new MachineLearningServicesImpl(clientObject.getMachineLearningServices(), this);
        }
        return machineLearningServices;
    }

    /** @return Resource collection API of Notebooks. */
    public Notebooks notebooks() {
        if (this.notebooks == null) {
            this.notebooks = new NotebooksImpl(clientObject.getNotebooks(), this);
        }
        return notebooks;
    }

    /** @return Resource collection API of StorageAccounts. */
    public StorageAccounts storageAccounts() {
        if (this.storageAccounts == null) {
            this.storageAccounts = new StorageAccountsImpl(clientObject.getStorageAccounts(), this);
        }
        return storageAccounts;
    }

    /** @return Resource collection API of WorkspaceConnections. */
    public WorkspaceConnections workspaceConnections() {
        if (this.workspaceConnections == null) {
            this.workspaceConnections = new WorkspaceConnectionsImpl(clientObject.getWorkspaceConnections(), this);
        }
        return workspaceConnections;
    }

    /**
     * @return Wrapped service client AzureMachineLearningWorkspaces providing direct access to the underlying
     *     auto-generated API implementation, based on Azure REST API.
     */
    public AzureMachineLearningWorkspaces serviceClient() {
        return this.clientObject;
    }
}
