feat: add async support for auth handler

robyn/authentication.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Optional
+from typing import Any, Coroutine, Optional, Union
 
 from robyn.robyn import Headers, Identity, Request, Response
 from robyn.status_codes import HTTP_401_UNAUTHORIZED
@@ -64,7 +64,7 @@ def unauthorized_response(self) -> Response:
         )
 
     @abstractmethod
-    def authenticate(self, request: Request) -> Optional[Identity]:
+    def authenticate(self, request: Request) -> Union[Optional[Identity], Coroutine[Any, Any, Optional[Identity]]]:
         """
         Authenticates the user.
         :param request: The request object.

robyn/router.py

@@ -290,10 +290,15 @@ def add_auth_middleware(self, endpoint: str):
 
         def decorator(handler):
             @wraps(handler)
-            def inner_handler(request: Request, *args):
+            async def inner_handler(request: Request, *args):
                 if not self.authentication_handler:
                     raise AuthenticationNotConfiguredError()
-                identity = self.authentication_handler.authenticate(request)
+
+                if inspect.iscoroutinefunction(self.authentication_handler.authenticate):
+                    identity = await self.authentication_handler.authenticate(request)
+                else:
+                    identity = self.authentication_handler.authenticate(request)
+
                 if identity is None:
                     return self.authentication_handler.unauthorized_response
                 request.identity = identity