71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from app.core.database import get_db
|
|
from app.api.deps import get_current_user
|
|
from models.api_models import UserRegister, UserLogin, UserResponse, LoginResponse
|
|
from models.database_models import User
|
|
from middleware.auth import AuthManager
|
|
import logging
|
|
|
|
router = APIRouter()
|
|
auth_manager = AuthManager()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# Authentication Routes
|
|
@router.post("/register", response_model=UserResponse)
|
|
def register_user(user_data: UserRegister, db: Session = Depends(get_db)):
|
|
"""Register a new user"""
|
|
# Check if email already exists
|
|
existing_email = db.query(User).filter(User.email == user_data.email).first()
|
|
if existing_email:
|
|
raise HTTPException(status_code=400, detail="Email already registered")
|
|
|
|
# Check if username already exists
|
|
existing_username = db.query(User).filter(User.username == user_data.username).first()
|
|
if existing_username:
|
|
raise HTTPException(status_code=400, detail="Username already taken")
|
|
|
|
# Hash password and create user
|
|
hashed_password = auth_manager.hash_password(user_data.password)
|
|
new_user = User(
|
|
email=user_data.email,
|
|
username=user_data.username,
|
|
hashed_password=hashed_password,
|
|
role="user",
|
|
is_active=True
|
|
)
|
|
|
|
db.add(new_user)
|
|
db.commit()
|
|
db.refresh(new_user)
|
|
|
|
logger.info(f"New user registered: {new_user.username}")
|
|
return new_user
|
|
|
|
|
|
@router.post("/login", response_model=LoginResponse)
|
|
def login_user(user_credentials: UserLogin, db: Session = Depends(get_db)):
|
|
"""Login user and return JWT token"""
|
|
user = auth_manager.authenticate_user(db, user_credentials.username, user_credentials.password)
|
|
if not user:
|
|
raise HTTPException(status_code=401, detail="Incorrect username or password")
|
|
|
|
# Create access token
|
|
token_data = auth_manager.create_access_token(user)
|
|
|
|
logger.info(f"User logged in: {user.username}")
|
|
|
|
return LoginResponse(
|
|
access_token=token_data["access_token"],
|
|
token_type=token_data["token_type"],
|
|
expires_in=token_data["expires_in"],
|
|
user=UserResponse.model_validate(user)
|
|
)
|
|
|
|
|
|
@router.get("/me", response_model=UserResponse)
|
|
def get_current_user_info(current_user: User = Depends(get_current_user)):
|
|
"""Get current user information"""
|
|
return UserResponse.model_validate(current_user)
|