1use futures::stream::StreamExt;
 2use std::env::args;
 3use std::process::exit;
 4use std::str::FromStr;
 5use tokio_xmpp::rustls;
 6use tokio_xmpp::Client;
 7use xmpp_parsers::jid::{BareJid, Jid};
 8use xmpp_parsers::message::{Lang, Message, MessageType};
 9use xmpp_parsers::presence::{Presence, Show as PresenceShow, Type as PresenceType};
10
11#[tokio::main]
12async fn main() {
13    env_logger::init();
14
15    rustls::crypto::aws_lc_rs::default_provider()
16        .install_default()
17        .expect("failed to install rustls crypto provider");
18
19    let args: Vec<String> = args().collect();
20    if args.len() != 3 {
21        println!("Usage: {} <jid> <password>", args[0]);
22        exit(1);
23    }
24    let jid = BareJid::from_str(&args[1]).expect(&format!("Invalid JID: {}", &args[1]));
25    let password = &args[2];
26
27    // Client instance
28    let mut client = Client::new(jid, password.to_owned());
29
30    // Main loop, processes events
31    while let Some(event) = client.next().await {
32        println!("event: {:?}", event);
33        if event.is_online() {
34            let jid = event
35                .get_jid()
36                .map(|jid| format!("{}", jid))
37                .unwrap_or("unknown".to_owned());
38            println!("Online at {}", jid);
39
40            let presence = make_presence();
41            client.send_stanza(presence.into()).await.unwrap();
42        } else if let Some(message) = event
43            .into_stanza()
44            .and_then(|stanza| Message::try_from(stanza).ok())
45        {
46            match (message.from, message.bodies.get("")) {
47                (Some(ref from), Some(body)) if body == "die" => {
48                    println!("Secret die command triggered by {}", from);
49                    break;
50                }
51                (Some(ref from), Some(body)) => {
52                    if message.type_ != MessageType::Error {
53                        // This is a message we'll echo
54                        let reply = make_reply(from.clone(), body.to_owned());
55                        client.send_stanza(reply.into()).await.unwrap();
56                    }
57                }
58                _ => {}
59            }
60        }
61    }
62
63    client.send_end().await.unwrap();
64}
65
66// Construct a <presence/>
67fn make_presence() -> Presence {
68    let mut presence = Presence::new(PresenceType::None);
69    presence.show = Some(PresenceShow::Chat);
70    presence
71        .statuses
72        .insert(Lang::from("en"), String::from("Echoing messages."));
73    presence
74}
75
76// Construct a chat <message/>
77fn make_reply(to: Jid, body: String) -> Message {
78    let mut message = Message::new(Some(to));
79    message.bodies.insert(Lang::default(), body);
80    message
81}