libstrophe 0.17.0

Rust ergonomic wrapper for libstrophe
Documentation 
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

// hack to emulate `extern crate libstrophe`
mod libstrophe {
	pub use crate::*;
}

fn version_handler(_ctx: &libstrophe::Context, conn: &mut libstrophe::Connection, stanza: &libstrophe::Stanza) -> bool {
	eprintln!("Received version request from {}", stanza.from().expect("Empty from"));

	let mut reply = stanza.reply();
	reply.set_stanza_type("result").expect("Cannot set stanza type");

	let mut query = libstrophe::Stanza::new();
	query.set_name("query").expect("Cannot set stanza name");
	if let Some(ns) = stanza.get_first_child().expect("No children").ns() {
		query.set_ns(ns).expect("Cannot set stanza ns");
	}

	let mut name = libstrophe::Stanza::new();
	name.set_name("name").expect("Cannot set stanza name");

	let mut text = libstrophe::Stanza::new();
	text.set_text("libstrophe example bot").expect("Cannot set stanza text");
	name.add_child(text).expect("Cannot add child");

	query.add_child(name).expect("Cannot add child");

	let mut version = libstrophe::Stanza::new();
	version.set_name("version").expect("Cannot set stanza name");

	let mut text = libstrophe::Stanza::new();
	text.set_text("1.0").expect("Cannot set stanza text");
	version.add_child(text).expect("Cannot add child");

	query.add_child(version).expect("Cannot add child");

	reply.add_child(query).expect("Cannot add child");

	conn.send(&reply);
	true
}

fn message_handler(_ctx: &libstrophe::Context, conn: &mut libstrophe::Connection, stanza: &libstrophe::Stanza) -> bool {
	let body = match stanza.get_child_by_name("body") {
		Some(body) => body,
		None => return true,
	};

	match stanza.stanza_type() {
		Some("error") | None => return true,
		_ => (),
	}

	let intext = body.text().expect("Cannot get body");

	eprintln!("Incoming message from {}: {}", stanza.from().expect("Cannot get from"), intext);

	let mut reply = stanza.reply();
	if reply.stanza_type().is_none() {
		reply.set_stanza_type("chat").expect("Cannot set type");
	}

	let (quit, replytext) = if intext == "quit" {
		(true, "bye!".to_owned())
	} else {
		(false, format!("{} to you too!", intext))
	};
	reply.set_body(replytext).expect("Cannot set body");

	conn.send(&reply);

	if quit {
		conn.disconnect();
	}

	true
}

/// Port of the [bot.c](https://github.com/strophe/libstrophe/blob/0.11.0/examples/bot.c) code
#[allow(dead_code)]
pub fn main() {
	env_logger::init();

	let jid = "test@example.com";
	let pass = "<password>";

	let conn_handler = move |ctx: &libstrophe::Context, conn: &mut libstrophe::Connection, evt: libstrophe::ConnectionEvent| {
		if let libstrophe::ConnectionEvent::Connect = evt {
			eprintln!("Connected");
			conn.handler_add(&version_handler, Some("jabber:iq:version"), Some("iq"), None);
			conn.handler_add(&message_handler, None, Some("message"), None);
			let pres = libstrophe::Stanza::new_presence();

			conn.send(&pres);
		} else {
			eprintln!("Disconnected");
			ctx.stop();
		}
	};

	let mut conn = libstrophe::Connection::new(libstrophe::Context::new_with_default_logger());
	conn.set_jid(jid);
	conn.set_pass(pass);
	let ctx = conn.connect_client(None, None, &conn_handler).expect("Cannot connect to XMPP server");
	ctx.run();
	libstrophe::shutdown();
}