from sqlalchemy.ext.asyncio import AsyncSession

from app.users import repository
from app.users.models import User


async def get_user_by_id(db: AsyncSession, user_id: int) -> User | None:
    return await repository.get_user_by_id(db, user_id)


async def get_user_by_email(db: AsyncSession, email: str) -> User | None:
    return await repository.get_user_by_email(db, email)


async def get_user_by_username(db: AsyncSession, username: str) -> User | None:
    return await repository.get_user_by_username(db, username)


async def update_user_profile(
    db: AsyncSession,
    user: User,
    *,
    username: str | None = None,
    avatar_url: str | None = None,
) -> User:
    if username is not None:
        user.username = username
    if avatar_url is not None:
        user.avatar_url = avatar_url
    await db.commit()
    await db.refresh(user)
    return user