from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.ext.asyncio import AsyncSession from app.auth.service import get_current_user from app.database.session import get_db from app.users.models import User from app.users.schemas import UserProfileUpdate, UserRead, UserSearchRead from app.users.service import get_user_by_id, get_user_by_username, search_users_by_username, update_user_profile router = APIRouter(prefix="/users", tags=["users"]) @router.get("/me", response_model=UserRead) async def read_me(current_user: User = Depends(get_current_user)) -> UserRead: return current_user @router.get("/search", response_model=list[UserSearchRead]) async def search_users( query: str, limit: int = 20, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_current_user), ) -> list[UserSearchRead]: if len(query.strip().lstrip("@")) < 2: return [] users = await search_users_by_username( db, query=query, limit=limit, exclude_user_id=current_user.id, ) return users @router.get("/{user_id}", response_model=UserRead) async def read_user(user_id: int, db: AsyncSession = Depends(get_db), _current_user: User = Depends(get_current_user)) -> UserRead: user = await get_user_by_id(db, user_id) if not user: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="User not found") return user @router.put("/profile", response_model=UserRead) async def update_profile( payload: UserProfileUpdate, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_current_user), ) -> UserRead: if payload.username and payload.username != current_user.username: username_owner = await get_user_by_username(db, payload.username) if username_owner: raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Username already taken") updated = await update_user_profile( db, current_user, username=payload.username, avatar_url=payload.avatar_url, ) return updated