Merge pull request #20 from cffls/plutus

Plutus functionalities

Author: cffls

integration-test/configs/local-alonzo/alonzo-genesis.json

@@ -1,9 +1,5 @@
 {
     "collateralPercentage": 1,
-    "maxBlockExUnits": {
-        "exUnitsMem": 500000000000,
-        "exUnitsSteps": 500000000000
-    },
     "maxCollateralInputs": 5,
     "maxValueSize": 4000,
     "costModels": {
@@ -177,12 +173,24 @@
         }
     },
     "executionPrices": {
-        "prMem": 0.1,
-        "prSteps": 0.1
+        "prSteps":
+        {
+            "numerator" :   721,
+            "denominator" : 10000000
+            },
+            "prMem":
+        {
+            "numerator" :   577,
+            "denominator" : 10000
+        }
     },
     "lovelacePerUTxOWord": 1,
     "maxTxExUnits": {
-        "exUnitsMem": 500000000000,
-        "exUnitsSteps": 500000000000
+        "exUnitsMem":   10000000,
+        "exUnitsSteps": 10000000000
+    },
+    "maxBlockExUnits": {
+        "exUnitsMem":   50000000,
+        "exUnitsSteps": 40000000000
     }
 }

integration-test/plutus_scripts/fortytwo.plutus

@@ -0,0 +1 @@
+59082459082101000033232332233223232333222323332223233333333222222223233322232333322223232332232333222323332223232332233223232333332222233223322332233223322332222223232533530313330063333573466e1cd55ce9baa0054800081248c98d4c120cd5ce0260248238231bad0043333573466e1cd55cea8012400046601264646464646464646464646666ae68cdc39aab9d500a480008cccccccccc05ccd40948c8c8cccd5cd19b8735573aa004900011980e981c1aba15002302a357426ae8940088c98d4c160cd5ce02e02c82b82b09aab9e5001137540026ae854028cd4094098d5d0a804999aa8163ae502b35742a010666aa058eb940acd5d0a80399a8128209aba15006335025335505204a75a6ae854014c8c8c8cccd5cd19b8735573aa0049000119a80f9919191999ab9a3370e6aae7540092000233502733504075a6ae854008c114d5d09aba25002232635305c3357380c00ba0b60b426aae7940044dd50009aba150023232323333573466e1cd55cea80124000466a04a66a080eb4d5d0a80118229aba135744a004464c6a60b866ae7018017416c1684d55cf280089baa001357426ae8940088c98d4c160cd5ce02e02c82b82b09aab9e5001137540026ae854010cd4095d71aba15003335025335505275c40026ae854008c0dcd5d09aba2500223263530543357380b00aa0a60a426ae8940044d5d1280089aba25001135744a00226ae8940044d5d1280089aba25001135744a00226aae7940044dd50009aba150023232323333573466e1d400520062301c3039357426aae79400c8cccd5cd19b875002480108c06cc10cd5d09aab9e500423333573466e1d400d20022301b302e357426aae7940148cccd5cd19b875004480008c078dd71aba135573ca00c464c6a609e66ae7014c14013813413012c1284d55cea80089baa001357426ae8940088c98d4c120cd5ce026024823823082409931a982399ab9c49010350543500048046135573ca00226ea80048848cc00400c0088004888888888848cccccccccc00402c02802402001c01801401000c00880048848cc00400c008800448848cc00400c0084800448848cc00400c0084800448848cc00400c00848004848888c010014848888c00c014848888c008014848888c004014800448c88c008dd6000990009aa81a911999aab9f0012500e233500d30043574200460066ae880080cc8c8c8c8cccd5cd19b8735573aa006900011998039919191999ab9a3370e6aae754009200023300d303135742a00466a02605a6ae84d5d1280111931a981b99ab9c03b038036035135573ca00226ea8004d5d0a801999aa805bae500a35742a00466a01eeb8d5d09aba25002232635303333573806e06806406226ae8940044d55cf280089baa00122212333001004003002200122123300100300220011335500175ceb44488c88c008dd5800990009aa81791191999aab9f0022500923350083355031300635573aa004600a6aae794008c010d5d100181709aba10011122002122122330010040031200112232323333573466e1d400520002350083005357426aae79400c8cccd5cd19b87500248008940208c98d4c0a8cd5ce01701581481401389aab9d500113754002242446004006224400224002464646666ae68cdc39aab9d5002480008cc018c01cd5d0a8011bad357426ae8940088c98d4c090cd5ce01401281181109aab9e50011375400244246600200600440024646666ae68cdc39aab9d5001480008dd71aba135573ca004464c6a604066ae7009008407c0784dd500089119191999ab9a3370ea00290021280391999ab9a3370ea004900111a80518031aba135573ca00846666ae68cdc3a801a40004a014464c6a604666ae7009c09008808408007c4d55cea80089baa00112122230030041122200211222001120012323333573466e1d40052002200623333573466e1d400920002006232635301b33573803e03803403203026aae74dd50008910010910009000919191919191999ab9a3370ea0029006100591999ab9a3370ea0049005100691999ab9a3370ea00690041198059bae35742a00a6eb4d5d09aba2500523333573466e1d4011200623300d375c6ae85401cdd71aba135744a00e46666ae68cdc3a802a400846602460286ae854024dd71aba135744a01246666ae68cdc3a8032400446028602a6ae84d55cf280591999ab9a3370ea00e900011809980b1aba135573ca018464c6a604066ae7009008407c07807407006c0680640604d55cea80209aab9e5003135573ca00426aae7940044dd500090911111118038041109111111198030048041091111111802804091111110020911111100191091111111980100480411091111111980080480410009191919191999ab9a3370ea002900111998041bad35742a0086eb4d5d0a8019bad357426ae89400c8cccd5cd19b875002480008c028c02cd5d09aab9e5006232635301133573802a02402001e01c26aae75400c4d5d1280089aab9e5001137540024244600400644424466600200a0080064002464646666ae68cdc3a800a40044600c6eb8d5d09aab9e500323333573466e1d4009200023008375c6ae84d55cf280211931a980599ab9c00f00c00a009008135573aa00226ea80048488c00800c8488c00400c800444888c8c8cccd5cd19b8735573aa0049000119aa80598031aba150023005357426ae8940088c98d4c020cd5ce00600480380309aab9e5001137540029309000900088910919800801801089000a4903505431001123230010012233003300200200133322222253353004333573466e1c00920540060051006133573892010e77726f6e672072656465656d65720000512200212200120011

