diff --git a/server.py b/server.py index b8db6fb..5b8d313 100644 --- a/server.py +++ b/server.py @@ -3051,12 +3051,15 @@ async def list_storage(request: Request, user: User = Depends(get_optional_user) accept = request.headers.get("accept", "") wants_json = "application/json" in accept and "text/html" not in accept - if not user: + # For browser sessions, also check cookie authentication + username = user.username if user else get_user_from_cookie(request) + + if not username: if wants_json: raise HTTPException(401, "Authentication required") return RedirectResponse(url="/login", status_code=302) - storages = await db.get_user_storage(user.username) + storages = await db.get_user_storage(username) # Add usage stats to each storage for storage in storages: @@ -3079,7 +3082,7 @@ async def list_storage(request: Request, user: User = Depends(get_optional_user) return {"storages": storages} # Default to HTML for browsers - return await ui_storage_page(user.username, storages, request) + return await ui_storage_page(username, storages, request) @app.post("/storage")