fmtr.tools 1.2.3__py3-none-any.whl → 1.2.5__py3-none-any.whl

fmtr/tools/__init__.py

@@ -29,6 +29,7 @@ from fmtr.tools.constants import Constants
 # Submodules
 from fmtr.tools.path_tools import Path, PackagePaths, AppPaths
 from fmtr.tools import ai_tools as ai
+from fmtr.tools import dns_tools as dns
 
 import fmtr.tools.setup_tools as setup
 from fmtr.tools.setup_tools import Setup, SetupPaths, Dependencies, Tools
@@ -172,11 +173,6 @@ try:
 except ImportError as exception:
     patterns = MissingExtraMockModule('patterns', exception)
 
-try:
-    from fmtr.tools import dns_tools as dns
-except ImportError as exception:
-    dns = MissingExtraMockModule('dns', exception)
-
 try:
     from fmtr.tools import http_tools as http
     from fmtr.tools.http_tools import Client

fmtr/tools/dns_tools/__init__.py

@@ -0,0 +1,6 @@
+from fmtr.tools.import_tools import MissingExtraMockModule
+
+try:
+    from fmtr.tools.dns_tools import server, client, dm
+except ImportError as exception:
+    server = client = dm = MissingExtraMockModule('dns', exception)

fmtr/tools/dns_tools/client.py

@@ -0,0 +1,78 @@
+import dns
+from dns import query
+from functools import cached_property
+from httpx_retries import Retry, RetryTransport
+
+from fmtr.tools import http_tools as http
+from fmtr.tools.dns_tools.dm import Exchange, Response, Request
+from fmtr.tools.logging_tools import logger
+
+RETRY_STRATEGY = Retry(
+    total=2,  # initial + 1 retry
+    allowed_methods={"GET", "POST"},
+    status_forcelist={502, 503, 504},
+    retry_on_exceptions=None,  # defaults to httpx.TransportError etc.
+    backoff_factor=0.25,  # short backoff (e.g. 0.25s, 0.5s)
+    max_backoff_wait=0.75,  # max total delay before giving up
+    backoff_jitter=0.1,  # small jitter to avoid retry bursts
+    respect_retry_after_header=False,  # DoH resolvers probably won't set this
+)
+
+
+class HTTPClientDoH(http.Client):
+    """
+
+    Base HTTP client for DoH-appropriate retry strategy.
+
+    """
+    TRANSPORT = RetryTransport(retry=RETRY_STRATEGY)
+
+
+class ClientBasePlain:
+    def __init__(self, host, port=53):
+        self.host = host
+        self.port = port
+
+    def resolve(self, exchange: Exchange):
+        with logger.span(f'UDP {self.host}:{self.port}'):
+            response = query.udp(q=exchange.request.message, where=self.host, port=self.port)
+            exchange.response_upstream = Response.from_message(response)
+
+
+class ClientDoH:
+    """
+
+    Base DoH client.
+
+    """
+
+    HEADERS = {"Content-Type": "application/dns-message"}
+    client = HTTPClientDoH()
+
+    def __init__(self, host, url):
+        self.host = host
+        self.url = url
+        self.bootstrap = ClientBasePlain('8.8.8.8')
+
+    @cached_property
+    def ip(self):
+        message = dns.message.make_query(self.host, dns.rdatatype.A, flags=0)
+        request = Request.from_message(message)
+        exchange = Exchange(request=request, ip=None, port=None)
+        self.bootstrap.resolve(exchange)
+        ip = next(iter(exchange.response_upstream.answer.items.keys())).address
+        return ip
+
+    def resolve(self, exchange: Exchange):
+        """
+
+        Resolve via DoH
+
+        """
+        request = exchange.request
+        headers = self.HEADERS | dict(Host=self.host)
+        url = self.url.format(host=self.ip)
+        response_doh = self.client.post(url, headers=headers, content=request.wire)
+        response_doh.raise_for_status()
+        response = Response.from_http(response_doh)
+        exchange.response_upstream = response

fmtr/tools/dns_tools/dm.py

