From 8e7b7ae31c95592855c8d62f6da9a699862a2ca2 Mon Sep 17 00:00:00 2001 From: Emmanuel Gil Peyrot Date: Sat, 27 May 2017 16:37:21 +0100 Subject: [PATCH] simplify the presence plugin using xmpp-parsers --- examples/client.rs | 4 +- src/plugins/presence.rs | 98 +++++++++-------------------------------- 2 files changed, 23 insertions(+), 79 deletions(-) diff --git a/examples/client.rs b/examples/client.rs index 290e69a7..b136da78 100644 --- a/examples/client.rs +++ b/examples/client.rs @@ -4,7 +4,7 @@ use xmpp::jid::Jid; use xmpp::client::ClientBuilder; use xmpp::plugins::stanza::StanzaPlugin; use xmpp::plugins::messaging::{MessagingPlugin, MessageEvent}; -use xmpp::plugins::presence::{PresencePlugin, Show}; +use xmpp::plugins::presence::{PresencePlugin, Type}; use xmpp::plugins::ping::{PingPlugin, PingEvent}; use std::env; @@ -20,7 +20,7 @@ fn main() { client.register_plugin(MessagingPlugin::new()); client.register_plugin(PresencePlugin::new()); client.register_plugin(PingPlugin::new()); - client.plugin::().set_presence(Show::Available, None).unwrap(); + client.plugin::().set_presence(Type::Available, None, None).unwrap(); client.main().unwrap(); /*loop { let event = client.next_event().unwrap(); diff --git a/src/plugins/presence.rs b/src/plugins/presence.rs index e63b5b3e..344dd1f6 100644 --- a/src/plugins/presence.rs +++ b/src/plugins/presence.rs @@ -1,56 +1,9 @@ +use std::collections::BTreeMap; + use error::Error; use plugin::PluginProxy; -use minidom::Element; - -use ns; - -use std::fmt; - -use std::str::FromStr; - -#[derive(Debug, Copy, Clone, PartialEq, Eq)] -pub enum Show { - Available, - Away, - ExtendedAway, - DoNotDisturb, - Chat, - Unavailable, -} - -impl fmt::Display for Show { - fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result { - match *self { - Show::Away => write!(fmt, "away"), - Show::ExtendedAway => write!(fmt, "xa"), - Show::DoNotDisturb => write!(fmt, "dnd"), - Show::Chat => write!(fmt, "chat"), - - // will never be seen inside a , maybe should crash? - Show::Available => write!(fmt, "available"), - Show::Unavailable => write!(fmt, "unavailable"), - } - } -} - -#[derive(Debug, Copy, Clone, PartialEq, Eq)] -pub struct InvalidShow; - -impl FromStr for Show { - type Err = InvalidShow; - - fn from_str(s: &str) -> Result { - Ok(match s { - "away" => Show::Away, - "xa" => Show::ExtendedAway, - "dnd" => Show::DoNotDisturb, - "chat" => Show::Chat, - - _ => { return Err(InvalidShow); } - }) - } -} +pub use xmpp_parsers::presence::{Presence, PresenceType as Type, Show}; pub struct PresencePlugin { proxy: PluginProxy, @@ -63,33 +16,24 @@ impl PresencePlugin { } } - pub fn set_presence(&self, show: Show, status: Option) -> Result<(), Error> { - if show == Show::Unavailable { - self.proxy.send(Element::builder("presence") - .ns(ns::CLIENT) - .attr("type", "unavailable") - .build()); - } - else { - let mut stanza = Element::builder("presence") - .ns(ns::CLIENT) - .build(); - if let Some(stat) = status { - let elem = Element::builder("status") - .ns(ns::CLIENT) - .append(stat) - .build(); - stanza.append_child(elem); - } - if show != Show::Available { - let mut elem = Element::builder("show") - .ns(ns::CLIENT) - .build(); - elem.append_text_node(show.to_string()); - stanza.append_child(elem); - } - self.proxy.send(stanza); - } + pub fn set_presence(&self, type_: Type, show: Option, status: Option) -> Result<(), Error> { + let presence = Presence { + from: None, + to: None, + id: None, + type_: type_, + show: show, + priority: 0i8, + statuses: { + let mut statuses = BTreeMap::new(); + if let Some(status) = status { + statuses.insert(String::new(), status); + } + statuses + }, + payloads: vec!(), + }; + self.proxy.send(presence.into()); Ok(()) } }