from abc import ABC, abstractmethod
from typing import Tuple

from sqlalchemy import and_, func, or_, select, update
from sqlalchemy.ext.asyncio import AsyncSession

from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import place_photos
from common.enums import ErrorType, MediaStatus
from common.logger import LOG
from common.utils.gtime import GTime


# 사진 CRUD.
class IMediaCRUD(ABC):
    @abstractmethod
    async def list_media(
        self, cdb: AsyncSession, place_id, status=None, unlabeled_only=False, unit_id=None, alt_required=False
    ) -> Tuple[ErrorType, list]:
        pass

    @abstractmethod
    async def apply_vision(self, cdb: AsyncSession, media_id, label, alt_text, confidence, status, ts) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def set_status(self, cdb: AsyncSession, place_id, media_id, status: int, ts) -> Tuple[ErrorType, int]:
        pass


class MediaCRUD(IMediaCRUD):
    async def list_media(
        self, cdb: AsyncSession, place_id, status=None, unlabeled_only: bool = False, unit_id=None, alt_required: bool = False
    ) -> Tuple[ErrorType, list]:
        """사진 목록."""
        try:
            conditions = [place_photos.place_id == place_id, place_photos.deleted == False]  # noqa: E712
            if status is not None:
                conditions.append(place_photos.status == status)
            if unlabeled_only:
                conditions.append(
                    or_(place_photos.alt_text.is_(None), func.btrim(place_photos.alt_text) == "")
                )
            if unit_id is not None:
                conditions.append(place_photos.unit_id == unit_id)
            if alt_required:
                # 공백만 있는 alt 도 빌더에선 '없음'이다 — 같은 기준으로 거른다.
                conditions.append(place_photos.alt_text.is_not(None))
                conditions.append(func.btrim(place_photos.alt_text) != "")
            query = select(place_photos).where(and_(*conditions)).order_by(place_photos.sort_order.asc(), place_photos.created_at.asc())
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            return (err_type, list(rows) if err_type == ErrorType.SUCCESS else [])
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, []

    async def apply_vision(self, cdb: AsyncSession, media_id, label, alt_text, confidence, status: int, ts) -> Tuple[ErrorType, int]:
        """Vision 분석 결과를 반영한다."""
        try:
            query = (
                update(place_photos)
                .where(place_photos.media_id == media_id, place_photos.deleted == False)  # noqa: E712
                .values(label=label, alt_text=alt_text, vision_confidence=confidence, status=status, updated_at=ts)
            )
            return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0

    async def set_status(self, cdb: AsyncSession, place_id, media_id, status: int, ts) -> Tuple[ErrorType, int]:
        """사람이 사진을 승인/반려한다."""
        try:
            query = (
                update(place_photos)
                .where(place_photos.media_id == media_id, place_photos.place_id == place_id, place_photos.deleted == False)  # noqa: E712
                .values(status=status, updated_at=ts)
            )
            return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0