Sulkta-OSS/pallas
3
0
Fork 0
Code Issues 1 Pull requests 4 Projects Releases Packages Wiki Activity Actions

feat(traverse): Add missing getters on output (#162)

Browse source
This commit is contained in:
Sebastien Guillemot 2022-08-07 03:48:32 +09:00 • committed by GitHub
parent 87160d02eb
commit c6a07e2577
2 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pallas-primitives/src/babbage/model.rs
View file

@ -93,7 +93,7 @@ pub use crate::alonzo::Coin;
pub use crate::alonzo::Value; pub use crate::alonzo::Value;
pub use crate::alonzo::TransactionOutput as LegacyTransacionOutput; pub use crate::alonzo::TransactionOutput as LegacyTransactionOutput;
pub use crate::alonzo::PoolKeyhash; pub use crate::alonzo::PoolKeyhash;
@ -285,7 +285,7 @@ pub struct TransactionBody {
#[derive(Debug, PartialEq, Clone)] #[derive(Debug, PartialEq, Clone)]
pub enum TransactionOutput { pub enum TransactionOutput {
Legacy(LegacyTransacionOutput), Legacy(LegacyTransactionOutput),
PostAlonzo(PostAlonzoTransactionOutput), PostAlonzo(PostAlonzoTransactionOutput),
} }

23
pallas-traverse/src/output.rs
View file

@ -2,7 +2,7 @@ use std::{borrow::Cow, ops::Deref};
use pallas_addresses::{Address, ByronAddress, Error as AddressError}; use pallas_addresses::{Address, ByronAddress, Error as AddressError};
use pallas_codec::minicbor; use pallas_codec::minicbor;
use pallas_primitives::{alonzo, babbage, byron}; use pallas_primitives::{alonzo, babbage::{self, DatumOption, ScriptRef}, byron};
use crate::{Era, MultiEraOutput}; use crate::{Era, MultiEraOutput};
@ -19,6 +19,27 @@ impl<'b> MultiEraOutput<'b> {
Self::Babbage(Box::new(Cow::Borrowed(output))) Self::Babbage(Box::new(Cow::Borrowed(output)))
} }
pub fn datum(&self) -> Option<DatumOption> {
match self {
MultiEraOutput::AlonzoCompatible(x) => x.datum_hash.map(DatumOption::Hash),
MultiEraOutput::Babbage(x) => match x.deref().deref() {
babbage::TransactionOutput::Legacy(x) => x.datum_hash.map(DatumOption::Hash),
babbage::TransactionOutput::PostAlonzo(x) => x.datum_option.clone(),
},
_ => None
}
}
pub fn script_ref(&self) -> Option<&ScriptRef> {
match &self {
MultiEraOutput::Babbage(x) => match x.deref().deref() {
babbage::TransactionOutput::Legacy(_) => None,
babbage::TransactionOutput::PostAlonzo(x) => x.script_ref.as_ref(),
},
_ => None
}
}
pub fn address(&self) -> Result<Address, AddressError> { pub fn address(&self) -> Result<Address, AddressError> {
match self { match self {
MultiEraOutput::AlonzoCompatible(x) => Address::from_bytes(&x.address), MultiEraOutput::AlonzoCompatible(x) => Address::from_bytes(&x.address),