mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 17:30:23 +08:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from starlette.responses import Response
|
||
|
|
|
||
|
|
from src.config import config
|
||
|
|
from src.core.exceptions import ErrorResponse
|
||
|
|
from src.services.auth.service import REFRESH_TOKEN_EXPIRATION_DAYS
|
||
|
|
|
||
|
|
|
||
|
|
def set_refresh_token_cookie(response: Response, refresh_token: str) -> None:
|
||
|
|
max_age_seconds = REFRESH_TOKEN_EXPIRATION_DAYS * 24 * 60 * 60
|
||
|
|
response.set_cookie(
|
||
|
|
key=config.auth_refresh_cookie_name,
|
||
|
|
value=refresh_token,
|
||
|
|
httponly=True,
|
||
|
|
secure=config.auth_refresh_cookie_secure,
|
||
|
|
samesite=config.auth_refresh_cookie_samesite,
|
||
|
|
path="/api/auth",
|
||
|
|
max_age=max_age_seconds,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def clear_refresh_token_cookie(response: Response) -> None:
|
||
|
|
response.delete_cookie(
|
||
|
|
key=config.auth_refresh_cookie_name,
|
||
|
|
path="/api/auth",
|
||
|
|
secure=config.auth_refresh_cookie_secure,
|
||
|
|
samesite=config.auth_refresh_cookie_samesite,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def error_response_with_cleared_cookie(exc: Exception) -> Response:
|
||
|
|
response = ErrorResponse.from_exception(exc)
|
||
|
|
clear_refresh_token_cookie(response)
|
||
|
|
return response
|