Add getReferenceInputsSizeForTxIds function

IntersectMBO · Apr 4, 2024 · a1e5545 · a1e5545
1 parent d45171b
commit a1e5545
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/cardano-api/internal/Cardano/Api/Tx/Body.hs b/cardano-api/internal/Cardano/Api/Tx/Body.hs
@@ -16,14 +16,7 @@
 {-# LANGUAGE TypeOperators #-}
 {-# LANGUAGE ViewPatterns #-}
 
-{- HLINT ignore "Avoid lambda using `infix`" -}
-{- HLINT ignore "Move brackets to avoid" -}
-{- HLINT ignore "Redundant flip" -}
-{- HLINT ignore "Use let" -}
-{- HLINT ignore "Use section" -}
-
 -- | Transaction bodies
---
 module Cardano.Api.Tx.Body (
     parseTxId,
     -- * Transaction bodies
@@ -82,6 +75,7 @@ module Cardano.Api.Tx.Body (
     TxIns,
     TxIx(..),
     genesisUTxOPseudoTxIn,
+    getReferenceInputsSizeForTxIds,
 
     -- * Transaction outputs
     CtxTx, CtxUTxO,
@@ -206,6 +200,7 @@ import qualified Cardano.Ledger.Alonzo.Scripts as Alonzo
 import qualified Cardano.Ledger.Alonzo.Tx as Alonzo (hashScriptIntegrity)
 import qualified Cardano.Ledger.Alonzo.TxWits as Alonzo
 import qualified Cardano.Ledger.Api as L
+import qualified Cardano.Ledger.Babbage.UTxO as L
 import           Cardano.Ledger.BaseTypes (StrictMaybe (..))
 import           Cardano.Ledger.Binary (Annotated (..))
 import qualified Cardano.Ledger.Binary as CBOR
@@ -3065,6 +3060,17 @@ genesisUTxOPseudoTxIn nw (GenesisUTxOKeyHash kh) =
              (Shelley.KeyHashObj kh)
              Shelley.StakeRefNull
 
+-- | Calculate the reference inputs size for provided set of transaction IDs and UTXOs.
+getReferenceInputsSizeForTxIds
+  :: forall era ledgerera. ShelleyLedgerEra era ~ ledgerera
+  => BabbageEraOnwards era
+  -> Ledger.UTxO ledgerera
+  -> Set TxIn
+  -> Int
+getReferenceInputsSizeForTxIds beo utxo txIds = babbageEraOnwardsConstraints beo $ do
+  let refScripts = L.getReferenceScriptsNonDistinct utxo (Set.map toShelleyTxIn txIds)
+  getSum $ foldMap (Sum .  SafeHash.originalBytesSize . snd) refScripts
+
 calculateExecutionUnitsLovelace :: Ledger.Prices -> ExecutionUnits -> Maybe L.Coin
 calculateExecutionUnitsLovelace prices eUnits =
   return $ Alonzo.txscriptfee prices (toAlonzoExUnits eUnits)

diff --git a/cardano-api/src/Cardano/Api.hs b/cardano-api/src/Cardano/Api.hs
@@ -342,6 +342,7 @@ module Cardano.Api (
     TxIns,
     TxIx(TxIx),
     renderTxIn,
+    getReferenceInputsSizeForTxIds,
 
     -- ** Transaction outputs
     CtxTx, CtxUTxO,