hostaddr 0.2.4

use super::{Domain, Host, ParseAsciiHostError, ParseHostError};

use core::{
  net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6},
  str::FromStr,
};

/// An error which can be returned when parsing a [`HostAddr`].
#[derive(Debug, thiserror::Error)]
pub enum ParseHostAddrError {
  /// Returned if the provided str does not contains a valid host.
  #[error(transparent)]
  Host(#[from] ParseHostError),
  /// Returned if the provided str does not contains a valid port.
  #[error(transparent)]
  Port(#[from] core::num::ParseIntError),
}

impl ParseHostAddrError {
  #[inline]
  const fn host() -> Self {
    Self::Host(ParseHostError(()))
  }
}

/// An error which can be returned when parsing a [`HostAddr`].
#[derive(Debug, thiserror::Error)]
pub enum ParseAsciiHostAddrError {
  /// Returned if the provided str does not contains a valid host.
  #[error(transparent)]
  Host(#[from] ParseAsciiHostError),
  /// Returned if the provided str does not contains a valid port.
  #[error(transparent)]
  Port(#[from] core::num::ParseIntError),
}

impl ParseAsciiHostAddrError {
  #[inline]
  const fn host() -> Self {
    Self::Host(ParseAsciiHostError(()))
  }
}

/// A host address, which consists of a [`Host`] and an optional port number.
#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct HostAddr<S> {
  /// The host name
  pub(super) host: Host<S>,
  /// The port number
  pub(super) port: Option<u16>,
}

#[cfg(feature = "cheap-clone")]
impl<S: cheap_clone::CheapClone> cheap_clone::CheapClone for HostAddr<S> {}

impl<S> core::fmt::Display for HostAddr<S>
where
  S: core::fmt::Display,
{
  fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
    match (self.host.is_ipv6(), self.port) {
      (true, Some(port)) => write!(f, "[{}]:{}", self.host, port),
      (_, Some(port)) => write!(f, "{}:{}", self.host, port),
      _ => write!(f, "{}", self.host),
    }
  }
}

impl<S> From<Domain<S>> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{Domain, HostAddr};
  ///
  /// let domain = Domain::<String>::try_from("example.com").unwrap();
  /// let host = HostAddr::<String>::from(domain);
  /// # }
  /// ```
  fn from(domain: Domain<S>) -> Self {
    Self::from_domain(domain)
  }
}

impl<S> From<(Domain<S>, u16)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{Domain, HostAddr};
  ///
  /// let domain = Domain::<String>::try_from("example.com").unwrap();
  /// let host = HostAddr::<String>::from((domain, 8080));
  /// # }
  /// ```
  fn from((host, port): (Domain<S>, u16)) -> Self {
    Self::from_domain(host).with_port(port)
  }
}

impl<S> From<(u16, Domain<S>)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{Domain, HostAddr};
  ///
  /// let domain = Domain::<String>::try_from("example.com").unwrap();
  /// let host = HostAddr::<String>::from((8080, domain));
  /// # }
  /// ```
  fn from((port, host): (u16, Domain<S>)) -> Self {
    Self::from_domain(host).with_port(port)
  }
}

impl<S> From<(IpAddr, u16)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::IpAddr;
  ///
  /// let ip = "127.0.0.1".parse::<IpAddr>().unwrap();
  /// let addr = HostAddr::<String>::from((ip, 8080));
  /// # }
  /// ```
  fn from((host, port): (IpAddr, u16)) -> Self {
    Self::from_ip_addr(host).with_port(port)
  }
}

impl<S> From<(u16, IpAddr)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::IpAddr;
  ///
  /// let ip = "::1".parse::<IpAddr>().unwrap();
  /// let addr = HostAddr::<String>::from((8080, ip));
  /// # }
  /// ```
  fn from((port, host): (u16, IpAddr)) -> Self {
    Self::from_ip_addr(host).with_port(port)
  }
}

impl<S> From<(Ipv4Addr, u16)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::Ipv4Addr;
  ///
  /// let ip = "127.0.0.1".parse::<Ipv4Addr>().unwrap();
  /// let addr = HostAddr::<String>::from((ip, 8080));
  /// # }
  /// ```
  fn from((host, port): (Ipv4Addr, u16)) -> Self {
    Self::from((IpAddr::V4(host), port))
  }
}

impl<S> From<(Ipv6Addr, u16)> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::Ipv6Addr;
  ///
  /// let ip = "::1".parse::<Ipv6Addr>().unwrap();
  /// let addr = HostAddr::<String>::from((ip, 8080));
  /// # }
  /// ```
  fn from((host, port): (Ipv6Addr, u16)) -> Self {
    Self::from((port, IpAddr::V6(host)))
  }
}

impl<S> From<SocketAddr> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::SocketAddr;
  ///
  /// let addr = "127.0.0.1:8080".parse::<SocketAddr>().unwrap();
  /// let host = HostAddr::<String>::from(addr);
  /// # }
  /// ```
  fn from(addr: SocketAddr) -> Self {
    Self::from_sock_addr(addr)
  }
}

impl<S> From<SocketAddrV4> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::SocketAddrV4;
  ///
  /// let addr = "127.0.0.1:8080".parse::<SocketAddrV4>().unwrap();
  /// let host = HostAddr::<String>::from(addr);
  /// # }
  /// ```
  fn from(addr: SocketAddrV4) -> Self {
    Self::from_sock_addr(SocketAddr::V4(addr))
  }
}

impl<S> From<SocketAddrV6> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::SocketAddrV6;
  ///
  /// let addr = "[::1]:8080".parse::<SocketAddrV6>().unwrap();
  /// let host = HostAddr::<String>::from(addr);
  /// # }
  /// ```
  fn from(addr: SocketAddrV6) -> Self {
    Self::from_sock_addr(SocketAddr::V6(addr))
  }
}

impl<S> From<IpAddr> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::IpAddr;
  ///
  /// let ip = "127.0.0.1".parse::<IpAddr>().unwrap();
  /// let addr = HostAddr::<String>::from(ip);
  /// # }
  /// ```
  fn from(ip: IpAddr) -> Self {
    Self::from_ip_addr(ip)
  }
}

impl<S> From<Ipv4Addr> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::Ipv4Addr;
  ///
  /// let ip = "127.0.0.1".parse::<Ipv4Addr>().unwrap();
  /// let addr = HostAddr::<String>::from(ip);
  /// # }
  /// ```
  fn from(ip: Ipv4Addr) -> Self {
    Self::from(IpAddr::V4(ip))
  }
}

impl<S> From<Ipv6Addr> for HostAddr<S> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::Ipv6Addr;
  ///
  /// let ip = "::1".parse::<Ipv6Addr>().unwrap();
  /// let addr = HostAddr::<String>::from(ip);
  /// # }
  /// ```
  fn from(ip: Ipv6Addr) -> Self {
    Self::from(IpAddr::V6(ip))
  }
}

impl<S> From<HostAddr<S>> for HostAddr<Domain<S>> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{HostAddr, Domain};
  /// use std::net::Ipv6Addr;
  ///
  /// let ip = "::1".parse::<Ipv6Addr>().unwrap();
  /// let addr = HostAddr::<String>::from(ip);
  /// let domain_addr: HostAddr<Domain<String>> = addr.into();
  /// # }
  /// ```
  #[inline]
  fn from(value: HostAddr<S>) -> Self {
    let (host, port) = value.into_components();
    match host {
      Host::Domain(domain) => Self {
        host: Host::Domain(Domain::new_unchecked(domain)),
        port,
      },
      Host::Ip(ip) => Self {
        host: Host::Ip(ip),
        port,
      },
    }
  }
}

impl<'a, S> From<&'a HostAddr<S>> for HostAddr<&'a Domain<S>> {
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{HostAddr, Domain};
  /// use std::net::Ipv6Addr;
  ///
  /// let ip = "::1".parse::<Ipv6Addr>().unwrap();
  /// let addr = HostAddr::<String>::from(ip);
  ///
  /// let domain_addr: HostAddr<&Domain<String>> = (&addr).into();
  /// # }
  /// ```
  #[inline]
  fn from(value: &'a HostAddr<S>) -> Self {
    let (host, port) = value.as_ref().into_components();
    match host {
      Host::Domain(domain) => Self {
        host: Host::Domain(Domain::from_ref_unchecked(domain)),
        port,
      },
      Host::Ip(ip) => Self {
        host: Host::Ip(ip),
        port,
      },
    }
  }
}

impl<S> HostAddr<S> {
  /// Create a new host address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host = HostAddr::<String>::new("example.com".parse().unwrap());
  /// println!("{}", host);
  /// # }
  /// ```
  #[inline]
  pub const fn new(host: Host<S>) -> Self {
    Self { host, port: None }
  }

  /// Create a new host address from a domain name
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{HostAddr, Domain};
  ///
  /// let host = HostAddr::<String>::from_domain("example.com".parse().unwrap());
  /// println!("{}", host);
  /// # }
  /// ```
  #[inline]
  pub fn from_domain(domain: Domain<S>) -> Self {
    Self {
      host: Host::Domain(domain.0),
      port: None,
    }
  }

  /// Create a new host address from an IP address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host = HostAddr::<String>::from_ip_addr("127.0.0.1".parse().unwrap());
  /// println!("{}", host);
  /// # }
  /// ```
  #[inline]
  pub const fn from_ip_addr(ip: IpAddr) -> Self {
    Self {
      host: Host::Ip(ip),
      port: None,
    }
  }

  /// Create a new host address from a `SocketAddr`
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host = HostAddr::<String>::from_sock_addr("127.0.0.1:8080".parse().unwrap());
  /// println!("{}", host);
  /// # }
  /// ```
  #[inline]
  pub const fn from_sock_addr(addr: SocketAddr) -> Self {
    Self {
      host: Host::Ip(addr.ip()),
      port: Some(addr.port()),
    }
  }

  /// Get the host name
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr: HostAddr<String> = "example.com:8080".parse().unwrap();
  /// println!("{}", addr.host());
  /// # }
  /// ```
  #[inline]
  pub const fn host(&self) -> &Host<S> {
    &self.host
  }

  /// Get the ip address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr: HostAddr<String> = HostAddr::from_ip_addr("127.0.0.1".parse().unwrap());
  /// println!("{}", addr.ip().unwrap());
  /// # }
  /// ```
  #[inline]
  pub const fn ip(&self) -> Option<&IpAddr> {
    self.host.ip()
  }

  /// Get the port number
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr: HostAddr<String> = "example.com:8080".parse().unwrap();
  ///
  /// assert_eq!(Some(8080), addr.port());
  /// # }
  /// ```
  #[inline]
  pub const fn port(&self) -> Option<u16> {
    self.port
  }

  /// Set the port number
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let mut host: HostAddr<String> = "example.com".parse().unwrap();
  /// host
  ///   .set_port(8080)
  ///   .set_host("example.org".parse().unwrap());
  /// assert_eq!(Some(8080), host.port());
  /// # }
  /// ```
  #[inline]
  pub const fn set_port(&mut self, port: u16) -> &mut Self {
    self.port = Some(port);
    self
  }

  /// Set the port number
  ///
  /// See also [`maybe_with_port`](Self::maybe_with_port).
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let mut host: HostAddr<String> = "example.com".parse().unwrap();
  /// host
  ///   .maybe_port(Some(8080))
  ///   .set_host("example.org".parse().unwrap());
  /// assert_eq!(Some(8080), host.port());
  /// # }
  /// ```
  #[inline]
  pub const fn maybe_port(&mut self, port: Option<u16>) -> &mut Self {
    self.port = port;
    self
  }

  /// Set the port number
  ///
  /// See also [`maybe_port`](Self::maybe_port).
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host  = "example.com".parse::<HostAddr<String>>().unwrap().maybe_with_port(Some(8080));
  /// assert_eq!(Some(8080), host.port());
  /// # }
  /// ```
  #[inline]
  pub const fn maybe_with_port(mut self, port: Option<u16>) -> Self {
    self.port = port;
    self
  }

  /// Set the port number
  ///
  /// See also [`set_port`](Self::set_port).
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host = "example.com".parse::<HostAddr<String>>().unwrap().with_port(8080);
  /// assert_eq!(Some(8080), host.port());
  /// # }
  /// ```
  #[inline]
  pub const fn with_port(mut self, port: u16) -> Self {
    self.port = Some(port);
    self
  }

  /// Set a default port if no port is currently set.
  ///
  /// If a port is already set, this method does nothing.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr = "example.com".parse::<HostAddr<String>>().unwrap()
  ///   .with_default_port(443);
  /// assert_eq!(Some(443), addr.port());
  ///
  /// let addr = "example.com:8080".parse::<HostAddr<String>>().unwrap()
  ///   .with_default_port(443);
  /// assert_eq!(Some(8080), addr.port());
  /// # }
  /// ```
  #[inline]
  pub const fn with_default_port(mut self, default: u16) -> Self {
    if self.port.is_none() {
      self.port = Some(default);
    }
    self
  }

  /// Clear the port number.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let mut addr: HostAddr<String> = "example.com:8080".parse().unwrap();
  /// addr.clear_port();
  /// assert_eq!(None, addr.port());
  /// # }
  /// ```
  #[inline]
  pub const fn clear_port(&mut self) -> &mut Self {
    self.port = None;
    self
  }

  /// Returns `true` if a port is set.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr: HostAddr<String> = "example.com:8080".parse().unwrap();
  /// assert!(addr.has_port());
  ///
  /// let addr: HostAddr<String> = "example.com".parse().unwrap();
  /// assert!(!addr.has_port());
  /// # }
  /// ```
  #[inline]
  pub const fn has_port(&self) -> bool {
    self.port.is_some()
  }

  /// Set the host name
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let mut addr: HostAddr<String> = "example.com".parse().unwrap();
  /// addr
  ///   .set_host("example.org".parse().unwrap())
  ///   .set_port(8080);
  /// assert_eq!("example.org", addr.as_ref().host().unwrap_domain().as_str());
  /// # }
  /// ```
  #[inline]
  pub fn set_host(&mut self, host: Host<S>) -> &mut Self {
    self.host = host;
    self
  }

  /// Set the host name
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let addr: HostAddr<String> = HostAddr::from_sock_addr("127.0.0.1:8080".parse().unwrap())
  ///   .with_host("example.com".parse().unwrap());
  /// assert_eq!("example.com", addr.as_ref().host().unwrap_domain().as_str());
  /// # }
  /// ```
  #[inline]
  pub fn with_host(mut self, host: Host<S>) -> Self {
    self.host = host;
    self
  }

  /// Returns `true` if the host is an IP address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = HostAddr::from_sock_addr("127.0.0.1:8080".parse().unwrap());
  /// assert!(host.is_ip());
  /// # }
  #[inline]
  pub const fn is_ip(&self) -> bool {
    self.host.is_ip()
  }

  /// Returns `true` if the host is an Ipv4 address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = HostAddr::from_sock_addr("127.0.0.1:8080".parse().unwrap());
  /// assert!(host.is_ipv4());
  /// # }
  /// ```
  #[inline]
  pub const fn is_ipv4(&self) -> bool {
    self.host.is_ipv4()
  }

  /// Returns `true` if the host is an Ipv6 address
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = HostAddr::from_sock_addr("[::1]:8080".parse().unwrap());
  /// assert!(host.is_ipv6());
  /// # }
  /// ```
  #[inline]
  pub const fn is_ipv6(&self) -> bool {
    self.host.is_ipv6()
  }

  /// Returns `true` if the host is a domain name
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = "example.com".parse().unwrap();
  /// assert!(host.is_domain());
  /// # }
  /// ```
  #[inline]
  pub const fn is_domain(&self) -> bool {
    self.host.is_domain()
  }

  /// Returns `true` if the host represents localhost.
  ///
  /// This method checks if the host is:
  /// - An IPv4 loopback address (127.0.0.0/8)
  /// - An IPv6 loopback address (::1)
  /// - The domain name "localhost"
  ///
  /// ## Example
  ///
  /// ```rust
  /// use hostaddr::HostAddr;
  ///
  /// let addr = HostAddr::try_from_ascii_str("127.0.0.1").unwrap();
  /// assert!(addr.is_localhost());
  ///
  /// let addr = HostAddr::try_from_ascii_str("::1").unwrap();
  /// assert!(addr.is_localhost());
  ///
  /// let addr = HostAddr::try_from_ascii_str("localhost").unwrap();
  /// assert!(addr.is_localhost());
  ///
  /// let addr = HostAddr::try_from_ascii_str("example.com").unwrap();
  /// assert!(!addr.is_localhost());
  /// ```
  #[inline]
  pub fn is_localhost(&self) -> bool
  where
    S: AsRef<str>,
  {
    match &self.host {
      Host::Ip(IpAddr::V4(ip)) => ip.is_loopback(),
      Host::Ip(IpAddr::V6(ip)) => ip.is_loopback(),
      Host::Domain(domain) => {
        let s = domain.as_ref();
        s.eq_ignore_ascii_case("localhost") || s.eq_ignore_ascii_case("localhost.")
      }
    }
  }

  /// Converts to a `SocketAddr` if the host is an IP address and a port is set.
  ///
  /// Returns `None` if the host is a domain name or if no port is set.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  /// use std::net::{IpAddr, SocketAddr};
  ///
  /// let addr: HostAddr<String> = "127.0.0.1:8080".parse().unwrap();
  /// assert_eq!(
  ///   Some(SocketAddr::new("127.0.0.1".parse::<IpAddr>().unwrap(), 8080)),
  ///   addr.to_socket_addr()
  /// );
  ///
  /// // Domain names cannot be converted to SocketAddr
  /// let addr: HostAddr<String> = "example.com:8080".parse().unwrap();
  /// assert_eq!(None, addr.to_socket_addr());
  ///
  /// // Missing port returns None
  /// let addr: HostAddr<String> = "127.0.0.1".parse().unwrap();
  /// assert_eq!(None, addr.to_socket_addr());
  /// # }
  /// ```
  #[inline]
  pub const fn to_socket_addr(&self) -> Option<SocketAddr> {
    match (&self.host, self.port) {
      (Host::Ip(ip), Some(port)) => Some(SocketAddr::new(*ip, port)),
      _ => None,
    }
  }

  /// Converts from `&HostAddr<S>` to `HostAddr<&S>`.
  ///
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use std::sync::Arc;
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<Arc<str>> = "example.com:8080".try_into().unwrap();
  /// assert_eq!("example.com", &**host.as_ref().host().unwrap_domain());
  /// # }
  /// ```
  #[inline]
  pub const fn as_ref(&self) -> HostAddr<&S> {
    HostAddr {
      host: self.host.as_ref(),
      port: self.port,
    }
  }

  /// Converts from `HostAddr<S>` (or `&HostAddr<S>`) to `HostAddr<&S::Target>`.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use std::sync::Arc;
  /// use hostaddr::HostAddr;
  ///
  /// let host = "example.com:9090".parse::<HostAddr<Arc<str>>>().unwrap();
  /// assert_eq!("example.com", host.as_deref().host().unwrap_domain());
  /// # }
  /// ```
  #[inline]
  pub fn as_deref(&self) -> HostAddr<&S::Target>
  where
    S: core::ops::Deref,
  {
    HostAddr {
      host: self.host.as_deref(),
      port: self.port,
    }
  }

  /// Consumes the `HostAddr` and returns the host name and port number.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = "example.com:8080".parse().unwrap();
  /// let (host, port) = host.into_components();
  /// assert_eq!("example.com", host.unwrap_domain().as_str());
  /// assert_eq!(Some(8080), port);
  /// # }
  /// ```
  #[inline]
  pub fn into_components(self) -> (Host<S>, Option<u16>) {
    let Self { host, port } = self;

    (host, port)
  }

  /// Unwraps the domain, panics if the host is an IP address.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = "example.com".parse().unwrap();
  /// let (domain, port) = host.unwrap_domain();
  /// assert_eq!("example.com", domain.as_str());
  /// assert_eq!(None, port);
  /// # }
  /// ```
  #[inline]
  pub fn unwrap_domain(self) -> (S, Option<u16>) {
    (self.host.unwrap_domain(), self.port)
  }

  /// Unwraps the IP address, panics if the host is a domain name.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = HostAddr::from_sock_addr("[::1]:8080".parse().unwrap());
  /// let (ip, port) = host.unwrap_ip();
  /// assert_eq!(ip, "::1".parse::<std::net::IpAddr>().unwrap());
  /// assert_eq!(Some(8080), port);
  /// # }
  /// ```
  #[inline]
  pub fn unwrap_ip(self) -> (IpAddr, Option<u16>) {
    (self.host.unwrap_ip(), self.port)
  }
}

impl<S> HostAddr<&S> {
  /// Maps an `HostAddr<&S>` to an `HostAddr<S>` by copying the contents of the
  /// host addr.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::{HostAddr, Buffer};
  ///
  /// let host: HostAddr<Buffer> = HostAddr::try_from("example.com").unwrap();
  /// assert_eq!("example.com", host.as_ref().copied().unwrap_domain().0.as_str());
  /// # }
  /// ```
  #[inline]
  pub const fn copied(self) -> HostAddr<S>
  where
    S: Copy,
  {
    HostAddr {
      host: self.host.copied(),
      port: self.port,
    }
  }

  /// Maps an `HostAddr<&S>` to an `HostAddr<S>` by cloning the contents of the
  /// host addr.
  ///
  /// ## Example
  ///
  /// ```rust
  /// # #[cfg(any(feature = "std", feature = "alloc"))]
  /// # {
  /// use hostaddr::HostAddr;
  ///
  /// let host: HostAddr<String> = "example.com".parse().unwrap();
  /// assert_eq!("example.com", host.as_ref().cloned().unwrap_domain().0.as_str());
  /// # }
  /// ```
  #[inline]
  pub fn cloned(self) -> HostAddr<S>
  where
    S: Clone,
  {
    HostAddr {
      host: self.host.cloned(),
      port: self.port,
    }
  }
}

macro_rules! try_from_str {
  ($convert:ident($s: ident)) => {{
    match try_parse_v6::<S>($s)? {
      Some(addr) => Ok(addr),
      None => {
        let mut parts = $s.splitn(2, ':');
        let host = parts.next().ok_or(ParseHostAddrError::host())?.$convert()?;
        let port = match parts.next() {
          Some(port) => Some(port.parse().map_err(ParseHostAddrError::Port)?),
          None => None,
        };

        Ok(Self { host, port })
      }
    }
  }};
}

impl<'a> HostAddr<&'a str> {
  /// Parses a [`HostAddr`] name from `&str`.
  ///
  /// Unlike `HostAddr::try_from` or `HostAddr::from_str`, this method does not perform any percent decoding
  /// or punycode decoding. If the input is not ASCII, it will return an error.
  ///
  /// ## Example
  ///
  /// ```rust
  /// use hostaddr::HostAddr;
  ///
  /// let host = HostAddr::try_from_ascii_str("example.com").unwrap();
  /// assert_eq!(host.unwrap_domain().0, "example.com");
  ///
  /// // This will return an error because the domain is not ASCII.
  /// assert!(HostAddr::try_from_ascii_str("测试.中国").is_err());
  ///
  /// // Thie will not return an error, even though the human-readable domain is not ASCII.
  /// let host = HostAddr::try_from_ascii_str("xn--0zwm56d.xn--fiqs8s").unwrap();
  /// assert_eq!(host.unwrap_domain().0, "xn--0zwm56d.xn--fiqs8s");
  /// ```
  #[inline]
  pub fn try_from_ascii_str(input: &'a str) -> Result<Self, ParseAsciiHostAddrError> {
    match try_parse_v6(input).map_err(|_| ParseAsciiHostAddrError::host())? {
      Some(addr) => Ok(addr),
      None => {
        if let Ok(ip) = input.parse() {
          return Ok(Self::from_ip_addr(ip));
        }

        let mut parts = input.splitn(2, ':');
        let host = Host::try_from_ascii_str(parts.next().ok_or(ParseAsciiHostAddrError::host())?)?;
        let port = match parts.next() {
          Some(port) => Some(port.parse().map_err(ParseAsciiHostAddrError::Port)?),
          None => None,
        };

        Ok(Self { host, port })
      }
    }
  }

  /// Converts the domain to a `HostAddr<&'a [u8]>`.
  ///
  /// ## Example
  ///
  /// ```rust
  /// use hostaddr::HostAddr;
  ///
  /// let addr = HostAddr::try_from_ascii_str("example.com").unwrap();
  /// assert_eq!(addr.as_bytes().unwrap_domain().0, b"example.com");
  /// ```
  #[inline]
  pub const fn as_bytes(&self) -> HostAddr<&'a [u8]> {
    HostAddr {
      host: self.host.as_bytes(),
      port: self.port,
    }
  }
}

impl<'a> HostAddr<&'a [u8]> {
  /// Parses a [`HostAddr`] from `&[u8]`.
  ///
  /// Unlike `HostAddr::try_from` or `HostAddr::from_str`, this method does not perform any percent decoding
  /// or punycode decoding. If the input is not ASCII, it will return an error.
  ///
  /// ## Example
  ///
  /// ```rust
  /// use hostaddr::HostAddr;
  ///
  /// let host = HostAddr::try_from_ascii_bytes(b"example.com").unwrap();
  /// assert_eq!(host.unwrap_domain().0, b"example.com");
  ///
  /// // This will return an error because the domain is not ASCII.
  /// assert!(HostAddr::try_from_ascii_bytes("测试.中国".as_bytes()).is_err());
  ///
  /// // Thie will not return an error, even though the human-readable domain is not ASCII.
  /// let host = HostAddr::try_from_ascii_bytes(b"xn--0zwm56d.xn--fiqs8s").unwrap();
  /// assert_eq!(host.unwrap_domain().0, b"xn--0zwm56d.xn--fiqs8s");
  /// ```
  #[inline]
  pub fn try_from_ascii_bytes(input: &'a [u8]) -> Result<Self, ParseAsciiHostAddrError> {
    let input_str = simdutf8::basic::from_utf8(input).map_err(|_| ParseAsciiHostError(()))?;
    match try_parse_v6(input_str).map_err(|_| ParseAsciiHostAddrError::host())? {
      Some(addr) => Ok(addr),
      None => {
        // if it's an pure ip address?
        if let Ok(ip) = IpAddr::from_str(input_str) {
          return Ok(Self::from_ip_addr(ip));
        }

        let mut parts = input_str.splitn(2, ':');
        let host = Host::try_from_ascii_bytes(
          parts
            .next()
            .map(|s| s.as_bytes())
            .ok_or(ParseAsciiHostAddrError::host())?,
        )?;
        let port = match parts.next() {
          Some(port) => Some(port.parse().map_err(ParseAsciiHostAddrError::Port)?),
          None => None,
        };

        Ok(Self { host, port })
      }
    }
  }

  /// Converts the domain to a `HostAddr<&'a str>`.
  ///
  /// ## Example
  ///
  /// ```rust
  /// use hostaddr::HostAddr;
  ///
  /// let addr = HostAddr::try_from_ascii_bytes(b"example.com").unwrap();
  /// assert_eq!(addr.as_str().unwrap_domain().0, "example.com");
  /// ```
  #[inline]
  pub const fn as_str(&self) -> HostAddr<&'a str> {
    HostAddr {
      host: self.host.as_str(),
      port: self.port,
    }
  }
}

impl<S> FromStr for HostAddr<S>
where
  Domain<S>: FromStr,
{
  type Err = ParseHostAddrError;

  fn from_str(s: &str) -> Result<Self, Self::Err> {
    try_from_str!(parse(s))
  }
}

impl<'a, S> TryFrom<&'a str> for HostAddr<S>
where
  Domain<S>: TryFrom<&'a str>,
{
  type Error = ParseHostAddrError;

  fn try_from(s: &'a str) -> Result<Self, Self::Error> {
    try_from_str!(try_into(s))
  }
}

impl<'a, S> TryFrom<(&'a str, u16)> for HostAddr<S>
where
  Domain<S>: TryFrom<&'a str>,
{
  type Error = ParseHostAddrError;

  fn try_from((s, port): (&'a str, u16)) -> Result<Self, Self::Error> {
    let host = Host::try_from(s)?;
    Ok(Self {
      host,
      port: Some(port),
    })
  }
}

fn try_parse_v6<S>(s: &str) -> Result<Option<HostAddr<S>>, ParseHostAddrError> {
  if let Some(without_prefix) = s.strip_prefix('[') {
    // Ipv6 and no port
    if let Some(ip) = without_prefix.strip_suffix(']') {
      return ip
        .parse()
        .map_err(|_| ParseHostAddrError::host())
        .map(|addr| Some(HostAddr::from_ip_addr(addr)));
    }

    // Ipv6 with port
    let mut parts = s.rsplitn(2, ':');

    let port = parts.next();
    let host = parts.next();

    match (host, port) {
      (Some(host), Some(_)) if host.ends_with("]") => {
        return s
          .parse()
          .map_err(|_| ParseHostAddrError::host())
          .map(|addr| Some(HostAddr::from_sock_addr(addr)));
      }
      _ => return Err(ParseHostAddrError::host()),
    }
  }

  Ok(None)
}

#[cfg(test)]
mod tests {
  use super::*;

  #[test]
  fn test_hostaddr_parsing() {
    #[cfg(any(feature = "std", feature = "alloc"))]
    {
      use std::string::String;

      let host: HostAddr<String> = "example.com".parse().unwrap();
      assert_eq!("example.com", host.as_ref().host().unwrap_domain());

      let host: HostAddr<String> = "example.com:8080".parse().unwrap();
      assert_eq!("example.com", host.as_ref().host().unwrap_domain());
      assert_eq!(Some(8080), host.port());

      let host: HostAddr<String> = "127.0.0.1:8080".parse().unwrap();
      assert_eq!(
        IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)),
        host.as_ref().host().unwrap_ip()
      );
      assert_eq!(Some(8080), host.port());

      let host: HostAddr<String> = "[::1]:8080".parse().unwrap();
      assert_eq!(
        IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
        host.as_ref().host().unwrap_ip()
      );
      assert_eq!(Some(8080), host.port());
    }

    let host: HostAddr<&str> = HostAddr::try_from_ascii_str("[::1]").unwrap();
    assert_eq!(
      IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );

    let host: HostAddr<&[u8]> = HostAddr::try_from_ascii_bytes(b"[::1]").unwrap();
    assert_eq!(
      IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );

    let host: HostAddr<&[u8]> = HostAddr::try_from_ascii_bytes(b"::1").unwrap();
    assert_eq!(
      IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );

    let host: HostAddr<&str> = HostAddr::try_from_ascii_str("::1").unwrap();
    assert_eq!(
      IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );
  }

  #[cfg(any(feature = "std", feature = "alloc"))]
  #[test]
  fn test_hostaddr_try_into() {
    use std::string::String;

    let host: HostAddr<String> = "example.com".try_into().unwrap();
    assert_eq!("example.com", host.as_ref().host().unwrap_domain());

    let host: HostAddr<String> = "example.com:8080".try_into().unwrap();
    assert_eq!("example.com", host.as_ref().host().unwrap_domain());
    assert_eq!(Some(8080), host.port());

    let host: HostAddr<String> = "127.0.0.1:8080".try_into().unwrap();
    assert_eq!(
      IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );
    assert_eq!(Some(8080), host.port());

    let host: HostAddr<String> = "[::1]:8080".try_into().unwrap();
    assert_eq!(
      IpAddr::V6(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1)),
      host.as_ref().host().unwrap_ip()
    );
    assert_eq!(Some(8080), host.port());
  }

  #[test]
  fn negative_try_parse_v6() {
    let _ = try_parse_v6::<&str>("[a]").unwrap_err();
    let _ = try_parse_v6::<&str>("[a]:8080").unwrap_err();
    let _ = try_parse_v6::<&str>("[a:8080").unwrap_err();
  }

  #[test]
  fn negative_try_from_ascii_bytes() {
    let err = HostAddr::try_from_ascii_bytes(b"example.com:aaa").unwrap_err();
    assert!(matches!(err, ParseAsciiHostAddrError::Port(_)));
  }

  #[test]
  fn negative_try_from_ascii_str() {
    let err = HostAddr::try_from_ascii_str("example.com:aaa").unwrap_err();
    assert!(matches!(err, ParseAsciiHostAddrError::Port(_)));
  }

  /// Regression test: IPv6 with port must display as `[::1]:port`, not `::1:port`.
  /// The old output was ambiguous and could not be parsed back.
  #[test]
  #[cfg(feature = "std")]
  fn ipv6_display_roundtrip() {
    let addr = HostAddr::<&str>::from_sock_addr("[::1]:8080".parse().unwrap());
    assert_eq!(addr.to_string(), "[::1]:8080");

    // Without port, no brackets
    let addr = HostAddr::<&str>::from_ip_addr("::1".parse().unwrap());
    assert_eq!(addr.to_string(), "::1");

    // IPv4 unchanged
    let addr = HostAddr::<&str>::from_sock_addr("127.0.0.1:3000".parse().unwrap());
    assert_eq!(addr.to_string(), "127.0.0.1:3000");

    // Roundtrip: display then parse
    #[cfg(any(feature = "std", feature = "alloc"))]
    {
      use std::string::String;
      let addr = HostAddr::<String>::from_sock_addr("[::1]:443".parse().unwrap());
      let displayed = addr.to_string();
      let reparsed: HostAddr<String> = displayed.parse().unwrap();
      assert_eq!(reparsed.port(), Some(443));
      assert!(reparsed.is_ipv6());
    }
  }
}