Use Write.Tx type for tx field of PartialTx. (#4235)

## Issue

ADP-3184

## Description

This PR adjusts the `tx` field of `PartialTx` to use the `Write.Tx`
(ledger) type instead of the equivalent type from `cardano-api`.

## Future work

Adjust `balanceTransaction` to return `Write.Tx` (ledger) instead of the
equivalent type from `cardano-api`.

lib/balance-tx/lib/internal/Internal/Cardano/Write/Tx.hs

@@ -52,6 +52,7 @@ module Internal.Cardano.Write.Tx
     , CardanoApi.ShelleyLedgerEra
     , cardanoEraFromRecentEra
     , shelleyBasedEraFromRecentEra
+    , asCardanoApiTx
     , fromCardanoApiTx
     , toCardanoApiUTxO
     , fromCardanoApiUTxO
@@ -791,14 +792,16 @@ outputs RecentEraBabbage = map sizedValue . toList . Babbage.btbOutputs
 --
 -- TODO [ADP-2353] Move to @cardano-api@ related module
 modifyLedgerBody
-    :: (Core.TxBody (CardanoApi.ShelleyLedgerEra cardanoEra) ->
+    :: forall cardanoEra. IsRecentEra cardanoEra
+    => (Core.TxBody (CardanoApi.ShelleyLedgerEra cardanoEra) ->
         Core.TxBody (CardanoApi.ShelleyLedgerEra cardanoEra))
-    -> CardanoApi.Tx cardanoEra
-    -> CardanoApi.Tx cardanoEra
-modifyLedgerBody f (CardanoApi.Tx body keyWits) = CardanoApi.Tx body' keyWits
+    -> Core.Tx (CardanoApi.ShelleyLedgerEra cardanoEra)
+    -> Core.Tx (CardanoApi.ShelleyLedgerEra cardanoEra)
+modifyLedgerBody f = asCardanoApiTx @cardanoEra modify
   where
-    body' =
-        case body of
+    modify (CardanoApi.Tx body keyWits) = CardanoApi.Tx body' keyWits
+      where
+        body' = case body of
             CardanoApi.ByronTxBody {} ->
                 error "Impossible: ByronTxBody in CardanoApi.ShelleyLedgerEra"
             CardanoApi.ShelleyTxBody
@@ -823,6 +826,16 @@ emptyTx era = withConstraints era $ Core.mkBasicTx Core.mkBasicTxBody
 -- Compatibility
 --------------------------------------------------------------------------------
 
+asCardanoApiTx
+    :: forall era. IsRecentEra era
+    => (CardanoApi.Tx era -> CardanoApi.Tx era)
+    -> Core.Tx (CardanoApi.ShelleyLedgerEra era)
+    -> Core.Tx (CardanoApi.ShelleyLedgerEra era)
+asCardanoApiTx f
+    = fromCardanoApiTx
+    . f
+    . toCardanoApiTx
+
 fromCardanoApiTx
     :: forall era. IsRecentEra era
     => CardanoApi.Tx era

lib/balance-tx/lib/internal/Internal/Cardano/Write/Tx/Balance.hs

@@ -228,7 +228,6 @@ import Internal.Cardano.Write.Tx
     , evaluateMinimumFee
     , evaluateTransactionBalance
     , feeOfBytes
-    , fromCardanoApiTx
     , getFeePerByte
     , isBelowMinimumCoinForTxOut
     , maxScriptExecutionCost
@@ -471,7 +470,7 @@ deriving instance Show (ErrBalanceTx era)
 -- and instead adjust the existing redeemer indexes ourselves when balancing,
 -- even though they are in an "unordered" set.
 data PartialTx era = PartialTx
-    { tx :: CardanoApi.Tx era
+    { tx :: Tx (ShelleyLedgerEra era)
     , inputs :: UTxO (ShelleyLedgerEra era)
       -- ^ NOTE: Can we rename this to something better? Perhaps 'extraUTxO'?
     , redeemers :: [Redeemer]
@@ -494,13 +493,13 @@ instance
         = nameF "PartialTx" $ mconcat
             [ nameF "inputs" (blockListF' "-" inF (Map.toList ins))
             , nameF "redeemers" (pretty redeemers)
-            , nameF "tx" (cardanoTxF tx)
+            , nameF "tx" (txF tx)
             ]
       where
         inF = build . show
 
-        cardanoTxF :: CardanoApi.Tx era -> Builder
-        cardanoTxF tx' = pretty $ pShow tx'
+        txF :: Tx (ShelleyLedgerEra era) -> Builder
+        txF tx' = pretty $ pShow tx'
 
 data UTxOIndex era = UTxOIndex
     { walletUTxO :: !W.UTxO
@@ -654,10 +653,13 @@ assignMinimalAdaQuantitiesToOutputsWithoutAda
     :: forall era
      . RecentEra era
     -> PParams (CardanoApi.ShelleyLedgerEra era)
-    -> CardanoApi.Tx era
-    -> CardanoApi.Tx era
-assignMinimalAdaQuantitiesToOutputsWithoutAda era pp = withConstraints era $
-    modifyLedgerBody $ over outputsTxBodyL $ fmap modifyTxOut
+    -> Tx (CardanoApi.ShelleyLedgerEra era)
+    -> Tx (CardanoApi.ShelleyLedgerEra era)
+assignMinimalAdaQuantitiesToOutputsWithoutAda era pp =
+    withConstraints era
+        $ modifyLedgerBody @era
+        $ over outputsTxBodyL
+        $ fmap modifyTxOut
   where
     modifyTxOut out = flip (modifyTxOutCoin era) out $ \c ->
         if c == mempty then computeMinimumCoinForTxOut era pp out else c
@@ -687,12 +689,12 @@ balanceTransactionWithSelectionStrategyAndNoZeroAdaAdjustment
     selectionStrategy
     ptx@(PartialTx partialTx inputUTxO redeemers)
     = do
-    guardExistingCollateral partialLedgerTx
-    guardExistingTotalCollateral partialLedgerTx
-    guardExistingReturnCollateral partialLedgerTx
+    guardExistingCollateral partialTx
+    guardExistingTotalCollateral partialTx
+    guardExistingReturnCollateral partialTx
     guardWalletUTxOConsistencyWith inputUTxO
 
-    (balance0, minfee0, _) <- balanceAfterSettingMinFee partialLedgerTx
+    (balance0, minfee0, _) <- balanceAfterSettingMinFee partialTx
 
     (extraInputs, extraCollateral', extraOutputs, s') <- do
 
@@ -729,7 +731,7 @@ balanceTransactionWithSelectionStrategyAndNoZeroAdaAdjustment
                 (recentEra @era)
                 protocolParameters
                 utxoAssumptions
-                (extractOutputsFromTx partialTx)
+                (extractOutputsFromTx (toCardanoApiTx partialTx))
                 redeemers
                 (UTxOSelection.fromIndexPair
                     (internalUtxoAvailable, externalSelectedUtxo))
@@ -828,8 +830,6 @@ balanceTransactionWithSelectionStrategyAndNoZeroAdaAdjustment
   where
     era = recentEra @era
 
-    partialLedgerTx = fromCardanoApiTx partialTx
-
     -- | Extract the inputs from the raw 'tx' of the 'Partialtx', with the
     -- corresponding 'TxOut' according to @combinedUTxO@.
     --
@@ -870,7 +870,7 @@ balanceTransactionWithSelectionStrategyAndNoZeroAdaAdjustment
       where
         txIns :: [TxIn]
         txIns = withConstraints (recentEra @era) $
-            Set.toList $ (fromCardanoApiTx tx) ^. (bodyTxL . inputsTxBodyL)
+            Set.toList $ tx ^. (bodyTxL . inputsTxBodyL)
 
     guardTxSize
         :: KeyWitnessCount
@@ -975,7 +975,7 @@ balanceTransactionWithSelectionStrategyAndNoZeroAdaAdjustment
         :: TxUpdate
         -> ExceptT (ErrBalanceTx era) m (Tx (ShelleyLedgerEra era))
     assembleTransaction update = ExceptT . pure $ do
-        tx' <- left ErrBalanceTxUpdateError $ updateTx era partialLedgerTx update
+        tx' <- left ErrBalanceTxUpdateError $ updateTx era partialTx update
         left ErrBalanceTxAssignRedeemers $
             assignScriptRedeemers
                 era pp timeTranslation combinedUTxO redeemers tx'

lib/wallet/api/http/Cardano/Wallet/Api/Http/Shelley/Server.hs

@@ -3427,7 +3427,7 @@ balanceTransaction
             $ body ^. #transaction
 
         pure $ Write.PartialTx
-            tx
+            (Write.fromCardanoApiTx tx)
             externalUTxO
             (fromApiRedeemer <$> body ^. #redeemers)
 

lib/wallet/src/Cardano/Wallet.hs

@@ -2356,7 +2356,7 @@ buildTransactionPure
             changeAddrGen
             (getState wallet)
             PartialTx
-                { tx = Cardano.Tx unsignedTxBody []
+                { tx = Write.fromCardanoApiTx (Cardano.Tx unsignedTxBody [])
                 , inputs = Write.UTxO mempty
                 , redeemers = []
                 }
@@ -2992,7 +2992,7 @@ transactionFee DBLayer{atomically, walletState} protocolParams
                 (Left preSelection)
 
         let ptx = PartialTx
-                { tx = Cardano.Tx unsignedTxBody []
+                { tx = Write.fromCardanoApiTx (Cardano.Tx unsignedTxBody [])
                 , inputs = Write.UTxO mempty
                 , redeemers = []
                 }