from fastapi import APIRouter, Depends, HTTPException, status from sqlalchemy.orm import Session from typing import List from ...core.database import get_db from ...core.security import get_password_hash from ...models.models import User from ...schemas import UserResponse, UserUpdate, MessageResponse from ...api.dependencies import get_current_active_user, get_admin_user router = APIRouter() @router.get("/me", response_model=UserResponse) async def get_current_user_profile( current_user: User = Depends(get_current_active_user) ): """Get current user's profile""" return current_user @router.put("/me", response_model=UserResponse) async def update_current_user_profile( user_update: UserUpdate, current_user: User = Depends(get_current_active_user), db: Session = Depends(get_db) ): """Update current user's profile""" update_data = user_update.model_dump(exclude_unset=True) # Check email uniqueness if email is being updated if 'email' in update_data and update_data['email'] != current_user.email: existing_user = db.query(User).filter(User.email == update_data['email']).first() if existing_user: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail="Email already registered" ) for field, value in update_data.items(): setattr(current_user, field, value) db.commit() db.refresh(current_user) return current_user @router.get("/", response_model=List[UserResponse]) async def list_users( skip: int = 0, limit: int = 100, current_user: User = Depends(get_admin_user), db: Session = Depends(get_db) ): """List all users (admin only)""" users = db.query(User).offset(skip).limit(limit).all() return users @router.get("/{user_id}", response_model=UserResponse) async def get_user( user_id: int, current_user: User = Depends(get_admin_user), db: Session = Depends(get_db) ): """Get user by ID (admin only)""" user = db.query(User).filter(User.id == user_id).first() if not user: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="User not found" ) return user @router.put("/{user_id}", response_model=UserResponse) async def update_user( user_id: int, user_update: UserUpdate, current_user: User = Depends(get_admin_user), db: Session = Depends(get_db) ): """Update user by ID (admin only)""" user = db.query(User).filter(User.id == user_id).first() if not user: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="User not found" ) update_data = user_update.model_dump(exclude_unset=True) for field, value in update_data.items(): setattr(user, field, value) db.commit() db.refresh(user) return user @router.delete("/{user_id}", response_model=MessageResponse) async def delete_user( user_id: int, current_user: User = Depends(get_admin_user), db: Session = Depends(get_db) ): """Delete user (admin only)""" user = db.query(User).filter(User.id == user_id).first() if not user: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="User not found" ) db.delete(user) db.commit() return {"message": "User deleted successfully"}