From 9776ef42cd96e9a2294097fb4a77be25bd67cf17 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 20 Feb 2023 09:03:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0websocket=20auth=20middleware?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/utils/middlewares.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 apps/utils/middlewares.py

diff --git a/apps/utils/middlewares.py b/apps/utils/middlewares.py
new file mode 100644
index 00000000..74be9af3
--- /dev/null
+++ b/apps/utils/middlewares.py
@@ -0,0 +1,20 @@
+from rest_framework_simplejwt.authentication import JWTAuthentication
+from asgiref.sync import sync_to_async
+
+@sync_to_async
+def _get_user(token: str):
+    jwt = JWTAuthentication()
+    return jwt.get_user(jwt.get_validated_token(token))
+
+class TokenAuthMiddleware:
+    def __init__(self, app) -> None:
+        self.app = app
+
+    async def __call__(self, scope, receive, send):
+        # Look up user from query string (you should also do things like
+        # checking if it is a valid user ID, or if scope["user"] is already
+        # populated).
+        from urllib.parse import parse_qs
+        token = parse_qs(str(scope["query_string"], 'UTF-8'))['token'][0]
+        scope['user'] = await _get_user(token)
+        return await self.app(scope, receive, send)
\ No newline at end of file