summaryrefslogtreecommitdiffstats
path: root/src/lib.rs
blob: 26636f7a8131bc35525d57e803ab6ab9be1d75d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
use std::{
    fmt::Display,
    net::{IpAddr, Ipv4Addr, Ipv6Addr},
};

pub mod netlink;
pub mod tmux;

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum Host {
    IpAddr(IpAddr),
    Hostname(String),
}

impl From<IpAddr> for Host {
    fn from(value: IpAddr) -> Self {
        match dns_lookup::lookup_addr(&value) {
            Ok(s) => Self::Hostname(s),
            Err(_) => Self::IpAddr(value),
        }
    }
}

impl From<Ipv4Addr> for Host {
    fn from(value: Ipv4Addr) -> Self {
        IpAddr::from(value).into()
    }
}

impl From<Ipv6Addr> for Host {
    fn from(value: Ipv6Addr) -> Self {
        IpAddr::from(value).into()
    }
}

impl From<String> for Host {
    fn from(value: String) -> Self {
        Self::Hostname(value)
    }
}

impl Display for Host {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Host::IpAddr(i) => write!(f, "{}", i),
            Host::Hostname(s) => write!(f, "{}", s),
        }
    }
}