from typing import Annotated
from uuid import UUID

from fastapi import Depends
from sqlalchemy.ext.asyncio.session import AsyncSession
from sqlmodel import select

from ..models.user import User
from ..providers.db_provider import get_session


class UserRepository:
    def __init__(self, session: Annotated[AsyncSession, Depends(get_session)]) -> None:
        self.session = session

    async def get_by_id(self, user_id: UUID):
        statement = select(User).where(User.id == user_id)
        result = await self.session.execute(statement)
        return result.scalar_one_or_none()

    async def get_by_email(self, email: str):
        statement = select(User).where(User.email == email)
        result = await self.session.execute(statement)
        return result.scalar_one_or_none()

    async def save(self, user: User) -> None:
        """
        Save a user to the database. If a user with that ID already exists, do an upsert.
        """
        existing = await self.get_by_id(user.id)
        if not existing:
            existing = user

        for key, value in user.model_dump(exclude_unset=True).items():
            setattr(user, key, value)

        try:
            self.session.add(user)
            await self.session.commit()
            await self.session.refresh(user)
        except Exception as e:
            await self.session.rollback()
            raise e