"""Конфигурация — pydantic models + YAML loader. Структура YAML: broker: host: localhost port: 1883 username_env: MQTT_USERNAME password_env: MQTT_PASSWORD instances: - name: livingroom_tv zmq_endpoint: tcp://127.0.0.1:5555 default_layout: quad ha_discovery: enabled: true prefix: homeassistant device_name: "CUDA Grid Composer" http: host: 0.0.0.0 port: 8080 log: level: INFO """ from __future__ import annotations import os import re from pathlib import Path from typing import Any, Self import yaml from pydantic import BaseModel, Field, field_validator _ENV_PATTERN = re.compile(r"\$\{(\w+)(?::-([^}]*))?\}") def _expand_env(obj: Any) -> Any: """Recursively replace ${VAR} and ${VAR:-default} в string values через os.environ. Используется при load YAML config — secrets вроде API tokens хранятся в .env (gitignored) и interpolate'ятся при start.""" if isinstance(obj, str): return _ENV_PATTERN.sub( lambda m: os.environ.get(m.group(1), m.group(2) or ""), obj, ) if isinstance(obj, dict): return {k: _expand_env(v) for k, v in obj.items()} if isinstance(obj, list): return [_expand_env(v) for v in obj] return obj class BrokerCfg(BaseModel): host: str = "localhost" port: int = 1883 client_id: str = "cuda-grid-controller" username_env: str | None = None password_env: str | None = None keepalive_sec: int = 30 @property def username(self) -> str | None: return os.environ.get(self.username_env) if self.username_env else None @property def password(self) -> str | None: return os.environ.get(self.password_env) if self.password_env else None class InstanceCfg(BaseModel): """Один FFmpeg pipeline = одна cuda_grid filter instance.""" name: str = Field(description="уникальное имя — становится частью HA entity ID") zmq_endpoint: str = Field( description="ZMQ endpoint видео-pipeline (cuda_grid + overlay filters)" ) audio_zmq_endpoint: str | None = Field( default=None, description="ZMQ endpoint отдельного audio sidecar (Phase 5d split-process). " "None = audio_set/intercom вызывают video pipeline (Phase 5a single source)", ) default_layout: str = "quad" filter_target: str = Field( default="Parsed_cuda_grid_0", description="Default ZMQ target для process_command (используется reload_icon и др.)", ) overlay_filter_targets: list[str] = Field( default_factory=list, description="Список cuda_grid instances в pipeline (e.g. quad/single/mpp). " "Overlay add/remove/clear broadcast ко всем — иначе при layout switch " "overlays исчезают (каждый cuda_grid имеет own overlay state). " "Если empty — fallback к [filter_target].", ) output_rtsp_url: str | None = Field( default=None, description="URL куда pipeline push'ит composed stream — controller read'ит для snapshot/preview endpoints", ) audio_sources: list["AudioSourceCfg"] = Field( default_factory=list, description="Audio sources для astreamselect switching (Phase 5b)", ) audio_filter_target: str = Field( default="astreamselect@as", description="Target filter для ZMQ команд переключения audio (должен соответствовать pipeline filter_complex)", ) music_volume_target: str = Field( default="volume@music", description="Target filter для управления громкостью music chain (Phase 5c ducking)", ) audio_output_volume_target: str = Field( default="volume@output_audio", description="ZMQ target volume filter в video pipeline (после astreamselect) для UI-mute audio. " "0 = mute, 1 = unmute. Phase 7 filter pipeline.", ) audio_input_select_target: str = Field( default="astreamselect@audio_input", description="ZMQ target astreamselect filter — выбор audio source в video pipeline. " "map=0 → live-audio (sidecar), map=1 → anullsrc (decouple от jittery RTSP). " "UI mute → map=1 + volume=0; unmute → map=0 + volume=1.", ) intercom_volume_target: str = Field( default="volume@intercom", description="Target filter для управления громкостью intercom (Phase 5c)", ) music_ducked_volume: float = Field( default=0.2, ge=0.0, le=1.0, description="Громкость music когда intercom активен (0.2 = -14 dB)", ) layout_filter_target: str = Field( default="cuda_grid@cg", description="ZMQ target cuda_grid filter (Phase 7) для runtime layout switching " "(set_layout ) и cell remapping (cell_map ).", ) main_cam_filter_target: str = Field( default="cuda_grid@cg", description="ZMQ target для cell_map command — single layout main camera. " "Phase 7: тот же cuda_grid@cg (был streamselect@main_cam).", ) mpp_main_filter_target: str = Field( default="cuda_grid@cg", description="ZMQ target для cell_map command — main_plus_preview main camera. " "Phase 7: тот же cuda_grid@cg (был streamselect@mpp_main).", ) max_cells: int = Field( default=4, ge=1, le=16, description="Кол-во input pads cuda_grid filter (Phase 7 max_cells option). " "Должно совпадать с max_cells в pipeline filter_complex.", ) layout_map: dict[str, int] = Field( default_factory=lambda: {"quad": 0, "single": 1, "main_plus_preview": 2}, description="Список доступных layouts (Phase 7: ключи используются для validation, " "значения игнорируются — cuda_grid resolve'ит layout по имени).", ) snapshot_history: "SnapshotHistoryCfg" = Field( default_factory=lambda: SnapshotHistoryCfg(), description="Periodic snapshot capture в shared volume", ) class SnapshotHistoryCfg(BaseModel): enabled: bool = False interval_sec: int = Field(default=60, ge=5, le=3600) keep_last: int = Field(default=120, ge=1, le=10000, description="Сколько последних snapshots держать (FIFO eviction)") dir: str = Field(default="/var/lib/cuda-grid/snapshots", description="Базовая директория; instance-name становится поддиректорией") class AudioSourceCfg(BaseModel): """Описание audio source в порядке как они добавлены в pipeline -i ...""" name: str = Field(description="Уникальное имя для API (e.g. 'europa_plus')") index: int = Field(ge=0, description="Index в astreamselect inputs (соответствует порядку -i в pipeline)") label: str | None = Field(default=None, description="UI-friendly label (default = name)") @field_validator("name") @classmethod def name_alnum(cls, v: str) -> str: if not v.replace("_", "").isalnum(): raise ValueError(f"instance name '{v}' must be alphanumeric + underscore") return v class HaDiscoveryCfg(BaseModel): enabled: bool = True prefix: str = "homeassistant" device_name: str = "CUDA Grid Composer" device_identifier: str = "cuda_grid_controller" class HttpCfg(BaseModel): host: str = "0.0.0.0" port: int = 8080 class LogCfg(BaseModel): level: str = "INFO" class Config(BaseModel): broker: BrokerCfg = BrokerCfg() instances: list[InstanceCfg] = [] ha_discovery: HaDiscoveryCfg = HaDiscoveryCfg() http: HttpCfg = HttpCfg() log: LogCfg = LogCfg() # Frigate bridge — late import чтобы избежать circular dep frigate: dict | None = None # parsed в FrigateBridgeCfg при runtime # Dynamic overlays (charts/chats) — late import тоже dynamic_overlays: dict | None = None # parsed в DynamicRenderer cfg # Stream watchdog — Phase 1 resilience (issue #3) watchdog: dict | None = None # parsed в WatchdogCfg при runtime icon_dir: str = Field( default="/var/lib/cuda-grid/icons", description="Shared volume куда controller пишет dynamic PNG; filter (`icon_dir=` option) читает оттуда", ) @classmethod def from_yaml(cls, path: Path | str) -> Self: with open(path) as f: data = yaml.safe_load(f) or {} data = _expand_env(data) return cls.model_validate(data)