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

// 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 chrono::FixedOffset;
use crate::date::DateTime;
use crate::iq::{IqGetPayload, IqResultPayload};
use crate::ns;
use crate::util::error::Error;
use minidom::Element;
use std::convert::TryFrom;
use std::str::FromStr;

generate_empty_element!(
    /// An entity time query.
    TimeQuery, "time", TIME
);

impl IqGetPayload for TimeQuery {}

/// An entity time result, containing an unique DateTime.
#[derive(Debug, Clone)]
pub struct TimeResult(pub DateTime);

impl IqResultPayload for TimeResult {}

impl TryFrom<Element> for TimeResult {
    type Error = Error;

    fn try_from(elem: Element) -> Result<TimeResult, Error> {
        check_self!(elem, "time", TIME);
        check_no_attributes!(elem, "time");

        let mut tzo = None;
        let mut utc = None;

        for child in elem.children() {
            if child.is("tzo", ns::TIME) {
                if tzo.is_some() {
                    return Err(Error::ParseError("More than one tzo element in time."));
                }
                check_no_children!(child, "tzo");
                check_no_attributes!(child, "tzo");
                // TODO: Add a FromStr implementation to FixedOffset to avoid this hack.
                let fake_date = String::from("2019-04-22T11:38:00") + &child.text();
                let date_time = DateTime::from_str(&fake_date)?;
                tzo = Some(date_time.timezone());
            } else if child.is("utc", ns::TIME) {
                if utc.is_some() {
                    return Err(Error::ParseError("More than one utc element in time."));
                }
                check_no_children!(child, "utc");
                check_no_attributes!(child, "utc");
                let date_time = DateTime::from_str(&child.text())?;
                if date_time.timezone() != FixedOffset::east(0) {
                    return Err(Error::ParseError("Non-UTC timezone for utc element."));
                }
                utc = Some(date_time);
            } else {
                return Err(Error::ParseError(
                    "Unknown child in time element.",
                ));
            }
        }

        let tzo = tzo.ok_or(Error::ParseError("Missing tzo child in time element."))?;
        let utc = utc.ok_or(Error::ParseError("Missing utc child in time element."))?;
        let date = utc.with_timezone(&tzo);

        Ok(TimeResult(date))
    }
}

impl From<TimeResult> for Element {
    fn from(time: TimeResult) -> Element {
        Element::builder("time")
            .ns(ns::TIME)
            .append(Element::builder("tzo")
                    .append(format!("{}", time.0.timezone()))
                    .build())
            .append(Element::builder("utc")
                    .append(time.0.with_timezone(&FixedOffset::east(0)).format("%FT%TZ"))
                    .build())
            .build()
    }
}

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

    // 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.0.timezone(), FixedOffset::west(6 * 3600));
        assert_eq!(time.0, DateTime::from_str("2006-12-19T12:58:35-05:00").unwrap());
        let elem2 = Element::from(time);
        assert_eq!(elem1, elem2);
    }
}