@@ -0,0 +1,110 @@
+import dns
+import httpx
+from dataclasses import dataclass
+from dns.message import Message
+from functools import cached_property
+from typing import Self, Optional
+
+
+@dataclass
+class BaseDNSData:
+    """
+
+    DNS response object.
+
+    """
+    wire: bytes
+
+    @cached_property
+    def message(self) -> Message:
+        return dns.message.from_wire(self.wire)
+
+    @classmethod
+    def from_message(cls, message: Message) -> Self:
+        return cls(message.to_wire())
+
+
+@dataclass
+class Response(BaseDNSData):
+    """
+
+    DNS response object.
+
+    """
+
+    http: Optional[httpx.Response] = None
+
+    @classmethod
+    def from_http(cls, response: httpx.Response) -> Self:
+        self = cls(response.content, http=response)
+        return self
+
+    @cached_property
+    def answer(self):
+        return self.message.answer[-1]
+
+
+@dataclass
+class Request(BaseDNSData):
+    """
+
+    DNS request object.
+
+    """
+    wire: bytes
+
+    @cached_property
+    def question(self):
+        return self.message.question[0]
+
+    @cached_property
+    def is_valid(self):
+        return len(self.message.question) != 0
+
+    @cached_property
+    def type(self):
+        return self.question.rdtype
+
+    @cached_property
+    def type_text(self):
+        return dns.rdatatype.to_text(self.type)
+
+    @cached_property
+    def name(self):
+        return self.question.name
+
+    @cached_property
+    def name_text(self):
+        return self.name.to_text()
+
+    @cached_property
+    def blackhole(self) -> Response:
+        blackhole = dns.message.make_response(self.message)
+        blackhole.flags |= dns.flags.RA
+        blackhole.set_rcode(dns.rcode.NXDOMAIN)
+        response = Response.from_message(blackhole)
+        return response
+
+
+@dataclass
+class Exchange:
+    """
+
+    Entire DNS exchange for a DNS Proxy: request -> upstream response -> response
+
+    """
+    ip: str
+    port: int
+
+    request: Request
+    response: Optional[Response] = None
+    response_upstream: Optional[Response] = None
+
+    @classmethod
+    def from_wire(cls, wire: bytes, ip: str, port: int) -> Self:
+        request = Request(wire)
+        return cls(request=request, ip=ip, port=port)
+
+    @cached_property
+    def client(self):
+        return f'{self.ip}:{self.port}'

fmtr/tools/dns_tools/server.py

@@ -0,0 +1,96 @@
+import socket
+
+from fmtr.tools import logger
+from fmtr.tools.dns_tools.client import ClientDoH
+from fmtr.tools.dns_tools.dm import Exchange, Response
+
+
+class ServerBasePlain:
+    """
+
+    Base for starting a plain DNS server
+
+    """
+
+    def __init__(self, host, port):
+        self.host = host
+        self.port = port
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+
+    def resolve(self, exchange: Exchange):
+        raise NotImplemented
+
+    def start(self):
+        """
+
+        Listen and resolve via overridden resolve method.
+
+        """
+        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        sock.bind((self.host, self.port))
+        print(f"Listening on {self.host}:{self.port}")
+        while True:
+            data, (ip, port) = sock.recvfrom(512)
+            exchange = Exchange.from_wire(data, ip=ip, port=port)
+            self.resolve(exchange)
+            sock.sendto(exchange.response.wire, (ip, port))
+
+
+class ServerBaseDoHProxy(ServerBasePlain):
+    """
+
+    Base for a DNS Proxy server
+
+    """
+
+    def __init__(self, host, port, client: ClientDoH):
+        super().__init__(host, port)
+        self.client = client
+
+    def process_question(self, exchange: Exchange):
+        return
+
+    def process_upstream(self, exchange: Exchange):
+        return
+
+    def from_upstream(self, exchange: Exchange) -> Exchange:
+
+        request = exchange.request
+        response_doh = self.client.post(self.URL, headers=self.HEADERS, content=request.wire)
+        response_doh.raise_for_status()
+        response = Response.from_http(response_doh)
+        exchange.response_upstream = response
+
+        return exchange
+
+    def resolve(self, exchange: Exchange):
+        """
+
+        Resolve a request, processing each stage, initial question, upstream response etc.
+        Subclasses can override the relevant processing methods to implement custom behaviour.
+
+        """
+
+        request = exchange.request
+
+        with logger.span(f'Handling request for {request.name_text} from {exchange.client}...'):
+
+            if not request.is_valid:
+                raise ValueError(f'Only one question per request is supported. Got {len(request.question)} questions.')
+
+            with logger.span(f'Processing question...'):
+                self.process_question(exchange)
+            if exchange.response:
+                return
+
+            with logger.span(f'Making upstream request for {request.name_text}...'):
+                self.client.resolve(exchange)
+
+            with logger.span(f'Processing upstream response...'):
+                self.process_upstream(exchange)
+
+            if exchange.response:
+                return
+
+            exchange.response = exchange.response_upstream
+            return

