use std::net::{Ipv4Addr, Ipv6Addr};

pub struct Ip4Wrapper {
    raw: Ipv4Addr,
}

pub struct Ip6Wrapper {
    raw: Ipv6Addr,
}

pub trait IpWrapper<IpType> {
    fn is_private(&self) -> bool;
    fn get_raw(&self) -> IpType;
}

impl Ip4Wrapper {
    pub fn new(ip: u32) -> Self {
        Self {
            raw: Ipv4Addr::from_bits(ip),
        }
    }
}

impl Ip6Wrapper {
    pub fn new(ip: Ipv6Addr) -> Self {
        Self {
            raw: ip,
        }
    }
}

impl IpWrapper<Ipv4Addr> for Ip4Wrapper {
    fn is_private(&self) -> bool {
        self.raw.is_private()
    }

    fn get_raw(&self) -> Ipv4Addr {
        self.raw
    }
}

impl IpWrapper<Ipv6Addr> for Ip6Wrapper {
    fn is_private(&self) -> bool {
        self.raw.is_unique_local()
    }

    fn get_raw(&self) -> Ipv6Addr {
        self.raw
    }
}