1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

use chrono::NaiveDate;
use serde::{self, Deserialize, Deserializer, Serialize, Serializer};
use std::fmt;
use std::ops::Sub;

const READABLE_DATE_FORMAT: &'static str = "%b %-e, %Y";

/// ReadableDate wraps a date and could be (de)serialize with format like
/// `Apr 4, 2010`.
#[derive(Debug, Clone, PartialEq, PartialOrd)]
pub struct ReadableDate(NaiveDate);

impl ReadableDate {
    pub fn new(year: i32, month: u32, day: u32) -> Self {
        Self(NaiveDate::from_ymd(year, month, day))
    }

    /// Create a `ReadableDate` for today.
    pub fn today() -> Self {
        Self(chrono::Local::today().naive_local())
    }
}

impl Default for ReadableDate {
    fn default() -> Self {
        Self::today()
    }
}

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

impl Serialize for ReadableDate {
    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        serializer.serialize_str(&format!("{}", self.0.format(READABLE_DATE_FORMAT)))
    }
}

impl<'de> Deserialize<'de> for ReadableDate {
    fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
        let s = String::deserialize(deserializer)?;
        Ok(Self(
            NaiveDate::parse_from_str(&s, READABLE_DATE_FORMAT)
                .map_err(serde::de::Error::custom)?,
        ))
    }
}

macro_rules! impl_sub {
    ($for:ty, $rhs:ty) => {
        impl Sub for $for {
            type Output = chrono::Duration;

            fn sub(self, other: $rhs) -> Self::Output {
                self.0 - other.0
            }
        }
    };
}

impl_sub! { ReadableDate, ReadableDate }
impl_sub! { &ReadableDate, &ReadableDate }

#[cfg(test)]
mod test {
    use super::*;

    #[test]
    fn date_sub() {
        assert_eq!(
            ReadableDate::new(2020, 02, 02) - ReadableDate::new(2020, 02, 01),
            chrono::Duration::days(1)
        );
    }
}