jamlib 3.0.0b13.dev0__tar.gz → 3.0.0b13.dev1__tar.gz

{jamlib-3.0.0b13.dev0/src/jamlib.egg-info → jamlib-3.0.0b13.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jamlib
-Version: 3.0.0b13.dev0
+Version: 3.0.0b13.dev1
 Summary: Simple and universal library for authorization.
 Author-email: Makridenko Adrian <adrianmakridenko@duck.com>, Ksenia Travnikova <kseniatravnikova@duck.com>
 License: Apache-2.0

{jamlib-3.0.0b13.dev0 → jamlib-3.0.0b13.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "jamlib"
-version = "3.0.0b13.dev0"
+version = "3.0.0b13.dev1"
 description = "Simple and universal library for authorization."
 authors = [
     {name = "Makridenko Adrian",email = "adrianmakridenko@duck.com"},

{jamlib-3.0.0b13.dev0 → jamlib-3.0.0b13.dev1}/src/jam/__init__.py

@@ -12,5 +12,5 @@ from jam.encoders import JsonEncoder
 from jam.instance import Jam
 
 
-__version__ = "3.0.0b13.dev0"
+__version__ = "3.0.0b13.dev1"
 __all__ = ["Jam", "JsonEncoder", "BaseJam", "BaseEncoder"]

jamlib-3.0.0b13.dev1/src/jam/ext/starlette/__init__.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+
+"""Starlette integration.
+
+Starlette docs: https://starlette.dev
+"""
+
+from .objects import BaseUser, SimpleUser
+from .backends import JWTBackend
+
+
+__all__ = [
+    "BaseUser",
+    "SimpleUser",
+    "JWTBackend",
+]

{jamlib-3.0.0b13.dev0 → jamlib-3.0.0b13.dev1}/src/jam/ext/starlette/backends.py

@@ -3,9 +3,17 @@
 from collections.abc import Callable
 from typing import Any
 
-from starlette.authentication import AuthenticationBackend, BaseUser
+from starlette.authentication import (
+    AuthCredentials,
+    AuthenticationBackend,
+    UnauthenticatedUser,
+)
+from starlette.authentication import BaseUser as StarletteBaseUser
+from starlette.requests import HTTPConnection
 
 from jam.exceptions import JamStarlettePluginConfigError
+from jam.ext.starlette.objects import BaseUser, SimpleUser
+from jam.jwt import JWT
 from jam.utils.config_maker import GENERIC_POINTER, __config_maker__
 
 
@@ -21,7 +29,7 @@ class BaseBackend(AuthenticationBackend):
         pointer: str = GENERIC_POINTER,
         cookie_name: str | None = None,
         header_name: str | None = None,
-        user: type[BaseUser] | None = None,
+        user: type[BaseUser] = SimpleUser,
         **kwargs,
     ) -> None:
         if not cookie_name and not header_name:
@@ -37,6 +45,25 @@ class BaseBackend(AuthenticationBackend):
         self._user = user
         self._config_setup(config, pointer, kwargs)
 
+    def _get_auth_token(self, connection: HTTPConnection) -> str | None:
+        if self._cookie_name:
+            token = connection.cookies.get(self._cookie_name, None)
+        elif self._header_name:
+            token_bear = connection.headers.get(self._header_name, None)
+            if token_bear:
+                token = token_bear.split("Bearer ")[1]  # noqa: E701
+            else:
+                token = None  # noqa: E701
+        else:
+            raise JamStarlettePluginConfigError(
+                message="cookie_name or header_name must be provided.",
+                details={
+                    "cookie_name": self._cookie_name,
+                    "header_name": self._header_name,
+                },
+            )
+        return token
+
     def _config_setup(
         self,
         config: dict[str, Any] | str | None,
@@ -63,3 +90,21 @@ class BaseBackend(AuthenticationBackend):
                     "error": str(e),
                 },
             )
+
+
+class JWTBackend(BaseBackend):
+    """JWT Backend for litestar."""
+
+    MODULE = JWT
+    _CONFIG_KEY = "jwt"
+
+    async def authenticate(  # noqa
+        self, conn: HTTPConnection
+    ) -> tuple[AuthCredentials, StarletteBaseUser] | None:
+        token = self._get_auth_token(conn)
+        if token:
+            data = self._auth.decode(token)
+            user = self._user.from_payload(data)
+            return AuthCredentials(["authenticated"]), user
+
+        return AuthCredentials(None), UnauthenticatedUser()

jamlib-3.0.0b13.dev1/src/jam/ext/starlette/objects.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from typing import Any, TypeVar
+
+from starlette.authentication import BaseUser as StarletteBaseUser
+
+
+User = TypeVar("User", bound="BaseUser")
+
+
+class BaseUser(ABC, StarletteBaseUser):
+    """BaseUser with builder from payload."""
+
+    @classmethod
+    @abstractmethod
+    def from_payload(cls, payload: dict[str, Any]) -> User:
+        """Create instance by payload.
+
+        Args:
+            payload (dict[str, Any]): Payload from auth
+
+        Example:
+        ```python
+
+        class MyUser(JamBaseUser):
+            id: int
+            username: str
+
+            @classmethod
+            def from_payload(cls, payload):
+                return cls(
+                    id=payload["id"],
+                    username=payload["username"]
+                )
+
+        payload = jwt.decode(token)
+        user = MyUser.from_payload(payload)
+        ```
+        """
+        raise NotImplementedError
+
+
+@dataclass(slots=True)
+class SimpleUser(BaseUser):
+    """Simple user with payload only."""
+
+    payload: dict[str, Any]
+
+    @classmethod
+    def from_payload(cls, payload: dict[str, Any]):  # noqa
+        return cls(payload=payload)

{jamlib-3.0.0b13.dev0 → jamlib-3.0.0b13.dev1/src/jamlib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jamlib
-Version: 3.0.0b13.dev0
+Version: 3.0.0b13.dev1
 Summary: Simple and universal library for authorization.
 Author-email: Makridenko Adrian <adrianmakridenko@duck.com>, Ksenia Travnikova <kseniatravnikova@duck.com>
 License: Apache-2.0

{jamlib-3.0.0b13.dev0 → jamlib-3.0.0b13.dev1}/src/jamlib.egg-info/SOURCES.txt

@@ -42,6 +42,7 @@ src/jam/ext/litestar/objects.py
 src/jam/ext/litestar/plugins.py
 src/jam/ext/starlette/__init__.py
 src/jam/ext/starlette/backends.py
+src/jam/ext/starlette/objects.py
 src/jam/jwt/__algorithms__.py
 src/jam/jwt/__base__.py
 src/jam/jwt/__init__.py

jamlib-3.0.0b13.dev0/src/jam/ext/starlette/__init__.py

@@ -1,6 +0,0 @@
-# -*- coding: utf-8 -*-
-
-"""Starlette integration.
-
-Starlette docs: https://starlette.dev
-"""