use crate::errors::ValidationError;
use crate::traits::ValueObject;
use std::net::Ipv4Addr;
pub type IpV4AddressInput = String;
pub type IpV4AddressOutput = String;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(transparent))]
pub struct IpV4Address(String);
impl ValueObject for IpV4Address {
type Input = IpV4AddressInput;
type Output = IpV4AddressOutput;
type Error = ValidationError;
fn new(value: Self::Input) -> Result<Self, Self::Error> {
let trimmed = value.trim();
if trimmed.is_empty() {
return Err(ValidationError::empty("IpV4Address"));
}
for octet in trimmed.split('.') {
if octet.len() > 1 && octet.starts_with('0') {
return Err(ValidationError::invalid("IpV4Address", trimmed));
}
}
trimmed
.parse::<Ipv4Addr>()
.map(|ip| Self(ip.to_string()))
.map_err(|_| ValidationError::invalid("IpV4Address", trimmed))
}
fn value(&self) -> &Self::Output {
&self.0
}
fn into_inner(self) -> Self::Input {
self.0
}
}
impl TryFrom<&str> for IpV4Address {
type Error = ValidationError;
fn try_from(value: &str) -> Result<Self, Self::Error> {
Self::new(value.to_owned())
}
}
impl std::fmt::Display for IpV4Address {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "{}", self.0)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn accepts_valid_ipv4() {
let ip = IpV4Address::new("192.168.1.1".into()).unwrap();
assert_eq!(ip.value(), "192.168.1.1");
}
#[test]
fn accepts_loopback() {
assert!(IpV4Address::new("127.0.0.1".into()).is_ok());
}
#[test]
fn accepts_all_zeros() {
assert!(IpV4Address::new("0.0.0.0".into()).is_ok());
}
#[test]
fn accepts_broadcast() {
assert!(IpV4Address::new("255.255.255.255".into()).is_ok());
}
#[test]
fn rejects_empty() {
assert!(IpV4Address::new(String::new()).is_err());
}
#[test]
fn rejects_out_of_range_octet() {
assert!(IpV4Address::new("256.0.0.1".into()).is_err());
}
#[test]
fn rejects_too_few_octets() {
assert!(IpV4Address::new("192.168.1".into()).is_err());
}
#[test]
fn rejects_leading_zeros() {
assert!(IpV4Address::new("192.168.001.001".into()).is_err());
}
#[test]
fn rejects_ipv6() {
assert!(IpV4Address::new("::1".into()).is_err());
}
#[test]
fn try_from_str() {
let ip: IpV4Address = "10.0.0.1".try_into().unwrap();
assert_eq!(ip.value(), "10.0.0.1");
}
}