add folder

nostr/pow.py

@@ -0,0 +1,54 @@
+import time
+from .event import Event
+from .key import PrivateKey
+
+def zero_bits(b: int) -> int:
+    n = 0
+
+    if b == 0:
+        return 8
+
+    while b >> 1:
+        b = b >> 1
+        n += 1
+
+    return 7 - n
+
+def count_leading_zero_bits(hex_str: str) -> int:
+    total = 0
+    for i in range(0, len(hex_str) - 2, 2):
+        bits = zero_bits(int(hex_str[i:i+2], 16))
+        total += bits
+
+        if bits != 8:
+            break
+
+    return total
+
+def mine_event(content: str, difficulty: int, public_key: str, kind: int, tags: list=[]) -> Event:
+    all_tags = [["nonce", "1", str(difficulty)]]
+    all_tags.extend(tags)
+
+    created_at = int(time.time())
+    event_id = Event.compute_id(public_key, created_at, kind, all_tags, content)
+    num_leading_zero_bits = count_leading_zero_bits(event_id)
+
+    attempts = 1
+    while num_leading_zero_bits < difficulty:
+        attempts += 1
+        all_tags[0][1] = str(attempts)
+        created_at = int(time.time())
+        event_id = Event.compute_id(public_key, created_at, kind, all_tags, content)
+        num_leading_zero_bits = count_leading_zero_bits(event_id)
+
+    return Event(public_key, content, created_at, kind, all_tags, event_id)
+
+def mine_key(difficulty: int) -> PrivateKey:
+    sk = PrivateKey()
+    num_leading_zero_bits = count_leading_zero_bits(sk.public_key.hex())
+
+    while num_leading_zero_bits < difficulty:
+        sk = PrivateKey()
+        num_leading_zero_bits = count_leading_zero_bits(sk.public_key.hex())
+
+    return sk