use core::fmt;
use core::str::FromStr;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default, Hash)]
#[non_exhaustive]
pub enum Network {
#[default]
Mainnet,
Testnet,
}
impl Network {
#[inline]
#[must_use]
pub const fn coin_type(self) -> u32 {
match self {
Self::Mainnet => 0,
Self::Testnet => 1,
}
}
#[inline]
#[must_use]
pub const fn name(self) -> &'static str {
match self {
Self::Mainnet => "mainnet",
Self::Testnet => "testnet",
}
}
}
impl fmt::Display for Network {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(self.name())
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
#[non_exhaustive]
pub struct ParseNetworkError;
impl fmt::Display for ParseNetworkError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("invalid network, expected: mainnet or testnet")
}
}
#[cfg(feature = "std")]
impl std::error::Error for ParseNetworkError {}
impl FromStr for Network {
type Err = ParseNetworkError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s.to_lowercase().as_str() {
"mainnet" | "main" | "bitcoin" => Ok(Self::Mainnet),
"testnet" | "test" | "testnet3" | "testnet4" => Ok(Self::Testnet),
_ => Err(ParseNetworkError),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn network_from_str() {
assert_eq!("mainnet".parse::<Network>().unwrap(), Network::Mainnet);
assert_eq!("main".parse::<Network>().unwrap(), Network::Mainnet);
assert_eq!("bitcoin".parse::<Network>().unwrap(), Network::Mainnet);
assert_eq!("testnet".parse::<Network>().unwrap(), Network::Testnet);
assert_eq!("test".parse::<Network>().unwrap(), Network::Testnet);
}
#[test]
fn network_from_str_case_insensitive() {
assert_eq!("MAINNET".parse::<Network>().unwrap(), Network::Mainnet);
assert_eq!("TESTNET".parse::<Network>().unwrap(), Network::Testnet);
}
#[test]
fn network_from_str_invalid() {
assert!("invalid".parse::<Network>().is_err());
assert!("".parse::<Network>().is_err());
}
#[test]
fn network_coin_type() {
assert_eq!(Network::Mainnet.coin_type(), 0);
assert_eq!(Network::Testnet.coin_type(), 1);
}
#[test]
fn network_default() {
assert_eq!(Network::default(), Network::Mainnet);
}
#[test]
fn network_display() {
assert_eq!(Network::Mainnet.to_string(), "mainnet");
assert_eq!(Network::Testnet.to_string(), "testnet");
}
}