use std::fmt;

#[derive(Clone, Eq, PartialEq)]
pub struct Name {
    name: String
}

impl Name {
    pub fn new(name: String) -> Self {
        Name {
            name,
        }
    }

    pub fn ends_with(&self, other: &Name) -> bool {
        self.name == other.name || self.name.ends_with(&(String::from(".") + &other.name))
    }
}

impl fmt::Display for Name {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.name)
    }
}

impl Ord for Name {
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        let mut labels = self.name.split('.').rev();
        let mut other_labels = other.name.split('.').rev();

        loop {
            match (labels.next(), other_labels.next()) {
                (Some(label), Some(other_label)) => match label.cmp(other_label) {
                    std::cmp::Ordering::Equal => (),
                    res => return res,
                },
                (None, Some(_)) => return std::cmp::Ordering::Less,
                (Some(_), None) => return std::cmp::Ordering::Greater,
                (None, None) => return std::cmp::Ordering::Equal,
            }
        }
    }
}

impl PartialOrd for Name {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}

#[derive(Eq, PartialEq)]
pub enum Rtype {
    A,
    Aaaa,
    Cname,
    Mx,
    Ns,
    Ptr,
    Soa,
    Srv,
    Txt
}

impl Rtype {
    pub fn value(&self) -> u16 {
        match self {
            Rtype::A => 1,
            Rtype::Aaaa => 28,
            Rtype::Cname => 5,
            Rtype::Mx => 15,
            Rtype::Ns => 2,
            Rtype::Ptr => 12,
            Rtype::Soa => 6,
            Rtype::Srv => 33,
            Rtype::Txt => 16,
        }
    }
}

impl Ord for Rtype {
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        self.value().cmp(&other.value())
    }
}

impl PartialOrd for Rtype {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}