56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
|
from fastapi import APIRouter, Depends, HTTPException, Response, status
|
||
|
|
from jose import JWTError
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from app.core.database import get_db
|
||
|
|
from app.core.security import create_access_token, decode_token
|
||
|
|
from app.modules.auth import schemas, service
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/register", response_model=schemas.RegisterResponse, status_code=201)
|
||
|
|
async def register(body: schemas.RegisterRequest, db: AsyncSession = Depends(get_db)):
|
||
|
|
token, business_id, user_id = await service.register_business(
|
||
|
|
db,
|
||
|
|
business_name=body.business_name,
|
||
|
|
business_type=body.business_type,
|
||
|
|
timezone=body.timezone,
|
||
|
|
email=body.email,
|
||
|
|
password=body.password,
|
||
|
|
)
|
||
|
|
return schemas.RegisterResponse(
|
||
|
|
access_token=token,
|
||
|
|
business_id=business_id,
|
||
|
|
user_id=user_id,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/login", response_model=schemas.TokenResponse)
|
||
|
|
async def login(body: schemas.LoginRequest, db: AsyncSession = Depends(get_db)):
|
||
|
|
token = await service.authenticate_user(db, body.email, body.password)
|
||
|
|
return schemas.TokenResponse(access_token=token)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/meta-callback", response_model=schemas.TokenResponse)
|
||
|
|
async def meta_callback(body: schemas.MetaCallbackRequest, db: AsyncSession = Depends(get_db)):
|
||
|
|
token = await service.meta_oauth_login(db, body.code, body.redirect_uri)
|
||
|
|
return schemas.TokenResponse(access_token=token)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/refresh", response_model=schemas.TokenResponse)
|
||
|
|
async def refresh(body: schemas.RefreshRequest):
|
||
|
|
try:
|
||
|
|
payload = decode_token(body.access_token)
|
||
|
|
except JWTError:
|
||
|
|
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Token inválido")
|
||
|
|
new_token = create_access_token(
|
||
|
|
{"sub": payload["sub"], "business_id": payload["business_id"]}
|
||
|
|
)
|
||
|
|
return schemas.TokenResponse(access_token=new_token)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/logout", status_code=204)
|
||
|
|
async def logout():
|
||
|
|
return Response(status_code=204)
|