Как реализовать Middleware для аутентификации WebSocket-соединений?

А, слушай, смотри, вот эта вся хуйня с вебсокетами, блядь. Вроде как открыл соединение, и понеслась, да? Но тут, сука, как в хорошем подъезде — просто так с улицы не зайдешь, нужен ключ, блядь. То есть, аутентификация, ёпта!

Вот представь: стучится к тебе какой-то левый клиент в сокет, а ты ему — «Иди нахуй, чувак, кто ты такой?». Для этого и нужен этот самый middleware, хитрая жопа, который на входе проверяет, свой ли пришел или манда с ушами.

Смотри, как это примерно в FastAPI обустроить можно:

from fastapi import WebSocket, WebSocketDisconnect, APIRouter
from fastapi.security import OAuth2PasswordBearer
from typing import Optional

router = APIRouter()
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token") # Ну, типа, откуда токены берутся

# Функция, которая решает — свой в доску или пидарас шерстяной
def verify_token(token: str) -> bool:
    # Тут, блядь, должна быть настоящая, ебаная проверка JWT или чего там
    # А пока так, для прикола
    return token == "valid_token"

async def websocket_auth_middleware(websocket: WebSocket) -> Optional[str]:
    """
    Вот он, сука, швейцар наш. Стоит на пороге и проверяет пропуска.
    Ждет, что первым делом ему в зубы токен положат.
    """
    try:
        # Ждем первое сообщение — в нем должен быть пароль, блядь
        token = await websocket.receive_text()
        if not verify_token(token):
            # А если хуйня какая-то пришла — захлопываем дверь перед носом!
            await websocket.close(code=1008, reason="Invalid authentication token")
            raise WebSocketDisconnect("Authentication failed")
        return token
    except WebSocketDisconnect:
        # Соединение уже отвалилось, нехуй тут делать
        raise

@router.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    try:
        # Вот тут вызываем нашего вышибалу
        token = await websocket_auth_middleware(websocket)
        # Если добрался сюда — значит свой, можно общаться
        await websocket.send_text(f"Hello, authenticated user with token: {token}")
        while True:
            data = await websocket.receive_text()
            await websocket.send_text(f"Message text was: {data}")
    except WebSocketDisconnect:
        print("WebSocket disconnected, authentication failed or client closed connection.")
    except Exception as e:
        print(f"An error occurred: {e}")

А теперь, блядь, ключевые моменты, чтобы не обосраться:

• Как токен передать? Да как угодно, ёпта! Можно в первом же сообщении, как у нас. Можно в query-параметре URL (ws://site/ws?token=тут_твой_хуй). Можно, если очень хочется, и в заголовках попробовать, но это уже танцы с бубном. Главное — договориться с клиентом, а то будет «я ему — токен, а он мне — хуй с горы».
• Если что-то пошло не так? Закрывай соединение, блядь, красиво! Есть специальные коды, например 1008 — «Нарушение политики, иди нахуй». Клиенту так понятнее, почему его послали.
• А если народу овердохуища? Ну, тут уже без брокера сообщений (типа Redis или RabbitMQ) будет тяжко. Они, эти брокеры, тоже могут с аутентификацией дружить, чтобы по всем серверам твой пропуск был действителен, а не только на одном подъезде.

Короче, смысл в том, чтобы не пускать в свой сокет кого попало, а то начнут хуйню слать, а ты потом разгребай. Ставь вышибалу на входе и спи спокойно, ёпта!