fastdate 0.3.36

use crate::error::Error;
use crate::{DateTime, get_digit, get_digit_unchecked};
use serde::{Deserialize, Deserializer, Serialize, Serializer};
use std::fmt::{Display, Formatter};
use std::str::FromStr;
use std::time::Duration;

#[derive(Clone, Debug, Eq, PartialEq, Hash)]
pub struct Time {
    /// 0...999999999
    pub nano: u32,
    /// 0...59
    pub sec: u8,
    /// 0...59
    pub minute: u8,
    /// 0...23
    pub hour: u8,
}

impl Time {
    /// Parse a time from bytes with a starting index, no check is performed for extract characters at
    /// the end of the string
    pub(crate) fn parse_bytes_partial(bytes: &[u8], offset: usize) -> Result<(Self, usize), Error> {
        if bytes.len() < offset {
            return Ok((
                Self {
                    nano: 0,
                    sec: 0,
                    minute: 0,
                    hour: 0,
                },
                0,
            ));
        }
        if bytes.len() - offset < 5 {
            return Err(Error::E("TooShort".to_string()));
        }
        let hour: u8;
        let minute: u8;
        unsafe {
            let h1 = get_digit_unchecked!(bytes, offset, "InvalidCharHour");
            let h2 = get_digit_unchecked!(bytes, offset + 1, "InvalidCharHour");
            hour = h1 * 10 + h2;
            let m1 = get_digit_unchecked!(bytes, offset + 3, "InvalidCharMinute");
            let m2 = get_digit_unchecked!(bytes, offset + 4, "InvalidCharMinute");
            minute = m1 * 10 + m2;
        }

        if hour > 23 {
            return Err(Error::E("OutOfRangeHour".to_string()));
        }

        if minute > 59 {
            return Err(Error::E("OutOfRangeMinute".to_string()));
        }
        let mut length: usize = 5;
        let (second, nano) = {
            let s1 = get_digit!(bytes, offset + 6, "InvalidCharSecond");
            let s2 = get_digit!(bytes, offset + 7, "InvalidCharSecond");
            let second = s1 * 10 + s2;
            if second > 59 {
                return Err(Error::E("OutOfRangeSecond".to_string()));
            }
            length = 8;
            let mut nano = 0;
            let frac_sep = bytes.get(offset + 8).copied();
            let mut number_buf = *b"         ";
            if frac_sep == Some(b'.') || frac_sep == Some(b',') {
                length = 9;
                let mut i: usize = 0;
                loop {
                    match bytes.get(offset + length + i) {
                        Some(c) if c.is_ascii_digit() => {
                            if i >= 9 {
                                return Err(Error::E("SecondFractionTooLong".to_string()));
                            }
                            number_buf[i] = *c;
                        }
                        _ => {
                            break;
                        }
                    }
                    i += 1;
                }
                if i == 0 {
                    return Err(Error::E("SecondFractionMissing".to_string()));
                }
                length += i;
            }
            let mut i = 0;
            for &item in number_buf.iter() {
                if item != b' ' {
                    //is number
                    let v = (item - b'0') as u32;
                    nano += v * 10_u32.pow(8 - i);
                    i += 1;
                }
            }
            (second, nano)
        };
        let t = Self {
            nano,
            sec: second,
            minute,
            hour,
        };
        Ok((t, length))
    }

    /// 0...999999999
    pub fn set_nano(mut self, arg: u32) -> Self {
        self.nano = arg;
        self
    }

    /// 0...999999
    pub fn set_micro(mut self, arg: u32) -> Self {
        self.nano = arg * 1000;
        self
    }
    /// 0...59
    pub fn set_sec(mut self, arg: u8) -> Self {
        self.sec = arg;
        self
    }
    /// 0...59
    pub fn set_minute(mut self, arg: u8) -> Self {
        self.minute = arg;
        self
    }
    /// 0...23
    pub fn set_hour(mut self, arg: u8) -> Self {
        self.hour = arg;
        self
    }
    /// get nano
    pub fn get_nano(&self) -> u32 {
        self.nano
    }
    /// get micro secs
    pub fn get_micro(&self) -> u32 {
        self.nano / 1000
    }

    /// get sec
    pub fn get_sec(&self) -> u8 {
        self.sec
    }

    /// minute
    pub fn get_minute(&self) -> u8 {
        self.minute
    }

    /// get hour
    pub fn get_hour(&self) -> u8 {
        self.hour
    }

