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

from module_business.entity.do.master_do import BizUserSite, BizUserWarehouse


class AccessDao:
    """仓库与客户现场数据权限数据库操作层。"""

    @classmethod
    async def user_has_warehouse(
        cls,
        db: AsyncSession,
        user_id: int,
        warehouse_id: int,
        *,
        for_update: bool = False,
    ) -> bool:
        """判断用户是否关联指定仓库。"""
        statement = select(BizUserWarehouse.user_id).where(
            BizUserWarehouse.user_id == user_id,
            BizUserWarehouse.warehouse_id == warehouse_id,
        )
        if for_update:
            statement = statement.with_for_update()
        mapped_user_id = (await db.execute(statement)).scalar_one_or_none()

        return mapped_user_id is not None

    @classmethod
    async def user_has_site(cls, db: AsyncSession, user_id: int, site_id: int) -> bool:
        """判断用户是否关联指定客户现场。"""
        mapped_user_id = (
            await db.execute(
                select(BizUserSite.user_id).where(
                    BizUserSite.user_id == user_id,
                    BizUserSite.site_id == site_id,
                )
            )
        ).scalar_one_or_none()

        return mapped_user_id is not None

    @classmethod
    async def list_user_warehouse_ids(cls, db: AsyncSession, user_id: int) -> list[int]:
        """获取用户关联的仓库 ID 列表。"""
        result = await db.execute(select(BizUserWarehouse.warehouse_id).where(BizUserWarehouse.user_id == user_id))

        return list(result.scalars().all())

    @classmethod
    async def list_user_site_ids(cls, db: AsyncSession, user_id: int) -> list[int]:
        """获取用户关联的客户现场 ID 列表。"""
        result = await db.execute(select(BizUserSite.site_id).where(BizUserSite.user_id == user_id))

        return list(result.scalars().all())

    @classmethod
    async def replace_warehouse_users(cls, db: AsyncSession, warehouse_id: int, user_ids: list[int]) -> None:
        """完整替换指定仓库的用户关联，仅刷新当前事务。"""
        await db.execute(delete(BizUserWarehouse).where(BizUserWarehouse.warehouse_id == warehouse_id))
        db.add_all([BizUserWarehouse(user_id=user_id, warehouse_id=warehouse_id) for user_id in user_ids])
        await db.flush()

    @classmethod
    async def replace_site_users(cls, db: AsyncSession, site_id: int, user_ids: list[int]) -> None:
        """完整替换指定客户现场的用户关联，仅刷新当前事务。"""
        await db.execute(delete(BizUserSite).where(BizUserSite.site_id == site_id))
        db.add_all([BizUserSite(user_id=user_id, site_id=site_id) for user_id in user_ids])
        await db.flush()