openassets-tapyrus 0.2.2

use std::fmt::{self, Display, Formatter};
use tapyrus::consensus::encode;
use tapyrus::network::constants::Network;
use tapyrus::util::address::Payload;
use tapyrus::util::base58;

/// A Open Assets Address
#[derive(Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Address {
    pub network: Network,
    pub payload: Payload,
}

const NAMESPACE: u8 = 0x13;

impl Address {
    pub fn new(
        payload: Payload,
        network: tapyrus::network::constants::Network,
    ) -> Result<Self, encode::Error> {
        Ok(Address { payload, network })
    }

    pub fn to_btc_addr(&self) -> Result<tapyrus::Address, encode::Error> {
        Ok(tapyrus::Address {
            network: self.network,
            payload: self.payload.clone(),
        })
    }
}

impl Display for Address {
    fn fmt(&self, fmt: &mut Formatter) -> fmt::Result {
        let mut prefixed = [0; 22];
        prefixed[0] = NAMESPACE;
        prefixed[1] = match self.network {
            tapyrus::network::constants::Network::Prod => 0,
            tapyrus::network::constants::Network::Dev => 111,
        };
        match self.payload {
            Payload::PubkeyHash(ref hash) => {
                prefixed[2..].copy_from_slice(&hash[..]);
                base58::check_encode_slice_to_fmt(fmt, &prefixed[..])
            }
            Payload::ScriptHash(ref hash) => {
                prefixed[2..].copy_from_slice(&hash[..]);
                base58::check_encode_slice_to_fmt(fmt, &prefixed[..])
            }
        }
    }
}

pub trait OAAddressConverter {
    fn to_oa_address(&self) -> Result<Address, encode::Error>;
}

impl OAAddressConverter for tapyrus::Address {
    fn to_oa_address(&self) -> Result<Address, encode::Error> {
        Address::new(self.payload.clone(), self.network)
    }
}

#[cfg(test)]
mod tests {
    use openassets::address::OAAddressConverter;
    use std::str::FromStr;
    use std::string::ToString;

    #[test]
    fn test_oa_address_converter() {
        let addr = tapyrus::Address::from_str("1F2AQr6oqNtcJQ6p9SiCLQTrHuM9en44H8").unwrap();
        assert_eq!(
            "akQz3f1v9JrnJAeGBC4pNzGNRdWXKan4U6E",
            addr.to_oa_address().unwrap().to_string()
        );
        assert_eq!(addr, addr.to_oa_address().unwrap().to_btc_addr().unwrap());

        let testnet_addr =
            tapyrus::Address::from_str("mkgW6hNYBctmqDtTTsTJrsf2Gh2NPtoCU4").unwrap();
        assert_eq!(
            "bWvePLsBsf6nThU3pWVZVWjZbcJCYQxHCpE",
            testnet_addr.to_oa_address().unwrap().to_string()
        );
        assert_eq!(
            testnet_addr,
            testnet_addr.to_oa_address().unwrap().to_btc_addr().unwrap()
        );
    }
}