ovdashboard/api/ovdashboard_api/routers/v1/image.py

"""
Router "image" provides:

- listing image files
- finding image files by name prefix
- getting image files in a uniform format by name prefix
"""

import re
from io import BytesIO
from logging import getLogger
from typing import Iterator

from fastapi import APIRouter, Depends
from fastapi.responses import StreamingResponse
from PIL import Image

from ...core.config import Config, ImageUIConfig, get_config
from ...core.dav_common import webdav_ensure_files, webdav_ensure_path
from ...core.webdav import WebDAV
from ._common import FileNameLister, PrefixFinder, PrefixUnique

_logger = getLogger(__name__)

router = APIRouter(prefix="/image", tags=["image"])

image_lister = FileNameLister(
    path_name="image_dir",
    re=re.compile(
        r"\.(gif|jpe?g|tiff?|png|bmp)$",
        flags=re.IGNORECASE,
    ),
)

image_finder = PrefixFinder(image_lister)
image_unique = PrefixUnique(image_finder)


@router.on_event("startup")
async def start_router() -> None:
    _logger.debug(f"{router.prefix} router starting.")

    if not webdav_ensure_path(await image_lister.remote_path):
        webdav_ensure_files(
            await image_lister.remote_path,
            "img1.jpg",
            "img2.jpg",
            "img3.jpg",
        )


@router.get(
    "/list",
    response_model=list[str],
    responses=image_lister.responses,
)
async def list_images(
    names: Iterator[str] = Depends(image_lister),
) -> list[str]:
    return list(names)


@router.get(
    "/find/{prefix}",
    response_model=list[str],
    responses=image_finder.responses,
)
async def find_images(
    names: Iterator[str] = Depends(image_finder),
) -> list[str]:
    return list(names)


@router.get(
    "/get/{prefix}",
    response_class=StreamingResponse,
    responses=image_unique.responses,
)
async def get_image(
    prefix: str,
    name: str = Depends(image_unique),
) -> StreamingResponse:
    cfg = await get_config()
    img = Image.open(
        BytesIO(await WebDAV.read_bytes(f"{await image_lister.remote_path}/{name}"))
    )

    img_buffer = BytesIO()
    img.save(img_buffer, **cfg.image.save_params)
    img_buffer.seek(0)

    return StreamingResponse(
        content=img_buffer,
        media_type="image/jpeg",
        headers={"Content-Disposition": f"filename={prefix}.jpg"},
    )


@router.get(
    "/config",
    response_model=ImageUIConfig,
)
async def get_ui_config(
    cfg: Config = Depends(get_config),
) -> ImageUIConfig:
    return cfg.image