From c2824326ba6a05de304959708f1290385569c0d0 Mon Sep 17 00:00:00 2001
From: Kyle Johns <kjohns1337@gmail.com>
Date: Wed, 12 Oct 2022 20:02:34 -0400
Subject: [PATCH] Add Mock option to Tx Building (#11)

* mock option implemented but need CardanoSharp lib update before proceeding

* ok updated cardanosharp for the new mocking feature and got cli to capture the --mock-witness-count value.

* updated the approach used to remove mocked vkey witnesses.

* updated cardanosharp. this update will remove mocks for us after calculating fee

* updated cs wallet and corrected the mock

* removed unnecessary edit

* with current implementation of cs wallet, it expects TransactionWitnessSet to be initialized
---
 Src/ConsoleTool/Constants.cs                                | 1 +
 Src/ConsoleTool/Cscli.ConsoleTool.csproj                    | 2 +-
 Src/ConsoleTool/ShowBaseHelpCommand.cs                      | 2 +-
 .../Transaction/BuildSimplePaymentTransactionCommand.cs     | 6 +++++-
 4 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/Src/ConsoleTool/Constants.cs b/Src/ConsoleTool/Constants.cs
index 04bdedf..d8018ff 100644
--- a/Src/ConsoleTool/Constants.cs
+++ b/Src/ConsoleTool/Constants.cs
@@ -70,6 +70,7 @@ public static class Constants
         { "--signing-keys", "signingKeys" },
         { "--send-all", "sendAll" },
         { "--out-file", "outFile" },
+        { "--mock-witness-count", "mockWitnessCount"}
         //{ "--output-format", "outputFormat" },
     };
 }
\ No newline at end of file
diff --git a/Src/ConsoleTool/Cscli.ConsoleTool.csproj b/Src/ConsoleTool/Cscli.ConsoleTool.csproj
index 55ea4f0..89fe9c1 100644
--- a/Src/ConsoleTool/Cscli.ConsoleTool.csproj
+++ b/Src/ConsoleTool/Cscli.ConsoleTool.csproj
@@ -32,7 +32,7 @@
 	</ItemGroup>
 	<ItemGroup>
 		<PackageReference Include="CardanoSharp.Koios.Sdk" Version="5.0.2" />
-		<PackageReference Include="CardanoSharp.Wallet" Version="2.9.0" />
+		<PackageReference Include="CardanoSharp.Wallet" Version="2.14.0" />
 		<PackageReference Include="Microsoft.Extensions.Configuration" Version="6.0.1" />
 		<PackageReference Include="Microsoft.Extensions.Configuration.Binder" Version="6.0.0" />
 		<PackageReference Include="Microsoft.Extensions.Configuration.CommandLine" Version="6.0.0" />
diff --git a/Src/ConsoleTool/ShowBaseHelpCommand.cs b/Src/ConsoleTool/ShowBaseHelpCommand.cs
index 85a06eb..80970a5 100644
--- a/Src/ConsoleTool/ShowBaseHelpCommand.cs
+++ b/Src/ConsoleTool/ShowBaseHelpCommand.cs
@@ -40,7 +40,7 @@ query info address --network <network> --address <payment_address>
     query info transaction --network <network> --tx-id <transaction_id>
 
 Transaction Commands:
-    BETA: transaction simple-payment build --network <network> --from <address> --to <address> (--ada <ada_amount> | --lovelaces <lovelace_amount> | --send-all true) [--ttl <slot_no>] [--signing-key <from_addr_payment_key>] [--submit true] [--message ""<string>""] [--out-file <output_path>] 
+    BETA: transaction simple-payment build --network <network> --from <address> --to <address> (--ada <ada_amount> | --lovelaces <lovelace_amount> | --send-all true) [--ttl <slot_no>] [--mock-witness-count <mock_witness_count>] [--signing-key <from_addr_payment_key>] [--submit true] [--message ""<string>""] [--out-file <output_path>] 
     transaction view --network <network> --cbor-hex <hex_string>
     transaction sign --cbor-hex <hex_string> --signing-keys <comma_separated_bech32_skeys> [--out-file <output_path>]
     transaction submit --network <network> --cbor-hex <hex_string>
diff --git a/Src/ConsoleTool/Transaction/BuildSimplePaymentTransactionCommand.cs b/Src/ConsoleTool/Transaction/BuildSimplePaymentTransactionCommand.cs
index 8d5d05d..4b38b56 100644
--- a/Src/ConsoleTool/Transaction/BuildSimplePaymentTransactionCommand.cs
+++ b/Src/ConsoleTool/Transaction/BuildSimplePaymentTransactionCommand.cs
@@ -10,6 +10,7 @@
 using CardanoSharp.Wallet.Utilities;
 using Cscli.ConsoleTool.Koios;
 using System.Text.Json;
+using CardanoSharp.Wallet.Models.Transactions;
 using static Cscli.ConsoleTool.Constants;
 
 namespace Cscli.ConsoleTool.Transaction;
@@ -24,6 +25,7 @@ public class BuildSimplePaymentTransactionCommand : ICommand
     public decimal Ada { get; init; } 
     public bool SendAll { get; init; } = false; 
     public uint Ttl { get; set; } // Slot for transaction expiry
+    public int? MockWitnessCount { get; set; } // Slot for transaction expiry
     public string? Message { get; init; } // Onchain Metadata 674 standard
     public bool Submit { get; init; } // Submits Transaction to Koios node
     public string? OutFile { get; init; } // cardano-cli compatible transaction file (signed only)
@@ -87,8 +89,10 @@ public async ValueTask<CommandResult> ExecuteAsync(CancellationToken ct)
             txBuilder.SetAuxData(auxDataBuilder);
         }
         var tx = txBuilder.Build();
+        if(MockWitnessCount is not null && MockWitnessCount > 0)
+            tx.TransactionWitnessSet = new TransactionWitnessSet();
         // Fee Calculation
-        var fee = tx.CalculateAndSetFee(protocolParams.MinFeeA, protocolParams.MinFeeB);
+        var fee = tx.CalculateAndSetFee(protocolParams.MinFeeA, protocolParams.MinFeeB, MockWitnessCount ?? 0);
         tx.TransactionBody.TransactionOutputs.Last().Value.Coin -= fee;
         var txCborBytes = tx.Serialize();
         if (!string.IsNullOrWhiteSpace(OutFile))