feat(SDK): add Execute.add_from_receipt (#14)

Author: fubuloubu

ruffsack/messages/execute.py

@@ -25,6 +25,9 @@ class ExecuteBase(EIP712Message):
 
 
 class Execute(ManagerAccessMixin):
+    MAX_CALLS = 8
+    MAX_CALLDATA_SIZE = 16_388
+
     def __init__(
         self,
         sack: "Ruffsack | None" = None,
@@ -54,6 +57,17 @@ class Execute(ExecuteBase):
         self.message = Execute(parent=parent)
 
     def add_raw(self, target: "AddressType", value: int = 0, data: bytes = b"") -> Self:
+        if len(self.message.calls) >= self.MAX_CALLS:
+            raise RuntimeError(
+                "Ruffsack does not support more than 8 calls per execute transaction."
+            )
+
+        if len(data) > self.MAX_CALLDATA_SIZE:
+            raise RuntimeError(
+                "Ruffsack calls do not support data field larger than"
+                f" {self.MAX_CALLDATA_SIZE} bytes."
+            )
+
         self.message.calls.append(Call(target=target, value=value, data=data))
         return self
 
@@ -64,6 +78,13 @@ def add(self, call, *args, value: int = 0) -> Self:
             data=call.encode_input(*args),
         )
 
+    def add_from_receipt(self, receipt: "ReceiptAPI") -> Self:
+        return self.add_raw(
+            target=receipt.receiver,
+            value=receipt.value,
+            data=receipt.data,
+        )
+
     def __call__(
         self, sack: "Ruffsack | None" = None, **txn_args
     ) -> "ReceiptAPI | None":

tests/test_execute.py

@@ -0,0 +1,25 @@
+import pytest
+from ape.utils import ZERO_ADDRESS
+from packaging.version import Version
+from ruffsack.messages import Execute
+
+
+def test_size_limits():
+    # NOTE: Use `Execute` directly to avoid parametrized fixture setup
+    txn = Execute(
+        parent=b"\x00" * 32,
+        version=Version("0.1"),
+        address=ZERO_ADDRESS,
+        chain_id=1,
+    )
+
+    with pytest.raises(RuntimeError):
+        # Can't add `.data` larger than `Execute.MAX_CALLDATA_SIZE`
+        txn.add_raw(ZERO_ADDRESS, data=b"\x00" * (Execute.MAX_CALLDATA_SIZE + 1))
+
+    for _ in range(Execute.MAX_CALLS):
+        txn.add_raw(ZERO_ADDRESS, data=b"\x00" * Execute.MAX_CALLDATA_SIZE)
+
+    with pytest.raises(RuntimeError):
+        # Can't add more than `Execute.MAX_CALLS`
+        txn.add_raw(ZERO_ADDRESS, data=b"\x00" * Execute.MAX_CALLDATA_SIZE)