Added GetAccountState query

ouroboros-consensus-cardano/changelog.d/20240503_143529_joosep.jaager_accountstate_query.md

@@ -0,0 +1,4 @@
+### Breaking
+
+- Add `GetAccountState` query
+

ouroboros-consensus-cardano/src/shelley/Ouroboros/Consensus/Shelley/Ledger/Query.hs

@@ -87,6 +87,7 @@ import           Ouroboros.Consensus.Shelley.Protocol.Abstract (ProtoCrypto)
 import           Ouroboros.Consensus.Util (ShowProxy (..))
 import           Ouroboros.Network.Block (Serialised (..), decodePoint,
                      encodePoint, mkSerialised)
+import Cardano.Ledger.Shelley.LedgerState (AccountState)
 
 {-------------------------------------------------------------------------------
   QueryLedger
@@ -297,6 +298,9 @@ data instance BlockQuery (ShelleyBlock proto era) :: Type -> Type where
   -- longer used (because mainnet has hard-forked to a newer version), it can be
   -- removed.
 
+  GetAccountState 
+    :: BlockQuery (ShelleyBlock proto era) AccountState
+
 instance (Typeable era, Typeable proto)
   => ShowProxy (BlockQuery (ShelleyBlock proto era)) where
 
@@ -434,6 +438,8 @@ instance (ShelleyCompatible proto era, ProtoCrypto proto ~ crypto)
           SL.queryCommitteeMembersState coldCreds hotCreds statuses st
         GetFilteredVoteDelegatees stakeCreds ->
           getFilteredVoteDelegatees st stakeCreds
+        GetAccountState ->
+          SL.queryAccountState st
     where
       lcfg    = configLedger $ getExtLedgerCfg cfg
       globals = shelleyLedgerGlobals lcfg
@@ -585,6 +591,8 @@ instance SameDepIndex (BlockQuery (ShelleyBlock proto era)) where
     | otherwise
     = Nothing
   sameDepIndex GetFilteredVoteDelegatees {} _ = Nothing
+  sameDepIndex GetAccountState {} GetAccountState {} = Just Refl
+  sameDepIndex GetAccountState {} _ = Nothing
 
 deriving instance Eq   (BlockQuery (ShelleyBlock proto era) result)
 deriving instance Show (BlockQuery (ShelleyBlock proto era) result)
@@ -620,6 +628,7 @@ instance ShelleyCompatible proto era => ShowQuery (BlockQuery (ShelleyBlock prot
       GetDRepStakeDistr {}                       -> show
       GetCommitteeMembersState {}                -> show
       GetFilteredVoteDelegatees {}               -> show
+      GetAccountState {}                         -> show
 
 -- | Is the given query supported by the given 'ShelleyNodeToClientVersion'?
 querySupportedVersion :: BlockQuery (ShelleyBlock proto era) result -> ShelleyNodeToClientVersion -> Bool
@@ -653,6 +662,7 @@ querySupportedVersion = \case
     GetDRepStakeDistr {}                       -> (>= v8)
     GetCommitteeMembersState {}                -> (>= v8)
     GetFilteredVoteDelegatees {}               -> (>= v8)
+    GetAccountState {}                         -> (>= v8)
     -- WARNING: when adding a new query, a new @ShelleyNodeToClientVersionX@
     -- must be added. See #2830 for a template on how to do this.
   where
@@ -774,6 +784,8 @@ encodeShelleyQuery query = case query of
       CBOR.encodeListLen 4 <> CBOR.encodeWord8 27 <> toCBOR coldCreds <> toCBOR hotCreds <> LC.toEraCBOR @era statuses
     GetFilteredVoteDelegatees stakeCreds ->
       CBOR.encodeListLen 2 <> CBOR.encodeWord8 28 <> LC.toEraCBOR @era stakeCreds
+    GetAccountState ->
+      CBOR.encodeListLen 1 <> CBOR.encodeWord8 29
 
 decodeShelleyQuery ::
      forall era proto. ShelleyBasedEra era
@@ -830,6 +842,7 @@ decodeShelleyQuery = do
         return $ SomeSecond $ GetCommitteeMembersState coldCreds hotCreds statuses
       (2, 28) -> requireCG $ do
         SomeSecond . GetFilteredVoteDelegatees <$> LC.fromEraCBOR @era
+      (1, 29) -> requireCG $ return $ SomeSecond GetAccountState
       _       -> failmsg "invalid"
 
 encodeShelleyResult ::
@@ -866,6 +879,7 @@ encodeShelleyResult v query = case query of
     GetDRepStakeDistr {}                       -> LC.toEraCBOR @era
     GetCommitteeMembersState {}                -> LC.toEraCBOR @era
     GetFilteredVoteDelegatees {}               -> LC.toEraCBOR @era
+    GetAccountState {}                         -> LC.toEraCBOR @era
 
 decodeShelleyResult ::
      forall proto era result. ShelleyCompatible proto era
@@ -902,6 +916,7 @@ decodeShelleyResult v query = case query of
     GetDRepStakeDistr {}                       -> LC.fromEraCBOR @era
     GetCommitteeMembersState {}                -> LC.fromEraCBOR @era
     GetFilteredVoteDelegatees {}               -> LC.fromEraCBOR @era
+    GetAccountState {}                         -> LC.fromEraCBOR @era
 
 currentPParamsEnDecoding ::
      forall era s.