82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
from fastapi import APIRouter, Depends, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.auth.schemas import (
|
|
AuthUserResponse,
|
|
LoginRequest,
|
|
MessageResponse,
|
|
RegisterRequest,
|
|
RequestPasswordResetRequest,
|
|
ResendVerificationRequest,
|
|
ResetPasswordRequest,
|
|
TokenResponse,
|
|
VerifyEmailRequest,
|
|
)
|
|
from app.auth.service import (
|
|
get_current_user,
|
|
get_email_sender,
|
|
login_user,
|
|
register_user,
|
|
request_password_reset,
|
|
resend_verification_email,
|
|
reset_password,
|
|
verify_email,
|
|
)
|
|
from app.database.session import get_db
|
|
from app.email.service import EmailService
|
|
from app.users.models import User
|
|
|
|
router = APIRouter(prefix="/auth", tags=["auth"])
|
|
|
|
|
|
@router.post("/register", response_model=MessageResponse, status_code=status.HTTP_201_CREATED)
|
|
async def register(
|
|
payload: RegisterRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
email_service: EmailService = Depends(get_email_sender),
|
|
) -> MessageResponse:
|
|
await register_user(db, payload, email_service)
|
|
return MessageResponse(message="Registration successful. Verification email sent.")
|
|
|
|
|
|
@router.post("/login", response_model=TokenResponse)
|
|
async def login(payload: LoginRequest, db: AsyncSession = Depends(get_db)) -> TokenResponse:
|
|
return await login_user(db, payload)
|
|
|
|
|
|
@router.post("/verify-email", response_model=MessageResponse)
|
|
async def verify_email_endpoint(payload: VerifyEmailRequest, db: AsyncSession = Depends(get_db)) -> MessageResponse:
|
|
await verify_email(db, payload)
|
|
return MessageResponse(message="Email verified successfully.")
|
|
|
|
|
|
@router.post("/resend-verification", response_model=MessageResponse)
|
|
async def resend_verification(
|
|
payload: ResendVerificationRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
email_service: EmailService = Depends(get_email_sender),
|
|
) -> MessageResponse:
|
|
await resend_verification_email(db, payload, email_service)
|
|
return MessageResponse(message="If the account exists, a verification email was sent.")
|
|
|
|
|
|
@router.post("/request-password-reset", response_model=MessageResponse)
|
|
async def request_password_reset_endpoint(
|
|
payload: RequestPasswordResetRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
email_service: EmailService = Depends(get_email_sender),
|
|
) -> MessageResponse:
|
|
await request_password_reset(db, payload, email_service)
|
|
return MessageResponse(message="If the account exists, a reset email was sent.")
|
|
|
|
|
|
@router.post("/reset-password", response_model=MessageResponse)
|
|
async def reset_password_endpoint(payload: ResetPasswordRequest, db: AsyncSession = Depends(get_db)) -> MessageResponse:
|
|
await reset_password(db, payload)
|
|
return MessageResponse(message="Password reset successfully.")
|
|
|
|
|
|
@router.get("/me", response_model=AuthUserResponse)
|
|
async def me(current_user: User = Depends(get_current_user)) -> AuthUserResponse:
|
|
return current_user
|