import uuid
from typing import Sequence

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

from app.models.acknowledgement import Acknowledgement, AcknowledgementType


async def create(db: AsyncSession, ack: Acknowledgement, *, commit: bool = True) -> Acknowledgement:
    db.add(ack)
    if commit:
        await db.commit()
        await db.refresh(ack)
    return ack


async def get_existing(
    db: AsyncSession,
    distribution_id: uuid.UUID,
    user_id: uuid.UUID,
    ack_type: AcknowledgementType,
) -> Acknowledgement | None:
    result = await db.execute(
        select(Acknowledgement).where(
            Acknowledgement.distribution_id == distribution_id,
            Acknowledgement.user_id == user_id,
            Acknowledgement.ack_type == ack_type,
        )
    )
    return result.scalar_one_or_none()


async def list_by_distribution(db: AsyncSession, distribution_id: uuid.UUID) -> Sequence[Acknowledgement]:
    result = await db.execute(select(Acknowledgement).where(Acknowledgement.distribution_id == distribution_id))
    return result.scalars().all()