xmpp/
delay.rs

1// Copyright (c) 2023 xmpp-rs contributors.
2//
3// This Source Code Form is subject to the terms of the Mozilla Public
4// License, v. 2.0. If a copy of the MPL was not distributed with this
5// file, You can obtain one at http://mozilla.org/MPL/2.0/.
6
7use chrono::{DateTime, Utc};
8use tokio_xmpp::{
9    jid::Jid,
10    parsers::{delay::Delay, message::Message, ns},
11};
12
13/// Time information associated with a stanza.
14///
15/// Contains information about when the message was received, and any claim about when it was sent.
16#[derive(Debug, Clone)]
17pub struct StanzaTimeInfo {
18    /// Time information when the message was received by the library
19    pub received: DateTime<Utc>,
20
21    /// Time information claimed by the sender or an intermediary.
22    ///
23    /// **Warning**: this has security implications. See [XEP-0203 security section](https://xmpp.org/extensions/xep-0203.html#security).
24    pub delays: Vec<Delay>,
25}
26
27impl StanzaTimeInfo {
28    pub fn delay_from(&self, jid: &Jid) -> Option<&Delay> {
29        self.delays.iter().find(|delay| {
30            if let Some(from) = &delay.from {
31                return from == jid;
32            }
33            return false;
34        })
35    }
36}
37
38/// Parsing a [`Message`], store the current time it was processed, as well [XEP-0203](https://xmpp.org/extensions/xep-0203.html#protocol)
39/// [`Delay`] contained in the message's payloads.
40///
41/// Specifically, this method will look for any `<delay/>` element in the message's payloads. If they were found,
42/// they will be added to the [`StanzaTimeInfo`] result.
43pub fn message_time_info(message: &Message) -> StanzaTimeInfo {
44    let mut delays = vec![];
45
46    // Scan the message payloads for XEP-0203 delays.
47    for payload in &message.payloads {
48        if payload.is("delay", ns::DELAY) {
49            match Delay::try_from(payload.clone()) {
50                Ok(delay) => delays.push(delay),
51                Err(e) => {
52                    error!("Wrong <delay> format in payload from {}:{}\n{:?}\nUsing received time only.",
53                    message.from.as_ref().unwrap().to_owned(),
54                    e,
55                    payload);
56                }
57            }
58        }
59    }
60
61    StanzaTimeInfo {
62        received: Utc::now(),
63        delays,
64    }
65}
xmpp/delay.rs

xmpp/
delay.rs
