Configure arbitrary NPM endpoint lists

dns_sync/config.py

@@ -79,6 +79,15 @@ def parse_endpoint_list(
     return tuple(endpoints)
 
 
+def endpoint_host(endpoint: Endpoint) -> str:
+    parsed = urlsplit(f"//{endpoint.address}")
+    if not parsed.hostname:
+        raise RuntimeError(
+            f"Invalid endpoint address: {endpoint.address!r}"
+        )
+    return parsed.hostname
+
+
 def env_bool(name: str, default: bool) -> bool:
     value = os.getenv(name)
     if value is None:
@@ -90,6 +99,7 @@ def env_bool(name: str, default: bool) -> bool:
 class AppConfig:
     source_name: str
     provider_name: str
+    npm_endpoints: tuple[Endpoint, ...]
     proxy_ip: str
     proxy_api_port: int
     proxy_api_scheme: str
@@ -101,18 +111,16 @@ class AppConfig:
 
     @property
     def proxy_url(self) -> str:
-        return f"{self.proxy_api_scheme}://{self.proxy_ip}:{self.proxy_api_port}"
+        return (
+            f"{self.proxy_api_scheme}://"
+            f"{self.npm_endpoints[0].address}"
+        )
 
     @classmethod
     def from_env(cls, include_disabled: bool = False) -> "AppConfig":
         source_name = os.getenv("PROXY", "npm").strip().lower()
         provider_name = os.getenv("DNS_PROVIDER", "verizon").strip().lower()
 
-        raw_proxy_ip = os.getenv("PROXY_IP")
-        if not raw_proxy_ip:
-            raise RuntimeError("PROXY_IP is required")
-        proxy_ip = normalize_ipv4(raw_proxy_ip, "PROXY_IP")
-
         try:
             port = int(os.getenv("PROXY_API_PORT", "81"))
         except ValueError as exc:
@@ -124,10 +132,42 @@ class AppConfig:
         if scheme not in {"http", "https"}:
             raise RuntimeError("PROXY_API_SCHEME must be http or https")
 
-        target_ip = normalize_ipv4(
-            os.getenv("DNS_TARGET_IP", proxy_ip),
-            "DNS_TARGET_IP",
-        )
+        raw_proxy_ip = os.getenv("PROXY_IP")
+        raw_npm_endpoints = os.getenv("NPM_ENDPOINTS")
+
+        if raw_npm_endpoints is None:
+            if not raw_proxy_ip:
+                raise RuntimeError(
+                    "NPM_ENDPOINTS or legacy PROXY_IP is required"
+                )
+            proxy_ip = normalize_ipv4(raw_proxy_ip, "PROXY_IP")
+            npm_endpoints = parse_endpoint_list("npm", proxy_ip, port)
+        else:
+            npm_endpoints = parse_endpoint_list(
+                "npm",
+                raw_npm_endpoints,
+                port,
+            )
+            if raw_proxy_ip:
+                proxy_ip = normalize_ipv4(raw_proxy_ip, "PROXY_IP")
+            else:
+                proxy_ip = endpoint_host(npm_endpoints[0])
+
+        raw_target_ip = os.getenv("DNS_TARGET_IP")
+        if raw_target_ip:
+            target_ip = normalize_ipv4(raw_target_ip, "DNS_TARGET_IP")
+        elif raw_proxy_ip:
+            target_ip = proxy_ip
+        elif len(npm_endpoints) == 1:
+            target_ip = normalize_ipv4(
+                endpoint_host(npm_endpoints[0]),
+                "DNS_TARGET_IP",
+            )
+        else:
+            raise RuntimeError(
+                "DNS_TARGET_IP is required when multiple NPM_ENDPOINTS "
+                "are configured without legacy PROXY_IP"
+            )
 
         domain_matching = env_bool("DOMAIN_MATCHING", False)
         raw_domain = os.getenv("DOMAIN")
@@ -147,6 +187,7 @@ class AppConfig:
         return cls(
             source_name=source_name,
             provider_name=provider_name,
+            npm_endpoints=npm_endpoints,
             proxy_ip=proxy_ip,
             proxy_api_port=port,
             proxy_api_scheme=scheme,