# ========================================================================== #
#                                                                            #
#    KVMD - The The main Pi-KVM daemon.                                      #
#                                                                            #
#    Copyright (C) 2018  Maxim Devaev <mdevaev@gmail.com>                    #
#                                                                            #
#    This program is free software: you can redistribute it and/or modify    #
#    it under the terms of the GNU General Public License as published by    #
#    the Free Software Foundation, either version 3 of the License, or       #
#    (at your option) any later version.                                     #
#                                                                            #
#    This program is distributed in the hope that it will be useful,         #
#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
#    GNU General Public License for more details.                            #
#                                                                            #
#    You should have received a copy of the GNU General Public License       #
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
# ========================================================================== #


import os
import asyncio

from typing import Dict

import dbus  # pylint: disable=import-error
import dbus.exceptions  # pylint: disable=import-error

from ...yamlconf.loader import load_yaml_file


# =====
class InfoManager:
    def __init__(
        self,
        meta_path: str,
        extras_path: str,
        loop: asyncio.AbstractEventLoop,
    ) -> None:

        self.__meta_path = meta_path
        self.__extras_path = extras_path

        self.__loop = loop

    async def get_meta(self) -> Dict:
        return (await self.__loop.run_in_executor(None, load_yaml_file, self.__meta_path))

    async def get_extras(self) -> Dict:
        return (await self.__loop.run_in_executor(None, self.__sync_get_extras))

    def __sync_get_extras(self) -> Dict:
        try:
            bus = dbus.SystemBus()

            def is_enabled(daemon: str) -> bool:
                obj = bus.get_object("org.freedesktop.systemd1", "/org/freedesktop/systemd1")
                get_unit_state = obj.get_dbus_method("GetUnitFileState", "org.freedesktop.systemd1.Manager")
                return (get_unit_state(daemon + ".service") in ["enabled", "enabled-runtime", "static", "indirect", "generated"])

        except dbus.exceptions.DBusException:
            is_enabled = (lambda daemon: True)

        extras: Dict[str, Dict] = {}
        for app in os.listdir(self.__extras_path):
            if app[0] != "." and os.path.isdir(os.path.join(self.__extras_path, app)):
                extras[app] = load_yaml_file(os.path.join(self.__extras_path, app, "manifest.yaml"))
                daemon = extras[app].get("daemon", "")
                if isinstance(daemon, str) and daemon.strip():
                    extras[app]["enabled"] = is_enabled(daemon.strip())
        return extras