import uuid

from fastapi import Depends

from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import place_photos, places
from common.enums import DBWRType, ErrorType, MediaStatus, SourceType
from common.models.gmodel import UserInfo
from crud.media_crud import IMediaCRUD, MediaCRUD
from crud.place_crud import PlaceCRUD
from router.v1.media.protocol import MediaData, Res_MediaList


def _is_publishable(row) -> bool:
    """이 사진이 지금 사이트에 실릴 수 있는가."""
    return row.status == MediaStatus.APPROVED.value and bool((row.alt_text or "").strip()) and bool((row.url or "").strip())


class MediaService:
    """사진 조회."""

    def __init__(self, crud: IMediaCRUD = Depends(MediaCRUD), place_crud: PlaceCRUD = Depends(PlaceCRUD)):
        self.crud = crud
        self.place_crud = place_crud

    # 사업장 로드(회사 스코프)
    async def _load_place(self, user_info: UserInfo, place_id: str):
        err_type, place = await DB_SESSION_MNG.execute_lambda(
            places.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.place_crud.get_place(s, uuid.UUID(user_info.user_id), uuid.UUID(place_id)),
        )
        if err_type != ErrorType.SUCCESS:
            return ErrorType.PLACE_NOT_FOUND, None
        return ErrorType.SUCCESS, place

    # 조회
    async def list_media(self, user_info: UserInfo, place_id: str, unit_id=None, publishable_only: bool = False) -> Res_MediaList:
        """사진 목록."""
        res = Res_MediaList()
        err_type, _place = await self._load_place(user_info, place_id)
        if err_type != ErrorType.SUCCESS:
            res.result.SetResult(err_type)
            return res

        # publishable_only 의 승인 조건만 CRUD 에 넘기고, alt 조건은 목록 계산과 함께 아래에서 건다.
        status = MediaStatus.APPROVED.value if publishable_only else None
        list_err, rows = await DB_SESSION_MNG.execute_lambda(
            place_photos.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.crud.list_media(
                s, uuid.UUID(place_id), status, False, unit_id, publishable_only
            ),
        )
        if list_err != ErrorType.SUCCESS:
            res.result.SetResult(list_err)
            return res

        items = []
        for r in rows:
            data = MediaData.model_validate(r)
            data.publishable = _is_publishable(r)
            items.append(data)
        res.media = items
        res.publishable = sum(1 for x in items if x.publishable)
        # 사람 확인 큐에 남은 수 — 관리 화면의 '검토할 것' 배지.
        res.pending_review = sum(1 for r in rows if r.status == MediaStatus.PENDING_REVIEW.value)
        # 재게시 권리(1-2)가 '불가'로 결론나면 통째로 빠질 사진 수.
        res.crawled = sum(1 for r in rows if r.source_type == SourceType.CRAWL.value)
        return res