xmpp/iq/
get.rs

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
// Copyright (c) 2023 xmpp-rs contributors.
//
// 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 tokio_xmpp::{
    jid::Jid,
    minidom::Element,
    parsers::{
        disco::DiscoInfoQuery,
        iq::Iq,
        ns,
        stanza_error::{DefinedCondition, ErrorType, StanzaError},
    },
};

use crate::{Agent, Event};

pub async fn handle_iq_get(
    agent: &mut Agent,
    _events: &mut Vec<Event>,
    from: Jid,
    _to: Option<Jid>,
    id: String,
    payload: Element,
) {
    if payload.is("query", ns::DISCO_INFO) {
        let query = DiscoInfoQuery::try_from(payload);
        match query {
            Ok(query) => {
                let mut disco_info = agent.disco.clone();
                disco_info.node = query.node;
                let iq = Iq::from_result(id, Some(disco_info)).with_to(from).into();
                let _ = agent.client.send_stanza(iq).await;
            }
            Err(err) => {
                let error = StanzaError::new(
                    ErrorType::Modify,
                    DefinedCondition::BadRequest,
                    "en",
                    &format!("{}", err),
                );
                let iq = Iq::from_error(id, error).with_to(from).into();
                let _ = agent.client.send_stanza(iq).await;
            }
        }
    } else {
        // We MUST answer unhandled get iqs with a service-unavailable error.
        let error = StanzaError::new(
            ErrorType::Cancel,
            DefinedCondition::ServiceUnavailable,
            "en",
            "No handler defined for this kind of iq.",
        );
        let iq = Iq::from_error(id, error).with_to(from).into();
        let _ = agent.client.send_stanza(iq).await;
    }
}