[alonzo] - AlonzoTxSeq

IntersectMBO · Feb 11, 2025 · 50817c5 · 50817c5
1 parent 30282f8
commit 50817c5
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 2 deletions.
diff --git a/eras/alonzo/impl/src/Cardano/Ledger/Alonzo/TxSeq/Internal.hs b/eras/alonzo/impl/src/Cardano/Ledger/Alonzo/TxSeq/Internal.hs
@@ -32,9 +32,11 @@ import qualified Cardano.Crypto.Hash as Hash
 import Cardano.Ledger.Alonzo.Era
 import Cardano.Ledger.Alonzo.Tx (AlonzoEraTx (..), IsValid (..), alonzoSegwitTx)
 import Cardano.Ledger.Binary (
+  Annotated (..),
   Annotator,
   DecCBOR (..),
   EncCBORGroup (..),
+  decodeAnnotated,
   encCBOR,
   encodeFoldableEncoder,
   encodeFoldableMapEncoder,
@@ -44,14 +46,14 @@ import Cardano.Ledger.Binary (
   withSlice,
  )
 import Cardano.Ledger.Core
-import Cardano.Ledger.Shelley.BlockChain (constructMetadata)
+import Cardano.Ledger.Shelley.BlockChain (constructMetadata, indexLookupSeq)
 import Control.Monad (unless)
 import Data.ByteString (ByteString)
 import Data.ByteString.Builder (shortByteString, toLazyByteString)
 import qualified Data.ByteString.Lazy as BSL
 import Data.Coerce (coerce)
 import qualified Data.Map.Strict as Map
-import Data.Maybe.Strict (strictMaybeToMaybe)
+import Data.Maybe.Strict (maybeToStrictMaybe, strictMaybeToMaybe)
 import Data.Proxy (Proxy (..))
 import qualified Data.Sequence as Seq
 import Data.Sequence.Strict (StrictSeq)
@@ -230,6 +232,65 @@ instance AlonzoEraTx era => DecCBOR (Annotator (AlonzoTxSeq era)) where
         <*> auxDataAnn
         <*> isValAnn
 
+instance
+  ( AlonzoEraTx era
+  , DecCBOR (TxBody era)
+  , DecCBOR (TxWits era)
+  , DecCBOR (TxAuxData era)
+  ) =>
+  DecCBOR (AlonzoTxSeq era)
+  where
+  decCBOR = do
+    Annotated bodies bodiesBs <- decodeAnnotated decCBOR
+    Annotated wits witsBs <- decodeAnnotated decCBOR
+    Annotated auxDataMap auxDataBs <- decodeAnnotated decCBOR
+    let b = length bodies
+        inRange x = (0 <= x) && (x <= (b - 1))
+        w = length wits
+    unless
+      (all inRange (Map.keysSet auxDataMap))
+      ( fail
+          ( "Some Auxiliarydata index is not in the range: 0 .. "
+              ++ show (b - 1)
+          )
+      )
+    let auxData = maybeToStrictMaybe <$> indexLookupSeq b auxDataMap
+    Annotated isValidIdxs isValidBs <- decodeAnnotated decCBOR
+    let vs = alignedValidFlags b isValidIdxs
+    unless
+      (b == w)
+      ( fail $
+          "different number of transaction bodies ("
+            <> show b
+            <> ") and witness sets ("
+            <> show w
+            <> ")"
+      )
+    unless
+      (all inRange isValidIdxs)
+      ( fail
+          ( "Some IsValid index is not in the range: 0 .. "
+              ++ show (b - 1)
+              ++ ", "
+              ++ show isValidIdxs
+          )
+      )
+    let mkTx body wt isValid ad =
+          mkBasicTx body
+            & witsTxL .~ wt
+            & auxDataTxL .~ ad
+            & isValidTxL .~ isValid
+    let txs =
+          StrictSeq.forceToStrict $
+            Seq.zipWith4 mkTx bodies wits vs auxData
+    pure $
+      AlonzoTxSeqRaw
+        txs
+        bodiesBs
+        witsBs
+        auxDataBs
+        isValidBs
+
 --------------------------------------------------------------------------------
 -- Internal utility functions
 --------------------------------------------------------------------------------

diff --git a/eras/shelley/impl/CHANGELOG.md b/eras/shelley/impl/CHANGELOG.md
@@ -2,6 +2,7 @@
 
 ## 1.16.0.0
 
+* Add `indexLookupSeq`
 * Add `segWitTx`
 * Rename `segwitTx` to `segWitAnnTx`
 * Move `AccountState` to `Cardano.Ledger.State`

diff --git a/eras/shelley/impl/src/Cardano/Ledger/Shelley/BlockChain.hs b/eras/shelley/impl/src/Cardano/Ledger/Shelley/BlockChain.hs
@@ -19,6 +19,7 @@
 module Cardano.Ledger.Shelley.BlockChain (
   ShelleyTxSeq (ShelleyTxSeq, txSeqTxns', TxSeq'),
   constructMetadata,
+  indexLookupSeq,
   txSeqTxns,
   bbHash,
   bBodySize,