astreum 0.2.9__tar.gz → 0.2.11__tar.gz

{astreum-0.2.9/src/astreum.egg-info → astreum-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: astreum
-Version: 0.2.9
+Version: 0.2.11
 Summary: Python library to interact with the Astreum blockchain and its Lispeum virtual machine.
 Author-email: "Roy R. O. Okello" <roy@stelar.xyz>
 Project-URL: Homepage, https://github.com/astreum/lib

{astreum-0.2.9 → astreum-0.2.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "astreum"
-version = "0.2.9"
+version = "0.2.11"
 authors = [
   { name="Roy R. O. Okello", email="roy@stelar.xyz" },
 ]

{astreum-0.2.9 → astreum-0.2.11}/src/astreum/node.py

@@ -403,7 +403,102 @@ class Transaction:
 
     def __hash__(self) -> int:
         return int.from_bytes(self.tx_hash, 'big')
+
+class Block:
+    def __init__(
+        self,
+        *,
+        number: int,
+        prev_block_hash: bytes,
+        timestamp: int,
+        accounts_hash: bytes,
+        total_astre_burned: int,
+        tx_limit: int,
+        tx_root: bytes,
+        vdf_difficulty: int,
+        vdf_output: bytes,
+        vdf_proof: bytes,
+        validator_pk: bytes,
+        signature: bytes,
+    ) -> None:
+        self.accounts_hash = accounts_hash
+        self.number = int(number)
+        self.prev_block_hash = prev_block_hash
+        self.timestamp = int(timestamp)
+        self.total_astre_burned = int(total_astre_burned)
+        self.tx_limit = int(tx_limit)
+        self.tx_root = tx_root
+        self.validator_pk = validator_pk
+        self.vdf_difficulty = int(vdf_difficulty)
+        self.vdf_output = vdf_output
+        self.vdf_proof = vdf_proof
+        self.signature = signature
+        self.body_hash = self._compute_body_hash()
+
+    def _body_fields_without_sig(self) -> list:
+        return [
+            self.accounts_hash,
+            self.number,
+            self.prev_block_hash,
+            self.timestamp,
+            self.total_astre_burned,
+            self.tx_limit,
+            self.tx_root,
+            self.validator_pk,
+            self.vdf_difficulty,
+            self.vdf_output,
+            self.vdf_proof,
+        ]
+
+    def _compute_body_hash(self) -> bytes:
+        return blake3.blake3(encode(self._body_fields_without_sig())).digest()
     
+    def to_bytes(self) -> bytes:
+        return encode(self._body_fields_without_sig() + [self.signature])
+
+    @classmethod
+    def from_bytes(cls, blob: bytes) -> "Block":
+        (
+            accounts_hash,
+            number,
+            prev_block_hash,
+            timestamp,
+            total_astre_burned,
+            tx_limit,
+            tx_root,
+            validator_pk,
+            vdf_difficulty,
+            vdf_output,
+            vdf_proof,
+            signature
+        ) = decode(blob)
+        return cls(
+            number=int(number),
+            prev_block_hash=prev_block_hash,
+            timestamp=int(timestamp),
+            accounts_hash=accounts_hash,
+            total_astre_burned=int(total_astre_burned),
+            tx_limit=int(tx_limit),
+            tx_root=tx_root,
+            vdf_difficulty=int(vdf_difficulty),
+            vdf_output=vdf_output,
+            vdf_proof=vdf_proof,
+            validator_pk=validator_pk,
+            signature=signature,
+        )
+
+    @property
+    def hash(self) -> bytes:
+        return blake3.blake3(self.body_hash + self.signature).digest()
+
+    def verify_block_signature(self) -> bool:
+        try:
+            pub = ed25519.Ed25519PublicKey.from_public_bytes(self.validator_pk)
+            pub.verify(self.signature, self.body_hash)
+            return True
+        except Exception:
+            return False
+  
 class Node:
     def __init__(self, config: dict = {}):
         self._machine_setup()

astreum-0.2.11/src/astreum/utils/patricia.py

@@ -0,0 +1,45 @@
+import blake3
+from typing import Callable, Dict, List, Optional
+from astreum import format
+
+EMPTY_HASH = b"\x00" * 32
+
+class PatriciaNode:
+    def __init__(
+        self,
+        key_len: int,
+        key_bits: bytes,
+        value: bytes,
+        children: List[bytes]|None = None
+    ):
+        self.key_len   = key_len
+        self.key_bits  = key_bits
+        self.value     = value
+        self.children  = children
+        self._hash: bytes | None = None
+
+    def to_bytes(self) -> bytes:
+        key_field = bytes([self.key_len]) + self.key_bits
+        return format.encode([key_field, self.value, self.children])
+
+    @classmethod
+    def from_bytes(cls, blob: bytes) -> "PatriciaNode":
+        key_field, value, children = format.decode(blob)
+        key_len   = key_field[0]
+        key_bits  = key_field[1:]
+        return cls(key_len, key_bits, value, children)
+    
+    def hash(self) -> bytes:
+        if self._hash is None:
+            self._hash = blake3.blake3(self.to_bytes()).digest()
+        return self._hash
+
+class PatriciaTrie:
+    def __init__(
+        self,
+        node_get: Callable[[bytes], Optional[bytes]],
+        root_hash: Optional[bytes] = None,
+    ) -> None:
+        self._node_get = node_get
+        self.nodes: Dict[bytes, bytes] = {}
+        self.root_hash: Optional[bytes] = root_hash

{astreum-0.2.9 → astreum-0.2.11/src/astreum.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: astreum
-Version: 0.2.9
+Version: 0.2.11
 Summary: Python library to interact with the Astreum blockchain and its Lispeum virtual machine.
 Author-email: "Roy R. O. Okello" <roy@stelar.xyz>
 Project-URL: Homepage, https://github.com/astreum/lib

{astreum-0.2.9 → astreum-0.2.11}/src/astreum.egg-info/SOURCES.txt

@@ -21,4 +21,6 @@ src/astreum/crypto/x25519.py
 src/astreum/lispeum/__init__.py
 src/astreum/lispeum/parser.py
 src/astreum/lispeum/tokenizer.py
+src/astreum/utils/__init__.py
+src/astreum/utils/patricia.py
 tests/test_node_machine.py