mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
98 lines
3 KiB
Python
98 lines
3 KiB
Python
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from loguru import logger
|
||
|
|
|
||
|
|
from api.db import db_client
|
||
|
|
from api.db.models import UserModel
|
||
|
|
from api.schemas.auth import AuthResponse, LoginRequest, SignupRequest, UserResponse
|
||
|
|
from api.services.auth.depends import create_user_configuration_with_mps_key, get_user
|
||
|
|
from api.utils.auth import create_jwt_token, hash_password, verify_password
|
||
|
|
|
||
|
|
router = APIRouter(
|
||
|
|
prefix="/auth",
|
||
|
|
tags=["auth"],
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/signup", response_model=AuthResponse)
|
||
|
|
async def signup(request: SignupRequest):
|
||
|
|
# Check if email is already taken
|
||
|
|
existing_user = await db_client.get_user_by_email(request.email)
|
||
|
|
if existing_user:
|
||
|
|
raise HTTPException(status_code=409, detail="Email already registered")
|
||
|
|
|
||
|
|
# Hash password and create user
|
||
|
|
hashed = hash_password(request.password)
|
||
|
|
user = await db_client.create_user_with_email(
|
||
|
|
email=request.email,
|
||
|
|
password_hash=hashed,
|
||
|
|
name=request.name,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create organization for the user
|
||
|
|
org_provider_id = f"org_{user.provider_id}"
|
||
|
|
organization, _ = await db_client.get_or_create_organization_by_provider_id(
|
||
|
|
org_provider_id=org_provider_id, user_id=user.id
|
||
|
|
)
|
||
|
|
|
||
|
|
# Link user to organization
|
||
|
|
await db_client.add_user_to_organization(user.id, organization.id)
|
||
|
|
await db_client.update_user_selected_organization(user.id, organization.id)
|
||
|
|
|
||
|
|
# Create default service configuration
|
||
|
|
try:
|
||
|
|
mps_config = await create_user_configuration_with_mps_key(
|
||
|
|
user.id, organization.id, user.provider_id
|
||
|
|
)
|
||
|
|
if mps_config:
|
||
|
|
await db_client.update_user_configuration(user.id, mps_config)
|
||
|
|
except Exception:
|
||
|
|
logger.warning(
|
||
|
|
"Failed to create default configuration for OSS user", exc_info=True
|
||
|
|
)
|
||
|
|
|
||
|
|
# Create JWT token
|
||
|
|
token = create_jwt_token(user.id, request.email)
|
||
|
|
|
||
|
|
return AuthResponse(
|
||
|
|
token=token,
|
||
|
|
user=UserResponse(
|
||
|
|
id=user.id,
|
||
|
|
email=user.email,
|
||
|
|
name=request.name,
|
||
|
|
organization_id=organization.id,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/login", response_model=AuthResponse)
|
||
|
|
async def login(request: LoginRequest):
|
||
|
|
# Look up user by email
|
||
|
|
user = await db_client.get_user_by_email(request.email)
|
||
|
|
if not user or not user.password_hash:
|
||
|
|
raise HTTPException(status_code=401, detail="Invalid email or password")
|
||
|
|
|
||
|
|
# Verify password
|
||
|
|
if not verify_password(request.password, user.password_hash):
|
||
|
|
raise HTTPException(status_code=401, detail="Invalid email or password")
|
||
|
|
|
||
|
|
# Create JWT token
|
||
|
|
token = create_jwt_token(user.id, user.email)
|
||
|
|
|
||
|
|
return AuthResponse(
|
||
|
|
token=token,
|
||
|
|
user=UserResponse(
|
||
|
|
id=user.id,
|
||
|
|
email=user.email,
|
||
|
|
organization_id=user.selected_organization_id,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/me", response_model=UserResponse)
|
||
|
|
async def get_current_user(user: UserModel = Depends(get_user)):
|
||
|
|
return UserResponse(
|
||
|
|
id=user.id,
|
||
|
|
email=user.email,
|
||
|
|
organization_id=user.selected_organization_id,
|
||
|
|
)
|