mirror of
https://github.com/evgen-app/chess_rpg_backend.git
synced 2024-09-21 06:38:47 +03:00
34 lines
889 B
Python
34 lines
889 B
Python
from channels.db import database_sync_to_async
|
|
from django.core.exceptions import PermissionDenied
|
|
|
|
from game.models import Player
|
|
from game.services.jwt import read_jwt
|
|
|
|
|
|
@database_sync_to_async
|
|
def get_player(headers):
|
|
# WARNING headers type is bytes
|
|
if b"authorization" not in headers or not headers[b"authorization"]:
|
|
return False
|
|
|
|
jwt = headers[b"authorization"].decode()
|
|
payload = read_jwt(jwt)
|
|
|
|
if not payload or "id" not in payload:
|
|
return False
|
|
|
|
return payload["id"]
|
|
|
|
|
|
class HeaderAuthMiddleware:
|
|
"""Custom middleware to read user auth token from string."""
|
|
|
|
def __init__(self, app):
|
|
# Store the ASGI application we were passed
|
|
self.app = app
|
|
|
|
async def __call__(self, scope, receive, send):
|
|
scope["player"] = await get_player(dict(scope["headers"]))
|
|
|
|
return await self.app(scope, receive, send)
|