{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE PatternSynonyms #-}
{-# LANGUAGE TypeFamilies #-}
{-# OPTIONS_GHC -Wno-orphans #-}

module Cardano.Ledger.Mary.TxCert () where

import Cardano.Ledger.Mary.Era (MaryEra)
import Cardano.Ledger.Mary.PParams ()
import Cardano.Ledger.Shelley.TxCert (
  EraTxCert (..),
  PoolCert (..),
  ShelleyDelegCert (..),
  ShelleyEraTxCert (..),
  ShelleyTxCert (..),
  getScriptWitnessShelleyTxCert,
  getVKeyWitnessShelleyTxCert,
  shelleyTotalDepositsTxCerts,
  shelleyTotalRefundsTxCerts,
  upgradeShelleyTxCert,
  pattern RegTxCert,
  pattern UnRegTxCert,
 )

instance EraTxCert MaryEra where
  type TxCert MaryEra = ShelleyTxCert MaryEra

  upgradeTxCert :: EraTxCert (PreviousEra MaryEra) =>
TxCert (PreviousEra MaryEra)
-> Either (TxCertUpgradeError MaryEra) (TxCert MaryEra)
upgradeTxCert = ShelleyTxCert MaryEra -> Either Void (ShelleyTxCert MaryEra)
forall a b. b -> Either a b
Right (ShelleyTxCert MaryEra -> Either Void (ShelleyTxCert MaryEra))
-> (ShelleyTxCert AllegraEra -> ShelleyTxCert MaryEra)
-> ShelleyTxCert AllegraEra
-> Either Void (ShelleyTxCert MaryEra)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ShelleyTxCert AllegraEra -> ShelleyTxCert MaryEra
forall era1 era2. ShelleyTxCert era1 -> ShelleyTxCert era2
upgradeShelleyTxCert

  getVKeyWitnessTxCert :: TxCert MaryEra -> Maybe (KeyHash Witness)
getVKeyWitnessTxCert = TxCert MaryEra -> Maybe (KeyHash Witness)
ShelleyTxCert MaryEra -> Maybe (KeyHash Witness)
forall era. ShelleyTxCert era -> Maybe (KeyHash Witness)
getVKeyWitnessShelleyTxCert

  getScriptWitnessTxCert :: TxCert MaryEra -> Maybe ScriptHash
getScriptWitnessTxCert = TxCert MaryEra -> Maybe ScriptHash
ShelleyTxCert MaryEra -> Maybe ScriptHash
forall era. ShelleyTxCert era -> Maybe ScriptHash
getScriptWitnessShelleyTxCert

  mkRegPoolTxCert :: StakePoolParams -> TxCert MaryEra
mkRegPoolTxCert = PoolCert -> ShelleyTxCert MaryEra
forall era. PoolCert -> ShelleyTxCert era
ShelleyTxCertPool (PoolCert -> ShelleyTxCert MaryEra)
-> (StakePoolParams -> PoolCert)
-> StakePoolParams
-> ShelleyTxCert MaryEra
forall b c a. (b -> c) -> (a -> b) -> a -> c
. StakePoolParams -> PoolCert
RegPool

  getRegPoolTxCert :: TxCert MaryEra -> Maybe StakePoolParams
getRegPoolTxCert (ShelleyTxCertPool (RegPool StakePoolParams
poolParams)) = StakePoolParams -> Maybe StakePoolParams
forall a. a -> Maybe a
Just StakePoolParams
poolParams
  getRegPoolTxCert TxCert MaryEra
_ = Maybe StakePoolParams
forall a. Maybe a
Nothing

  mkRetirePoolTxCert :: KeyHash StakePool -> EpochNo -> TxCert MaryEra
mkRetirePoolTxCert KeyHash StakePool
poolId EpochNo
epochNo = PoolCert -> ShelleyTxCert MaryEra
forall era. PoolCert -> ShelleyTxCert era
ShelleyTxCertPool (PoolCert -> ShelleyTxCert MaryEra)
-> PoolCert -> ShelleyTxCert MaryEra
forall a b. (a -> b) -> a -> b
$ KeyHash StakePool -> EpochNo -> PoolCert
RetirePool KeyHash StakePool
poolId EpochNo
epochNo

  getRetirePoolTxCert :: TxCert MaryEra -> Maybe (KeyHash StakePool, EpochNo)
getRetirePoolTxCert (ShelleyTxCertPool (RetirePool KeyHash StakePool
poolId EpochNo
epochNo)) = (KeyHash StakePool, EpochNo) -> Maybe (KeyHash StakePool, EpochNo)
forall a. a -> Maybe a
Just (KeyHash StakePool
poolId, EpochNo
epochNo)
  getRetirePoolTxCert TxCert MaryEra
_ = Maybe (KeyHash StakePool, EpochNo)
forall a. Maybe a
Nothing

  lookupRegStakeTxCert :: TxCert MaryEra -> Maybe (Credential Staking)
lookupRegStakeTxCert = \case
    RegTxCert Credential Staking
c -> Credential Staking -> Maybe (Credential Staking)
forall a. a -> Maybe a
Just Credential Staking
c
    TxCert MaryEra
_ -> Maybe (Credential Staking)
forall a. Maybe a
Nothing
  lookupUnRegStakeTxCert :: TxCert MaryEra -> Maybe (Credential Staking)
lookupUnRegStakeTxCert = \case
    UnRegTxCert Credential Staking
c -> Credential Staking -> Maybe (Credential Staking)
forall a. a -> Maybe a
Just Credential Staking
c
    TxCert MaryEra
_ -> Maybe (Credential Staking)
forall a. Maybe a
Nothing

  getTotalDepositsTxCerts :: forall (f :: * -> *).
Foldable f =>
PParams MaryEra
-> (KeyHash StakePool -> Bool) -> f (TxCert MaryEra) -> Coin
getTotalDepositsTxCerts = PParams MaryEra
-> (KeyHash StakePool -> Bool) -> f (TxCert MaryEra) -> Coin
forall era (f :: * -> *).
(EraPParams era, Foldable f, EraTxCert era) =>
PParams era
-> (KeyHash StakePool -> Bool) -> f (TxCert era) -> Coin
shelleyTotalDepositsTxCerts

  getTotalRefundsTxCerts :: forall (f :: * -> *).
Foldable f =>
PParams MaryEra
-> (Credential Staking -> Maybe Coin)
-> (Credential DRepRole -> Maybe Coin)
-> f (TxCert MaryEra)
-> Coin
getTotalRefundsTxCerts PParams MaryEra
pp Credential Staking -> Maybe Coin
lookupStakeDeposit Credential DRepRole -> Maybe Coin
_ = PParams MaryEra
-> (Credential Staking -> Maybe Coin) -> f (TxCert MaryEra) -> Coin
forall era (f :: * -> *).
(EraPParams era, Foldable f, EraTxCert era) =>
PParams era
-> (Credential Staking -> Maybe Coin) -> f (TxCert era) -> Coin
shelleyTotalRefundsTxCerts PParams MaryEra
pp Credential Staking -> Maybe Coin
lookupStakeDeposit

instance ShelleyEraTxCert MaryEra where
  mkRegTxCert :: Credential Staking -> TxCert MaryEra
mkRegTxCert = ShelleyDelegCert -> ShelleyTxCert MaryEra
forall era. ShelleyDelegCert -> ShelleyTxCert era
ShelleyTxCertDelegCert (ShelleyDelegCert -> ShelleyTxCert MaryEra)
-> (Credential Staking -> ShelleyDelegCert)
-> Credential Staking
-> ShelleyTxCert MaryEra
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Credential Staking -> ShelleyDelegCert
ShelleyRegCert

  getRegTxCert :: TxCert MaryEra -> Maybe (Credential Staking)
getRegTxCert (ShelleyTxCertDelegCert (ShelleyRegCert Credential Staking
c)) = Credential Staking -> Maybe (Credential Staking)
forall a. a -> Maybe a
Just Credential Staking
c
  getRegTxCert TxCert MaryEra
_ = Maybe (Credential Staking)
forall a. Maybe a
Nothing

  mkUnRegTxCert :: Credential Staking -> TxCert MaryEra
mkUnRegTxCert = ShelleyDelegCert -> ShelleyTxCert MaryEra
forall era. ShelleyDelegCert -> ShelleyTxCert era
ShelleyTxCertDelegCert (ShelleyDelegCert -> ShelleyTxCert MaryEra)
-> (Credential Staking -> ShelleyDelegCert)
-> Credential Staking
-> ShelleyTxCert MaryEra
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Credential Staking -> ShelleyDelegCert
ShelleyUnRegCert

  getUnRegTxCert :: TxCert MaryEra -> Maybe (Credential Staking)
getUnRegTxCert (ShelleyTxCertDelegCert (ShelleyUnRegCert Credential Staking
c)) = Credential Staking -> Maybe (Credential Staking)
forall a. a -> Maybe a
Just Credential Staking
c
  getUnRegTxCert TxCert MaryEra
_ = Maybe (Credential Staking)
forall a. Maybe a
Nothing

  mkDelegStakeTxCert :: Credential Staking -> KeyHash StakePool -> TxCert MaryEra
mkDelegStakeTxCert Credential Staking
c KeyHash StakePool
kh = ShelleyDelegCert -> ShelleyTxCert MaryEra
forall era. ShelleyDelegCert -> ShelleyTxCert era
ShelleyTxCertDelegCert (ShelleyDelegCert -> ShelleyTxCert MaryEra)
-> ShelleyDelegCert -> ShelleyTxCert MaryEra
forall a b. (a -> b) -> a -> b
$ Credential Staking -> KeyHash StakePool -> ShelleyDelegCert
ShelleyDelegCert Credential Staking
c KeyHash StakePool
kh

  getDelegStakeTxCert :: TxCert MaryEra -> Maybe (Credential Staking, KeyHash StakePool)
getDelegStakeTxCert (ShelleyTxCertDelegCert (ShelleyDelegCert Credential Staking
c KeyHash StakePool
kh)) = (Credential Staking, KeyHash StakePool)
-> Maybe (Credential Staking, KeyHash StakePool)
forall a. a -> Maybe a
Just (Credential Staking
c, KeyHash StakePool
kh)
  getDelegStakeTxCert TxCert MaryEra
_ = Maybe (Credential Staking, KeyHash StakePool)
forall a. Maybe a
Nothing

  mkGenesisDelegTxCert :: AtMostEra "Babbage" MaryEra => GenesisDelegCert -> TxCert MaryEra
mkGenesisDelegTxCert = GenesisDelegCert -> TxCert MaryEra
GenesisDelegCert -> ShelleyTxCert MaryEra
forall era. GenesisDelegCert -> ShelleyTxCert era
ShelleyTxCertGenesisDeleg

  getGenesisDelegTxCert :: AtMostEra "Babbage" MaryEra =>
TxCert MaryEra -> Maybe GenesisDelegCert
getGenesisDelegTxCert (ShelleyTxCertGenesisDeleg GenesisDelegCert
c) = GenesisDelegCert -> Maybe GenesisDelegCert
forall a. a -> Maybe a
Just GenesisDelegCert
c
  getGenesisDelegTxCert TxCert MaryEra
_ = Maybe GenesisDelegCert
forall a. Maybe a
Nothing

  mkMirTxCert :: AtMostEra "Babbage" MaryEra => MIRCert -> TxCert MaryEra
mkMirTxCert = MIRCert -> TxCert MaryEra
MIRCert -> ShelleyTxCert MaryEra
forall era. MIRCert -> ShelleyTxCert era
ShelleyTxCertMir

  getMirTxCert :: AtMostEra "Babbage" MaryEra => TxCert MaryEra -> Maybe MIRCert
getMirTxCert (ShelleyTxCertMir MIRCert
c) = MIRCert -> Maybe MIRCert
forall a. a -> Maybe a
Just MIRCert
c
  getMirTxCert TxCert MaryEra
_ = Maybe MIRCert
forall a. Maybe a
Nothing