1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
from fastapi import APIRouter, Depends, Request
from sqlalchemy.ext.asyncio import AsyncSession
from app.auth.jwt import decode_token
from app.models.user import User
from app.utils.db import get_async_session
router = APIRouter(tags=["auth"])
async def get_current_user_from_cookie(
request: Request,
session: AsyncSession = Depends(get_async_session),
) -> dict:
token = request.cookies.get("access_token")
if not token:
return {"authenticated": False, "user": None}
try:
payload = decode_token(token)
sub = payload.get("sub")
if sub is None:
return {"authenticated": False, "user": None}
user_id = int(sub)
except ValueError, TypeError:
return {"authenticated": False, "user": None}
user = await User.get_user_by_id(user_id, session=session)
if not user or user.token_version != payload.get("token_version"):
return {"authenticated": False, "user": None}
# Return authenticated user
return {
"authenticated": True,
"user": {
"id": user.id,
"username": user.username,
"password": user.has_password,
"google_id": user.google_id,
"email": user.email,
"premium": user.premium,
"is_banned": user.is_banned,
"is_moderator": user.is_moderator,
},
}
@router.get("/me")
async def read_current_user(
user_info: dict = Depends(get_current_user_from_cookie),
):
# Directly return the dict to the frontend
return user_info
|
