from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from app.database.database import get_db
from app.models import User
from app.schemas.user import UserResponse, UserUpdate
from app.services.auth import verify_token

router = APIRouter(prefix="/api/users", tags=["users"])


def get_current_user(token: str, db: Session = Depends(get_db)) -> User:
    user_id = verify_token(token)
    if user_id is None:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail="Invalid token",
        )
    user = db.query(User).filter(User.id == user_id).first()
    if not user:
        raise HTTPException(status_code=404, detail="User not found")
    return user


@router.get("/me", response_model=UserResponse)
def get_current_user_profile(token: str, db: Session = Depends(get_db)):
    user = get_current_user(token, db)
    return user


@router.put("/me", response_model=UserResponse)
def update_user_profile(token: str, user_update: UserUpdate, db: Session = Depends(get_db)):
    user = get_current_user(token, db)

    for field, value in user_update.dict(exclude_unset=True).items():
        setattr(user, field, value)

    db.commit()
    db.refresh(user)
    return user


@router.get("/{user_id}", response_model=UserResponse)
def get_user_by_id(user_id: int, db: Session = Depends(get_db)):
    user = db.query(User).filter(User.id == user_id).first()
    if not user:
        raise HTTPException(status_code=404, detail="User not found")
    return user