GNServer 0.0.0.0.24__py3-none-any.whl → 0.0.0.0.26__py3-none-any.whl

GNServer/_client.py

@@ -101,7 +101,11 @@ class GNExceptions:
 
 
 from KeyisBClient.gn import GNRequest, GNResponse
+from KeyisBTools import TTLDict
 
+from KeyisBTools.cryptography.sign import M1
+
+m1 = M1()
 
 
 
@@ -117,9 +121,13 @@ class AsyncClient:
         self._active_connections: Dict[str, QuicClient] = {}
 
         self.__dns_client: Optional[AsyncClient] = None
+        self._dns_cache: TTLDict = TTLDict()
 
-    
-    async def getCoreDNS(self, domain: str) -> str:
+    async def getCoreDNS(self, domain: str, token: Optional[bytes] = None, use_cache: bool = True, keep_alive: bool = False) -> str:
+        if use_cache:
+            resuilt = self._dns_cache.get(domain)
+            if resuilt is not None:
+                return resuilt
 
         if ':' in domain and domain.split('.')[-1].split(':')[0].isdigit() and domain.split(':')[-1].isdigit():
             return domain
@@ -128,57 +136,73 @@ class AsyncClient:
             if self.__dns_client is None:
                 self.__dns_client = AsyncClient()
 
-            r1 = await self.__dns_client.request(GNRequest('GET', Url(f'gn://{self.__dns_core__ipv4}/getIp?d={domain}')), keep_alive=False)
+            if token is not None:
+                payload = {'sign': {'alg': 'KeyisB-c-s-m1', 'data': m1.sign(token)}}
+            else:
+                payload = None
+
+            r1 = await self.__dns_client.request(GNRequest('GET', Url(f'gn://{self.__dns_core__ipv4}/getIp?d={domain}'), payload=payload), keep_alive=keep_alive)
 
             if r1.command != 'ok':
                 raise GNExceptions.ConnectionError.dns.data
 
             r1_data = r1.payload
 
-            return r1_data['ip'] + ':' + str(r1_data['port'])
+            result = r1_data['ip'] + ':' + str(r1_data['port'])
+
+            self._dns_cache.set(domain, result, r1_data.get('ttl'))
+
+            return result
 
         except httpx.TimeoutException:
             raise GNExceptions.ConnectionError.dns.timeout
         except:
             raise GNExceptions.ConnectionError.dns.connection
 
-    
+    async def getGNDNS(self, domain: str, token: Optional[bytes] = None, use_cache: bool = True, keep_alive: bool = False) -> str:
+        if use_cache:
+            resuilt = self._dns_cache.get(domain)
+            if resuilt is not None:
+                return resuilt
 
-    async def getGNDNS(self, domain: str, token: Optional[str] = None) -> str:
-        
         if ':' in domain and domain.split('.')[-1].split(':')[0].isdigit() and domain.split(':')[-1].isdigit():
             return domain
 
-
         if self.__dns_gn__ipv4 is None:
             self.__dns_gn__ipv4 = await self.getCoreDNS('dns.gn')
 
-
         try:
             if self.__dns_client is None:
                 self.__dns_client = AsyncClient()
             
             if token is not None:
-                payload = {'token': token}
+                payload = {'sign': {'alg': 'KeyisB-c-s-m1', 'data': m1.sign(token)}}
             else:
                 payload = None
 
-            r1 = await self.__dns_client.request(GNRequest('GET', Url(f'gn://{self.__dns_gn__ipv4}/getIp?d={domain}'), payload=payload))
+            r1 = await self.__dns_client.request(GNRequest('GET', Url(f'gn://{self.__dns_gn__ipv4}/getIp?d={domain}'), payload=payload), keep_alive=keep_alive)
 
             if r1.command != 'ok':
                 raise GNExceptions.ConnectionError.dns.data
 
             r1_data = r1.payload
 
-            return r1_data['ip'] + ':' + str(r1_data['port'])
+            result = r1_data['ip'] + ':' + str(r1_data['port'])
+
+            self._dns_cache.set(domain, result, r1_data.get('ttl'))
+
+            return result
 
         except httpx.TimeoutException:
             raise GNExceptions.ConnectionError.dns.timeout
         except:
             raise GNExceptions.ConnectionError.dns.connection
 
-
-
+    async def getDNS(self, domain: str, token: Optional[str] = None, use_cache: bool = True, keep_alive: bool = False):
+        if domain.endswith(('.core', '.gw', '.gn', '.cdn', '.sys', '.gwis', '.abs')):
+            return await self.getCoreDNS(domain=domain, use_cache=use_cache, keep_alive=keep_alive)
+        else:
+            return await self.getGNDNS(domain=domain, token=token, use_cache=use_cache, keep_alive=keep_alive)
 
     def addRequestCallback(self, callback: Callable, name: str):
         self.__request_callbacks[name] = callback
@@ -199,7 +223,7 @@ class AsyncClient:
                     print('дождались')
                     if c.status == 'active':
                         return c
-                    elif c.status == 'connecting': # если очень дого подключаемся, то кидаем ошибку
+                    elif c.status == 'connecting': # если очень долго подключаемся, то кидаем ошибку
                         await self.disconnect(domain)
                         raise GNExceptions.ConnectionError.client.timeout
                     elif c.status == 'disconnect':
@@ -215,7 +239,7 @@ class AsyncClient:
         c.status = 'connecting'
         self._active_connections[domain] = c
 
-        data = await self.getCoreDNS(domain)
+        data = await self.getDNS(domain)
 
         data = data.split(':')
 

