refac: token handling

silentoplayz · Nov 6, 2024 · 4616b50 · 4616b50
1 parent 60df959
commit 4616b50
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
@@ -2224,7 +2224,14 @@ async def get_app_config(request: Request):
     user = None
     if "token" in request.cookies:
         token = request.cookies.get("token")
-        data = decode_token(token)
+        try:
+            data = decode_token(token)
+        except Exception as e:
+            log.debug(e)
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="Invalid token",
+            )
         if data is not None and "id" in data:
             user = Users.get_user_by_id(data["id"])
 

diff --git a/backend/open_webui/utils/utils.py b/backend/open_webui/utils/utils.py
@@ -91,7 +91,15 @@ def get_current_user(
         return get_current_user_by_api_key(token)
 
     # auth by jwt token
-    data = decode_token(token)
+
+    try:
+        data = decode_token(token)
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Invalid token",
+        )
+
     if data is not None and "id" in data:
         user = Users.get_user_by_id(data["id"])
         if user is None: