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 futures::StreamExt;
 8use tokio_xmpp::{
 9    parsers::{
10        disco::DiscoInfoQuery, iq::Iq, message::Message, presence::Presence, roster::Roster,
11    },
12    Event as TokioXmppEvent,
13};
14
15use crate::{iq, message, presence, Agent, Event};
16
17/// Wait for new events.
18///
19/// # Returns
20///
21/// - `Some(events)` if there are new events; multiple may be returned at once.
22/// - `None` if the underlying stream is closed.
23pub async fn wait_for_events(agent: &mut Agent) -> Option<Vec<Event>> {
24    if let Some(event) = agent.client.next().await {
25        let mut events = Vec::new();
26
27        match event {
28            TokioXmppEvent::Online { resumed: false, .. } => {
29                let presence = Agent::make_initial_presence(&agent.disco, &agent.node).into();
30                let _ = agent.client.send_stanza(presence).await;
31                events.push(Event::Online);
32                // TODO: only send this when the ContactList feature is enabled.
33                let iq = Iq::from_get(
34                    "roster",
35                    Roster {
36                        ver: None,
37                        items: vec![],
38                    },
39                )
40                .into();
41                let _ = agent.client.send_stanza(iq).await;
42
43                // Query account disco to know what bookmarks spec is used
44                let iq = Iq::from_get("disco-account", DiscoInfoQuery { node: None }).into();
45                let _ = agent.client.send_stanza(iq).await;
46                agent.awaiting_disco_bookmarks_type = true;
47            }
48            TokioXmppEvent::Online { resumed: true, .. } => {}
49            TokioXmppEvent::Disconnected(e) => {
50                events.push(Event::Disconnected(e));
51            }
52            TokioXmppEvent::Stanza(elem) => {
53                if elem.is("iq", "jabber:client") {
54                    let iq = Iq::try_from(elem).unwrap();
55                    let new_events = iq::handle_iq(agent, iq).await;
56                    events.extend(new_events);
57                } else if elem.is("message", "jabber:client") {
58                    let message = Message::try_from(elem).unwrap();
59                    let new_events = message::handle_message(agent, message).await;
60                    events.extend(new_events);
61                } else if elem.is("presence", "jabber:client") {
62                    let presence = Presence::try_from(elem).unwrap();
63                    let new_events = presence::receive::handle_presence(agent, presence).await;
64                    events.extend(new_events);
65                } else if elem.is("error", "http://etherx.jabber.org/streams") {
66                    println!("Received a fatal stream error: {}", String::from(&elem));
67                } else {
68                    panic!("Unknown stanza: {}", String::from(&elem));
69                }
70            }
71        }
72
73        Some(events)
74    } else {
75        None
76    }
77}