use crate::{DomainRef, IPAddress};
#[derive(Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Hash, Debug)]
pub enum HostRef<'a> {
Name(DomainRef<'a>),
Address(IPAddress),
}
impl<'a> From<DomainRef<'a>> for HostRef<'a> {
fn from(domain: DomainRef<'a>) -> Self {
Self::Name(domain)
}
}
impl<'a, A: Into<IPAddress>> From<A> for HostRef<'a> {
fn from(ip: A) -> Self {
Self::Address(ip.into())
}
}
impl<'a> HostRef<'a> {
pub const fn is_domain(&self) -> bool {
matches!(self, Self::Name(_))
}
pub const fn is_ip(&self) -> bool {
matches!(self, Self::Address(_))
}
}
#[cfg(test)]
mod tests {
use crate::{DomainRef, HostRef, IPAddress, IPv4Address};
#[test]
fn construction() {
let result: HostRef = DomainRef::LOCALHOST.into();
let expected: HostRef = HostRef::Name(DomainRef::LOCALHOST);
assert_eq!(result, expected);
let result: HostRef = IPAddress::V4(IPv4Address::LOCALHOST).into();
let expected: HostRef = HostRef::Address(IPAddress::V4(IPv4Address::LOCALHOST));
assert_eq!(result, expected);
}
#[test]
fn matching() {
let host: HostRef = DomainRef::LOCALHOST.into();
assert!(host.is_domain());
assert!(!host.is_ip());
let host: HostRef = IPv4Address::LOCALHOST.into();
assert!(!host.is_domain());
assert!(host.is_ip());
}
}