picky 7.0.0-rc.23

use picky_asn1::date::{Date, GeneralizedTime, UTCTime, UTCTimeRepr};
use picky_asn1_x509::validity::Time;
use std::fmt;

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct UtcDate(GeneralizedTime);

impl UtcDate {
    #[inline]
    pub fn new(year: u16, month: u8, day: u8, hour: u8, minute: u8, second: u8) -> Option<Self> {
        Some(Self(GeneralizedTime::new(year, month, day, hour, minute, second)?))
    }

    #[inline]
    pub fn ymd(year: u16, month: u8, day: u8) -> Option<Self> {
        Some(Self(GeneralizedTime::new(year, month, day, 0, 0, 0)?))
    }

    #[cfg(any(feature = "time_conversion", feature = "chrono_conversion"))]
    #[inline]
    pub fn now() -> Self {
        #[cfg(feature = "time_conversion")]
        {
            Self(time::OffsetDateTime::now_utc().into())
        }
        #[cfg(all(feature = "chrono_conversion", not(feature = "time_conversion")))]
        {
            Self(chrono::offset::Utc::now().into())
        }
    }

    #[inline]
    pub fn year(&self) -> u16 {
        self.0.year()
    }

    #[inline]
    pub fn month(&self) -> u8 {
        self.0.month()
    }

    #[inline]
    pub fn day(&self) -> u8 {
        self.0.day()
    }

    #[inline]
    pub fn hour(&self) -> u8 {
        self.0.hour()
    }

    #[inline]
    pub fn minute(&self) -> u8 {
        self.0.minute()
    }

    #[inline]
    pub fn second(&self) -> u8 {
        self.0.second()
    }
}

impl From<UtcDate> for UTCTime {
    fn from(date: UtcDate) -> Self {
        unsafe {
            UTCTime::new_unchecked(
                date.0.year(),
                date.0.month(),
                date.0.day(),
                date.0.hour(),
                date.0.minute(),
                date.0.second(),
            )
        }
    }
}

impl From<UTCTime> for UtcDate {
    fn from(date: Date<UTCTimeRepr>) -> Self {
        Self(unsafe {
            GeneralizedTime::new_unchecked(
                date.year(),
                date.month(),
                date.day(),
                date.hour(),
                date.minute(),
                date.second(),
            )
        })
    }
}

impl From<UtcDate> for GeneralizedTime {
    fn from(date: UtcDate) -> GeneralizedTime {
        date.0
    }
}

impl From<GeneralizedTime> for UtcDate {
    fn from(date: GeneralizedTime) -> Self {
        Self(date)
    }
}

impl From<UtcDate> for Time {
    fn from(date: UtcDate) -> Self {
        // Time is used to encode validity period.
        // As per RFC 5280,
        // > CAs conforming to this profile MUST always encode certificate
        // > validity dates through the year 2049 as UTCTime; certificate validity
        // > dates in 2050 or later MUST be encoded as GeneralizedTime.
        // > Conforming applications MUST be able to process validity dates that
        // > are encoded in either UTCTime or GeneralizedTime.
        if date.year() >= 2050 {
            Self::Generalized(Into::<GeneralizedTime>::into(date).into())
        } else {
            Self::Utc(Into::<UTCTime>::into(date).into())
        }
    }
}

impl From<Time> for UtcDate {
    fn from(time: Time) -> Self {
        match time {
            Time::Utc(utc) => utc.0.into(),
            Time::Generalized(gen_time) => gen_time.0.into(),
        }
    }
}

impl fmt::Display for UtcDate {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(
            f,
            "{:04}-{:02}-{:02} {:02}:{:02}:{:02}",
            self.year(),
            self.month(),
            self.day(),
            self.hour(),
            self.minute(),
            self.second()
        )
    }
}

#[cfg(feature = "time_conversion")]
mod time_convert {
    use super::*;
    use time::OffsetDateTime;

    impl From<OffsetDateTime> for UtcDate {
        fn from(dt: OffsetDateTime) -> Self {
            Self(dt.into())
        }
    }

    impl TryFrom<UtcDate> for OffsetDateTime {
        type Error = time::error::ComponentRange;

        fn try_from(date: UtcDate) -> Result<Self, Self::Error> {
            date.0.try_into()
        }
    }
}

#[cfg(feature = "chrono_conversion")]
mod chrono_convert {
    use super::*;
    use chrono::{DateTime, Utc};

    impl From<DateTime<Utc>> for UtcDate {
        fn from(dt: DateTime<Utc>) -> Self {
            Self(dt.into())
        }
    }

    impl From<UtcDate> for DateTime<Utc> {
        fn from(date: UtcDate) -> Self {
            date.0.into()
        }
    }
}