feat: fill in cert redeemers

txpipe · Jul 15, 2024 · 4a12d53 · 4a12d53
1 parent bb30164
commit 4a12d53
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/pallas-traverse/src/witnesses.rs b/pallas-traverse/src/witnesses.rs
@@ -174,6 +174,13 @@ impl<'b> MultiEraTx<'b> {
         })
     }
 
+    pub fn find_certificate_redeemer(&self, certificate_order: u32) -> Option<MultiEraRedeemer> {
+        self.redeemers().into_iter().find(|r| {
+            r.tag() == pallas_primitives::conway::RedeemerTag::Cert
+                && r.index() == certificate_order
+        })
+    }
+
     pub fn plutus_v2_scripts(&self) -> &[PlutusV2Script] {
         match self {
             Self::Byron(_) => &[],

diff --git a/pallas-utxorpc/src/lib.rs b/pallas-utxorpc/src/lib.rs
@@ -191,7 +191,12 @@ impl<C: LedgerContext> Mapper<C> {
         }
     }
 
-    pub fn map_cert(&self, x: &trv::MultiEraCert) -> u5c::Certificate {
+    pub fn map_cert(
+        &self,
+        x: &trv::MultiEraCert,
+        tx: &trv::MultiEraTx,
+        order: u32,
+    ) -> u5c::Certificate {
         let inner = match x.as_alonzo().unwrap() {
             babbage::Certificate::StakeRegistration(a) => {
                 u5c::certificate::Certificate::StakeRegistration(self.map_stake_credential(a))
@@ -279,15 +284,16 @@ impl<C: LedgerContext> Mapper<C> {
 
         u5c::Certificate {
             certificate: inner.into(),
-            redeemer: None, // TODO
+            redeemer: tx
+                .find_certificate_redeemer(order)
+                .map(|r| self.map_redeemer(&r)),
         }
     }
 
     pub fn map_withdrawals(
         &self,
         x: &(&[u8], u64),
         tx: &trv::MultiEraTx,
-        // lexicographical order of the input we're mapping
         order: u32,
     ) -> u5c::Withdrawal {
         u5c::Withdrawal {
@@ -542,7 +548,12 @@ impl<C: LedgerContext> Mapper<C> {
                 .map(|(order, i)| self.map_tx_input(i, tx, order as u32, &resolved))
                 .collect(),
             outputs: tx.outputs().iter().map(|x| self.map_tx_output(x)).collect(),
-            certificates: tx.certs().iter().map(|x| self.map_cert(x)).collect(),
+            certificates: tx
+                .certs()
+                .iter()
+                .enumerate()
+                .map(|(order, x)| self.map_cert(x, tx, order as u32))
+                .collect(),
             withdrawals: tx
                 .withdrawals_sorted_set()
                 .iter()