Convert CertState to a type family (#4861)

Resolves #4693

Co-authored-by: Alexey Kuleshevich <alexey.kuleshevich@iohk.io>

Author: Lucsanszky

eras/allegra/impl/CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## 1.7.0.0
 
+* Converted `CertState` to a type family
 * Made the fields of predicate failures and environments lazy
 * Add `Era era` constraint to `NoThunks` instance for `TimeLock`
 * Remove `Era era` constraint from:

eras/allegra/impl/cardano-ledger-allegra.cabal

@@ -34,6 +34,7 @@ library
 
   hs-source-dirs: src
   other-modules:
+    Cardano.Ledger.Allegra.CertState
     Cardano.Ledger.Allegra.Era
     Cardano.Ledger.Allegra.PParams
     Cardano.Ledger.Allegra.Rules.Bbody

eras/allegra/impl/src/Cardano/Ledger/Allegra/CertState.hs

@@ -0,0 +1,32 @@
+{-# LANGUAGE DataKinds #-}
+{-# LANGUAGE TypeFamilies #-}
+{-# OPTIONS_GHC -Wno-orphans #-}
+
+module Cardano.Ledger.Allegra.CertState () where
+
+import Cardano.Ledger.Allegra.Era (AllegraEra)
+import Cardano.Ledger.CertState
+import Cardano.Ledger.Shelley.CertState
+import Data.Coerce (coerce)
+
+instance EraCertState AllegraEra where
+  type CertState AllegraEra = ShelleyCertState AllegraEra
+
+  mkCertState = mkShelleyCertState
+
+  upgradeCertState = coerce
+
+  certDStateL = shelleyCertDStateL
+  {-# INLINE certDStateL #-}
+
+  certVStateL = shelleyCertVStateL
+  {-# INLINE certVStateL #-}
+
+  certPStateL = shelleyCertPStateL
+  {-# INLINE certPStateL #-}
+
+  obligationCertState = shelleyObligationCertState
+
+  certsTotalDepositsTxBody = shelleyCertsTotalDepositsTxBody
+
+  certsTotalRefundsTxBody = shelleyCertsTotalRefundsTxBody

eras/allegra/impl/src/Cardano/Ledger/Allegra/Rules/Utxo.hs

@@ -39,7 +39,7 @@ import Cardano.Ledger.BaseTypes (
  )
 import Cardano.Ledger.Binary (DecCBOR (..), EncCBOR (..), serialize)
 import Cardano.Ledger.Binary.Coders
-import Cardano.Ledger.CertState (certDState, dsGenDelegs)
+import Cardano.Ledger.CertState (EraCertState (..))
 import Cardano.Ledger.Coin (Coin)
 import Cardano.Ledger.Rules.ValidationMode (Test, runTest)
 import qualified Cardano.Ledger.Shelley.LedgerState as Shelley
@@ -175,13 +175,14 @@ utxoTransition ::
   , InjectRuleFailure "UTXO" AllegraUtxoPredFailure era
   , InjectRuleFailure "UTXO" Shelley.ShelleyUtxoPredFailure era
   , EraRule "UTXO" era ~ AllegraUTXO era
+  , EraCertState era
   ) =>
   TransitionRule (EraRule "UTXO" era)
 utxoTransition = do
   TRC (Shelley.UtxoEnv slot pp certState, utxos, tx) <- judgmentContext
   let Shelley.UTxOState utxo _ _ ppup _ _ = utxos
       txBody = tx ^. bodyTxL
-      genDelegs = dsGenDelegs (certDState certState)
+      genDelegs = certState ^. Shelley.certDStateL . Shelley.dsGenDelegsL
 
   {- ininterval slot (txvld tx) -}
   runTest $ validateOutsideValidityIntervalUTxO slot txBody
@@ -308,6 +309,7 @@ instance
   , GovState era ~ ShelleyGovState era
   , InjectRuleFailure "UTXO" AllegraUtxoPredFailure era
   , InjectRuleFailure "UTXO" Shelley.ShelleyUtxoPredFailure era
+  , EraCertState era
   ) =>
   STS (AllegraUTXO era)
   where

eras/allegra/impl/src/Cardano/Ledger/Allegra/Rules/Utxow.hs

@@ -14,6 +14,7 @@ import Cardano.Ledger.Allegra.Core
 import Cardano.Ledger.Allegra.Era (AllegraEra, AllegraUTXOW)
 import Cardano.Ledger.Allegra.Rules.Utxo (AllegraUTXO, AllegraUtxoPredFailure)
 import Cardano.Ledger.BaseTypes
+import Cardano.Ledger.CertState
 import Cardano.Ledger.Shelley.LedgerState (UTxOState)
 import Cardano.Ledger.Shelley.Rules (
   ShelleyPpupPredFailure,
@@ -58,6 +59,7 @@ instance
   , Signal (EraRule "UTXO" era) ~ Tx era
   , EraRule "UTXOW" era ~ AllegraUTXOW era
   , InjectRuleFailure "UTXOW" ShelleyUtxowPredFailure era
+  , EraCertState era
   ) =>
   STS (AllegraUTXOW era)
   where

eras/allegra/impl/src/Cardano/Ledger/Allegra/Transition.hs

@@ -5,6 +5,7 @@
 
 module Cardano.Ledger.Allegra.Transition (TransitionConfig (..)) where
 
+import Cardano.Ledger.Allegra.CertState ()
 import Cardano.Ledger.Allegra.Era
 import Cardano.Ledger.Allegra.Translation ()
 import Cardano.Ledger.Genesis (NoGenesis (..))

eras/allegra/impl/src/Cardano/Ledger/Allegra/Translation.hs

@@ -12,15 +12,16 @@
 
 module Cardano.Ledger.Allegra.Translation (shelleyToAllegraAVVMsToDelete) where
 
+import Cardano.Ledger.Allegra.CertState ()
 import Cardano.Ledger.Allegra.Era (AllegraEra)
 import Cardano.Ledger.Allegra.Tx ()
 import Cardano.Ledger.Binary (DecoderError)
 import Cardano.Ledger.CertState (CommitteeState (..))
 import Cardano.Ledger.Genesis (NoGenesis (..))
 import Cardano.Ledger.Shelley (ShelleyEra)
+import Cardano.Ledger.Shelley.CertState (ShelleyCertState)
 import Cardano.Ledger.Shelley.Core
 import Cardano.Ledger.Shelley.LedgerState (
-  CertState (..),
   DState (..),
   EpochState (..),
   LedgerState (..),
@@ -143,14 +144,7 @@ instance TranslateEra AllegraEra VState where
 instance TranslateEra AllegraEra PState where
   translateEra _ PState {..} = pure PState {..}
 
-instance TranslateEra AllegraEra CertState where
-  translateEra ctxt ls =
-    pure
-      CertState
-        { certDState = translateEra' ctxt $ certDState ls
-        , certPState = translateEra' ctxt $ certPState ls
-        , certVState = translateEra' ctxt $ certVState ls
-        }
+instance TranslateEra AllegraEra ShelleyCertState
 
 instance TranslateEra AllegraEra LedgerState where
   translateEra ctxt ls =

eras/allegra/impl/src/Cardano/Ledger/Allegra/UTxO.hs

@@ -6,6 +6,7 @@
 
 module Cardano.Ledger.Allegra.UTxO () where
 
+import Cardano.Ledger.Allegra.CertState ()
 import Cardano.Ledger.Allegra.Core
 import Cardano.Ledger.Allegra.Era (AllegraEra)
 import Cardano.Ledger.Shelley.UTxO (

eras/alonzo/impl/CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## 1.13.0.0
 
+* Converted `CertState` to a type family
 * Remove `reapplyAlonzoTx` as no longer needed.
 * Add `TxInfoResult` data family, `mkTxInfoResult` and `lookupTxInfoResult` to `EraPlutusContext`
 * Add `lookupTxInfoResultImpossible` helper
@@ -15,6 +16,7 @@
 
 ### `testlib`
 
+* Converted `CertState` to a type family
 * Expose `alonzoFixupFees`
 
 ## 1.12.0.0

eras/alonzo/impl/cardano-ledger-alonzo.cabal

@@ -48,6 +48,7 @@ library
 
   hs-source-dirs: src
   other-modules:
+    Cardano.Ledger.Alonzo.CertState
     Cardano.Ledger.Alonzo.Era
     Cardano.Ledger.Alonzo.Rules.Bbody
     Cardano.Ledger.Alonzo.Rules.Deleg