lib: add a component connection method
This commit is contained in:
parent
430cd4b44d
commit
c0f3fc4afb
7 changed files with 416 additions and 3 deletions
103
examples/echo_component.rs
Normal file
103
examples/echo_component.rs
Normal file
|
|
@ -0,0 +1,103 @@
|
|||
extern crate futures;
|
||||
extern crate tokio_core;
|
||||
extern crate tokio_xmpp;
|
||||
extern crate jid;
|
||||
extern crate minidom;
|
||||
extern crate xmpp_parsers;
|
||||
extern crate try_from;
|
||||
|
||||
use std::env::args;
|
||||
use std::process::exit;
|
||||
use std::str::FromStr;
|
||||
use try_from::TryFrom;
|
||||
use tokio_core::reactor::Core;
|
||||
use futures::{Future, Stream, Sink, future};
|
||||
use tokio_xmpp::Component;
|
||||
use minidom::Element;
|
||||
use xmpp_parsers::presence::{Presence, Type as PresenceType, Show as PresenceShow};
|
||||
use xmpp_parsers::message::{Message, MessageType};
|
||||
use jid::Jid;
|
||||
|
||||
fn main() {
|
||||
let args: Vec<String> = args().collect();
|
||||
if args.len() < 3 || args.len() > 5 {
|
||||
println!("Usage: {} <jid> <password> [server] [port]", args[0]);
|
||||
exit(1);
|
||||
}
|
||||
let jid = &args[1];
|
||||
let password = &args[2];
|
||||
let server = &args.get(3).unwrap().parse().unwrap_or("127.0.0.1".to_owned());
|
||||
let port: u16 = args.get(4).unwrap().parse().unwrap_or(5347u16);
|
||||
|
||||
// tokio_core context
|
||||
let mut core = Core::new().unwrap();
|
||||
// Component instance
|
||||
println!("{} {} {} {} {:?}", jid, password, server, port, core.handle());
|
||||
let component = Component::new(jid, password, server, port, core.handle()).unwrap();
|
||||
|
||||
// Make the two interfaces for sending and receiving independent
|
||||
// of each other so we can move one into a closure.
|
||||
println!("Got it: {}", component.jid);
|
||||
let (sink, stream) = component.split();
|
||||
// Wrap sink in Option so that we can take() it for the send(self)
|
||||
// to consume and return it back when ready.
|
||||
let mut sink = Some(sink);
|
||||
let mut send = move |stanza| {
|
||||
sink = Some(
|
||||
sink.take().
|
||||
expect("sink")
|
||||
.send(stanza)
|
||||
.wait()
|
||||
.expect("sink.send")
|
||||
);
|
||||
};
|
||||
// Main loop, processes events
|
||||
let done = stream.for_each(|event| {
|
||||
if event.is_online() {
|
||||
println!("Online!");
|
||||
|
||||
let presence = make_presence(Jid::from_str("test@component.linkmauve.fr/coucou").unwrap(), Jid::from_str("linkmauve@linkmauve.fr").unwrap());
|
||||
send(presence);
|
||||
} else if let Some(message) = event.into_stanza()
|
||||
.and_then(|stanza| Message::try_from(stanza).ok())
|
||||
{
|
||||
// This is a message we'll echo
|
||||
match (message.from, message.bodies.get("")) {
|
||||
(Some(from), Some(body)) =>
|
||||
if message.type_ != MessageType::Error {
|
||||
let reply = make_reply(from, body);
|
||||
send(reply);
|
||||
},
|
||||
_ => (),
|
||||
}
|
||||
}
|
||||
|
||||
Box::new(future::ok(()))
|
||||
});
|
||||
|
||||
// Start polling `done`
|
||||
match core.run(done) {
|
||||
Ok(_) => (),
|
||||
Err(e) => {
|
||||
println!("Fatal: {}", e);
|
||||
()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Construct a <presence/>
|
||||
fn make_presence(from: Jid, to: Jid) -> Element {
|
||||
let mut presence = Presence::new(PresenceType::None);
|
||||
presence.from = Some(from);
|
||||
presence.to = Some(to);
|
||||
presence.show = PresenceShow::Chat;
|
||||
presence.statuses.insert(String::from("en"), String::from("Echoing messages."));
|
||||
presence.into()
|
||||
}
|
||||
|
||||
// Construct a chat <message/>
|
||||
fn make_reply(to: Jid, body: &str) -> Element {
|
||||
let mut message = Message::new(Some(to));
|
||||
message.bodies.insert(String::new(), body.to_owned());
|
||||
message.into()
|
||||
}
|
||||
Loading…
Reference in a new issue