Added sizedCollateralReturnTxBodyF

Author: Soupstraw

eras/babbage/impl/src/Cardano/Ledger/Babbage/TxBody.hs

@@ -90,6 +90,7 @@ import Cardano.Ledger.MemoBytes (
   eqRaw,
   getMemoRawType,
   getMemoSafeHash,
+  getterMemoRawType,
   lensMemoRawType,
   mkMemoizedEra,
   zipMemoRawType,
@@ -119,6 +120,8 @@ class (AlonzoEraTxBody era, BabbageEraTxOut era) => BabbageEraTxBody era where
 
   sizedCollateralReturnTxBodyL :: Lens' (TxBody TopTx era) (StrictMaybe (Sized (TxOut era)))
 
+  sizedCollateralReturnTxBodyF :: SimpleGetter (TxBody l era) (StrictMaybe (Sized (TxOut era)))
+
   allSizedOutputsTxBodyF :: SimpleGetter (TxBody l era) (StrictSeq (Sized (TxOut era)))
 
 -- ======================================
@@ -228,7 +231,7 @@ allSizedOutputsBabbageTxBodyF ::
 allSizedOutputsBabbageTxBodyF =
   to $ \txBody ->
     let txOuts = txBody ^. sizedOutputsTxBodyL
-     in case txBody ^. sizedCollateralReturnTxBodyL of
+     in case txBody ^. sizedCollateralReturnTxBodyF of
           SNothing -> txOuts
           SJust collTxOut -> txOuts |> collTxOut
 {-# INLINEABLE allSizedOutputsBabbageTxBodyF #-}
@@ -356,6 +359,8 @@ instance BabbageEraTxBody BabbageEra where
       txBodyRaw {btbrCollateralReturn = collateralReturn}
   {-# INLINE sizedCollateralReturnTxBodyL #-}
 
+  sizedCollateralReturnTxBodyF = getterMemoRawType (\BabbageTxBodyRaw {btbrCollateralReturn} -> btbrCollateralReturn)
+
   allSizedOutputsTxBodyF = allSizedOutputsBabbageTxBodyF
   {-# INLINE allSizedOutputsTxBodyF #-}
 

libs/cardano-ledger-core/src/Cardano/Ledger/Core/TxLevel.hs

@@ -24,7 +24,7 @@ module Cardano.Ledger.Core.TxLevel (
 import Cardano.Ledger.Core.Era (Era (..))
 import Control.Monad.Trans.Class (lift)
 import Control.Monad.Trans.Fail.String
-import Data.Functor.Identity (runIdentity)
+import Data.Functor.Identity (Identity (..), runIdentity)
 import Data.Kind (Type)
 import Data.Typeable
 import GHC.Stack