fmtr/tools/pattern_tools.py

@@ -1,9 +1,8 @@
+import regex as re
 from dataclasses import dataclass
 from functools import cached_property
 from typing import List
 
-import regex as re
-
 from fmtr.tools.logging_tools import logger
 
 
@@ -158,7 +157,11 @@ class Rewriter:
 
                 previous = new
 
-            logger.debug(f'Finished rewriting: {get_history_str()}')
+        if len(history) == 1:
+            history_str = 'No rewrites performed.'
+        else:
+            history_str = get_history_str()
+        logger.debug(f'Finished rewriting: {history_str}')
 
         return previous
 

fmtr/tools/version

@@ -1 +1 @@
-1.2.3
+1.2.5

{fmtr_tools-1.2.3.dist-info → fmtr_tools-1.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fmtr.tools
-Version: 1.2.3
+Version: 1.2.5
 Summary: Collection of high-level tools to simplify everyday development tasks, with a focus on AI/ML
 Home-page: https://github.com/fmtr/tools
 Author: Frontmatter
@@ -128,60 +128,60 @@ Requires-Dist: logfire[httpx]; extra == "http"
 Provides-Extra: setup
 Requires-Dist: setuptools; extra == "setup"
 Provides-Extra: all
-Requires-Dist: deepmerge; extra == "all"
-Requires-Dist: pydantic-settings; extra == "all"
-Requires-Dist: sentence_transformers; extra == "all"
-Requires-Dist: diskcache; extra == "all"
-Requires-Dist: google-auth; extra == "all"
-Requires-Dist: faker; extra == "all"
-Requires-Dist: tinynetrc; extra == "all"
-Requires-Dist: sre_yield; extra == "all"
-Requires-Dist: html2text; extra == "all"
-Requires-Dist: json_repair; extra == "all"
-Requires-Dist: pymupdf; extra == "all"
-Requires-Dist: peft; extra == "all"
-Requires-Dist: appdirs; extra == "all"
-Requires-Dist: google-api-python-client; extra == "all"
 Requires-Dist: distributed; extra == "all"
-Requires-Dist: openai; extra == "all"
-Requires-Dist: yamlscript; extra == "all"
-Requires-Dist: fastapi; extra == "all"
+Requires-Dist: pydevd-pycharm; extra == "all"
+Requires-Dist: ollama; extra == "all"
+Requires-Dist: sre_yield; extra == "all"
+Requires-Dist: flet-webview; extra == "all"
+Requires-Dist: dask[bag]; extra == "all"
+Requires-Dist: semver; extra == "all"
+Requires-Dist: diskcache; extra == "all"
 Requires-Dist: tabulate; extra == "all"
 Requires-Dist: bokeh; extra == "all"
+Requires-Dist: openai; extra == "all"
+Requires-Dist: pytest-cov; extra == "all"
 Requires-Dist: uvicorn[standard]; extra == "all"
-Requires-Dist: pydantic-ai[logfire,openai]; extra == "all"
-Requires-Dist: transformers[sentencepiece]; extra == "all"
-Requires-Dist: docker; extra == "all"
-Requires-Dist: pymupdf4llm; extra == "all"
-Requires-Dist: torchaudio; extra == "all"
-Requires-Dist: pydevd-pycharm; extra == "all"
-Requires-Dist: regex; extra == "all"
+Requires-Dist: pyyaml; extra == "all"
 Requires-Dist: Unidecode; extra == "all"
-Requires-Dist: huggingface_hub; extra == "all"
+Requires-Dist: json_repair; extra == "all"
+Requires-Dist: google-auth-oauthlib; extra == "all"
+Requires-Dist: pymupdf4llm; extra == "all"
+Requires-Dist: google-auth; extra == "all"
+Requires-Dist: fastapi; extra == "all"
 Requires-Dist: contexttimer; extra == "all"
-Requires-Dist: pandas; extra == "all"
-Requires-Dist: flet[all]; extra == "all"
+Requires-Dist: tinynetrc; extra == "all"
 Requires-Dist: flet-video; extra == "all"
-Requires-Dist: pyyaml; extra == "all"
-Requires-Dist: google-auth-httplib2; extra == "all"
-Requires-Dist: ollama; extra == "all"
-Requires-Dist: setuptools; extra == "all"
+Requires-Dist: yamlscript; extra == "all"
+Requires-Dist: appdirs; extra == "all"
 Requires-Dist: logfire[httpx]; extra == "all"
-Requires-Dist: openpyxl; extra == "all"
-Requires-Dist: dask[bag]; extra == "all"
-Requires-Dist: semver; extra == "all"
-Requires-Dist: logfire; extra == "all"
-Requires-Dist: pytest-cov; extra == "all"
-Requires-Dist: filetype; extra == "all"
-Requires-Dist: tokenizers; extra == "all"
-Requires-Dist: flet-webview; extra == "all"
-Requires-Dist: google-auth-oauthlib; extra == "all"
+Requires-Dist: huggingface_hub; extra == "all"
+Requires-Dist: transformers[sentencepiece]; extra == "all"
+Requires-Dist: torchvision; extra == "all"
 Requires-Dist: httpx_retries; extra == "all"
+Requires-Dist: pymupdf; extra == "all"
+Requires-Dist: docker; extra == "all"
+Requires-Dist: regex; extra == "all"
+Requires-Dist: tokenizers; extra == "all"
 Requires-Dist: pydantic; extra == "all"
+Requires-Dist: logfire; extra == "all"
+Requires-Dist: filetype; extra == "all"
 Requires-Dist: logfire[fastapi]; extra == "all"
+Requires-Dist: html2text; extra == "all"
+Requires-Dist: flet[all]; extra == "all"
+Requires-Dist: torchaudio; extra == "all"
+Requires-Dist: setuptools; extra == "all"
 Requires-Dist: httpx; extra == "all"
-Requires-Dist: torchvision; extra == "all"
 Requires-Dist: dnspython[doh]; extra == "all"
+Requires-Dist: deepmerge; extra == "all"
+Requires-Dist: google-auth-httplib2; extra == "all"
+Requires-Dist: sentence_transformers; extra == "all"
+Requires-Dist: peft; extra == "all"
+Requires-Dist: google-api-python-client; extra == "all"
+Requires-Dist: openpyxl; extra == "all"
+Requires-Dist: faker; extra == "all"
+Requires-Dist: pandas; extra == "all"
+Requires-Dist: pydantic-settings; extra == "all"
+Requires-Dist: pydantic-ai[logfire,openai]; extra == "all"
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{fmtr_tools-1.2.3.dist-info → fmtr_tools-1.2.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-fmtr/tools/__init__.py,sha256=yXBnDJFPfv6qj4_KuLpMXoDZ92aiEzugLs0vcM31D5k,5770
+fmtr/tools/__init__.py,sha256=vCxKR0SymygpO-EtCGPmGZEQv5O1goM7Jr8P4m-eJsQ,5676
 fmtr/tools/api_tools.py,sha256=w8Zrp_EwN5KlUghwLoTCXo4z1irg5tAsReCqDLjASfE,2133
 fmtr/tools/async_tools.py,sha256=ewz757WcveQJd-G5SVr2JDOQVbdLGecCgl-tsBGVZz4,284
 fmtr/tools/augmentation_tools.py,sha256=-6ESbO4CDlKqVOV1J1V6qBeoBMzbFIinkDHRHnCBej0,55
@@ -8,7 +8,6 @@ fmtr/tools/data_modelling_tools.py,sha256=0BFm-F_cYzVTxftWQwORkPd0FM2BTLVh9-s0-r
 fmtr/tools/dataclass_tools.py,sha256=0Gt6KeLhtPgubo_2tYkIVqB8oQ91Qzag8OAGZDdjvMU,1209
 fmtr/tools/datatype_tools.py,sha256=3P4AWIFGkJ-UqvXlj0Jc9IvkIIgTOE9jRrOk3NVbpH8,1508
 fmtr/tools/debugging_tools.py,sha256=_xzqS0V5BpL8d06j-jVQjGgI7T020QsqVXKAKMz7Du8,2082
-fmtr/tools/dns_tools.py,sha256=7OoELFd9lZ3ULdJ6h8QwHYFNUYS9atqYrflKEzzFW-g,5461
 fmtr/tools/docker_tools.py,sha256=rdaZje2xhlmnfQqZnR7IHgRdWncTLjrJcViUTt5oEwk,617
 fmtr/tools/environment_tools.py,sha256=ZO2e8pTzbQ8jNXnmKpaZF7_3qkVM6U5iqku5YSwOszE,1849
 fmtr/tools/function_tools.py,sha256=_oW3-HZXMst2pcU-I-U7KTMmzo0g9MIQKmX-c2_NEoE,858
@@ -31,7 +30,7 @@ fmtr/tools/netrc_tools.py,sha256=PpNpz_mWlQi6VHGromKwFfTyLpHUXsd4LY6-OKLCbeI,376
 fmtr/tools/openai_tools.py,sha256=6SUgejgzUzmlKKct2_ePXntvMegu3FJgfk9x7aqtqYc,742
 fmtr/tools/packaging_tools.py,sha256=FlgOTnDRHZWQL2iR-wucTsyGEHRE-MlddKL30MPmUqE,253
 fmtr/tools/parallel_tools.py,sha256=QEb_gN1StkxsqYaH4HSjiJX8Y3gpb2uKNsOzG4uFpaM,3071
-fmtr/tools/pattern_tools.py,sha256=Tv67jftyfjp04QlRMD4pwS1IV_SIXCtDTD_uIlMZiSM,4229
+fmtr/tools/pattern_tools.py,sha256=T9f2wVi_0lPuj5npcxs0yBU91LAgWGHwtrdN6rUZKm8,4357
 fmtr/tools/pdf_tools.py,sha256=xvv9B84uAF81rFJRnXhSsxYuP42vY9ZdPVFrSMVe8G8,4069
 fmtr/tools/platform_tools.py,sha256=7p69CmAHe_sF68Fx9uVhns1k5EewTHTWgUYzkl6ZQKA,308
 fmtr/tools/process_tools.py,sha256=Ysh5Dk2QFBhXQerArjKdt7xZd3JrN5Ho02AaOjH0Nnw,1425
@@ -45,12 +44,16 @@ fmtr/tools/tabular_tools.py,sha256=tpIpZzYku1HcJrHZJL6BC39LmN3WUWVhFbK2N7nDVmE,1
 fmtr/tools/tokenization_tools.py,sha256=me-IBzSLyNYejLybwjO9CNB6Mj2NYfKPaOVThXyaGNg,4268
 fmtr/tools/tools.py,sha256=CAsApa1YwVdNE6H66Vjivs_mXYvOas3rh7fPELAnTpk,795
 fmtr/tools/unicode_tools.py,sha256=yS_9wpu8ogNoiIL7s1G_8bETFFO_YQlo4LNPv1NLDeY,52
-fmtr/tools/version,sha256=xH9bGLikMOaYuf4V5R9hGZhOeDNLzz9F4hDTDDfvL54,5
+fmtr/tools/version,sha256=Cl1xpTAk2dakEAYnido_4VW6RMyhUhB5H7iuCcdJQmw,5
 fmtr/tools/version_tools.py,sha256=yNs_CGqWpqE4jbK9wsPIi14peJVXYbhIcMqHAFOw3yE,1480
 fmtr/tools/yaml_tools.py,sha256=9kuYChqJelWQIjGlSnK4iDdOWWH06P0gp9jIcRrC3UI,1903
 fmtr/tools/ai_tools/__init__.py,sha256=JZrLuOFNV1A3wvJgonxOgz_4WS-7MfCuowGWA5uYCjs,372
 fmtr/tools/ai_tools/agentic_tools.py,sha256=acSEPFS-aguDXanWGs3fAAlRyJSYPZW7L-Kb2qDLm-I,4300
 fmtr/tools/ai_tools/inference_tools.py,sha256=2UP2gXEyOJUjyyV6zmFIYmIxUsh1rXkRH0IbFvr2bRs,11908
+fmtr/tools/dns_tools/__init__.py,sha256=PwHxnpiy6_isQfUmz_5V1hL0dcPaA6ItqvoGWW8MOfk,222
+fmtr/tools/dns_tools/client.py,sha256=zYMUc63iFRSPtCnhqx_nvsHHf-VQrLt3Vor0qxImfLs,2390
+fmtr/tools/dns_tools/dm.py,sha256=mvXacq6QJ86G0S0tkzJFhU7bOaSJytvsMNlxs5X9hfE,2236
+fmtr/tools/dns_tools/server.py,sha256=8oGCJZ-xTJhJ0gq21ro8Z0ZpfRYIdoWy7bLm9LOQcU0,2764
 fmtr/tools/entrypoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fmtr/tools/entrypoints/cache_hfh.py,sha256=fQNs4J9twQuZH_Yj98-oOvEX7-LrSUP3kO8nzw2HrHs,60
 fmtr/tools/entrypoints/ep_test.py,sha256=B8HfWISfSgw_xVX475CbJGh_QnpOe9MH65H8qGjTWbY,46
@@ -70,9 +73,9 @@ fmtr/tools/tests/test_environment.py,sha256=iHaiMQfECYZPkPKwfuIZV9uHuWe3aE-p_dN_
 fmtr/tools/tests/test_json.py,sha256=IeSP4ziPvRcmS8kq7k9tHonC9rN5YYq9GSNT2ul6Msk,287
 fmtr/tools/tests/test_path.py,sha256=AkZQa6_8BQ-VaCyL_J-iKmdf2ZaM-xFYR37Kun3k4_g,2188
 fmtr/tools/tests/test_yaml.py,sha256=jc0TwwKu9eC0LvFGNMERdgBue591xwLxYXFbtsRwXVM,287
-fmtr_tools-1.2.3.dist-info/licenses/LICENSE,sha256=FW9aa6vVN5IjRQWLT43hs4_koYSmpcbIovlKeAJ0_cI,10757
-fmtr_tools-1.2.3.dist-info/METADATA,sha256=ogcH2Hf0wFsWkOT5H3KOLMlR_7DNWGl_BxecAmUthZc,15943
-fmtr_tools-1.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fmtr_tools-1.2.3.dist-info/entry_points.txt,sha256=fSQrDGNctdQXbUxpMWYVfVQ0mhZMDyaEDG3D3a0zOSc,278
-fmtr_tools-1.2.3.dist-info/top_level.txt,sha256=LXem9xCgNOD72tE2gRKESdiQTL902mfFkwWb6-dlwEE,5
-fmtr_tools-1.2.3.dist-info/RECORD,,
+fmtr_tools-1.2.5.dist-info/licenses/LICENSE,sha256=FW9aa6vVN5IjRQWLT43hs4_koYSmpcbIovlKeAJ0_cI,10757
+fmtr_tools-1.2.5.dist-info/METADATA,sha256=XH7DX192P5cYVOFAn4DR6Yh1lkAniyNoe6GWJA4p_ug,15943
+fmtr_tools-1.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fmtr_tools-1.2.5.dist-info/entry_points.txt,sha256=fSQrDGNctdQXbUxpMWYVfVQ0mhZMDyaEDG3D3a0zOSc,278
+fmtr_tools-1.2.5.dist-info/top_level.txt,sha256=LXem9xCgNOD72tE2gRKESdiQTL902mfFkwWb6-dlwEE,5
+fmtr_tools-1.2.5.dist-info/RECORD,,

fmtr/tools/dns_tools.py

@@ -1,221 +0,0 @@
-import dns
-import httpx
-import socket
-from dataclasses import dataclass
-from dns.message import Message
-from functools import cached_property
-from httpx_retries import Retry, RetryTransport
-from typing import Optional, Self
-
-from fmtr.tools import Client, logger
-
-RETRY_STRATEGY = Retry(
-    total=2,  # initial + 1 retry
-    allowed_methods={"GET", "POST"},
-    status_forcelist={502, 503, 504},
-    retry_on_exceptions=None,  # defaults to httpx.TransportError etc.
-    backoff_factor=0.25,  # short backoff (e.g. 0.25s, 0.5s)
-    max_backoff_wait=0.75,  # max total delay before giving up
-    backoff_jitter=0.1,  # small jitter to avoid retry bursts
-    respect_retry_after_header=False,  # DoH resolvers probably won't set this
-)
-
-
-class HTTPClientDoH(Client):
-    """
-
-    Base HTTP client for DoH-appropriate retry strategy.
-
-    """
-    TRANSPORT = RetryTransport(retry=RETRY_STRATEGY)
-
-
-@dataclass
-class BaseDNSData:
-    """
-
-    DNS response object.
-
-    """
-    wire: bytes
-
-    @cached_property
-    def message(self) -> Message:
-        return dns.message.from_wire(self.wire)
-
-    @classmethod
-    def from_message(cls, message: Message) -> Self:
-        return cls(message.to_wire())
-
-
-@dataclass
-class Response(BaseDNSData):
-    """
-
-    DNS response object.
-
-    """
-
-    http: Optional[httpx.Response] = None
-
-    @classmethod
-    def from_http(cls, response: httpx.Response) -> Self:
-        self = cls(response.content, http=response)
-        return self
-
-
-@dataclass
-class Request(BaseDNSData):
-    """
-
-    DNS request object.
-
-    """
-    wire: bytes
-
-    @cached_property
-    def question(self):
-        return self.message.question[0]
-
-    @cached_property
-    def is_valid(self):
-        return len(self.message.question) != 0
-
-    @cached_property
-    def type(self):
-        return self.question.rdtype
-
-    @cached_property
-    def type_text(self):
-        return dns.rdatatype.to_text(self.type)
-
-    @cached_property
-    def name(self):
-        return self.question.name
-
-    @cached_property
-    def name_text(self):
-        return self.name.to_text()
-
-    @cached_property
-    def blackhole(self) -> Response:
-        blackhole = dns.message.make_response(self.message)
-        blackhole.flags |= dns.flags.RA
-        blackhole.set_rcode(dns.rcode.NXDOMAIN)
-        response = Response.from_message(blackhole)
-        return response
-
-
-@dataclass
-class Exchange:
-    """
-
-    Entire DNS exchange for a DNS Proxy: request -> upstream response -> response
-
-    """
-    ip: str
-    port: int
-
-    request: Request
-    response: Optional[Response] = None
-    response_upstream: Optional[Response] = None
-
-    @classmethod
-    def from_wire(cls, wire: bytes, ip: str, port: int) -> Self:
-        request = Request(wire)
-        return cls(request=request, ip=ip, port=port)
-
-    @cached_property
-    def client(self):
-        return f'{self.ip}:{self.port}'
-
-
-class BasePlain:
-    """
-
-    Base for starting a plain DNS server
-
-    """
-
-    def __init__(self, host, port):
-        self.host = host
-        self.port = port
-        self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-
-    def resolve(self, exchange: Exchange):
-        raise NotImplemented
-
-    def start(self):
-        """
-
-        Listen and resolve via overridden resolve method.
-
-        """
-        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-        sock.bind((self.host, self.port))
-        print(f"Listening on {self.host}:{self.port}")
-        while True:
-            data, (ip, port) = sock.recvfrom(512)
-            exchange = Exchange.from_wire(data, ip=ip, port=port)
-            self.resolve(exchange)
-            sock.sendto(exchange.response.wire, (ip, port))
-
-
-class BaseDoHProxy(BasePlain):
-    """
-
-    Base for a DNS Proxy server
-
-    """
-
-    URL = None
-    HEADERS = {"Content-Type": "application/dns-message"}
-    client = HTTPClientDoH()
-
-    def process_question(self, exchange: Exchange):
-        return
-
-    def process_upstream(self, exchange: Exchange):
-        return
-
-    def from_upstream(self, exchange: Exchange) -> Exchange:
-
-        request = exchange.request
-        response_doh = self.client.post(self.URL, headers=self.HEADERS, content=request.wire)
-        response_doh.raise_for_status()
-        response = Response.from_http(response_doh)
-        exchange.response_upstream = response
-
-        return exchange
-
-    def resolve(self, exchange: Exchange):
-        """
-
-        Resolve a request, processing each stage, initial question, upstream response etc.
-        Subclasses can override the relevant processing methods to implement custom behaviour.
-
-        """
-
-        request = exchange.request
-
-        with logger.span(f'Handling request for {request.name_text} from {exchange.client}...'):
-
-            if not request.is_valid:
-                raise ValueError(f'Only one question per request is supported. Got {len(request.question)} questions.')
-
-            with logger.span(f'Processing question...'):
-                self.process_question(exchange)
-            if exchange.response:
-                return
-
-            with logger.span(f'Making upstream request for {request.name_text}...'):
-                self.from_upstream(exchange)
-
-            with logger.span(f'Processing upstream response...'):
-                self.process_upstream(exchange)
-
-            if exchange.response:
-                return
-
-            exchange.response = exchange.response_upstream
-            return