|
| 1 | +"""Authentication endpoints.""" |
| 2 | +from fastapi import APIRouter, Depends, HTTPException, status, Request |
| 3 | +from sqlalchemy.orm import Session |
| 4 | +from ....core.database import get_db |
| 5 | +from ....schemas.user import UserCreate, UserLogin, UserResponse, Token, TokenRefresh |
| 6 | +from ....services.auth_service import AuthService |
| 7 | + |
| 8 | +router = APIRouter() |
| 9 | + |
| 10 | + |
| 11 | +@router.post("/register", response_model=UserResponse, status_code=status.HTTP_201_CREATED) |
| 12 | +def register(user_data: UserCreate, db: Session = Depends(get_db)): |
| 13 | + """Register a new user.""" |
| 14 | + auth_service = AuthService(db) |
| 15 | + try: |
| 16 | + user = auth_service.create_user( |
| 17 | + email=user_data.email, |
| 18 | + password=user_data.password, |
| 19 | + first_name=user_data.first_name, |
| 20 | + last_name=user_data.last_name |
| 21 | + ) |
| 22 | + return user |
| 23 | + except ValueError as e: |
| 24 | + raise HTTPException(status_code=400, detail=str(e)) |
| 25 | + |
| 26 | + |
| 27 | +@router.post("/login", response_model=Token) |
| 28 | +def login( |
| 29 | + credentials: UserLogin, |
| 30 | + request: Request, |
| 31 | + db: Session = Depends(get_db) |
| 32 | +): |
| 33 | + """Authenticate user and get tokens.""" |
| 34 | + auth_service = AuthService(db) |
| 35 | + user = auth_service.authenticate(credentials.email, credentials.password) |
| 36 | + |
| 37 | + if not user: |
| 38 | + raise HTTPException( |
| 39 | + status_code=status.HTTP_401_UNAUTHORIZED, |
| 40 | + detail="Invalid credentials" |
| 41 | + ) |
| 42 | + |
| 43 | + if not user.is_active: |
| 44 | + raise HTTPException( |
| 45 | + status_code=status.HTTP_403_FORBIDDEN, |
| 46 | + detail="User account is disabled" |
| 47 | + ) |
| 48 | + |
| 49 | + tokens = auth_service.create_session( |
| 50 | + user, |
| 51 | + ip_address=request.client.host if request.client else None, |
| 52 | + user_agent=request.headers.get("user-agent") |
| 53 | + ) |
| 54 | + |
| 55 | + return tokens |
| 56 | + |
| 57 | + |
| 58 | +@router.post("/refresh", response_model=Token) |
| 59 | +def refresh_token(token_data: TokenRefresh, db: Session = Depends(get_db)): |
| 60 | + """Refresh access token.""" |
| 61 | + auth_service = AuthService(db) |
| 62 | + tokens = auth_service.refresh_access_token(token_data.refresh_token) |
| 63 | + |
| 64 | + if not tokens: |
| 65 | + raise HTTPException( |
| 66 | + status_code=status.HTTP_401_UNAUTHORIZED, |
| 67 | + detail="Invalid refresh token" |
| 68 | + ) |
| 69 | + |
| 70 | + return tokens |
| 71 | + |
| 72 | + |
| 73 | +@router.post("/logout") |
| 74 | +def logout(token_data: TokenRefresh, db: Session = Depends(get_db)): |
| 75 | + """Logout and invalidate session.""" |
| 76 | + auth_service = AuthService(db) |
| 77 | + success = auth_service.logout(token_data.refresh_token) |
| 78 | + |
| 79 | + return {"success": success} |
0 commit comments