implement date parsing as FromStr trait
This commit is contained in:
		@ -1,6 +1,7 @@
 | 
			
		||||
use serde::de::Error;
 | 
			
		||||
use serde::Serialize;
 | 
			
		||||
use serde::{Deserialize, Deserializer, Serializer};
 | 
			
		||||
use std::str::FromStr;
 | 
			
		||||
 | 
			
		||||
#[derive(Clone, Debug, Eq, PartialEq, Hash, Ord, PartialOrd)]
 | 
			
		||||
pub struct Date {
 | 
			
		||||
@ -14,23 +15,49 @@ impl Date {
 | 
			
		||||
    pub fn new(year: u16, month: u8, day: u8) -> Self {
 | 
			
		||||
        Date { year, month, day }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
    pub fn from_str(input: &str) -> Option<Self> {
 | 
			
		||||
#[derive(Clone, PartialEq, Eq, Debug)]
 | 
			
		||||
pub enum DateParseError {
 | 
			
		||||
    InvalidNumber,
 | 
			
		||||
    MonthOutOfRange,
 | 
			
		||||
    DayOutOfRange,
 | 
			
		||||
    WrongFormat,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl FromStr for Date {
 | 
			
		||||
    type Err = DateParseError;
 | 
			
		||||
 | 
			
		||||
    fn from_str(input: &str) -> Result<Date, DateParseError> {
 | 
			
		||||
        let mut parts = input.splitn(3, "-");
 | 
			
		||||
 | 
			
		||||
        let year = parts.next()?.parse::<u16>().ok()?;
 | 
			
		||||
        let month = parts.next()?.parse::<u8>().ok()?;
 | 
			
		||||
        let day = parts.next()?.parse::<u8>().ok()?;
 | 
			
		||||
        let year = parts
 | 
			
		||||
            .next()
 | 
			
		||||
            .ok_or(DateParseError::WrongFormat)?
 | 
			
		||||
            .parse::<u16>()
 | 
			
		||||
            .map_err(|_| DateParseError::InvalidNumber)?;
 | 
			
		||||
 | 
			
		||||
        let month = parts
 | 
			
		||||
            .next()
 | 
			
		||||
            .ok_or(DateParseError::WrongFormat)?
 | 
			
		||||
            .parse::<u8>()
 | 
			
		||||
            .map_err(|_| DateParseError::InvalidNumber)?;
 | 
			
		||||
 | 
			
		||||
        let day = parts
 | 
			
		||||
            .next()
 | 
			
		||||
            .ok_or(DateParseError::WrongFormat)?
 | 
			
		||||
            .parse::<u8>()
 | 
			
		||||
            .map_err(|_| DateParseError::InvalidNumber)?;
 | 
			
		||||
 | 
			
		||||
        if month > 12 {
 | 
			
		||||
            return None;
 | 
			
		||||
            return Err(DateParseError::DayOutOfRange);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if day > 31 {
 | 
			
		||||
            return None;
 | 
			
		||||
            return Err(DateParseError::MonthOutOfRange);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        Some(Date { year, month, day })
 | 
			
		||||
        Ok(Date { year, month, day })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -54,8 +81,8 @@ impl<'de> Deserialize<'de> for Date {
 | 
			
		||||
        let data = String::deserialize(deserializer)?;
 | 
			
		||||
 | 
			
		||||
        match Date::from_str(&data) {
 | 
			
		||||
            Some(date) => Ok(date),
 | 
			
		||||
            None => Err(D::Error::custom(
 | 
			
		||||
            Ok(date) => Ok(date),
 | 
			
		||||
            Err(_) => Err(D::Error::custom(
 | 
			
		||||
                "Expected a String in this format: YYYY-MM-DD.",
 | 
			
		||||
            )),
 | 
			
		||||
        }
 | 
			
		||||
@ -76,14 +103,20 @@ mod test {
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn from_str() {
 | 
			
		||||
        assert_eq!(Some(Date::new(2019, 10, 10)), Date::from_str("2019-10-10"));
 | 
			
		||||
        assert_eq!(Some(Date::new(2019, 1, 1)), Date::from_str("2019-1-1"));
 | 
			
		||||
        assert_eq!(None, Date::from_str("2019-1-1-444"));
 | 
			
		||||
        assert_eq!(Ok(Date::new(2019, 10, 10)), Date::from_str("2019-10-10"));
 | 
			
		||||
 | 
			
		||||
        assert_eq!(Ok(Date::new(2019, 1, 1)), Date::from_str("2019-1-1"));
 | 
			
		||||
 | 
			
		||||
        assert!(Date::from_str("2019-1-1-444").is_err());
 | 
			
		||||
 | 
			
		||||
        assert_eq!(
 | 
			
		||||
            Some(Date::new(2019, 1, 1)),
 | 
			
		||||
            Ok(Date::new(2019, 1, 1)),
 | 
			
		||||
            Date::from_str("2019-0000000000001-00000001")
 | 
			
		||||
        );
 | 
			
		||||
        assert_eq!(None, Date::from_str("400-400-400"));
 | 
			
		||||
 | 
			
		||||
        assert!(Date::from_str("XXX-400-400").is_err());
 | 
			
		||||
 | 
			
		||||
        assert!(Date::from_str("400-400-400").is_err());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -114,13 +114,13 @@ impl FromStr for QueryFilter {
 | 
			
		||||
                }
 | 
			
		||||
                "a" | "after" => {
 | 
			
		||||
                    let date = Date::from_str(filter_arg)
 | 
			
		||||
                        .ok_or(QueryFilterSyntaxError::DateArgumentInvalid)?;
 | 
			
		||||
                        .map_err(|_| QueryFilterSyntaxError::DateArgumentInvalid)?;
 | 
			
		||||
 | 
			
		||||
                    Ok(QueryFilter::SpecialAfterDate(date))
 | 
			
		||||
                }
 | 
			
		||||
                "b" | "bef" | "before" => {
 | 
			
		||||
                    let date = Date::from_str(filter_arg)
 | 
			
		||||
                        .ok_or(QueryFilterSyntaxError::DateArgumentInvalid)?;
 | 
			
		||||
                        .map_err(|_| QueryFilterSyntaxError::DateArgumentInvalid)?;
 | 
			
		||||
 | 
			
		||||
                    Ok(QueryFilter::SpecialBeforeDate(date))
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
		Reference in New Issue
	
	Block a user