New Addon

Google BK
2023-04-23 18:17:03 +07:00
parent 1c7fd2b476
commit 702d080a1e
248 changed files with 72745 additions and 0 deletions
@@ -0,0 +1,58 @@
+import socket
+from typing import Any, Dict, List
+
+import aiodns
+from aiohttp.resolver import AsyncResolver
+from injector import inject, singleton
+
+from ..config import Config, Setting
+from ..logger import getLogger
+
+logger = getLogger(__name__)
+
+TTL_HOURS = 12
+
+
+@singleton
+class Resolver(AsyncResolver):
+    @inject
+    def __init__(self, config: Config):
+        super().__init__()
+        self.config = config
+        self._original_dns = self._resolver
+        self.setAlternateResolver()
+        config.subscribe(self.updateConfig)
+
+    async def resolve(self, host: str, port: int = 0,
+                      family: int = socket.AF_INET) -> List[Dict[str, Any]]:
+        if host == self.config.get(Setting.DRIVE_HOST_NAME) and len(self.config.get(Setting.DRIVE_IPV4)) > 0 and family == 0:
+            # return the "mocked" drive address instead.
+            return [{
+                'family': 0,
+                'flags': socket.AddressInfo.AI_NUMERICHOST,
+                'port': port,
+                'proto': 0,
+                'host': self.config.get(Setting.DRIVE_IPV4),
+                'hostname': host
+            }]
+        addresses = await super().resolve(host, port, family)
+        return addresses
+
+    def updateConfig(self):
+        if self._alt_ns != self.config.get(Setting.ALTERNATE_DNS_SERVERS):
+            self.setAlternateResolver()
+            self._resolver = self._alt_dns
+
+    def setAlternateResolver(self):
+        if len(self.config.get(Setting.ALTERNATE_DNS_SERVERS)) > 0:
+            self._alt_dns = aiodns.DNSResolver(loop=self._loop, nameservers=self.config.get(
+                Setting.ALTERNATE_DNS_SERVERS).split(","))
+        else:
+            self._alt_dns = self._original_dns
+        self._alt_ns = self.config.get(Setting.ALTERNATE_DNS_SERVERS)
+
+    def toggle(self):
+        if self._resolver == self._original_dns:
+            self._resolver = self._alt_dns
+        else:
+            self._resolver = self._original_dns