WithdrawalRecord (#349)

txpipe · Jul 23, 2022 · 44cacbe · 44cacbe
1 parent a559934
commit 44cacbe
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 4 deletions.
diff --git a/src/mapper/babbage.rs b/src/mapper/babbage.rs
@@ -89,6 +89,10 @@ impl EventWriter {
                     .collect_plutus_datum_records(&witnesses.plutus_data)?
                     .into();
             }
+
+            if let Some(withdrawals) = &body.withdrawals {
+                record.withdrawals = self.collect_withdrawal_records(withdrawals).into();
+            }
         }
 
         Ok(record)

diff --git a/src/mapper/collect.rs b/src/mapper/collect.rs
@@ -1,9 +1,9 @@
 use pallas::{
-    codec::utils::MaybeIndefArray,
+    codec::utils::{KeyValuePairs, MaybeIndefArray},
     ledger::primitives::{
         alonzo::{
-            AuxiliaryData, MintedBlock, Multiasset, NativeScript, PlutusData, PlutusScript,
-            Redeemer, TransactionInput, VKeyWitness, Value,
+            AuxiliaryData, Coin, MintedBlock, Multiasset, NativeScript, PlutusData, PlutusScript,
+            Redeemer, RewardAccount, TransactionInput, VKeyWitness, Value,
         },
         babbage::{
             LegacyTransacionOutput, PlutusV2Script, PostAlonzoTransactionOutput, TransactionOutput,
@@ -16,7 +16,7 @@ use crate::{
     model::{
         MetadataRecord, MintRecord, NativeWitnessRecord, OutputAssetRecord, PlutusDatumRecord,
         PlutusRedeemerRecord, PlutusWitnessRecord, TransactionRecord, TxInputRecord,
-        TxOutputRecord, VKeyWitnessRecord,
+        TxOutputRecord, VKeyWitnessRecord, WithdrawalRecord,
     },
     Error,
 };
@@ -88,6 +88,19 @@ impl EventWriter {
             .collect()
     }
 
+    pub fn collect_withdrawal_records(
+        &self,
+        withdrawls: &KeyValuePairs<RewardAccount, Coin>,
+    ) -> Vec<WithdrawalRecord> {
+        withdrawls
+            .iter()
+            .map(|(reward_account, coin)| WithdrawalRecord {
+                reward_account: reward_account.to_hex(),
+                coin: coin.into(),
+            })
+            .collect()
+    }
+
     pub fn collect_metadata_records(
         &self,
         aux_data: &AuxiliaryData,

diff --git a/src/mapper/map.rs b/src/mapper/map.rs
@@ -442,6 +442,10 @@ impl EventWriter {
                     .collect_plutus_datum_records(&witnesses.plutus_data)?
                     .into();
             }
+
+            if let Some(withdrawals) = &body.withdrawals {
+                record.withdrawals = self.collect_withdrawal_records(withdrawals).into();
+            }
         }
 
         Ok(record)

diff --git a/src/model.rs b/src/model.rs
@@ -146,6 +146,12 @@ impl From<MintRecord> for EventData {
     }
 }
 
+#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
+pub struct WithdrawalRecord {
+    pub reward_account: String,
+    pub coin: u64,
+}
+
 #[derive(Serialize, Deserialize, Debug, Clone, Default, PartialEq, Eq)]
 pub struct TransactionRecord {
     pub hash: String,
@@ -168,6 +174,7 @@ pub struct TransactionRecord {
     pub plutus_witnesses: Option<Vec<PlutusWitnessRecord>>,
     pub plutus_redeemers: Option<Vec<PlutusRedeemerRecord>>,
     pub plutus_data: Option<Vec<PlutusDatumRecord>>,
+    pub withdrawals: Option<Vec<WithdrawalRecord>>,
 }
 
 impl From<TransactionRecord> for EventData {