"""회원가입, 이메일 인증 및 회사 검색 API.""" from datetime import datetime from fastapi import APIRouter, HTTPException, Query from common_util.common_util_auth import generate_otp, hash_password, verify_password from common_util.common_util_auth_repository import ( complete_registration, consume_otp, create_registration, get_active_otp, get_user_by_email, refresh_pending_registration, replace_otp, search_companies, ) from common_util.common_util_email import send_email_background from common_util.common_util_email_templates import otp_email from .A07_Register_Schema import RegisterRequest, RegisterVerifyRequest router = APIRouter(prefix="/api/auth", tags=["Authentication"]) @router.get("/companies") async def find_companies(q: str = Query(min_length=1, max_length=100)): return {"status": "success", "companies": await search_companies(q.strip())} @router.post("/register/request", status_code=202) async def request_registration(payload: RegisterRequest): email = str(payload.email).lower() data = payload.model_dump(exclude={"password_confirm"}) data["email"] = email password_hash = hash_password(payload.password) existing = await get_user_by_email(email) if existing: # 이미 이메일 인증까지 마친 계정이면 재가입 불가 if existing["status"] != "PENDING_EMAIL": raise HTTPException(status_code=409, detail="이미 등록된 이메일입니다.") # 미인증 계정이면 최신 요청으로 갱신하고 새 인증 코드를 발급 user_id = existing["id"] await refresh_pending_registration(user_id, data, password_hash) else: user_id = await create_registration(data, password_hash) code = generate_otp() await replace_otp(user_id, "REGISTER", hash_password(code)) subject, html = otp_email(code, "회원가입") send_email_background(email, subject, html) return {"status": "success", "message": "인증 코드를 발송했습니다."} @router.post("/register/verify") async def verify_registration(payload: RegisterVerifyRequest): user = await get_user_by_email(str(payload.email).lower()) if not user or user["status"] != "PENDING_EMAIL": raise HTTPException(status_code=400, detail="인증 대기 중인 가입 요청이 없습니다.") otp = await get_active_otp(user["id"], "REGISTER") if ( not otp or otp["expires_at"] < datetime.utcnow() or not verify_password(payload.otp_code, otp["otp_hash"]) ): raise HTTPException(status_code=400, detail="인증 코드가 유효하지 않습니다.") await consume_otp(otp["id"]) await complete_registration(user["id"], bool(user["is_master"])) # 회원가입 시점에는 회사가 없으므로(NO_COMPANY) 마스터 알림은 발송하지 않는다. # 회사 생성/연결은 로그인 후 B01_Dashboard에서 진행한다. return { "status": "success", "account_status": "NO_COMPANY", }