from fastapi import Depends, HTTPException, status from fastapi.security import HTTPBearer from jose import jwt, JWTError from sqlalchemy.orm import Session from app.core.config import settings from app.db.session import get_db from app.models.user import User from app.schemas.user import TokenData security = HTTPBearer() def get_current_user( db: Session = Depends(get_db), token: str = Depends(security) ) -> User: credentials_exception = HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Could not validate credentials", headers={"WWW-Authenticate": "Bearer"}, ) try: payload = jwt.decode( token.credentials, settings.SECRET_KEY, algorithms=[settings.ALGORITHM] ) email: str = payload.get("sub") if email is None: raise credentials_exception token_data = TokenData(email=email) except JWTError: raise credentials_exception user = db.query(User).filter(User.email == token_data.email).first() if user is None: raise credentials_exception return user def get_current_active_user(current_user: User = Depends(get_current_user)) -> User: if not current_user.is_active: raise HTTPException(status_code=400, detail="Inactive user") return current_user def get_current_recruiter(current_user: User = Depends(get_current_active_user)) -> User: from app.models.user import UserRole if current_user.role != UserRole.RECRUITER and current_user.role != UserRole.ADMIN: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail="Not enough permissions" ) return current_user