Extract enacted treasury withdrawals

Author: kderme

cardano-db-sync/src/Cardano/DbSync/Era/Universal/Insert/LedgerEvent.hs

@@ -19,14 +19,15 @@ import Cardano.DbSync.Era.Cardano.Insert (insertEpochSyncTime)
 import qualified Cardano.DbSync.Era.Shelley.Generic as Generic
 import Cardano.DbSync.Era.Universal.Adjust (adjustEpochRewards)
 import Cardano.DbSync.Era.Universal.Epoch (insertPoolDepositRefunds, insertProposalRefunds, insertRewardRests, insertRewards)
-import Cardano.DbSync.Era.Universal.Insert.GovAction (updateDropped)
+import Cardano.DbSync.Era.Universal.Insert.GovAction
 import Cardano.DbSync.Era.Universal.Validate (validateEpochRewards)
 import Cardano.DbSync.Error
 import Cardano.DbSync.Ledger.Event
 import Cardano.DbSync.Types
 import Cardano.DbSync.Util
 import Cardano.Prelude
 import Cardano.Slotting.Slot (EpochNo (..))
+import Control.Monad.Extra (whenJust)
 import Control.Monad.Trans.Control (MonadBaseControl)
 import qualified Data.Map.Strict as Map
 import qualified Data.Set as Set
@@ -101,6 +102,12 @@ insertNewEpochLedgerEvents syncEnv currentEpochNo@(EpochNo curEpoch) =
           let en' = filter (\e -> Set.notMember (garGovActionId e) uncl) en
               ex' = filter (\e -> Set.notMember (garGovActionId e) uncl) ex
           insertProposalRefunds ntw (subFromCurrentEpoch 1) currentEpochNo cache (en' <> ex') -- TODO: check if they are disjoint to avoid double entries.
+          forM_ en $ \gar -> whenJust (garMTreasury gar) $ \treasuryMap -> do
+            gaId <- resolveGovActionProposal (garGovActionId gar)
+            lift $ void $ DB.updateGovActionEnacted gaId (unEpochNo currentEpochNo)
+            forM_ (Map.toList treasuryMap) $ \(_ra, _c) ->
+              -- TODO: Add reward
+              pure ()
         LedgerMirDist rwd -> do
           unless (Map.null rwd) $ do
             let rewards = Map.toList rwd

cardano-db-sync/src/Cardano/DbSync/Ledger/Event.hs

@@ -20,7 +20,7 @@ module Cardano.DbSync.Ledger.Event (
   splitDeposits,
 ) where
 
-import Cardano.Db hiding (AdaPots, EpochNo, SyncState, epochNo)
+import Cardano.Db hiding (AdaPots, EpochNo, SyncState, TreasuryWithdrawals, epochNo)
 import qualified Cardano.DbSync.Era.Shelley.Generic as Generic
 import Cardano.DbSync.Era.Shelley.Generic.Tx.Shelley
 import Cardano.DbSync.Types
@@ -29,8 +29,8 @@ import Cardano.Ledger.Address (RewardAccount)
 import qualified Cardano.Ledger.Allegra.Rules as Allegra
 import Cardano.Ledger.Alonzo.Rules (AlonzoBbodyEvent (..), AlonzoUtxoEvent (..), AlonzoUtxowEvent (..))
 import qualified Cardano.Ledger.Alonzo.Rules as Alonzo
-import Cardano.Ledger.Api (GovActionId, GovActionState (..), ProposalProcedure (..))
 import Cardano.Ledger.Coin (Coin (..))
+import Cardano.Ledger.Conway.Governance
 import Cardano.Ledger.Conway.Rules as Conway
 import qualified Cardano.Ledger.Core as Ledger
 import Cardano.Ledger.SafeHash (SafeHash)
@@ -83,6 +83,7 @@ data GovActionRefunded = GovActionRefunded
   { garGovActionId :: GovActionId StandardCrypto
   , garDeposit :: Coin
   , garReturnAddr :: RewardAccount StandardCrypto
+  , garMTreasury :: Maybe (Map (RewardAccount StandardCrypto) Coin)
   , garIsEnacted :: Bool -- True for enacted, False for retired, possibly redundant
   }
   deriving (Eq)
@@ -267,8 +268,13 @@ toLedgerEventConway evt hasRewards =
         { garGovActionId = gasId gas
         , garDeposit = pProcDeposit $ gasProposalProcedure gas
         , garReturnAddr = pProcReturnAddr $ gasProposalProcedure gas
+        , garMTreasury = mWithrawal
         , garIsEnacted = isEnacted
         }
+      where
+        mWithrawal = case pProcGovAction (gasProposalProcedure gas) of
+          TreasuryWithdrawals mp _ -> Just mp
+          _ -> Nothing
 
 instance All ConvertLedgerEvent xs => ConvertLedgerEvent (HardForkBlock xs) where
   toLedgerEvent hasRewards =