integration-test/test/test_mint_nft.py renamed to integration-test/test/test_all.py

@@ -5,40 +5,43 @@
 import tempfile
 import time
 
+import cbor2
 from retry import retry
 
 from pycardano import *
 
 
-class TestMintNFT:
+@retry(tries=10, delay=6)
+def check_chain_context(chain_context):
+    print(f"Current chain tip: {chain_context.last_block_slot}")
+
+
+class TestAll:
     # Define chain context
     NETWORK = Network.TESTNET
 
     OGMIOS_WS = "ws://localhost:1337"
 
     chain_context = OgmiosChainContext(OGMIOS_WS, Network.TESTNET)
 
-    @retry(tries=10, delay=6)
-    def check_ogmios(self):
-        print(f"Current chain tip: {self.chain_context.last_block_slot}")
+    check_chain_context(chain_context)
 
-    def test_mint(self):
-        self.check_ogmios()
-        chain_context = OgmiosChainContext(self.OGMIOS_WS, Network.TESTNET)
-
-        payment_key_path = os.environ.get("PAYMENT_KEY")
-        extended_key_path = os.environ.get("EXTENDED_PAYMENT_KEY")
-        if not payment_key_path or not extended_key_path:
-            raise Exception(
-                "Cannot find payment key. Please specify environment variable PAYMENT_KEY and extended_key_path"
-            )
-        payment_skey = PaymentSigningKey.load(payment_key_path)
-        payment_vkey = PaymentVerificationKey.from_signing_key(payment_skey)
-        extended_payment_skey = PaymentExtendedSigningKey.load(extended_key_path)
-        extended_payment_vkey = PaymentExtendedVerificationKey.from_signing_key(
-            extended_payment_skey
+    payment_key_path = os.environ.get("PAYMENT_KEY")
+    extended_key_path = os.environ.get("EXTENDED_PAYMENT_KEY")
+    if not payment_key_path or not extended_key_path:
+        raise Exception(
+            "Cannot find payment key. Please specify environment variable PAYMENT_KEY and extended_key_path"
         )
-        address = Address(payment_vkey.hash(), network=self.NETWORK)
+    payment_skey = PaymentSigningKey.load(payment_key_path)
+    payment_vkey = PaymentVerificationKey.from_signing_key(payment_skey)
+    extended_payment_skey = PaymentExtendedSigningKey.load(extended_key_path)
+    extended_payment_vkey = PaymentExtendedVerificationKey.from_signing_key(
+        extended_payment_skey
+    )
+
+    @retry(tries=2, delay=6)
+    def test_mint(self):
+        address = Address(self.payment_vkey.hash(), network=self.NETWORK)
 
         # Load payment keys or create them if they don't exist
         def load_or_create_key_pair(base_dir, base_name):
@@ -74,10 +77,10 @@ def load_or_create_key_pair(base_dir, base_name):
         pub_key_policy_1 = ScriptPubkey(policy_vkey.hash())
 
         # A policy that requires a signature from the extended payment key
-        pub_key_policy_2 = ScriptPubkey(extended_payment_vkey.hash())
+        pub_key_policy_2 = ScriptPubkey(self.extended_payment_vkey.hash())
 
         # A time policy that disallows token minting after 10000 seconds from last block
-        must_before_slot = InvalidHereAfter(chain_context.last_block_slot + 10000)
+        must_before_slot = InvalidHereAfter(self.chain_context.last_block_slot + 10000)
 
         # Combine two policies using ScriptAll policy
         policy = ScriptAll([pub_key_policy_1, pub_key_policy_2, must_before_slot])
@@ -125,7 +128,7 @@ def load_or_create_key_pair(base_dir, base_name):
         """Build transaction"""
 
         # Create a transaction builder
-        builder = TransactionBuilder(chain_context)
+        builder = TransactionBuilder(self.chain_context)
 
         # Add our own address as the input address
         builder.add_input_address(address)
@@ -143,7 +146,7 @@ def load_or_create_key_pair(base_dir, base_name):
         builder.auxiliary_data = auxiliary_data
 
         # Calculate the minimum amount of lovelace that need to hold the NFT we are going to mint
-        min_val = min_lovelace(Value(0, my_nft), chain_context)
+        min_val = min_lovelace(Value(0, my_nft), self.chain_context)
 
         # Send the NFT to our own address
         nft_output = TransactionOutput(address, Value(min_val, my_nft))
@@ -154,19 +157,21 @@ def load_or_create_key_pair(base_dir, base_name):
 
         """Sign transaction and add witnesses"""
         # Sign the transaction body hash using the payment signing key
-        payment_signature = payment_skey.sign(tx_body.hash())
+        payment_signature = self.payment_skey.sign(tx_body.hash())
 
         # Sign the transaction body hash using the extended payment signing key
-        extended_payment_signature = extended_payment_skey.sign(tx_body.hash())
+        extended_payment_signature = self.extended_payment_skey.sign(tx_body.hash())
 
         # Sign the transaction body hash using the policy signing key because we are minting new tokens
         policy_signature = policy_skey.sign(tx_body.hash())
 
         # Add verification keys and their signatures to the witness set
         vk_witnesses = [
-            VerificationKeyWitness(payment_vkey, payment_signature),
+            VerificationKeyWitness(self.payment_vkey, payment_signature),
             VerificationKeyWitness(policy_vkey, policy_signature),
-            VerificationKeyWitness(extended_payment_vkey, extended_payment_signature),
+            VerificationKeyWitness(
+                self.extended_payment_vkey, extended_payment_signature
+            ),
         ]
 
         # Create final signed transaction
@@ -185,11 +190,11 @@ def load_or_create_key_pair(base_dir, base_name):
 
         # Submit signed transaction to the network
         print("############### Submitting transaction ###############")
-        chain_context.submit_tx(signed_tx.to_cbor())
+        self.chain_context.submit_tx(signed_tx.to_cbor())
 
         time.sleep(3)
 
-        utxos = chain_context.utxos(str(address))
+        utxos = self.chain_context.utxos(str(address))
         found_nft = False
 
         for utxo in utxos:
@@ -207,7 +212,7 @@ def load_or_create_key_pair(base_dir, base_name):
             ),
         )
 
