import uuid
from typing import Sequence

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

from app.models.drug_shipment import DrugShipment
from app.schemas.drug_shipment import DrugShipmentCreate, DrugShipmentUpdate


async def create_shipment(
    db: AsyncSession,
    study_id: uuid.UUID,
    shipment_in: DrugShipmentCreate,
    created_by: uuid.UUID | None,
) -> DrugShipment:
    shipment = DrugShipment(
        study_id=study_id,
        direction=shipment_in.direction,
        site_name=shipment_in.site_name,
        drug_desc=shipment_in.drug_desc,
        ship_date=shipment_in.ship_date,
        carrier=shipment_in.carrier,
        tracking_no=shipment_in.tracking_no,
        from_party=shipment_in.from_party,
        to_party=shipment_in.to_party,
        status=shipment_in.status,
        remark=shipment_in.remark,
        created_by=created_by,
    )
    db.add(shipment)
    await db.commit()
    await db.refresh(shipment)
    return shipment


async def get_shipment(db: AsyncSession, shipment_id: uuid.UUID) -> DrugShipment | None:
    result = await db.execute(select(DrugShipment).where(DrugShipment.id == shipment_id))
    return result.scalar_one_or_none()


async def list_shipments(
    db: AsyncSession,
    study_id: uuid.UUID,
    site_name: str | None = None,
    tracking_no: str | None = None,
    status: str | None = None,
    skip: int = 0,
    limit: int = 100,
) -> Sequence[DrugShipment]:
    stmt = select(DrugShipment).where(DrugShipment.study_id == study_id)
    if site_name:
        stmt = stmt.where(DrugShipment.site_name.ilike(f"%{site_name}%"))
    if tracking_no:
        stmt = stmt.where(DrugShipment.tracking_no.ilike(f"%{tracking_no}%"))
    if status:
        stmt = stmt.where(DrugShipment.status == status)
    stmt = stmt.order_by(DrugShipment.created_at.desc()).offset(skip).limit(limit)
    result = await db.execute(stmt)
    return result.scalars().all()


async def update_shipment(
    db: AsyncSession, shipment: DrugShipment, shipment_in: DrugShipmentUpdate
) -> DrugShipment:
    update_data = shipment_in.model_dump(exclude_unset=True)
    for key, value in update_data.items():
        setattr(shipment, key, value)
    await db.commit()
    await db.refresh(shipment)
    return shipment


async def delete_shipment(db: AsyncSession, shipment: DrugShipment) -> None:
    await db.delete(shipment)
    await db.commit()