chore: Ignore resolving reference scripts if reference scripts are added through withReferenceSctip() and V3 tests

satran004 · satran004 · commit 514da26c100a · 2024-09-05T23:32:15.000+08:00
diff --git a/quicktx/src/it/java/com/bloxbean/cardano/client/quicktx/ScriptTxV3IT.java b/quicktx/src/it/java/com/bloxbean/cardano/client/quicktx/ScriptTxV3IT.java
@@ -1,28 +1,23 @@
 package com.bloxbean.cardano.client.quicktx;
 
 import com.bloxbean.cardano.client.address.AddressProvider;
-import com.bloxbean.cardano.client.api.TransactionEvaluator;
 import com.bloxbean.cardano.client.api.exception.ApiException;
 import com.bloxbean.cardano.client.api.model.Amount;
-import com.bloxbean.cardano.client.api.model.EvaluationResult;
 import com.bloxbean.cardano.client.api.model.Result;
 import com.bloxbean.cardano.client.api.model.Utxo;
 import com.bloxbean.cardano.client.backend.model.TxContentRedeemers;
 import com.bloxbean.cardano.client.common.model.Networks;
 import com.bloxbean.cardano.client.function.helper.ScriptUtxoFinders;
 import com.bloxbean.cardano.client.function.helper.SignerProviders;
-import com.bloxbean.cardano.client.plutus.spec.BigIntPlutusData;
-import com.bloxbean.cardano.client.plutus.spec.ExUnits;
-import com.bloxbean.cardano.client.plutus.spec.PlutusV3Script;
-import com.bloxbean.cardano.client.plutus.spec.RedeemerTag;
+import com.bloxbean.cardano.client.plutus.spec.*;
+import com.bloxbean.cardano.client.spec.EraSerializationConfig;
 import com.bloxbean.cardano.client.util.JsonUtil;
 import org.junit.jupiter.api.Test;
 
 import java.math.BigInteger;
 import java.util.List;
 import java.util.Optional;
 import java.util.Random;