{gnserver-0.0.0.0.24.dist-info → gnserver-0.0.0.0.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GNServer
-Version: 0.0.0.0.24
+Version: 0.0.0.0.26
 Summary: GNServer
 Home-page: https://github.com/KeyisB/libs/tree/main/GNServer
 Author: KeyisB

gnserver-0.0.0.0.26.dist-info/RECORD

@@ -0,0 +1,9 @@
+GNServer/___client.py,sha256=hmeUL2Vqp-BnwJeRcLZAaIfRNVxBrRRB_AFk9ofkei4,25459
+GNServer/__init__.py,sha256=V50sMYrrPdOGuI1iJm-SW7izhX-eggDH16AHvtIKjmM,1480
+GNServer/_app.py,sha256=WYK16u-G6TDEMBa3MVaJP3-HzTh4NIcmzsyOdxaSsm0,30664
+GNServer/_client.py,sha256=TMBVo1C0nj7uSbnQHXoQDFMRB9Mb_tIc9C0zrIaOOlc,29869
+gnserver-0.0.0.0.26.dist-info/licenses/LICENSE,sha256=WH_t7dKZyWJ5Ld07eYIkUG4Tv6zZWXtAdsUqYAUesn0,1084
+gnserver-0.0.0.0.26.dist-info/METADATA,sha256=bC3T30RqCGzP7Y1QSjq_0MRXmD6gVIh0YcSLM3kl-UM,805
+gnserver-0.0.0.0.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gnserver-0.0.0.0.26.dist-info/top_level.txt,sha256=-UOUBuD4u7Qkb1o5PdcwyA3kx8xCH2lwy0tJHi26Wb4,9
+gnserver-0.0.0.0.26.dist-info/RECORD,,

GNServer/tools/__init__.py

@@ -1 +0,0 @@
-from ._models import TTLDict

GNServer/tools/_models.py

@@ -1,74 +0,0 @@
-import time
-import asyncio
-from typing import Optional
-
-
-
-class TTLDict:
-    def __init__(self, default_ttl: int = 60, cleanup_interval: int = 300):
-        """
-        :param default_ttl: TTL по умолчанию (сек), если при записи не указан
-        :param cleanup_interval: периодическая очистка от просроченных ключей (сек), по умолчанию 5 мин
-        """
-
-        self._store = {}
-        self._default_ttl = default_ttl
-        self._cleanup_interval = cleanup_interval
-        self._task = None
-
-    def set(self, key, value, ttl: Optional[int] = None):
-        if ttl is None:
-            ttl = self._default_ttl
-        expire_at = time.monotonic() + ttl
-        self._store[key] = (value, expire_at)
-
-        if self._task is None:
-            loop = asyncio.get_running_loop()
-            self._task = loop.create_task(self._cleanup_worker())
-
-    def get(self, key):
-        now = time.monotonic()
-        item = self._store.get(key)
-        if not item:
-            return None
-
-        value, expire_at = item
-        if expire_at < now:
-            del self._store[key]
-            return None
-        return value
-
-    def __setitem__(self, key, value):
-        self.set(key, value, self._default_ttl)
-
-    def __getitem__(self, key):
-        return self.get(key)
-
-    def __contains__(self, key):
-        return self.get(key) is not None
-
-    def __len__(self):
-        return len(self._store)
-
-    def __repr__(self):
-        return f"<TTLDict size={len(self._store)}>"
-
-    async def _cleanup_worker(self):
-        while True:
-            await asyncio.sleep(self._cleanup_interval)
-            self.cleanup()
-
-    def cleanup(self):
-        """Удалить все просроченные ключи"""
-        now = time.monotonic()
-        expired = [k for k, (_, exp) in self._store.items() if exp < now]
-        for k in expired:
-            self._store.pop(k, None)
-
-    async def stop(self):
-        """Остановить фон очистки"""
-        self._task.cancel()
-        try:
-            await self._task
-        except asyncio.CancelledError:
-            pass

gnserver-0.0.0.0.24.dist-info/RECORD

@@ -1,11 +0,0 @@
-GNServer/___client.py,sha256=hmeUL2Vqp-BnwJeRcLZAaIfRNVxBrRRB_AFk9ofkei4,25459
-GNServer/__init__.py,sha256=V50sMYrrPdOGuI1iJm-SW7izhX-eggDH16AHvtIKjmM,1480
-GNServer/_app.py,sha256=WYK16u-G6TDEMBa3MVaJP3-HzTh4NIcmzsyOdxaSsm0,30664
-GNServer/_client.py,sha256=9zw3DbkVHjNr-DNhlLV92r2sNJ-VY9Ahc4QjP1sUCII,28443
-GNServer/tools/__init__.py,sha256=itqkS5iBB2GEHqz8H-htqgd55rUi6utnuKVAzBBByCM,28
-GNServer/tools/_models.py,sha256=1V94cbNHQGAl5l9DJbGvvkm1gmsgTEMgkjzlnZk8ymw,2264
-gnserver-0.0.0.0.24.dist-info/licenses/LICENSE,sha256=WH_t7dKZyWJ5Ld07eYIkUG4Tv6zZWXtAdsUqYAUesn0,1084
-gnserver-0.0.0.0.24.dist-info/METADATA,sha256=kVi7DeXW2P6Zgk8y3ZcUuDNFiS_gRvrsR8exkZPJ3vM,805
-gnserver-0.0.0.0.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gnserver-0.0.0.0.24.dist-info/top_level.txt,sha256=-UOUBuD4u7Qkb1o5PdcwyA3kx8xCH2lwy0tJHi26Wb4,9
-gnserver-0.0.0.0.24.dist-info/RECORD,,