import logging
import os.path

from osbenchmark.builder.downloaders.downloader import Downloader
from osbenchmark.builder.utils.binary_keys import BinaryKeys
from osbenchmark.exceptions import ExecutorError


class OpenSearchDistributionDownloader(Downloader):
    def __init__(self, provision_config_instance, executor, path_manager, distribution_repository_provider):
        super().__init__(executor)
        self.logger = logging.getLogger(__name__)
        self.provision_config_instance = provision_config_instance
        self.path_manager = path_manager
        self.distribution_repository_provider = distribution_repository_provider

    def download(self, host):
        binary_path = self._fetch_binary(host)
        return {BinaryKeys.OPENSEARCH: binary_path}

    def _fetch_binary(self, host):
        download_url = self.distribution_repository_provider.get_download_url(host)
        distribution_path = self._create_distribution_path(host, download_url)
        opensearch_version = self.provision_config_instance.variables["distribution"]["version"]

        is_binary_present = self._is_binary_present(host, distribution_path)
        is_cache_enabled = self.distribution_repository_provider.is_cache_enabled()

        if is_binary_present and is_cache_enabled:
            self.logger.info("Skipping download for version [%s]. Found existing binary at [%s].", opensearch_version,
                             distribution_path)
        else:
            self._download_opensearch(host, distribution_path, download_url, opensearch_version)

        return distribution_path

    def _create_distribution_path(self, host, download_url):
        distribution_root_path = os.path.join(self.provision_config_instance.variables["node"]["root"]["dir"], "distributions")
        self.path_manager.create_path(host, distribution_root_path)

        distribution_binary_name = self.distribution_repository_provider.get_file_name_from_download_url(download_url)
        return os.path.join(distribution_root_path, distribution_binary_name)

    def _is_binary_present(self, host, distribution_path):
        try:
            self.executor.execute(host, f"test -f {distribution_path}")
            return True
        except ExecutorError:
            return False

    def _download_opensearch(self, host, distribution_path, download_url, opensearch_version):
        self.logger.info("Resolved download URL [%s] for version [%s]", download_url, opensearch_version)
        self.logger.info("Starting download of OpenSearch [%s]", opensearch_version)

        try:
            self.executor.execute(host, f"curl -o {distribution_path} {download_url}")
        except ExecutorError as e:
            self.logger.exception("Exception downloading OpenSearch distribution for version [%s] from [%s].",
                                  opensearch_version, download_url)
            raise e

        self.logger.info("Successfully downloaded OpenSearch [%s].", opensearch_version)