# ======================================================================= #
#  Copyright (C) 2020 - 2024 Dominik Willner <th33xitus@gmail.com>        #
#                                                                         #
#  This file is part of KIAUH - Klipper Installation And Update Helper    #
#  https://github.com/dw-0/kiauh                                          #
#                                                                         #
#  This file may be distributed under the terms of the GNU GPLv3 license  #
# ======================================================================= #

import configparser
from pathlib import Path
from typing import Union

from utils.logger import Logger


# noinspection PyMethodMayBeStatic
class ConfigManager:
    def __init__(self, cfg_file: Path):
        self.config_file = cfg_file
        self.config = CustomConfigParser()

        if cfg_file.is_file():
            self.read_config()

    def read_config(self) -> None:
        if not self.config_file:
            Logger.print_error("Unable to read config file. File not found.")
            return

        self.config.read_file(open(self.config_file, "r"))

    def write_config(self) -> None:
        with open(self.config_file, "w") as cfg:
            self.config.write(cfg)

    def get_value(self, section: str, key: str, silent=True) -> Union[str, bool, None]:
        if not self.config.has_section(section):
            if not silent:
                log = f"Section not defined. Unable to read section: [{section}]."
                Logger.print_error(log)
            return None

        if not self.config.has_option(section, key):
            if not silent:
                log = f"Option not defined in section [{section}]. Unable to read option: '{key}'."
                Logger.print_error(log)
            return None

        value = self.config.get(section, key)
        if value == "True" or value == "true":
            return True
        elif value == "False" or value == "false":
            return False
        else:
            return value

    def set_value(self, section: str, key: str, value: str):
        self.config.set(section, key, value)


class CustomConfigParser(configparser.ConfigParser):
    """
    A custom ConfigParser class overwriting the write() method of configparser.Configparser.
    Key and value will be delimited by a ": ".
    Note the whitespace AFTER the colon, which is the whole reason for that overwrite.
    """

    def write(self, fp, space_around_delimiters=False):
        if self._defaults:
            fp.write("[%s]\n" % configparser.DEFAULTSECT)
            for key, value in self._defaults.items():
                fp.write("%s: %s\n" % (key, str(value).replace("\n", "\n\t")))
            fp.write("\n")
        for section in self._sections:
            fp.write("[%s]\n" % section)
            for key, value in self._sections[section].items():
                if key == "__name__":
                    continue
                if (value is not None) or (self._optcre == self.OPTCRE):
                    key = ": ".join((key, str(value).replace("\n", "\n\t")))
                fp.write("%s\n" % key)
            fp.write("\n")