- src/middleware/auth.py

  - add NEED_AUTH_REGEX_URL
  - refactor AuthMiddleware

Author: Meoyoug

src/middleware/auth.py

@@ -1,17 +1,31 @@
+import re
+
 from fastapi import HTTPException
 from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoint
 from starlette.requests import Request
 from starlette.responses import JSONResponse, Response
 
 from src.services.auth import AuthService
 
+NEED_AUTH_REGEX_URL = [
+    r"^/users/me$",
+    r"^/users/me/profile_image$",
+    r"^/users/me/reviews$",
+    r"^/reviews$",
+    r"^/reviews/\d+$",
+    r"^/reviews/\d+/is_liked$",
+    r"^/likes/reviews/\d+/like$",
+    r"^/likes/reviews/\d+/unlike$",
+]
+
 
 class AuthMiddleware(BaseHTTPMiddleware):
     async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response:
         try:
-            if request.url.path.startswith("/users"):
-                if request.url.path not in ["/users/login", "/users", "/users/search"]:
+            for url in NEED_AUTH_REGEX_URL:
+                if re.match(url, request.url.path):
                     request = await AuthService().get_current_user(request)
+                    break
             response: Response = await call_next(request)
             return response
         except HTTPException as e: