feat: reimplement exclude option

1 files changed, 10 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 26636f7..51ade36 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,8 @@
 use std::{
+    convert::Infallible,
     fmt::Display,
     net::{IpAddr, Ipv4Addr, Ipv6Addr},
+    str::FromStr,
 };
 
 pub mod netlink;
@@ -12,6 +14,14 @@ pub enum Host {
     Hostname(String),
 }
 
+impl FromStr for Host {
+    type Err = Infallible;
+
+    fn from_str(s: &str) -> Result<Self, Self::Err> {
+        Ok(IpAddr::from_str(s).map_or_else(|_| s.to_string().into(), Host::from))
+    }
+}
+
 impl From<IpAddr> for Host {
     fn from(value: IpAddr) -> Self {
         match dns_lookup::lookup_addr(&value) {