spakky-fastapi 0.7.4__py3-none-any.whl → 0.7.7__py3-none-any.whl

spakky_fastapi/jwt_auth.py

@@ -0,0 +1,82 @@
+from typing import Any, TypeVar, Protocol, Awaitable, runtime_checkable
+from inspect import signature
+from logging import Logger
+from dataclasses import InitVar, field, dataclass
+
+from fastapi import Body, Depends, Response, status
+from fastapi.security import OAuth2PasswordBearer
+from spakky.aop.advice import Around
+from spakky.aop.advisor import IAsyncAdvisor
+from spakky.aop.aspect import AsyncAspect
+from spakky.bean.autowired import autowired
+from spakky.core.annotation import FunctionAnnotation
+from spakky.core.types import AsyncFunc, P
+from spakky.cryptography.error import InvalidJWTFormatError, JWTDecodingError
+from spakky.cryptography.jwt import JWT
+from spakky.cryptography.key import Key
+
+R_co = TypeVar("R_co", covariant=True)
+
+
+@runtime_checkable
+class IAuthenticatedFunction(Protocol[P, R_co]):
+    __defaults__: tuple[Any, ...] | None
+    __kwdefaults__: dict[str, Any] | None
+
+    # pylint: disable=no-self-argument
+    def __call__(
+        self_,  # type: ignore
+        self: Any,
+        token: JWT,
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> Awaitable[R_co]:
+        raise NotImplementedError
+
+
+@dataclass
+class JWTAuth(FunctionAnnotation):
+    token_url: InitVar[str]
+    authenticator: OAuth2PasswordBearer = field(init=False)
+
+    def __post_init__(self, token_url: str) -> None:
+        self.authenticator = OAuth2PasswordBearer(tokenUrl=token_url)
+
+    def __call__(
+        self, obj: IAuthenticatedFunction[P, R_co]
+    ) -> IAuthenticatedFunction[P, R_co]:
+        parameters = list(signature(obj).parameters.values())
+        if obj.__defaults__ is not None:
+            extra = obj.__defaults__
+        else:
+            extra = tuple(Body() for x in parameters[1:] if x.name != "token")
+        obj.__defaults__ = (Depends(self.authenticator),) + extra
+        return super().__call__(obj)
+
+
+@AsyncAspect()
+class AsyncJWTAuthAdvisor(IAsyncAdvisor):
+    __logger: Logger
+    __key: Key
+
+    @autowired
+    def __init__(self, logger: Logger, key: Key) -> None:
+        super().__init__()
+        self.__logger = logger
+        self.__key = key
+
+    @Around(JWTAuth.contains)
+    async def around_async(self, joinpoint: AsyncFunc, *args: Any, **kwargs: Any) -> Any:
+        token: str = kwargs["token"]
+        try:
+            jwt: JWT = JWT(token=token)
+        except (InvalidJWTFormatError, JWTDecodingError):
+            return Response(content=None, status_code=status.HTTP_401_UNAUTHORIZED)
+        else:
+            if jwt.is_expired:
+                return Response(content=None, status_code=status.HTTP_401_UNAUTHORIZED)
+            if jwt.verify(self.__key) is False:
+                return Response(content=None, status_code=status.HTTP_401_UNAUTHORIZED)
+            self.__logger.info(f"[{type(self).__name__}] {jwt.payload!r}")
+            kwargs["token"] = jwt
+            return await joinpoint(*args, **kwargs)

{spakky_fastapi-0.7.4.dist-info → spakky_fastapi-0.7.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spakky-fastapi
-Version: 0.7.4
+Version: 0.7.7
 Summary: Highly abstracted Framework core to use DDD & DI/IoC & AOP & Etc...
 Author: Spakky
 Author-email: sejong418@icloud.com
@@ -10,7 +10,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: fastapi (>=0.109.2,<0.110.0)
-Requires-Dist: spakky-core (>=0.7.4)
+Requires-Dist: spakky-core (>=0.7.7)
 Requires-Dist: websockets (>=12.0,<13.0)
 Description-Content-Type: text/markdown
 

{spakky_fastapi-0.7.4.dist-info → spakky_fastapi-0.7.7.dist-info}/RECORD

@@ -1,8 +1,9 @@
 spakky_fastapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spakky_fastapi/error.py,sha256=fAJuCZPxARY0SBhG9b-xsa2nvfXD7P49jZG1BzhRs3I,99
+spakky_fastapi/jwt_auth.py,sha256=9P_qG6yoCVQb3xYHxtY9Wdh77faJSSjUoJ6P-K5j0A0,2880
 spakky_fastapi/post_processor.py,sha256=xTKJgsX4pEMDFZRl_uVac7HTYrc2q2ylXDM3iQYVRtA,2833
 spakky_fastapi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spakky_fastapi/routing.py,sha256=3OhnnDIzaDeNhuIMWpPG17--98ti6LsgdfENsWr332g,19181
-spakky_fastapi-0.7.4.dist-info/METADATA,sha256=JScz0phDaFgydYZodog29NpEQQXhGRmoUcHom7paDzM,1660
-spakky_fastapi-0.7.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-spakky_fastapi-0.7.4.dist-info/RECORD,,
+spakky_fastapi-0.7.7.dist-info/METADATA,sha256=ts-8994w9eMn0EAv5SHXOVQ--ldDVDTIzNfOnOxQBak,1660
+spakky_fastapi-0.7.7.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+spakky_fastapi-0.7.7.dist-info/RECORD,,