-        builder = TransactionBuilder(chain_context)
+        builder = TransactionBuilder(self.chain_context)
         builder.add_input_address(address)
 
         builder.add_output(nft_to_send)
@@ -216,11 +221,11 @@ def load_or_create_key_pair(base_dir, base_name):
 
         """Sign transaction and add witnesses"""
         # Sign the transaction body hash using the payment signing key
-        payment_signature = payment_skey.sign(tx_body.hash())
+        payment_signature = self.payment_skey.sign(tx_body.hash())
 
         # Add verification keys and their signatures to the witness set
         vk_witnesses = [
-            VerificationKeyWitness(payment_vkey, payment_signature),
+            VerificationKeyWitness(self.payment_vkey, payment_signature),
         ]
 
         # Create final signed transaction
@@ -236,11 +241,11 @@ def load_or_create_key_pair(base_dir, base_name):
 
         # Submit signed transaction to the network
         print("############### Submitting transaction ###############")
-        chain_context.submit_tx(signed_tx.to_cbor())
+        self.chain_context.submit_tx(signed_tx.to_cbor())
 
         time.sleep(3)
 
-        utxos = chain_context.utxos(str(address))
+        utxos = self.chain_context.utxos(str(address))
         found_nft = False
 
         for utxo in utxos:
@@ -249,3 +254,132 @@ def load_or_create_key_pair(base_dir, base_name):
                 found_nft = True
 
         assert found_nft, f"Cannot find target NFT in address: {address}"
