from uuid import UUID

from sqlalchemy import select, func
from sqlalchemy.ext.asyncio import AsyncSession

from app.models.evidence_image import EvidenceImage, SourceApp, PageType
from app.repositories.base import BaseRepository


class ImageRepository(BaseRepository[EvidenceImage]):
    def __init__(self, session: AsyncSession):
        super().__init__(EvidenceImage, session)

    async def find_by_hash(self, file_hash: str) -> EvidenceImage | None:
        result = await self.session.execute(
            select(EvidenceImage).where(EvidenceImage.file_hash == file_hash)
        )
        return result.scalar_one_or_none()

    async def list_by_case(
        self,
        case_id: UUID,
        source_app: SourceApp | None = None,
        page_type: PageType | None = None,
    ) -> list[EvidenceImage]:
        query = select(EvidenceImage).where(EvidenceImage.case_id == case_id)
        if source_app:
            query = query.where(EvidenceImage.source_app == source_app)
        if page_type:
            query = query.where(EvidenceImage.page_type == page_type)
        query = query.order_by(EvidenceImage.uploaded_at.desc())
        result = await self.session.execute(query)
        return list(result.scalars().all())

    async def count_by_case(self, case_id: UUID) -> int:
        result = await self.session.execute(
            select(func.count()).select_from(EvidenceImage).where(EvidenceImage.case_id == case_id)
        )
        return result.scalar() or 0