fedimovies/mitra-utils/src/did_pkh.rs
2023-04-24 17:35:32 +02:00

82 lines
2.4 KiB
Rust

/// https://github.com/w3c-ccg/did-pkh/blob/main/did-pkh-method-draft.md
use std::fmt;
use std::str::FromStr;
use regex::Regex;
use super::{caip2::ChainId, currencies::Currency, did::DidParseError};
// https://github.com/ChainAgnostic/CAIPs/blob/master/CAIPs/caip-10.md#syntax
const DID_PKH_RE: &str = r"did:pkh:(?P<network>[-a-z0-9]{3,8}):(?P<chain>[-a-zA-Z0-9]{1,32}):(?P<address>[a-zA-Z0-9]{1,64})";
#[derive(Clone, Debug, PartialEq)]
pub struct DidPkh {
pub chain_id: ChainId,
pub address: String,
}
impl DidPkh {
pub fn from_address(currency: &Currency, address: &str) -> Self {
let chain_id = match currency {
Currency::Ethereum => ChainId::ethereum_mainnet(),
Currency::Monero => unimplemented!(),
};
let address = currency.normalize_address(address);
Self { chain_id, address }
}
pub fn currency(&self) -> Option<Currency> {
self.chain_id.currency()
}
}
impl fmt::Display for DidPkh {
fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
let did_str = format!(
"did:pkh:{}:{}:{}",
self.chain_id.namespace, self.chain_id.reference, self.address,
);
write!(formatter, "{}", did_str)
}
}
impl FromStr for DidPkh {
type Err = DidParseError;
fn from_str(value: &str) -> Result<Self, Self::Err> {
let did_pkh_re = Regex::new(DID_PKH_RE).unwrap();
let caps = did_pkh_re.captures(value).ok_or(DidParseError)?;
let did_pkh = Self {
chain_id: ChainId {
namespace: caps["network"].to_string(),
reference: caps["chain"].to_string(),
},
address: caps["address"].to_string(),
};
Ok(did_pkh)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_did_pkh_string_conversion() {
let address = "0xB9C5714089478a327F09197987f16f9E5d936E8a";
let ethereum = Currency::Ethereum;
let did = DidPkh::from_address(&ethereum, address);
assert_eq!(did.currency().unwrap(), ethereum);
assert_eq!(did.address, address.to_lowercase());
let did_str = did.to_string();
assert_eq!(
did_str,
"did:pkh:eip155:1:0xb9c5714089478a327f09197987f16f9e5d936e8a",
);
let did: DidPkh = did_str.parse().unwrap();
assert_eq!(did.address, address.to_lowercase());
}
}