vjdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZe dGdd ZGd d ejZGd d eZGddeZGddZdS)z Email verification service for quick registration and account activation. Supports 6-digit secure numeric OTPs, activation link tokens, cooldown enforcement, and pluggable senders. ) annotationsN)AnyDictListOptionalTuple) dataclassfieldT)slotscveZdZUdZded<ded<ded<ded<ded<ded <d Zd ed <d Zded<dddZdS)VerificationCodez!Activation code and token record.stremail account_idcodetokenfloat created_at expires_atrintfailed_attemptsFbool is_consumedN current_timeOptional[float]returncF||ntj}||jkSN)timer)selfrnows 7C:\Projects\FreeExile\server\auth\email_verification.py is_expiredzVerificationCode.is_expireds#*6llDIKKT_$$r)rrrr)__name__ __module__ __qualname____doc____annotations__rrr#r$r"r r s++JJJOOO IIIJJJOK%%%%%%%r$r c8eZdZdZej d d d ZdS) EmailSenderz)Abstract interface for email dispatching.Nto_emailrsubjectbody html_body Optional[str]rrcdSrr*r r-r.r/r0s r" send_emailzEmailSender.send_email"s  r$r r-rr.rr/rr0r1rr)r%r&r'r(abcabstractmethodr4r*r$r"r,r,sF33 $(        r$r,c&eZdZdZddZ ddd ZdS)MockEmailSenderz;In-memory email sender for unit tests and local simulation.rNonecg|_dSr) sent_messages)r s r"__init__zMockEmailSender.__init__0s35r$Nr-rr.r/r0r1rc |j||||pdttjddS)N)tor.r/r0 timestampT)r<appendrrr3s r"r4zMockEmailSender.send_email3sQ !!"bTY[[)) # #    tr$)rr:rr5)r%r&r'r(r=r4r*r$r"r9r9-sLEE6666$( r$r9ceZdZdZ d d d ZdS)ConsoleEmailSenderz!Development console email sender.Nr-rr.r/r0r1rrc 8td|d|d|ddS)Nz [DEV EMAIL SENDER] To: z | Subject:  T)printr3s r"r4zConsoleEmailSender.send_emailGs3 S(SSSS4SSSTTTtr$rr5)r%r&r'r(r4r*r$r"rDrDDs8++$( r$rDcJeZdZdZ ddd ZddZddZd dZd!dZd"dZ dS)#EmailVerificationServicezHManages email activation OTPs, link tokens, cooldowns, and verification.< email_senderr,code_ttl_secondsrcooldown_seconds max_attemptsrr:cZ||_||_||_||_i|_i|_dSr)rMrNrOrP_records _last_sent)r rMrNrOrPs r"r=z!EmailVerificationService.__init__Us6) 0 0(57 ,.r$rrcN|Sr)striplowerr rs r"_normalize_emailz)EmailVerificationService._normalize_emailcs{{}}""$$$r$Tuple[bool, float]c||}|j|}|dStj|z }|j|z }|dkrd|fSdS)N)TgrF)rXrSgetrrO)r r norm_email last_timeelapsed remainings r" can_resendz#EmailVerificationService.can_resendfsk**511 O'' 33  9)++ ))G3 q==)# #yr$rr c ||}tj}||\}}|s tdt |dt jdd}t jd}||jz} t|||||| } | |j |<||j |<d} d|d |jd zd |d |d } |j || | | S)z.Generate secure OTP and send activation email.zRate limited: Please wait z% seconds before requesting a new codei@B06d )rrrrrru<[FreeExile] Mã Kích Hoạt Tài Khoản Cổ Võ Hắc ÁmuyChào mừng chiến binh lưu đày đến với FreeExile! Mã xác thực kích hoạt tài khoản của bạn là: u' Mã này có hiệu lực trong vòng rKuj phút. Hoặc nhấp vào liên kết sau để kích hoạt nhanh: https://freeexile.io/activate?email=z&token=uM Nếu bạn không thực hiện yêu cầu này, vui lòng bỏ qua email.)r-r.r/)rXrr` ValueErrorrsecrets randbelow token_urlsaferNr rRrSrMr4) r rrr\r!can_sendr_rrrrecordr.r/s r"send_activation_emailz.EmailVerificationService.send_activation_emailrsZ**511 ikk"ooj99) qo#i..ooopp p#I..44%b))400 !!!    %+ j!&) #P [JN [ [595Jb5P [ [4>  [ [GL  [ [ [  $$j'PT$UUU r$Optional[VerificationCode]c\|j||Sr)rRr[rXrWs r"get_active_recordz*EmailVerificationService.get_active_records&}  !6!6u!=!=>>>r$ code_or_tokenTuple[bool, str]c||}|j|}|sdS|jrdS|rdS|j|jkrdS|}tj |j |}tj |j |}|s|r d|_dS|xjdz c_td|j|jz }d d |d fS) zHVerify OTP or activation token with single-use and max retry protection.)Fz3No active verification request found for this email)Fz'Verification code has already been used)Fz7Verification code has expired. Please request a new one)Fz2Too many failed attempts. Code locked for securityT)TOKrFzIncorrect code. z attempts remaining) rXrRr[rr#rrPrUrecompare_digestrrmax) r rrnr\ri cleaned_input is_code_matchis_token_matchremaining_attemptss r" verify_codez$EmailVerificationService.verify_codes**511 "":.. POO   DCC      TSS  !T%6 6 6NN%++-- .v{MJJ  / mLL  N !%F :!# D$58N$NOOP);PPPPPr$N)rJrKrL) rMr,rNrrOrrPrrr:)rrrr)rrrrY)rrrrrr )rrrrk)rrrnrrro) r%r&r'r(r=rXr`rjrmryr*r$r"rIrIRsRR !$ " / / / / /%%%%    $$$$L????QQQQQQr$rI)r( __future__rr6rertypingrrrrr dataclassesr r r ABCr,r9rDrIr*r$r"r~s #"""""  33333333333333((((((((  % % % % % % % %      #'    k.        dQdQdQdQdQdQdQdQdQdQr$