import functools
import os
from contextlib import suppress
from typing import Optional, TypeVar, cast

import fmf
import fmf.utils

import tmt.log
import tmt.utils
from tmt._compat.pathlib import Path
from tmt._compat.pydantic import ValidationError
from tmt.config.models.hardware import HardwareConfig
from tmt.config.models.link import LinkConfig
from tmt.config.models.themes import Theme, ThemeConfig
from tmt.container import MetadataContainer

MetadataContainerT = TypeVar('MetadataContainerT', bound='MetadataContainer')

# Config directory
DEFAULT_CONFIG_DIR = Path('~/.config/tmt')


def effective_config_dir() -> Path:
    """
    Find out what the actual config directory is.

    If ``TMT_CONFIG_DIR`` variable is set, it is used. Otherwise,
    :py:const:`DEFAULT_CONFIG_DIR` is picked.
    """

    if 'TMT_CONFIG_DIR' in os.environ:
        return Path(os.environ['TMT_CONFIG_DIR']).expanduser()

    return DEFAULT_CONFIG_DIR.expanduser()


class Config:
    """
    User configuration
    """

    def __init__(self, logger: tmt.log.Logger) -> None:
        """
        Initialize config directory path
        """

        self.path = effective_config_dir()
        self.logger = logger

        try:
            self.path.mkdir(parents=True, exist_ok=True)
        except OSError as error:
            raise tmt.utils.GeneralError(f"Failed to create config '{self.path}'.") from error

    @property
    def _last_run_symlink(self) -> Path:
        return self.path / 'last-run'

    @property
    def last_run(self) -> Optional[Path]:
        """
        Get the last run workdir path
        """

        return self._last_run_symlink.resolve() if self._last_run_symlink.is_symlink() else None

    @last_run.setter
    def last_run(self, workdir: Path) -> None:
        """
        Set the last run to the given run workdir
        """

        with suppress(OSError):
            self._last_run_symlink.unlink()

        try:
            self._last_run_symlink.symlink_to(workdir)
        except FileExistsError:
            # Race when tmt runs in parallel
            self.logger.warning(
                f"Unable to mark '{workdir}' as the last run, "
                "'tmt run --last' might not pick the right run directory."
            )
        except OSError as error:
            raise tmt.utils.GeneralError(f"Unable to save last run '{self.path}'.") from error

    @functools.cached_property
    def fmf_tree(self) -> Optional[fmf.Tree]:
        """
        Return the configuration tree
        """

        try:
            return fmf.Tree(self.path)
        except fmf.utils.RootError:
            self.logger.debug(f"Config tree not found in '{self.path}'.")

            return None

    def _parse_config_subtree(
        self, path: str, model: type[MetadataContainerT]
    ) -> Optional[MetadataContainerT]:
        if self.fmf_tree is None:
            return None

        subtree = cast(Optional[fmf.Tree], self.fmf_tree.find(path))

        if not subtree:
            self.logger.debug(f"Config path '{path}' not found in '{self.path}'.")

            return None

        try:
            return model.model_validate(subtree.data)

        except ValidationError as error:
            raise tmt.utils.SpecificationError(
                f"Invalid configuration in '{subtree.name}'."
            ) from error

    @functools.cached_property
    def link(self) -> Optional[LinkConfig]:
        """
        Return the link configuration, if present.
        """

        return self._parse_config_subtree('/link', LinkConfig)

    @functools.cached_property
    def theme(self) -> Theme:
        theme_config = self._parse_config_subtree('/theme', ThemeConfig)

        if theme_config is None:
            return ThemeConfig.get_default_theme(logger=self.logger)

        return theme_config.get_active_theme(logger=self.logger)

    @property
    def hardware(self) -> Optional[HardwareConfig]:
        """
        Return the hardware configuration, if present.
        """

        return self._parse_config_subtree('/hardware', HardwareConfig)