generate Network <-> Magic From and TryFrom with a macro

This commit is contained in:
Antoni Spaanderman 2024-06-09 00:23:28 +02:00
parent f260b097f4
commit 76826313a1
No known key found for this signature in database
GPG Key ID: AE0B68E552E5DF8C
1 changed files with 29 additions and 22 deletions

View File

@ -243,14 +243,14 @@ impl FromStr for Magic {
}
}
macro_rules! generate_network_magic_conversion {
($(Network::$network:ident => Magic::$magic:ident,)*) => {
impl From<Network> for Magic {
fn from(network: Network) -> Magic {
match network {
// Note: new network entries must explicitly be matched in `try_from` below.
Network::Bitcoin => Magic::BITCOIN,
Network::Testnet => Magic::TESTNET,
Network::Signet => Magic::SIGNET,
Network::Regtest => Magic::REGTEST,
$(
Network::$network => Magic::$magic,
)*
}
}
}
@ -260,15 +260,22 @@ impl TryFrom<Magic> for Network {
fn try_from(magic: Magic) -> Result<Self, Self::Error> {
match magic {
// Note: any new network entries must be matched against here.
Magic::BITCOIN => Ok(Network::Bitcoin),
Magic::TESTNET => Ok(Network::Testnet),
Magic::SIGNET => Ok(Network::Signet),
Magic::REGTEST => Ok(Network::Regtest),
$(
Magic::$magic => Ok(Network::$network),
)*
_ => Err(UnknownMagicError(magic)),
}
}
}
};
}
generate_network_magic_conversion! {
Network::Bitcoin => Magic::BITCOIN,
Network::Testnet => Magic::TESTNET,
Network::Signet => Magic::SIGNET,
Network::Regtest => Magic::REGTEST,
}
impl fmt::Display for Magic {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {