ledgrab/server/src/wled_controller/api/dependencies.py

"""Dependency injection for API routes.

Uses a registry dict instead of individual module-level globals.
All getter function signatures remain unchanged for FastAPI Depends() compatibility.
"""

from typing import Any, Dict, TypeVar

from wled_controller.core.processing.processor_manager import ProcessorManager
from wled_controller.storage.database import Database
from wled_controller.storage import DeviceStore
from wled_controller.storage.template_store import TemplateStore
from wled_controller.storage.postprocessing_template_store import PostprocessingTemplateStore
from wled_controller.storage.picture_source_store import PictureSourceStore
from wled_controller.storage.output_target_store import OutputTargetStore
from wled_controller.storage.color_strip_store import ColorStripStore
from wled_controller.storage.audio_source_store import AudioSourceStore
from wled_controller.storage.audio_template_store import AudioTemplateStore
from wled_controller.storage.value_source_store import ValueSourceStore
from wled_controller.storage.automation_store import AutomationStore
from wled_controller.storage.scene_preset_store import ScenePresetStore
from wled_controller.storage.sync_clock_store import SyncClockStore
from wled_controller.storage.color_strip_processing_template_store import (
    ColorStripProcessingTemplateStore,
)
from wled_controller.storage.gradient_store import GradientStore
from wled_controller.storage.weather_source_store import WeatherSourceStore
from wled_controller.storage.asset_store import AssetStore
from wled_controller.core.automations.automation_engine import AutomationEngine
from wled_controller.core.weather.weather_manager import WeatherManager
from wled_controller.core.backup.auto_backup import AutoBackupEngine
from wled_controller.core.processing.sync_clock_manager import SyncClockManager
from wled_controller.core.update.update_service import UpdateService
from wled_controller.storage.home_assistant_store import HomeAssistantStore
from wled_controller.core.home_assistant.ha_manager import HomeAssistantManager

T = TypeVar("T")

# Central dependency registry — keyed by type or string label
_deps: Dict[str, Any] = {}


def _get(key: str, label: str) -> Any:
    """Get a dependency by key, raising RuntimeError if not initialized."""
    dep = _deps.get(key)
    if dep is None:
        raise RuntimeError(f"{label} not initialized")
    return dep


# ── Typed getters (unchanged signatures for FastAPI Depends()) ──────────


def get_device_store() -> DeviceStore:
    return _get("device_store", "Device store")


def get_template_store() -> TemplateStore:
    return _get("template_store", "Template store")


def get_pp_template_store() -> PostprocessingTemplateStore:
    return _get("pp_template_store", "Postprocessing template store")


def get_picture_source_store() -> PictureSourceStore:
    return _get("picture_source_store", "Picture source store")


def get_output_target_store() -> OutputTargetStore:
    return _get("output_target_store", "Output target store")


def get_color_strip_store() -> ColorStripStore:
    return _get("color_strip_store", "Color strip store")


def get_audio_source_store() -> AudioSourceStore:
    return _get("audio_source_store", "Audio source store")


def get_audio_template_store() -> AudioTemplateStore:
    return _get("audio_template_store", "Audio template store")


def get_value_source_store() -> ValueSourceStore:
    return _get("value_source_store", "Value source store")


def get_processor_manager() -> ProcessorManager:
    return _get("processor_manager", "Processor manager")


def get_automation_store() -> AutomationStore:
    return _get("automation_store", "Automation store")


def get_scene_preset_store() -> ScenePresetStore:
    return _get("scene_preset_store", "Scene preset store")


def get_automation_engine() -> AutomationEngine:
    return _get("automation_engine", "Automation engine")


def get_auto_backup_engine() -> AutoBackupEngine:
    return _get("auto_backup_engine", "Auto-backup engine")


def get_sync_clock_store() -> SyncClockStore:
    return _get("sync_clock_store", "Sync clock store")