    /// display time and return len
    pub fn display_time(&self, start: usize, buf: &mut [u8]) -> usize {
        buf[start] = b'0' + (self.hour / 10);
        buf[start + 1] = b'0' + (self.hour % 10);
        buf[start + 3] = b'0' + (self.minute / 10);
        buf[start + 4] = b'0' + (self.minute % 10);
        buf[start + 6] = b'0' + (self.sec / 10);
        buf[start + 7] = b'0' + (self.sec % 10);
        let mut real_len = start + 1 + 8 + 8 + 1;
        buf[start + 8] = b'.';
        buf[start + 9] = b'0' + (self.nano / 100000000 % 10) as u8;
        buf[start + 10] = b'0' + (self.nano / 10000000 % 10) as u8;
        buf[start + 11] = b'0' + (self.nano / 1000000 % 10) as u8;
        buf[start + 12] = b'0' + (self.nano / 100000 % 10) as u8;
        buf[start + 13] = b'0' + (self.nano / 10000 % 10) as u8;
        buf[start + 14] = b'0' + (self.nano / 1000 % 10) as u8;
        buf[start + 15] = b'0' + (self.nano / 100 % 10) as u8;
        buf[start + 16] = b'0' + (self.nano / 10 % 10) as u8;
        buf[start + 17] = b'0' + (self.nano % 10) as u8;
        if self.nano == 0 {
            real_len -= 10;
        } else {
            let current = real_len;
            let mut last_zero = false;
            for i in 0..8 {
                let i = current - 1 - i;
                if buf[i] == b'0' {
                    last_zero = true;
                }
                if last_zero && buf[i] == b'0' {
                    real_len -= 1;
                } else {
                    break;
                }
            }
        }
        real_len
    }
}

impl From<Duration> for Time {
    fn from(d: Duration) -> Self {
        let hour = (d.as_secs() / 3600) as u8;
        let min = (d.as_secs() / 60 - (hour as u64 * 60)) as u8;
        let sec = (d.as_secs() - hour as u64 * 3600u64 - min as u64 * 60u64) as u8;
        let micros = d
            - Duration::from_secs(hour as u64 * 3600u64)
            - Duration::from_secs(min as u64 * 60u64)
            - Duration::from_secs(sec as u64);
        Self {
            nano: micros.as_nanos() as u32,
            sec,
            minute: min,
            hour,
        }
    }
}

impl From<Time> for Duration {
    fn from(d: Time) -> Self {
        Duration::from_secs(d.hour as u64 * 60 * 60)
            + Duration::from_secs(d.minute as u64 * 60)
            + Duration::from_secs(d.sec as u64)
            + Duration::from_nanos(d.nano as u64)
    }
}

impl FromStr for Time {
    type Err = Error;

    /// from RFC3339Micro = "15:04:05.999999999"
    fn from_str(s: &str) -> Result<Time, Error> {
        //"00:00:00.000000";
        Ok(Time::parse_bytes_partial(s.as_bytes(), 0)?.0)
    }
}

impl Display for Time {
    /// fmt RFC3339Micro = "2006-01-02T15:04:05.999999999"
    fn fmt(&self, f: &mut Formatter) -> std::fmt::Result {
        let mut buf: [u8; 18] = *b"00:00:00.000000000";
        let len = self.display_time(0, &mut buf);
        f.write_str(std::str::from_utf8(&buf[..len]).unwrap())
    }
}

impl Serialize for Time {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        serializer.serialize_str(&self.to_string())
    }
}

impl<'de> Deserialize<'de> for Time {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        use serde::de::Error;
        Time::from_str(&String::deserialize(deserializer)?)
            .map_err(|e| Error::custom(e.to_string()))
    }
}

impl From<DateTime> for Time {
    fn from(arg: DateTime) -> Self {
        Time {
            nano: arg.nano(),
            sec: arg.sec(),
            minute: arg.minute(),
            hour: arg.hour(),
        }
    }
}

#[cfg(test)]
mod test {
    use crate::time::Time;

    #[test]
    fn test_time_empty3() {
        let d = Time::parse_bytes_partial("111".as_bytes(), 100).unwrap();
        assert_eq!(
            d.0,
            Time {
                nano: 0,
                sec: 0,
                minute: 0,
                hour: 0,
            }
        );
    }

    #[test]
    fn test_display_time() {
        let d = Time {
            nano: 123456,
            sec: 0,
            minute: 0,
            hour: 8,
        };
        let mut buf: [u8; 18] = *b"00:00:00.000000000";
        d.display_time(0, &mut buf);
        assert_eq!(
            "08:00:00.000123456",
            String::from_utf8(buf.to_vec()).unwrap()
        );
    }
}