66 lines
1.8 KiB
Rust
66 lines
1.8 KiB
Rust
|
use plugin::{Plugin, PluginReturn, PluginProxy};
|
||
|
|
use event::Event;
|
||
|
|
use minidom::Element;
|
||
|
|
use error::Error;
|
||
|
|
use jid::Jid;
|
||
|
|
use ns;
|
||
|
|
|
||
|
|
#[derive(Debug)]
|
||
|
|
pub struct PingEvent {
|
||
|
|
pub from: Jid,
|
||
|
|
pub to: Jid,
|
||
|
|
pub id: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Event for PingEvent {}
|
||
|
|
|
||
|
|
pub struct PingPlugin {
|
||
|
|
proxy: PluginProxy,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl PingPlugin {
|
||
|
|
pub fn new() -> PingPlugin {
|
||
|
|
PingPlugin {
|
||
|
|
proxy: PluginProxy::new(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn send_ping(&self, to: &Jid) -> Result<(), Error> {
|
||
|
|
let mut elem = Element::builder("iq")
|
||
|
|
.attr("type", "get")
|
||
|
|
.attr("to", to.to_string())
|
||
|
|
.build();
|
||
|
|
elem.append_child(Element::builder("ping").ns(ns::PING).build());
|
||
|
|
self.proxy.send(elem);
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn reply_ping(&self, event: &PingEvent) {
|
||
|
|
let reply = Element::builder("iq")
|
||
|
|
.attr("type", "result")
|
||
|
|
.attr("to", event.from.to_string())
|
||
|
|
.attr("id", event.id.to_string())
|
||
|
|
.build();
|
||
|
|
self.proxy.send(reply);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Plugin for PingPlugin {
|
||
|
|
fn get_proxy(&mut self) -> &mut PluginProxy {
|
||
|
|
&mut self.proxy
|
||
|
|
}
|
||
|
|
|
||
|
|
fn handle(&mut self, elem: &Element) -> PluginReturn {
|
||
|
|
if elem.is("iq", ns::CLIENT) && elem.attr("type") == Some("get") {
|
||
|
|
if elem.has_child("ping", ns::PING) {
|
||
|
|
self.proxy.dispatch(PingEvent { // TODO: safety!!!
|
||
|
|
from: elem.attr("from").unwrap().parse().unwrap(),
|
||
|
|
to: elem.attr("to").unwrap().parse().unwrap(),
|
||
|
|
id: elem.attr("id").unwrap().parse().unwrap(),
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
PluginReturn::Continue
|
||
|
|
}
|
||
|
|
}
|