31 lines
844 B
Python
31 lines
844 B
Python
"""Channels base classes"""
|
|
from channels.exceptions import DenyConnection
|
|
from channels.generic.websocket import JsonWebsocketConsumer
|
|
from structlog import get_logger
|
|
|
|
from authentik.api.auth import token_from_header
|
|
from authentik.core.models import User
|
|
|
|
LOGGER = get_logger()
|
|
|
|
|
|
class AuthJsonConsumer(JsonWebsocketConsumer):
|
|
"""Authorize a client with a token"""
|
|
|
|
user: User
|
|
|
|
def connect(self):
|
|
headers = dict(self.scope["headers"])
|
|
if b"authorization" not in headers:
|
|
LOGGER.warning("WS Request without authorization header")
|
|
raise DenyConnection()
|
|
|
|
raw_header = headers[b"authorization"]
|
|
|
|
token = token_from_header(raw_header)
|
|
if not token:
|
|
LOGGER.warning("Failed to authenticate")
|
|
raise DenyConnection()
|
|
|
|
self.user = token.user
|