Modus/src/services/config.py

"""
Centralized configuration service with file watching and dynamic reloads.

This service encapsulates the logic for loading the application's JSON config,
watching for file changes, and notifying registered listeners when the config
changes. It is implemented as a singleton and intended to be reused by any
module that needs dynamic configuration.
"""

from typing import Any, Callable, Dict, List, Optional

from fabric.utils import Gio, GLib, logger, os
from tomlkit import document as toml_document
from tomlkit import dump as toml_dump
from tomlkit import load as toml_load
from tomlkit.items import Array, Bool, Integer, String

from utils.utils import toml_file


class ConfigService:
    """Singleton service handling config state and reload notifications."""

    _instance: Optional["ConfigService"] = None

    def __new__(cls):
        if cls._instance is None:
            cls._instance = super().__new__(cls)
        return cls._instance

    def __init__(self):
        if getattr(self, "_initialized", False):
            return

        self._initialized = True
        self._config: Any = {}
        self._reload_callbacks: List[
            Callable[[Dict[str, Any], Dict[str, Any]], None]
        ] = []
        self._config_file: str = toml_file("config.toml")
        self._monitors: List[Gio.FileMonitor] = []
        self._reload_pending: bool = False
        self.RELOAD_DELAY_MS: int = 100

        self._load_config()
        self._last_notified_config = self._config.copy()
        self._setup_monitors()

    def get(self, key: str, default: Any = None) -> Any:
        return self._config.get(key, default)

    def get_all(self) -> Dict[str, Any]:
        return self._config.copy()

    def has_changed(self, key: str, old_config: Dict[str, Any]) -> bool:
        return self._config.get(key) != old_config.get(key)

    def set(self, key: str, value: Any) -> None:
        """Set a config value (local state only)."""
        self._config[key] = value

    def register_reload_callback(
        self, callback: Callable[[Dict[str, Any], Dict[str, Any]], None]
    ) -> None:
        if callback not in self._reload_callbacks:
            self._reload_callbacks.append(callback)

    @staticmethod
    def _dict_to_toml(d: dict) -> Any:
        doc = toml_document()
        for k, v in d.items():
            if isinstance(v, bool):
                doc[k] = Bool(v)
            elif isinstance(v, int):
                doc[k] = Integer(v)
            elif isinstance(v, str):
                doc[k] = String(v, String.STANDARD)
            elif isinstance(v, list):
                arr = Array()
                for item in v:
                    if isinstance(item, bool):
                        arr.append(Bool(item))
                    elif isinstance(item, int):
                        arr.append(Integer(item))
                    elif isinstance(item, str):
                        arr.append(String(item, String.STANDARD))
                    else:
                        arr.append(item)
                arr.set_multiline(True) if len(v) > 1 else None
                doc[k] = arr
            else:
                doc[k] = v
        return doc

    def _load_config(self) -> None:
        try:
            if os.path.exists(self._config_file):
                with open(self._config_file) as f:
                    self._config = toml_load(f)
            else:
                try:
                    from utils.constants import DEFAULT

                    os.makedirs(os.path.dirname(self._config_file), exist_ok=True)
                    self._config = self._dict_to_toml(DEFAULT)
                    with open(self._config_file, "w") as f:
                        toml_dump(self._config, f)
                    logger.info(
                        f"[ConfigService] Generated default config at {self._config_file}"
                    )
                except ImportError as e:
                    logger.error(f"[ConfigService] Failed to import defaults: {e}")
                    self._config = {}
        except Exception as e:
            logger.error(f"[ConfigService] Failed to load config: {e}")
            self._config = {}

    def _setup_monitors(self) -> None:
        files_to_watch = [self._config_file, os.path.dirname(self._config_file)]
        for file_path in files_to_watch:
            if not os.path.exists(file_path):
                continue
            try:
                gio_file = Gio.File.new_for_path(file_path)
                monitor = gio_file.monitor_file(Gio.FileMonitorFlags.NONE, None)
                monitor.connect("changed", self._on_file_changed, file_path)
                self._monitors.append(monitor)
            except Exception as e:
                logger.error(f"[ConfigService] Failed to monitor {file_path}: {e}")

    def _on_file_changed(self, monitor, file, _other_file, event_type, file_path: str):
        # Trigger on various change events to be more robust across editors/OSs
        valid_events = [
            Gio.FileMonitorEvent.CHANGES_DONE_HINT,
            Gio.FileMonitorEvent.CHANGED,
            Gio.FileMonitorEvent.CREATED,
            Gio.FileMonitorEvent.DELETED,
        ]
        if event_type in valid_events and not self._reload_pending:
            self._reload_pending = True
            GLib.timeout_add(self.RELOAD_DELAY_MS, self._reload_config)

    def _reload_config(self) -> bool:
        try:
            self._reload_pending = False
            old_config = self._last_notified_config.copy()
            self._load_config()

            # Always notify listeners on reload request to be safe
            for callback in list(self._reload_callbacks):
                try:
                    callback(self._config, old_config)
                except Exception as e:
                    logger.error(f"[ConfigService] Callback failed: {e}")

            self._last_notified_config = self._config.copy()
            return False
        except Exception as e:
            logger.error(f"[ConfigService] Failed to reload configuration: {e}")
            self._reload_pending = False
            return False


_service: Optional[ConfigService] = None


def start_config_service() -> ConfigService:
    global _service
    if _service is None:
        _service = ConfigService()
    return _service


# Convenience helpers for concise usage
def config() -> ConfigService:
    """Get the singleton service (concise alias)."""
    return start_config_service()


def get_config(key: str, default: Any = None) -> Any:
    """Get a single config value from the singleton service."""
    return start_config_service().get(key, default)


def get_config_all() -> Dict[str, Any]:
    """Get the entire current configuration state."""
    return start_config_service().get_all()


def on_config_change(
    callback: Callable[[Dict[str, Any], Dict[str, Any]], None],
) -> None:
    """Register a reload callback on the singleton service."""
    start_config_service().register_reload_callback(callback)