From 0ada7cb4c74f8f72ad15271b8b6553c27c2f1c86 Mon Sep 17 00:00:00 2001
From: Santiago Carmuega <santiago@carmuega.me>
Date: Fri, 17 Jun 2022 08:31:26 -0300
Subject: [PATCH] feat(traverse): Add output-at helper method (#124)

---
 pallas-traverse/src/lib.rs |  2 +-
 pallas-traverse/src/tx.rs  | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/pallas-traverse/src/lib.rs b/pallas-traverse/src/lib.rs
index 2c707d1..a87bb14 100644
--- a/pallas-traverse/src/lib.rs
+++ b/pallas-traverse/src/lib.rs
@@ -52,7 +52,7 @@ pub enum MultiEraBlock<'b> {
     Byron(Box<Cow<'b, byron::MintedBlock<'b>>>),
 }
 
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 #[non_exhaustive]
 pub enum MultiEraTx<'b> {
     AlonzoCompatible(Box<Cow<'b, alonzo::MintedTx<'b>>>, Era),
diff --git a/pallas-traverse/src/tx.rs b/pallas-traverse/src/tx.rs
index 9774bdb..abae439 100644
--- a/pallas-traverse/src/tx.rs
+++ b/pallas-traverse/src/tx.rs
@@ -67,6 +67,21 @@ impl<'b> MultiEraTx<'b> {
         }
     }
 
+    pub fn output_at(&self, index: usize) -> Option<MultiEraOutput> {
+        match self {
+            MultiEraTx::AlonzoCompatible(x, _) => x
+                .transaction_body
+                .outputs
+                .get(index)
+                .map(MultiEraOutput::from_alonzo_compatible),
+            MultiEraTx::Byron(x) => x
+                .transaction
+                .outputs
+                .get(index)
+                .map(MultiEraOutput::from_byron),
+        }
+    }
+
     pub fn inputs(&self) -> Vec<MultiEraInput> {
         match self {
             MultiEraTx::AlonzoCompatible(x, _) => x