From b4614ab185538398111824f67ad3567ccda7e8da Mon Sep 17 00:00:00 2001
From: Jonathan Knowles <jonathan.knowles@iohk.io>
Date: Tue, 14 Mar 2023 10:15:13 +0000
Subject: [PATCH] Add missing constraints within `Write.Tx`.

---
 lib/wallet/src/Cardano/Wallet/Write/Tx.hs | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/lib/wallet/src/Cardano/Wallet/Write/Tx.hs b/lib/wallet/src/Cardano/Wallet/Write/Tx.hs
index 15fab2eeca3..255637faa53 100644
--- a/lib/wallet/src/Cardano/Wallet/Write/Tx.hs
+++ b/lib/wallet/src/Cardano/Wallet/Write/Tx.hs
@@ -883,13 +883,13 @@ evaluateMinimumFee era pp tx kwc =
 -- is not automatically the minimum fee.
 --
 evaluateTransactionBalance
-    :: Babbage.ShelleyEraTxBody (ShelleyLedgerEra era)
-    => RecentEra era
+    :: RecentEra era
     -> Core.PParams (Cardano.ShelleyLedgerEra era)
     -> Shelley.UTxO (Cardano.ShelleyLedgerEra era)
     -> Core.TxBody (Cardano.ShelleyLedgerEra era)
     -> Core.Value (Cardano.ShelleyLedgerEra era)
 evaluateTransactionBalance era pp utxo txBody' =
+    withShelleyEraTxBodyConstraint era $
     withCLIConstraint era $
         Shelley.evaluateTransactionBalance pp utxo isNewPool txBody'
   where
@@ -903,19 +903,29 @@ evaluateTransactionBalance era pp utxo txBody' =
 -- Module-internal helpers
 --------------------------------------------------------------------------------
 
+withShelleyEraTxBodyConstraint
+    :: RecentEra era
+    -> ((Babbage.ShelleyEraTxBody (ShelleyLedgerEra era)) => a)
+    -> a
+withShelleyEraTxBodyConstraint era a = case era of
+    RecentEraAlonzo -> a
+    RecentEraBabbage -> a
+    RecentEraConway -> a
+
 withStandardCryptoConstraint
     :: RecentEra era
     -> ((Crypto (Cardano.ShelleyLedgerEra era) ~ StandardCrypto) => a)
     -> a
 withStandardCryptoConstraint era a = case era of
-    RecentEraBabbage -> a
     RecentEraAlonzo -> a
+    RecentEraBabbage -> a
+    RecentEraConway -> a
 
 withCLIConstraint
     :: RecentEra era
     -> (CLI (ShelleyLedgerEra era) => a)
     -> a
 withCLIConstraint era a = case era of
-    RecentEraConway -> a
-    RecentEraBabbage -> a
     RecentEraAlonzo -> a
+    RecentEraBabbage -> a
+    RecentEraConway -> a