P1i6dZddlZddlmZddlmZmZddlmZeje Z GddeZ Gd d e Z Gd d e Zy) z Custom JWT authentication backend that automatically tracks user activity. This extends Simple JWT's authentication to fire signals whenever JWT tokens are validated, ensuring systematic activity tracking for API requests. N)JWTAuthentication) InvalidToken TokenError)fire_jwt_authenticated_signalc,eZdZdZfdZfdZxZS)!ActivityTrackingJWTAuthenticationz Custom JWT authentication that automatically tracks user activity. This extends the default Simple JWT authentication to fire a signal whenever a user is successfully authenticated via JWT token validation. ct||}||\}} t|||S|S#t$r#}tj d|Yd}~|Sd}~wwxYw)zX Authenticate the request and fire activity tracking signal on success. Nz(Error firing JWT authentication signal: )super authenticater Exceptionloggererror)selfrequestresultuservalidated_tokene __class__s 5/home/cursorai/projects/iching/main/authentication.pyr z.ActivityTrackingJWTAuthentication.authenticatesq %g.  $* !D/ M-dG<  v  M GsKLL  Ms ) AAAc t||S#t$r5}tj d|t |j dd}~wwxYw)z Validate the given raw token and return a validated token instance. Override to add additional logging if needed. zJWT token validation failed: rN)r get_validated_tokenrrdebugrargs)r raw_tokenrrs rrz5ActivityTrackingJWTAuthentication.get_validated_token,sO  *7.y9 9 * LL8< =qvvay) ) *s A0A  A)__name__ __module__ __qualname____doc__r r __classcell__rs@rr r s( * *r c"eZdZdZfdZxZS)JWTCookieAuthenticationz JWT authentication that reads tokens from cookies instead of headers. This is useful for web applications that want to use JWT tokens stored in HTTP-only cookies for additional security. c|jjd}|t| |St |t r|j dS|S)zU Extract the JWT token from cookies instead of Authorization header. access_tokenutf-8)COOKIESgetr get_raw_token isinstancestrencode)rrrrs rr+z%JWTCookieAuthentication.get_raw_tokenAsN OO''7  7(1 1,6y#,Fy(UIUr#rrrr r+r!r"s@rr%r%9s V Vr#r%c"eZdZdZfdZxZS)HybridJWTAuthenticationz Hybrid JWT authentication that tries both header and cookie methods. This provides maximum flexibility by supporting both: - Authorization header (for API clients) - HTTP-only cookies (for web applications) ct||}||S|jjd}|#t |t r|j dS|Sy)zY Try to extract JWT token from Authorization header first, then cookies. Nr'r()r r+r)r*r,r-r.)rr header_token cookie_tokenrs rr+z%HybridJWTAuthentication.get_raw_tokenXs_ w,W5  # **>:  #3=lC3P<&&w/ bVb br#r/r"s@rr1r1Osr#r1)r logging'rest_framework_simplejwt.authenticationr#rest_framework_simplejwt.exceptionsrrsignalsr getLoggerrrr r%r1r#rr;sTEH2   8 $&*(9&*RV?V,?r#