1use std::fmt;
2
3use serde::Deserialize;
4use serde::Deserializer;
5use serde::Serialize;
6use serde::Serializer;
7use serde::de::Visitor;
8
9use crate::DateTime;
10
11#[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
12impl Serialize for DateTime {
13  fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
14    if self.nanos == 0 {
15      serializer.collect_str(&self.format("%Y-%m-%dT%H:%M:%S%z"))
16    } else if self.nanos % 1_000 == 0 {
17      serializer.collect_str(&self.format("%Y-%m-%dT%H:%M:%S%.6f%z"))
18    } else {
19      serializer.collect_str(&self.format("%Y-%m-%dT%H:%M:%S%.9f%z"))
20    }
21  }
22}
23
24struct DateTimeVisitor;
25
26impl Visitor<'_> for DateTimeVisitor {
27  type Value = DateTime;
28
29  #[cfg(not(tarpaulin_include))]
30  fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
31    formatter.write_str("a YYYY-MM-DD HH:MM:SS date string")
32  }
33
34  fn visit_str<E: serde::de::Error>(self, s: &str) -> Result<Self::Value, E> {
35    s.parse().map_err(E::custom)
36  }
37}
38
39#[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
40impl<'de> Deserialize<'de> for DateTime {
41  fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
42    deserializer.deserialize_str(DateTimeVisitor)
43  }
44}
45
46#[cfg(test)]
47mod tests {
48  use serde_test::Token;
49  use serde_test::assert_tokens;
50
51  use crate::DateTime;
52  use crate::datetime;
53
54  #[test]
55  fn test_serde() {
56    assert_tokens(&datetime! { 2012-04-21 11:00:00 }, &[Token::Str("2012-04-21T11:00:00")]);
57    assert_tokens(&DateTime::ymd(2024, 7, 4).hms(15, 30, 45).nanos(123_456_000).build(), &[
58      Token::Str("2024-07-04T15:30:45.123456"),
59    ]);
60    assert_tokens(&DateTime::ymd(2024, 7, 4).hms(15, 30, 45).nanos(123_456_789).build(), &[
61      Token::Str("2024-07-04T15:30:45.123456789"),
62    ]);
63  }
64
65  #[cfg(feature = "tz")]
66  #[test]
67  fn test_serde_tz() {
68    assert_tokens(&datetime! { 2012-04-21 11:00:00 us::EASTERN }, &[Token::Str(
69      "2012-04-21T11:00:00-0400",
70    )]);
71    assert_tokens(&datetime! { 2012-04-21 11:00:00 europe::BERLIN }, &[Token::Str(
72      "2012-04-21T11:00:00+0200",
73    )]);
74  }
75}