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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

// Copyright (c) 2019 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
use std::str::FromStr;

use chrono::{DateTime, FixedOffset, Utc};

use crate::core::{error::Error, FromXml, IntoXml, TextCodec};
use crate::date::Xep0082;
use crate::iq::{IqGetPayload, IqResultPayload};
use crate::ns::TIME;

struct ColonSeparatedOffset;

impl TextCodec<FixedOffset> for ColonSeparatedOffset {
    fn decode(s: &str) -> Result<FixedOffset, Error> {
        Ok(FixedOffset::from_str(s)?)
    }

    fn encode(value: FixedOffset) -> Option<String> {
        let offset = value.local_minus_utc();
        let nminutes = offset / 60;
        let nseconds = offset % 60;
        let nhours = nminutes / 60;
        let nminutes = nminutes % 60;
        if nseconds == 0 {
            Some(format!("{:+03}:{:02}", nhours, nminutes))
        } else {
            Some(format!("{:+03}:{:02}:{:02}", nhours, nminutes, nseconds))
        }
    }
}

/// An entity time query.
#[derive(FromXml, IntoXml, Debug, PartialEq, Clone)]
#[xml(namespace = TIME, name = "time")]
pub struct TimeQuery;

impl IqGetPayload for TimeQuery {}

/// An entity time result, containing an unique DateTime.
///
/// Note that this type can be converted to and from [`chrono::DateTime`]
/// (with [`chrono::Utc`] and [`chrono::FixedOffset`] timezones), which is
/// probably more friendly than using it directly.
#[derive(FromXml, IntoXml, Debug, PartialEq, Clone)]
#[xml(namespace = TIME, name = "time")]
pub struct TimeResult {
    /// The UTC offset
    #[xml(child(namespace = TIME, name = "tzo", extract(text(codec = ColonSeparatedOffset))))]
    pub tz_offset: FixedOffset,

    /// The UTC timestamp
    #[xml(child(namespace = TIME, name = "utc", extract(text(codec = Xep0082))))]
    pub utc: DateTime<Utc>,
}

impl IqResultPayload for TimeResult {}

impl From<TimeResult> for chrono::DateTime<FixedOffset> {
    fn from(other: TimeResult) -> Self {
        other.utc.with_timezone(&other.tz_offset)
    }
}

impl From<TimeResult> for chrono::DateTime<Utc> {
    fn from(other: TimeResult) -> Self {
        other.utc.into()
    }
}

impl From<chrono::DateTime<FixedOffset>> for TimeResult {
    fn from(other: chrono::DateTime<FixedOffset>) -> Self {
        let tz_offset = *other.offset();
        let utc = other.with_timezone(&Utc);
        TimeResult {
            tz_offset,
            utc: utc.into(),
        }
    }
}

impl From<chrono::DateTime<Utc>> for TimeResult {
    fn from(other: chrono::DateTime<Utc>) -> Self {
        TimeResult {
            tz_offset: FixedOffset::east_opt(0).unwrap(),
            utc: other.into(),
        }
    }
}

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

    // DateTime’s size doesn’t depend on the architecture.
    #[test]
    fn test_size() {
        assert_size!(TimeQuery, 0);
        assert_size!(TimeResult, 16);
    }

    #[test]
    fn parse_response() {
        let elem: Element =
            "<time xmlns='urn:xmpp:time'><tzo>-06:00</tzo><utc>2006-12-19T17:58:35Z</utc></time>"
                .parse()
                .unwrap();
        let elem1 = elem.clone();
        let time = TimeResult::try_from(elem).unwrap();
        assert_eq!(time.tz_offset, FixedOffset::west_opt(6 * 3600).unwrap());
        assert_eq!(
            DateTime::<FixedOffset>::from(time.clone()),
            DateTime::<FixedOffset>::from_str("2006-12-19T12:58:35-05:00").unwrap()
        );
        assert_eq!(
            DateTime::<Utc>::from(time.clone()),
            DateTime::<Utc>::from_str("2006-12-19T12:58:35-05:00").unwrap()
        );
        let elem2 = Element::from(time);
        assert_eq!(elem1, elem2);
    }
}