New ModeTx GADT to help with default values across BuildTx/ViewTx

IntersectMBO · Mar 21, 2023 · 7a1ddcc · 7a1ddcc
1 parent 16e60df
commit 7a1ddcc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/cardano-api/src/Cardano/Api.hs b/cardano-api/src/Cardano/Api.hs
@@ -163,6 +163,7 @@ module Cardano.Api (
     defaultTxBodyContent,
     defaultTxFee,
     defaultTxValidityUpperBound,
+    defaultBuildTxWith,
     setTxIns,
     setTxInsCollateral,
     setTxInsReference,

diff --git a/cardano-api/src/Cardano/Api/TxBody.hs b/cardano-api/src/Cardano/Api/TxBody.hs
@@ -32,6 +32,7 @@ module Cardano.Api.TxBody (
     defaultTxBodyContent,
     defaultTxFee,
     defaultTxValidityUpperBound,
+    defaultBuildTxWith,
     setTxIns,
     setTxInsCollateral,
     setTxInsReference,
@@ -106,6 +107,7 @@ module Cardano.Api.TxBody (
     BuildTxWith(..),
     BuildTx,
     ViewTx,
+    ModeTx(..),
 
     -- * Era-dependent transaction body features
     CollateralSupportedInEra(..),
@@ -1325,6 +1327,16 @@ data BuildTxWith build a where
      ViewTx      ::      BuildTxWith ViewTx  a
      BuildTxWith :: a -> BuildTxWith BuildTx a
 
+defaultBuildTxWith :: ModeTx build -> a -> BuildTxWith build a
+defaultBuildTxWith inModeTx a =
+  case inModeTx of
+    InViewTx -> ViewTx
+    InBuildTx -> BuildTxWith a
+
+data ModeTx a where
+  InBuildTx :: ModeTx BuildTx
+  InViewTx :: ModeTx ViewTx
+
 instance Functor (BuildTxWith build) where
     fmap _ ViewTx = ViewTx
     fmap f (BuildTxWith x) = BuildTxWith (f x)
@@ -1765,8 +1777,8 @@ data TxBodyContent build era =
      }
      deriving (Eq, Show)
 
-defaultTxBodyContent :: IsCardanoEra era => TxBodyContent BuildTx era
-defaultTxBodyContent = TxBodyContent
+defaultTxBodyContent :: IsCardanoEra era => ModeTx mtx -> TxBodyContent mtx era
+defaultTxBodyContent mtx = TxBodyContent
     { txIns = []
     , txInsCollateral = TxInsCollateralNone
     , txInsReference = TxInsReferenceNone
@@ -1778,7 +1790,7 @@ defaultTxBodyContent = TxBodyContent
     , txMetadata = TxMetadataNone
     , txAuxScripts = TxAuxScriptsNone
     , txExtraKeyWits = TxExtraKeyWitnessesNone
-    , txProtocolParams = BuildTxWith Nothing
+    , txProtocolParams = defaultBuildTxWith mtx Nothing
     , txWithdrawals = TxWithdrawalsNone
     , txCertificates = TxCertificatesNone
     , txUpdateProposal = TxUpdateProposalNone