dominusnode-haystack 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

dominusnode_haystack/tools.py

@@ -42,6 +42,7 @@ Requires: httpx (``pip install httpx``), haystack-ai (``pip install haystack-ai`
 
 from __future__ import annotations
 
+import hashlib
 import ipaddress
 import json
 import math
@@ -381,6 +382,46 @@ def _sanitize_error(message: str) -> str:
     return _CREDENTIAL_RE.sub("***", message)
 
 
+def _count_leading_zero_bits(data: bytes) -> int:
+    """Count leading zero bits in a byte array."""
+    count = 0
+    for byte in data:
+        if byte == 0:
+            count += 8
+        else:
+            mask = 0x80
+            while mask and not (byte & mask):
+                count += 1
+                mask >>= 1
+            break
+    return count
+
+
+def _solve_pow(base_url: str) -> Optional[dict]:
+    """Solve a Proof-of-Work challenge for CAPTCHA-free registration."""
+    try:
+        pow_url = f"{base_url.rstrip('/')}/api/auth/pow/challenge"
+        with httpx.Client(timeout=30.0, follow_redirects=False) as client:
+            resp = client.post(pow_url, headers={"Content-Type": "application/json"})
+            if resp.status_code >= 400:
+                return None
+            challenge = resp.json()
+        prefix = challenge.get("prefix", "")
+        difficulty = challenge.get("difficulty", 20)
+        challenge_id = challenge.get("challengeId", "")
+        if not prefix or not challenge_id:
+            return None
+        nonce = 0
+        while nonce < 100_000_000:
+            h = hashlib.sha256((prefix + str(nonce)).encode()).digest()
+            if _count_leading_zero_bits(h) >= difficulty:
+                return {"challengeId": challenge_id, "nonce": str(nonce)}
+            nonce += 1
+        return None
+    except Exception:
+        return None
+
+
 # ---------------------------------------------------------------------------
 # Prototype pollution prevention
 # ---------------------------------------------------------------------------
@@ -1858,11 +1899,17 @@ class DominusNodeToolkit:
             return {"error": err}
 
         try:
+            body: dict = {"email": email, "password": password}
+            pow_result = _solve_pow(self.base_url)
+            if pow_result:
+                body["pow"] = pow_result
             result = self._api_request_public(
                 "POST",
                 "/api/auth/register",
-                {"email": email, "password": password},
+                body,
             )
+            if pow_result and isinstance(result, dict):
+                result["_pow_verified"] = True
             return result
         except Exception as e:
             return {"error": _sanitize_error(str(e))}

{dominusnode_haystack-1.2.0.dist-info → dominusnode_haystack-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dominusnode-haystack
-Version: 1.2.0
+Version: 1.2.1
 Summary: Dominus Node rotating proxy tools for Haystack AI pipelines
 Author-email: Dominus Node <dev@dominusnode.com>
 License: MIT

dominusnode_haystack-1.2.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+dominusnode_haystack/__init__.py,sha256=T_yPVVK5_muWIMrtagjrstVox8ycVJad5bg9U5QvwnI,163
+dominusnode_haystack/tools.py,sha256=ga1x6wXFBj0U77H5shT4usK5IK5vHEJLzjD7pUQy9sA,124990
+dominusnode_haystack-1.2.1.dist-info/licenses/LICENSE,sha256=NBkjfgN0tpbJjWf7uqsPCOMgq7uuSEW4uaGDCpB8HbI,1069
+dominusnode_haystack-1.2.1.dist-info/METADATA,sha256=DqQLd-0ISlpmYdN_IolcLp2Kf4C7Udm8JPr0SEiBIQs,968
+dominusnode_haystack-1.2.1.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+dominusnode_haystack-1.2.1.dist-info/top_level.txt,sha256=fQyiWIOvfOKV6wVVcQDoakEiJfQ3XumGBoHBrovFD0U,21
+dominusnode_haystack-1.2.1.dist-info/RECORD,,

dominusnode_haystack-1.2.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-dominusnode_haystack/__init__.py,sha256=T_yPVVK5_muWIMrtagjrstVox8ycVJad5bg9U5QvwnI,163
-dominusnode_haystack/tools.py,sha256=9-9mcgUFXzdY86Nj7J5iBqOb-RV8Hb6KIGKRuZtBFec,123338
-dominusnode_haystack-1.2.0.dist-info/licenses/LICENSE,sha256=NBkjfgN0tpbJjWf7uqsPCOMgq7uuSEW4uaGDCpB8HbI,1069
-dominusnode_haystack-1.2.0.dist-info/METADATA,sha256=lZRuo_moycs4WeKcDBvl6eEymWqcPZWsLjia6q6LGd4,968
-dominusnode_haystack-1.2.0.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-dominusnode_haystack-1.2.0.dist-info/top_level.txt,sha256=fQyiWIOvfOKV6wVVcQDoakEiJfQ3XumGBoHBrovFD0U,21
-dominusnode_haystack-1.2.0.dist-info/RECORD,,