ovdashboard/api/ovdashboard_api/config.py

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

from io import BytesIO
from logging import getLogger
from typing import Any

from pydantic import BaseModel
from tomli import loads as toml_loads
from tomli_w import dump as toml_dump
from webdav3.exceptions import RemoteResourceNotFound

from .dav_common import caldav_list
from .dav_file import DavFile
from .settings import SETTINGS

_logger = 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"

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

    @classmethod
    async def get(cls) -> "Config":
        """
        Load the configuration instance from the server using `TOML`.
        """

        dav_file = DavFile(SETTINGS.config_path)

        try:
            cfg = cls.parse_obj(
                toml_loads(await dav_file.as_string)
            )

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

            cfg = cls()
            cfg.calendar.aggregates["All Events"] = list(await caldav_list())

            buffer = BytesIO()
            toml_dump(cfg.dict(), buffer)
            buffer.seek(0)
            await dav_file.write(buffer.read())

        return cfg