-import java.util.Set;
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertTrue;
@@ -33,15 +28,15 @@ public class ScriptTxV3IT extends TestDataBaseIT {
     void alwaysTrueScript() throws ApiException {
         PlutusV3Script plutusScript = PlutusV3Script.builder()
                 .type("PlutusScriptV3")
-                .cborHex("49480100002221200101")
+                .cborHex("46450101002499")
                 .build();
 
         String scriptAddress = AddressProvider.getEntAddress(plutusScript, Networks.testnet()).toBech32();
         BigInteger scriptAmt = new BigInteger("2479280");
 
         Random rand = new Random();
         long randInt = System.currentTimeMillis();
-        BigIntPlutusData plutusData =  new BigIntPlutusData(BigInteger.valueOf(randInt)); //any random number
+        BigIntPlutusData plutusData = new BigIntPlutusData(BigInteger.valueOf(randInt)); //any random number
 
         Tx tx = new Tx();
         tx.payToContract(scriptAddress, Amount.lovelace(scriptAmt), plutusData)
@@ -55,7 +50,7 @@ void alwaysTrueScript() throws ApiException {
         System.out.println(result.getResponse());
         checkIfUtxoAvailable(result.getValue(), scriptAddress);
 
-        Optional<Utxo> optionalUtxo  = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, scriptAddress, plutusData);
+        Optional<Utxo> optionalUtxo = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, scriptAddress, plutusData);
         ScriptTx scriptTx = new ScriptTx()
                 .collectFrom(optionalUtxo.get(), plutusData)
                 .payToAddress(receiver1, Amount.lovelace(scriptAmt))
@@ -66,22 +61,6 @@ void alwaysTrueScript() throws ApiException {
                 .feePayer(sender2Addr)
                 .withSigner(SignerProviders.signerFrom(sender2))
                 .withRequiredSigners(sender2.getBaseAddress())
-//                .withTxEvaluator(!backendType.equals(BLOCKFROST)?
-//                        new AikenTransactionEvaluator(utxoSupplier, protocolParamsSupplier): null)
-                .withTxEvaluator(new TransactionEvaluator() {
-                    @Override
-                    public Result<List<EvaluationResult>> evaluateTx(byte[] cbor, Set<Utxo> inputUtxos) throws ApiException {
-                        EvaluationResult evaluationResult = new EvaluationResult();
-                        evaluationResult.setRedeemerTag(RedeemerTag.Spend);
-                        evaluationResult.setIndex(0);
-                        evaluationResult.setExUnits(ExUnits.builder()
-                                .mem(BigInteger.valueOf(42061))
-                                .steps(BigInteger.valueOf(14890343))
-                                .build()
-                        );
-                        return Result.success("success").withValue(List.of(evaluationResult));
-                    }
-                })
                 .withVerifier(txn -> {
                     System.out.println(JsonUtil.getPrettyJson(txn));
                     assertThat(txn.getBody().getRequiredSigners()).hasSize(1);
@@ -99,10 +78,253 @@ public Result<List<EvaluationResult>> evaluateTx(byte[] cbor, Set<Utxo> inputUtx
         List<TxContentRedeemers> redeemers = getBackendService().getTransactionService()
                 .getTransactionRedeemers(result1.getValue()).getValue();
 
+        try {
+            Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
         long gottenValue = getBackendService().getScriptService()
                 .getScriptDatum(redeemers.get(0).getRedeemerDataHash())
                 .getValue().getJsonValue().get("int").asLong();
         assertThat(gottenValue).isEqualTo(randInt);
     }
 
+    @Test
+    void referenceInputUtxo_guessSumScript() throws ApiException, InterruptedException {
+        EraSerializationConfig.INSTANCE.useConwayEraFormat();
+        //Sum Script
+        PlutusV3Script sumScript =
+                PlutusV3Script.builder()
+                        .cborHex("46450101002499")
+                        .build();
+        String sumScriptAddr = AddressProvider.getEntAddress(sumScript, Networks.testnet()).toBech32();
+        Amount sumScriptAmt = Amount.ada(4.0);
+        PlutusData sumScriptDatum = new BigIntPlutusData(BigInteger.valueOf(8)); //redeemer should be 36
+        PlutusData sumScriptRedeemer = new BigIntPlutusData(BigInteger.valueOf(36));
+
+        //Create a reference input and send lock amount at script address
+        Tx createRefInputTx = new Tx();
+        createRefInputTx.payToAddress(receiver1, List.of(Amount.ada(1.0)), sumScript)
+                .payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+
+        QuickTxBuilder quickTxBuilder = new QuickTxBuilder(backendService);
+        Result<String> result = quickTxBuilder.compose(createRefInputTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        System.out.println("Tx Response: " + result.getResponse());
+        assertTrue(result.isSuccessful());
+
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        Utxo refUtxo = Utxo.builder()
+                .txHash(result.getValue())
+                .outputIndex(0)
+                .build();
+
+        //Pay to script
+        Tx scriptPayTx = new Tx();
+        scriptPayTx.payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+        result = quickTxBuilder.compose(scriptPayTx)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        //Find the utxo for the script address
+        Optional<Utxo> sumUtxo = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, sumScriptAddr, sumScriptDatum);
+        ScriptTx scriptTx = new ScriptTx()
+                .collectFrom(sumUtxo.get(), sumScriptRedeemer)
+                .readFrom(refUtxo)
+                .payToAddress(receiver1, List.of(sumScriptAmt))
+                .withChangeAddress(sumScriptAddr, sumScriptDatum);
+
+        Result<String> result1 = quickTxBuilder.compose(scriptTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+//                .withTxEvaluator(!backendType.equals(BLOCKFROST)?
+//                        new AikenTransactionEvaluator(utxoSupplier, protocolParamsSupplier, scriptHash -> sumScript): null)
+                .completeAndWait(System.out::println);
+
+        System.out.println(result1.getResponse());
+        assertTrue(result1.isSuccessful());
+
+        checkIfUtxoAvailable(result1.getValue(), sender1Addr);
+    }
+
+
+    @Test
+    void referenceInputUtxo_guessSumScript_withRefScriptsCall() throws ApiException, InterruptedException {
+
+        EraSerializationConfig.INSTANCE.setUseConwayEraFormat(true);
+        //Sum Script
+        PlutusV3Script sumScript =
+                PlutusV3Script.builder()
+                        .cborHex("46450101002499")
+                        .build();
+        String sumScriptAddr = AddressProvider.getEntAddress(sumScript, Networks.testnet()).toBech32();
+        Amount sumScriptAmt = Amount.ada(4.0);
+        PlutusData sumScriptDatum = new BigIntPlutusData(BigInteger.valueOf(8)); //redeemer should be 36
+        PlutusData sumScriptRedeemer = new BigIntPlutusData(BigInteger.valueOf(36));
+
+        //Create a reference input and send lock amount at script address
+        Tx createRefInputTx = new Tx();
+        createRefInputTx.payToAddress(receiver1, List.of(Amount.ada(1.0)), sumScript)
+                .payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+
+        QuickTxBuilder quickTxBuilder = new QuickTxBuilder(backendService);
+        Result<String> result = quickTxBuilder.compose(createRefInputTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        System.out.println("Tx Response: " + result.getResponse());
+        assertTrue(result.isSuccessful());
+
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        Utxo refUtxo = Utxo.builder()
+                .txHash(result.getValue())
+                .outputIndex(0)
+                .build();
+
+        //Pay to script
+        Tx scriptPayTx = new Tx();
+        scriptPayTx.payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+        result = quickTxBuilder.compose(scriptPayTx)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        //Find the utxo for the script address
+        Optional<Utxo> sumUtxo = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, sumScriptAddr, sumScriptDatum);
+        ScriptTx scriptTx = new ScriptTx()
+                .collectFrom(sumUtxo.get(), sumScriptRedeemer)
+                .readFrom(refUtxo)
+                .payToAddress(receiver1, List.of(sumScriptAmt))
+                .withChangeAddress(sumScriptAddr, sumScriptDatum);
+
+        Result<String> result1 = quickTxBuilder.compose(scriptTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .withReferenceScripts(sumScript)
+//                .withTxEvaluator(!backendType.equals(BLOCKFROST)?
+//                        new AikenTransactionEvaluator(utxoSupplier, protocolParamsSupplier, scriptHash -> sumScript): null)
+                .completeAndWait(System.out::println);
+
+        System.out.println(result1.getResponse());
+        assertTrue(result1.isSuccessful());
+
+        checkIfUtxoAvailable(result1.getValue(), sender1Addr);
+    }
+
+    @Test
+    void plutusV2AndPlutusV3() throws ApiException, InterruptedException {
+
+        EraSerializationConfig.INSTANCE.setUseConwayEraFormat(true);
+        //Sum Script
+        PlutusV3Script sumScript =
+                PlutusV3Script.builder()
+                        .cborHex("46450101002499")
+                        .build();
+        String sumScriptAddr = AddressProvider.getEntAddress(sumScript, Networks.testnet()).toBech32();
+        Amount sumScriptAmt = Amount.ada(4.0);
+        PlutusData sumScriptDatum = new BigIntPlutusData(BigInteger.valueOf(8)); //redeemer should be 36
+        PlutusData sumScriptRedeemer = new BigIntPlutusData(BigInteger.valueOf(36));
+
+        //Create a reference input and send lock amount at script address
+        Tx createRefInputTx = new Tx();
+        createRefInputTx.payToAddress(receiver1, List.of(Amount.ada(1.0)), sumScript)
+                .payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+
+        QuickTxBuilder quickTxBuilder = new QuickTxBuilder(backendService);
+        Result<String> result = quickTxBuilder.compose(createRefInputTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        System.out.println("Tx Response: " + result.getResponse());
+        assertTrue(result.isSuccessful());
+
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        Utxo refUtxo = Utxo.builder()
+                .txHash(result.getValue())
+                .outputIndex(0)
+                .build();
+
+        //Pay to script
+        Tx scriptPayTx = new Tx();
+        scriptPayTx.payToContract(sumScriptAddr, sumScriptAmt, sumScriptDatum)
+                .from(sender1Addr);
+        result = quickTxBuilder.compose(scriptPayTx)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .completeAndWait(System.out::println);
+        //Required as backend service returns outdated utxo
+        if (result.isSuccessful()) {
+            checkIfUtxoAvailable(result.getValue(), sender1Addr);
+        }
+
+        PlutusV2Script alwaysSuccessScript = PlutusV2Script.builder()
+                .type("PlutusScriptV2")
+                .cborHex("49480100002221200101")
+                .build();
+
+        String alwaysSuccessV2ScriptAddress = AddressProvider.getEntAddress(alwaysSuccessScript, Networks.testnet()).toBech32();
+        BigInteger scriptAmt = new BigInteger("2479280");
+
+        Random rand = new Random();
+        int randInt = rand.nextInt();
+        BigIntPlutusData plutusData = new BigIntPlutusData(BigInteger.valueOf(randInt)); //any random number
+
+        Tx tx = new Tx();
+        tx.payToContract(alwaysSuccessV2ScriptAddress, Amount.lovelace(scriptAmt), plutusData)
+                .from(sender2Addr);
+
+        Result<String> result1 = quickTxBuilder.compose(tx)
+                .withSigner(SignerProviders.signerFrom(sender2))
+                .completeAndWait(System.out::println);
+
+        System.out.println(result1.getResponse());
+        checkIfUtxoAvailable(result1.getValue(), alwaysSuccessV2ScriptAddress);
+
+        //Find the utxo for the script address
+        Optional<Utxo> sumUtxo = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, sumScriptAddr, sumScriptDatum);
+        Optional<Utxo> alwaysSuccessUtxo = ScriptUtxoFinders.findFirstByInlineDatum(utxoSupplier, alwaysSuccessV2ScriptAddress, plutusData);
+        ScriptTx scriptTx = new ScriptTx()
+                .collectFrom(sumUtxo.get(), sumScriptRedeemer)
+                .collectFrom(alwaysSuccessUtxo.get(), PlutusData.unit())
+                .readFrom(refUtxo)
+                .payToAddress(receiver1, List.of(sumScriptAmt))
+                .attachSpendingValidator(alwaysSuccessScript)
+                .withChangeAddress(sumScriptAddr, sumScriptDatum);
+
+        Result<String> result2 = quickTxBuilder.compose(scriptTx)
+                .feePayer(sender1Addr)
+                .withSigner(SignerProviders.signerFrom(sender1))
+                .withReferenceScripts(sumScript)
+                .completeAndWait(System.out::println);
+
+        System.out.println(result2.getResponse());
+        assertTrue(result2.isSuccessful());
+
+        checkIfUtxoAvailable(result2.getValue(), sender1Addr);
+    }
 }
diff --git a/quicktx/src/main/java/com/bloxbean/cardano/client/quicktx/QuickTxBuilder.java b/quicktx/src/main/java/com/bloxbean/cardano/client/quicktx/QuickTxBuilder.java
@@ -17,6 +17,7 @@
 import com.bloxbean.cardano.client.function.TxSigner;
 import com.bloxbean.cardano.client.function.exception.TxBuildException;
 import com.bloxbean.cardano.client.function.helper.CollateralBuilders;
+import com.bloxbean.cardano.client.function.helper.ReferenceScriptResolver;
 import com.bloxbean.cardano.client.function.helper.ScriptCostEvaluators;
 import com.bloxbean.cardano.client.function.helper.ScriptBalanceTxProviders;
 import com.bloxbean.cardano.client.plutus.spec.PlutusScript;
@@ -99,16 +100,20 @@ public QuickTxBuilder(UtxoSupplier utxoSupplier,
     }
 
     /**
-     * Create QuickTxBuilder with backend service
+     * Create QuickTxBuilder from BackendService
      *
-     * @param backendService backend service
+     * @param backendService
      */
     public QuickTxBuilder(BackendService backendService) {
-        this(new DefaultUtxoSupplier(backendService.getUtxoService()),
-                new DefaultProtocolParamsSupplier(backendService.getEpochService()),
-                new DefaultScriptSupplier(backendService.getScriptService()),
-                new DefaultTransactionProcessor(backendService.getTransactionService())
-        );
+        this.utxoSupplier = new DefaultUtxoSupplier(backendService.getUtxoService());
+        this.protocolParamsSupplier = new DefaultProtocolParamsSupplier(backendService.getEpochService());
+        this.transactionProcessor = new DefaultTransactionProcessor(backendService.getTransactionService());
+
+        try {
+            this.backendScriptSupplier = new DefaultScriptSupplier(backendService.getScriptService());
+        } catch (UnsupportedOperationException e) {
+            //Not supported
+        }
     }
 
     /**
@@ -285,10 +290,9 @@ public Transaction build() {
                 txBuilder = txBuilder.andThen(addRequiredSignersBuilder());
             }
 
+            //set reference scripts if set
             if (referenceScripts != null && !referenceScripts.isEmpty()) {
-                txBuilder = txBuilder.andThen((context, transaction) -> {
-                    referenceScripts.forEach(script -> context.addRefScripts(script));
-                });
+                referenceScripts.forEach(script -> txBuilderContext.addRefScripts(script));
             }
 
             if (preBalanceTrasformer != null)
@@ -314,6 +318,11 @@ public Transaction build() {
             }
 
             if (containsScriptTx) {
+                //Resolve any reference scripts if any
+                if (referenceScripts == null || referenceScripts.isEmpty()) { //Resolve only if not set explicitly
+                    txBuilder = txBuilder.andThen(ReferenceScriptResolver.resolveReferenceScript());
+                }
+
                 txBuilder = txBuilder.andThen(((context, transaction) -> {
                     boolean negativeAmt = transaction.getBody().getOutputs()
                             .stream()
