xn-auth 0.2.39__tar.gz → 0.2.40__tar.gz

{xn_auth-0.2.39/xn_auth.egg-info → xn_auth-0.2.40}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xn-auth
-Version: 0.2.39
+Version: 0.2.40
 Summary: Auth adapter for XN-Api framework
 Author-email: Artemiev <mixartemev@gmail.com>
 License: MIT

{xn_auth-0.2.39 → xn_auth-0.2.40}/x_auth/controller.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
 
-from aiogram.utils.web_app import WebAppInitData, safe_parse_webapp_init_data
+from aiogram.utils.auth_widget import check_signature
+from aiogram.utils.web_app import WebAppInitData, safe_parse_webapp_init_data, WebAppUser
 from litestar import Response, post
 from litestar.connection import ASGIConnection
 from litestar.exceptions import NotAuthorizedException
@@ -8,7 +9,7 @@ from litestar.security.jwt import JWTCookieAuth
 
 from x_auth.middleware import JWTAuthMiddleware, Tok
 from x_auth.models import User
-from x_auth.types import AuthUser
+from x_auth.types import AuthUser, TgUser
 
 
 async def retrieve_user_handler(token: Tok, _cn: ASGIConnection) -> AuthUser:
@@ -33,20 +34,32 @@ class Auth:
             exclude=["/schema", "/auth", "/public"] + (exc_paths or []),
         )
 
-        @post("/auth/tma", tags=["Auth"], description="Gen JWToken from tg initData")
-        async def tma(tid: str) -> Response[user_model.in_type(True)]:
-            try:
-                twaid: WebAppInitData = safe_parse_webapp_init_data(self.jwt.token_secret, tid)
-            except ValueError:
-                raise NotAuthorizedException(detail="Tg Initdata invalid")
-            user_in = await user_model.tg2in(twaid.user)
+        async def user_proc(user: WebAppUser) -> Response[WebAppUser]:
+            user_in = await user_model.tg2in(user)
             db_user, cr = await user_model.update_or_create(**user_in.df_unq())  # on login: update user in db from tg
             res = self.jwt.login(
                 identifier=str(db_user.id),
                 token_extras={"role": db_user.role, "blocked": db_user.blocked},
-                response_body=user_model.validate(dict(db_user)),
+                response_body=user,
             )
             res.cookies[0].httponly = False
             return res
 
-        self.handler = tma
+        # login for api endpoint
+        @post("/auth/twa", tags=["Auth"], description="Gen JWToken from tg login widget")
+        async def twa(data: TgUser) -> Response[WebAppUser]:  # widget
+            dct = data.dump()
+            if not check_signature(self.jwt.token_secret, dct.pop("hash"), **dct):
+                raise NotAuthorizedException("Tg login widget data invalid")
+            return await user_proc(WebAppUser(**dct))
+
+        @post("/auth/tma", tags=["Auth"], description="Gen JWToken from tg initData")
+        async def tma(tid: str) -> Response[WebAppUser]:
+            try:
+                twaid: WebAppInitData = safe_parse_webapp_init_data(self.jwt.token_secret, tid)
+            except ValueError:
+                raise NotAuthorizedException(detail="Tg Initdata invalid")
+            return await user_proc(twaid.user)
+
+        self.tma_handler = tma
+        self.twa_handler = twa

{xn_auth-0.2.39 → xn_auth-0.2.40}/x_auth/models.py

@@ -49,6 +49,7 @@ class User(Model):
     username: OneToOneRelation[Username] = OneToOneField("models.Username", "user")
     username_id: int
     first_name: str | None = CharField(63)
+    pic: str | None = CharField(127, null=True)
     last_name: str | None = CharField(31, null=True)
     blocked: bool = BooleanField(default=False)
     lang: Lang | None = IntEnumField(Lang, default=Lang.ru, null=True)
@@ -65,6 +66,7 @@ class User(Model):
                 "last_name": u.last_name,
                 "username_id": un.id,
                 "lang": u.language_code and Lang[u.language_code],
+                "pic": u.photo_url and u.photo_url.replace("https://t.me/i/userpic/320/", "")[:-4],
             }
         )
         if blocked is not None:

xn_auth-0.2.40/x_auth/types.py

@@ -0,0 +1,61 @@
+from datetime import datetime
+from json import dumps
+from typing import Literal, Self
+
+from msgspec import Struct, to_builtins, convert
+
+from x_auth.enums import Role
+
+
+class Xs(Struct):
+    def dump(self, nones: bool = False) -> dict:
+        return {k: v for k, v in to_builtins(self).items() if nones or v is not None}
+
+    def json(self, nones: bool = False) -> str:
+        return dumps(self.dump())
+
+    @classmethod
+    def load(cls, obj, **kwargs) -> Self:
+        dct = dict(obj)
+        return convert({**dct, **kwargs}, cls)  # , strict=False
+
+
+class AuthUser(Struct):
+    id: int
+    blocked: bool
+    role: Role
+
+
+class Proxy(Struct):
+    id: str
+    username: str
+    password: str
+    proxy_address: str
+    port: int
+    valid: bool
+    last_verification: datetime
+    country_code: str
+    city_name: str
+    created_at: datetime
+
+
+class Replacement(Struct):
+    id: int
+    reason: Literal["auto_invalidated", "auto_out_of_rotation"]
+    replaced_with: str
+    replaced_with_port: int
+    replaced_with_country_code: str
+    proxy: str
+    proxy_port: int
+    proxy_country_code: str
+    created_at: datetime
+
+
+class TgUser(Xs):
+    id: int
+    first_name: str
+    auth_date: int
+    hash: str
+    username: str | None = None
+    photo_url: str | None = None
+    last_name: str | None = None

{xn_auth-0.2.39 → xn_auth-0.2.40/xn_auth.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xn-auth
-Version: 0.2.39
+Version: 0.2.40
 Summary: Auth adapter for XN-Api framework
 Author-email: Artemiev <mixartemev@gmail.com>
 License: MIT

xn_auth-0.2.39/x_auth/types.py

@@ -1,37 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from msgspec import Struct
-
-from x_auth.enums import Role
-
-
-class AuthUser(Struct):
-    id: int
-    blocked: bool
-    role: Role
-
-
-class Proxy(Struct):
-    id: str
-    username: str
-    password: str
-    proxy_address: str
-    port: int
-    valid: bool
-    last_verification: datetime
-    country_code: str
-    city_name: str
-    created_at: datetime
-
-
-class Replacement(Struct):
-    id: int
-    reason: Literal["auto_invalidated", "auto_out_of_rotation"]
-    replaced_with: str
-    replaced_with_port: int
-    replaced_with_country_code: str
-    proxy: str
-    proxy_port: int
-    proxy_country_code: str
-    created_at: datetime