from __future__ import annotations

from typing import TYPE_CHECKING, Protocol
from uuid import UUID

from sqlalchemy import select

from core.db.base_repository import BaseRepository
from models.automation_jobs import AutomationJob

if TYPE_CHECKING:
    from sqlalchemy.ext.asyncio import AsyncSession


class MemoryRepositoryLike(Protocol):
    async def get_job_by_id_and_owner(
        self, *, job_id: UUID, owner_id: UUID
    ) -> AutomationJob | None: ...


class MemoryRepository(BaseRepository[AutomationJob]):
    def __init__(self, session: AsyncSession) -> None:
        super().__init__(session=session, model=AutomationJob)

    async def get_job_by_id_and_owner(
        self, *, job_id: UUID, owner_id: UUID
    ) -> AutomationJob | None:
        stmt = (
            select(AutomationJob)
            .where(AutomationJob.id == job_id)
            .where(AutomationJob.owner_id == owner_id)
            .where(AutomationJob.deleted_at.is_(None))
        )
        result = await self._session.execute(stmt)
        return result.scalar_one_or_none()