import copy
import logging
import os.path

import yaml

import gen
import ssh.validate
from gen.build_deploy.bash import onprem_source
from pkgpanda.util import load_yaml, write_string, YamlParseError

log = logging.getLogger(__name__)

config_sample = """
---
# The name of your DC/OS cluster. Visable in the DC/OS user interface.
cluster_name: 'DC/OS'
master_discovery: static
exhibitor_storage_backend: 'static'
resolvers:
- 8.8.8.8
- 8.8.4.4
ssh_port: 22
process_timeout: 10000
bootstrap_url: file:///opt/dcos_install_tmp
"""


def normalize_config_validation(messages):
    """Accepts Gen error message format and returns a flattened dictionary
    of validation messages.

    :param messages: Gen validation messages
    :type messages: dict | None
    """
    validation = {}
    if 'errors' in messages:
        for key, errors in messages['errors'].items():
            validation[key] = errors['message']

    if 'unset' in messages:
        for key in messages['unset']:
            validation[key] = 'Must set {}, no way to calculate value.'.format(key)

    return validation


def make_default_config_if_needed(config_path):
    if os.path.exists(config_path):
        return

    write_string(config_path, config_sample)


class NoConfigError(Exception):
    pass


class Config():

    def __init__(self, config_path):
        self.config_path = config_path

        # Create the config file iff allowed and there isn't one provided by the user.

        self._config = self._load_config()
        if not isinstance(self._config, dict):
            # FIXME
            raise NotImplementedError()

    def _load_config(self):
        if self.config_path is None:
            return {}

        try:
            return load_yaml(self.config_path)
        except FileNotFoundError as ex:
            raise NoConfigError(
                "No config file found at {}. See the DC/OS documentation for the "
                "available configuration options. You can also use the GUI web installer (--web), "
                "which provides a guided configuration and installation for simple "
                "deployments.".format(self.config_path)) from ex
        except OSError as ex:
            raise NoConfigError(
                "Failed to open config file at {}: {}. See the DC/OS documentation to learn "
                "how to create a config file. You can also use the GUI web installer (--web), "
                "which provides a guided configuration and installation for simple "
                "deployments.".format(self.config_path, ex)) from ex
        except YamlParseError as ex:
            raise NoConfigError("Unable to load configuration file. {}".format(ex)) from ex

    def update(self, updates):
        # TODO(cmaloney): check that the updates are all for valid keys, keep
        # any ones for valid keys and throw out any for invalid keys, returning
        # errors for the invalid keys.
        self._config.update(updates)

    # TODO(cmaloney): Figure out a way for the installer being generated (Advanced AWS CF templates vs.
    # bash) to automatically set this in gen.generate rather than having to merge itself.
    def as_gen_format(self):
        return gen.stringify_configuration(self._config)

    def do_validate(self, include_ssh):
        user_arguments = self.as_gen_format()
        extra_sources = [onprem_source]
        extra_targets = []
        if include_ssh:
            extra_sources.append(ssh.validate.source)
            extra_targets.append(ssh.validate.get_target())

        sources, targets, _ = gen.get_dcosconfig_source_target_and_templates(user_arguments, [], extra_sources)
        targets = targets + extra_targets

        resolver = gen.internals.resolve_configuration(sources, targets)
        # TODO(cmaloney): kill this function and make the API return the structured
        # results api as was always intended rather than the flattened / lossy other
        # format. This will be an  API incompatible change. The messages format was
        # specifically so that there wouldn't be this sort of API incompatibility.
        return normalize_config_validation(resolver.status_dict)

    def get_yaml_str(self):
        return yaml.dump(self._config, default_flow_style=False, explicit_start=True)

    def write_config(self):
        assert self.config_path is not None

        write_string(self.config_path, self.get_yaml_str())

    def __getitem__(self, key: str):
        return self._config[key]

    def __contains__(self, key: str):
        return key in self._config

    # TODO(cmaloney): kill this, should use config target to set defaults. The config targets should
    # set these defaults.
    def hacky_default_get(self, *args, **kwargs):
        return self._config.get(*args, **kwargs)

    @property
    def config(self):
        return copy.copy(self._config)


def to_config(config_dict: dict):
    config = Config(None)
    config.update(config_dict)
    return config