+
+    @retry(tries=2, delay=6)
+    def test_plutus(self):
+
+        # ----------- Giver give ---------------
+
+        with open("plutus_scripts/fortytwo.plutus", "r") as f:
+            script_hex = f.read()
+            forty_two_script = cbor2.loads(bytes.fromhex(script_hex))
+
+        script_hash = plutus_script_hash(forty_two_script)
+
+        script_address = Address(script_hash, network=self.NETWORK)
+
+        giver_address = Address(self.payment_vkey.hash(), network=self.NETWORK)
+
+        builder = TransactionBuilder(self.chain_context)
+        builder.add_input_address(giver_address)
+        datum = PlutusData()  # A Unit type "()" in Haskell
+        builder.add_output(
+            TransactionOutput(script_address, 50000000, datum_hash=datum_hash(datum))
+        )
+
+        tx_body = builder.build(change_address=giver_address)
+        witness_set = builder.build_witness_set()
+
+        payment_signature = self.payment_skey.sign(tx_body.hash())
+
+        witness_set.vkey_witnesses = [
+            VerificationKeyWitness(self.payment_vkey, payment_signature),
+        ]
+
+        signed_tx = Transaction(
+            tx_body,
+            witness_set,
+        )
+
+        print("############### Transaction created ###############")
+        print(signed_tx)
+        print(signed_tx.to_cbor())
+        print("############### Submitting transaction ###############")
+        self.chain_context.submit_tx(signed_tx.to_cbor())
+        time.sleep(3)
+
+        # ----------- Fund taker a collateral UTxO ---------------
+
+        taker_address = Address(self.extended_payment_vkey.hash(), network=self.NETWORK)
+
+        builder = TransactionBuilder(self.chain_context)
+
+        builder.add_input_address(giver_address)
+        builder.add_output(TransactionOutput(taker_address, 5000000))
+
+        tx_body = builder.build(change_address=giver_address)
+        witness_set = builder.build_witness_set()
+
+        payment_signature = self.payment_skey.sign(tx_body.hash())
+
+        witness_set.vkey_witnesses = [
+            VerificationKeyWitness(self.payment_vkey, payment_signature),
+        ]
+
+        signed_tx = Transaction(
+            tx_body,
+            witness_set,
+        )
+
+        print("############### Transaction created ###############")
+        print(signed_tx)
+        print(signed_tx.to_cbor())
+        print("############### Submitting transaction ###############")
+        self.chain_context.submit_tx(signed_tx.to_cbor())
+        time.sleep(3)
+
+        # ----------- Taker take ---------------
+
+        redeemer = Redeemer(
+            RedeemerTag.SPEND, 42, ExecutionUnits(mem=10000000, steps=10000000000)
+        )
+
+        utxo_to_spend = self.chain_context.utxos(str(script_address))[0]
+
+        taker_address = Address(self.extended_payment_vkey.hash(), network=self.NETWORK)
+
+        builder = TransactionBuilder(self.chain_context)
+
+        builder.add_script_input(utxo_to_spend, forty_two_script, datum, redeemer)
+        take_output = TransactionOutput(taker_address, 25123456)
+        builder.add_output(take_output)
+
+        non_nft_utxo = None
+        for utxo in self.chain_context.utxos(str(taker_address)):
+            if isinstance(utxo.output.amount, int):
+                non_nft_utxo = utxo
+                break
+
+        builder.collaterals.append(non_nft_utxo)
+
+        tx_body = builder.build(change_address=taker_address)
+        witness_set = builder.build_witness_set()
+
+        collateral_signature = self.extended_payment_skey.sign(tx_body.hash())
+
+        witness_set.vkey_witnesses = [
+            VerificationKeyWitness(self.extended_payment_vkey, collateral_signature),
+        ]
+
+        signed_tx = Transaction(
+            tx_body,
+            witness_set,
+        )
+
+        print("############### Transaction created ###############")
+        print(signed_tx)
+        print(signed_tx.to_cbor())
+        print("############### Submitting transaction ###############")
+        self.chain_context.submit_tx(signed_tx.to_cbor())
+
+        time.sleep(3)
+
+        utxos = self.chain_context.utxos(str(taker_address))
+        found = False
+
+        for utxo in utxos:
+            output = utxo.output
+            if output == take_output:
+                found = True
+
+        assert found, f"Cannot find target UTxO in address: {taker_address}"

pycardano/exception.py

@@ -34,6 +34,10 @@ class InvalidTransactionException(PyCardanoException):
     pass
 
 
+class TransactionBuilderException(PyCardanoException):
+    pass
+
+
 class TransactionFailedException(PyCardanoException):
     pass
 

pycardano/hash.py

@@ -1,6 +1,6 @@
 """All type of hashes in Cardano ledger spec."""
 
-from typing import Type, TypeVar, Union
+from typing import TypeVar, Union
 
 from pycardano.serialization import CBORSerializable
 
@@ -10,6 +10,7 @@
     "TRANSACTION_HASH_SIZE",
     "DATUM_HASH_SIZE",
     "AUXILIARY_DATA_HASH_SIZE",
+    "SCRIPT_DATA_HASH_SIZE",
     "ConstrainedBytes",
     "VerificationKeyHash",
     "ScriptHash",
@@ -63,7 +64,7 @@ def to_primitive(self) -> bytes:
         return self.payload
 
     @classmethod
-    def from_primitive(cls: Type[T], value: Union[bytes, str]) -> T:
+    def from_primitive(cls: T, value: Union[bytes, str]) -> T:
         if isinstance(value, str):
             value = bytes.fromhex(value)
         return cls(value)