def get_sync_clock_manager() -> SyncClockManager:
    return _get("sync_clock_manager", "Sync clock manager")


def get_cspt_store() -> ColorStripProcessingTemplateStore:
    return _get("cspt_store", "Color strip processing template store")


def get_gradient_store() -> GradientStore:
    return _get("gradient_store", "Gradient store")


def get_weather_source_store() -> WeatherSourceStore:
    return _get("weather_source_store", "Weather source store")


def get_weather_manager() -> WeatherManager:
    return _get("weather_manager", "Weather manager")


def get_asset_store() -> AssetStore:
    return _get("asset_store", "Asset store")


def get_ha_store() -> HomeAssistantStore:
    return _get("ha_store", "Home Assistant store")


def get_ha_manager() -> HomeAssistantManager:
    return _get("ha_manager", "Home Assistant manager")


def get_database() -> Database:
    return _get("database", "Database")


def get_update_service() -> UpdateService:
    return _get("update_service", "Update service")


# ── Event helper ────────────────────────────────────────────────────────


def fire_entity_event(entity_type: str, action: str, entity_id: str) -> None:
    """Fire an entity_changed event via the ProcessorManager event bus.

    Args:
        entity_type: e.g. "device", "output_target", "color_strip_source"
        action: "created", "updated", or "deleted"
        entity_id: The entity's unique ID
    """
    pm = _deps.get("processor_manager")
    if pm is not None:
        pm.fire_event(
            {
                "type": "entity_changed",
                "entity_type": entity_type,
                "action": action,
                "id": entity_id,
            }
        )


# ── Initialization ──────────────────────────────────────────────────────


def init_dependencies(
    device_store: DeviceStore,
    template_store: TemplateStore,
    processor_manager: ProcessorManager,
    database: Database | None = None,
    pp_template_store: PostprocessingTemplateStore | None = None,
    picture_source_store: PictureSourceStore | None = None,
    output_target_store: OutputTargetStore | None = None,
    color_strip_store: ColorStripStore | None = None,
    audio_source_store: AudioSourceStore | None = None,
    audio_template_store: AudioTemplateStore | None = None,
    value_source_store: ValueSourceStore | None = None,
    automation_store: AutomationStore | None = None,
    scene_preset_store: ScenePresetStore | None = None,
    automation_engine: AutomationEngine | None = None,
    auto_backup_engine: AutoBackupEngine | None = None,
    sync_clock_store: SyncClockStore | None = None,
    sync_clock_manager: SyncClockManager | None = None,
    cspt_store: ColorStripProcessingTemplateStore | None = None,
    gradient_store: GradientStore | None = None,
    weather_source_store: WeatherSourceStore | None = None,
    weather_manager: WeatherManager | None = None,
    update_service: UpdateService | None = None,
    asset_store: AssetStore | None = None,
    ha_store: HomeAssistantStore | None = None,
    ha_manager: HomeAssistantManager | None = None,
):
    """Initialize global dependencies."""
    _deps.update(
        {
            "database": database,
            "device_store": device_store,
            "template_store": template_store,
            "processor_manager": processor_manager,
            "pp_template_store": pp_template_store,
            "picture_source_store": picture_source_store,
            "output_target_store": output_target_store,
            "color_strip_store": color_strip_store,
            "audio_source_store": audio_source_store,
            "audio_template_store": audio_template_store,
            "value_source_store": value_source_store,
            "automation_store": automation_store,
            "scene_preset_store": scene_preset_store,
            "automation_engine": automation_engine,
            "auto_backup_engine": auto_backup_engine,
            "sync_clock_store": sync_clock_store,
            "sync_clock_manager": sync_clock_manager,
            "cspt_store": cspt_store,
            "gradient_store": gradient_store,
            "weather_source_store": weather_source_store,
            "weather_manager": weather_manager,
            "update_service": update_service,
            "asset_store": asset_store,
            "ha_store": ha_store,
            "ha_manager": ha_manager,
        }
    )