Move handling of IqType::Set to iq::set module

This commit is contained in:
xmppftw@kl.netlib.re 2023-12-31 13:57:39 +01:00
parent d821d2ecb2
commit d3603d1e68
2 changed files with 38 additions and 13 deletions

View file

@ -4,15 +4,13 @@
// License, v. 2.0. If a copy of the MPL was not distributed with this // 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/. // file, You can obtain one at http://mozilla.org/MPL/2.0/.
use tokio_xmpp::parsers::{ use tokio_xmpp::parsers::iq::{Iq, IqType};
iq::{Iq, IqType},
stanza_error::{DefinedCondition, ErrorType, StanzaError},
};
use crate::{Agent, Event}; use crate::{Agent, Event};
pub mod get; pub mod get;
pub mod result; pub mod result;
pub mod set;
pub async fn handle_iq(agent: &mut Agent, iq: Iq) -> Vec<Event> { pub async fn handle_iq(agent: &mut Agent, iq: Iq) -> Vec<Event> {
let mut events = vec![]; let mut events = vec![];
@ -24,15 +22,8 @@ pub async fn handle_iq(agent: &mut Agent, iq: Iq) -> Vec<Event> {
get::handle_iq_get(agent, &mut events, from, iq.to, iq.id, payload).await; get::handle_iq_get(agent, &mut events, from, iq.to, iq.id, payload).await;
} else if let IqType::Result(Some(payload)) = iq.payload { } else if let IqType::Result(Some(payload)) = iq.payload {
result::handle_iq_result(agent, &mut events, from, iq.to, iq.id, payload).await; result::handle_iq_result(agent, &mut events, from, iq.to, iq.id, payload).await;
} else if let IqType::Set(_payload) = iq.payload { } else if let IqType::Set(payload) = iq.payload {
let error = StanzaError::new( set::handle_iq_set(agent, &mut events, from, iq.to, iq.id, payload).await;
ErrorType::Cancel,
DefinedCondition::ServiceUnavailable,
"en",
"No handler defined for this kind of iq.",
);
let iq = Iq::from_error(iq.id, error).with_to(from).into();
let _ = agent.client.send_stanza(iq).await;
} }
events events
} }

34
xmpp/src/iq/set.rs Normal file
View file

@ -0,0 +1,34 @@
// 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::{
parsers::{
iq::Iq,
stanza_error::{DefinedCondition, ErrorType, StanzaError},
},
Element, Jid,
};
use crate::{Agent, Event};
pub async fn handle_iq_set(
agent: &mut Agent,
_events: &mut Vec<Event>,
from: Jid,
_to: Option<Jid>,
id: String,
_payload: Element,
) {
// We MUST answer unhandled set 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;
}