added ada amount method on output

txpipe · Jun 20, 2022 · fba54db · fba54db
1 parent 1984488
commit fba54db
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pallas-traverse/src/output.rs b/pallas-traverse/src/output.rs
@@ -1,6 +1,6 @@
 use std::borrow::Cow;
 
-use pallas_primitives::{alonzo, byron};
+use pallas_primitives::{alonzo::{self, Value}, byron};
 
 use crate::MultiEraOutput;
 
@@ -22,6 +22,16 @@ impl<'b> MultiEraOutput<'b> {
         }
     }
 
+    pub fn ada_amount(&self) -> u64 {
+        match self {
+            MultiEraOutput::Byron(x) => x.amount,
+            MultiEraOutput::AlonzoCompatible(x) => match x.amount {
+                Value::Coin(c) => u64::from(c),
+                Value::Multiasset(c, _) => u64::from(c)
+            }
+        }
+    }
+
     pub fn as_alonzo(&self) -> Option<&alonzo::TransactionOutput> {
         match self {
             MultiEraOutput::Byron(_) => None,