ovdashboard/api/ovdashboard_api/core/config.py

"""
Python representation of the "config.txt" file inside the WebDAV directory.
"""

import logging
import tomllib
from typing import Any

from pydantic import BaseModel
from webdav3.exceptions import RemoteResourceNotFound

# from .caldav import CalDAV
from .settings import SETTINGS
from .webdav import WebDAV

_logger = logging.getLogger(__name__)


class TickerUIConfig(BaseModel):
    """
    Configuration for how the UI displays the ticker content.
    """

    color: str = "primary"


class TickerConfig(TickerUIConfig):
    """
    Section "[ticker]" in "config.txt".

    Combined configuration for the ticker.
    """

    file_name: str = "ticker"
    separator: str = " +++ "
    comment_marker: str = "#"


class ImageUIConfig(BaseModel):
    """
    Configuration for how the UI displays the image carousel.
    """

    height: int = 300
    contain: bool = False
    speed: int = 10000


class ImageConfig(ImageUIConfig):
    """
    Sections "[image*]" in "config.txt".
    """

    mode: str = "RGB"
    save_params: dict[str, Any] = {
        "format": "JPEG",
        "quality": 85,
    }


class CalendarUIConfig(BaseModel):
    """
    Configuration for how the UI displays the calendar carousel.
    """

    speed: int = 10000


class CalendarConfig(CalendarUIConfig):
    """
    Sections "[calendar*]" in "config.txt".
    """

    future_days: int = 365
    aggregates: dict[str, list[str]] = {}


class ServerUIConfig(BaseModel):
    """
    Section "[server]" in "config.txt".
    """

    name: str = "OEKZident"
    host: str = "https://oekzident.de"


class LogoUIConfig(BaseModel):
    """
    Section "[logo]" in "config.txt".
    """

    above: str = "Technisches Hilfswerk"
    below: str = "OV Musterstadt"


class Config(BaseModel):
    """
    Main representation of "config.txt".
    """

    image_dir: str = "image"
    text_dir: str = "text"
    file_dir: str = "file"

    logo: LogoUIConfig = LogoUIConfig()
    image: ImageConfig = ImageConfig()
    server: ServerUIConfig = ServerUIConfig()
    ticker: TickerConfig = TickerConfig()
    calendar: CalendarConfig = CalendarConfig()


async def get_config() -> Config:
    """
    Load the configuration instance from the server using `TOML`.
    """

    try:
        cfg_str = await WebDAV.read_str(SETTINGS.webdav.config_filename)
        cfg = Config.model_validate(tomllib.loads(cfg_str))

    except RemoteResourceNotFound:
        _logger.warning(
            f"Config file {SETTINGS.webdav.config_filename!r} not found, creating ..."
        )

        cfg = Config()
        # cfg.calendar.aggregates["All Events"] = list(await CalDAV.calendars)

        # await WebDAV.write_str(
        #     SETTINGS.webdav.config_filename,
        #     tomli_w.dumps(cfg.model_dump()),
        # )

    return cfg