# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------

import os
import functools
from devtools_testutils import PowerShellPreparer, AzureMgmtPreparer, get_credential

ENABLE_LOGGER = os.getenv("ENABLE_LOGGER", "False")
REGION = os.getenv("DOCUMENTINTELLIGENCE_LOCATION", None)


DocumentIntelligencePreparer = functools.partial(
    PowerShellPreparer,
    "documentintelligence",
    documentintelligence_endpoint="https://fakeendpoint.cognitiveservices.azure.com",
    documentintelligence_storage_container_sas_url="https://blob_sas_url",
    documentintelligence_testing_data_container_sas_url="https://blob_sas_url",
    documentintelligence_training_data_classifier_sas_url="https://blob_sas_url",
    documentintelligence_batch_training_data_container_sas_url="https://blob_sas_url",
    documentintelligence_batch_training_result_data_container_sas_url="https://blob_sas_url",
    documentintelligence_batch_training_async_result_data_container_sas_url="https://blob_sas_url",
    documentintelligence_resource_id="/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rgname/providers/Microsoft.CognitiveServices/accounts/documentintelligencename",
    documentintelligence_resource_region="region",
)


class GlobalClientPreparer(AzureMgmtPreparer):
    def __init__(self, client_cls, client_kwargs={}, **kwargs):
        super(GlobalClientPreparer, self).__init__(name_prefix="", random_name_length=42)
        self.client_kwargs = client_kwargs
        self.client_cls = client_cls

    def create_resource(self, name, **kwargs):
        if self.is_live:
            documentintelligence_endpoint = os.environ["DOCUMENTINTELLIGENCE_ENDPOINT"]
            polling_interval = 5
        else:
            documentintelligence_endpoint = "https://fakeendpoint.cognitiveservices.azure.com"
            polling_interval = 0

        client = self.client_cls(
            documentintelligence_endpoint,
            get_credential(),
            polling_interval=polling_interval,
            logging_enable=True if ENABLE_LOGGER == "True" else False,
            **self.client_kwargs
        )
        kwargs.update({"client": client})
        return kwargs


class GlobalClientPreparerAsync(AzureMgmtPreparer):
    def __init__(self, client_cls, client_kwargs={}, **kwargs):
        super(GlobalClientPreparerAsync, self).__init__(name_prefix="", random_name_length=42)
        self.client_kwargs = client_kwargs
        self.client_cls = client_cls

    def create_resource(self, name, **kwargs):
        if self.is_live:
            documentintelligence_endpoint = os.environ["DOCUMENTINTELLIGENCE_ENDPOINT"]
            polling_interval = 5
        else:
            documentintelligence_endpoint = "https://fakeendpoint.cognitiveservices.azure.com"
            polling_interval = 0

        client = self.client_cls(
            documentintelligence_endpoint,
            get_credential(is_async=True),
            polling_interval=polling_interval,
            logging_enable=True if ENABLE_LOGGER == "True" else False,
            **self.client_kwargs
        )
        kwargs.update({"client": client})
        return kwargs