rustdi 0.1.0

Modern WhatsApp Client Library for Rust
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

extern crate simple_logger;
#[macro_use]
extern crate log;
extern crate qrcode;
extern crate image;
extern crate bincode;
extern crate rustdi;
extern crate reqwest;
extern crate base64;

use std::fs::{File, OpenOptions};

use image::Luma;

use rustdi::connection;
use rustdi::connection::{DisconnectReason, PersistentSession, RustdiHandler, RustdiConnection, UserData, State};
use rustdi::message::ChatMessage;
use rustdi::Jid;


const SESSION_FILENAME: &str = "session.bin";

struct Handler {
    subscribed_jid: Jid
}

impl RustdiHandler for Handler {
    fn on_state_changed(&self, connection: &RustdiConnection<Handler>, state: State) {
        info!("new state: {:?}", state);
        if state == State::Connected {
            connection.subscribe_presence(&self.subscribed_jid);
        }

    }

    fn on_persistent_session_data_changed(&self, persistent_session: PersistentSession) {
        bincode::serialize_into(OpenOptions::new().create(true).write(true).open(SESSION_FILENAME).unwrap(), &persistent_session).unwrap();
    }
    fn on_user_data_changed(&self, _: &RustdiConnection<Handler>, user_data: UserData) {
        if let UserData::PresenceChange(jid, status, _) = user_data {
            if jid == self.subscribed_jid {
                info!("{} is now {:?}", jid.phonenumber().unwrap(), status);
            }
        }
    }
    fn on_disconnect(&self, _: DisconnectReason) {
        info!("disconnected");
    }
    fn on_message(&self, _: &RustdiConnection<Handler>, _: bool, _: Box<ChatMessage>) {}
}

fn main() {

    simple_logger::init_with_level(log::Level::Info).unwrap();
    let handler = Handler {subscribed_jid: Jid::from_phone_number("+49123456789".to_string()).unwrap()};

    if let Ok(file) = File::open(SESSION_FILENAME) {
        let (_, join_handle) = connection::with_persistent_session(bincode::deserialize_from(file).unwrap(), handler);
        join_handle.join().unwrap();
    } else {
        let (_, join_handle) = connection::new(|qr| { qr.render::<Luma<u8>>().module_dimensions(10, 10).build().save("login_qr.png").unwrap(); }, handler);
        join_handle.join().unwrap();
    }
}