blob: ddf4febb37aedd5ae61a12c02cf828154eac4cf3 (
plain)
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
|
use std::process::Command;
use std::fs::File;
use std::io::prelude::*;
extern crate toml;
extern crate websocket;
extern crate rustc_serialize;
use rustc_serialize::{Decodable};
use rustc_serialize::json::{Json, Decoder};
mod message;
use message::{Message, Push};
fn main() {
use websocket::{Receiver};
use websocket::client::request::Url;
use websocket::Client;
let mut cfg_file = File::open("config.toml").unwrap();
let mut s = String::new();
cfg_file.read_to_string(&mut s);
let cfg = toml::Parser::new(s.as_ref()).parse().unwrap();
let token = cfg.get("pushbullet").unwrap().as_table().unwrap().get("token").expect("Could not find 'token' in config.toml").as_str().unwrap();
let wss_url = format!("wss://stream.pushbullet.com/websocket/{}", token);
let url = Url::parse(wss_url.as_ref()).unwrap();
let request = Client::connect(url).unwrap();
let response = request.send().unwrap();
response.validate().unwrap();
let (mut sender, mut receiver) = response.begin().split();
for message in receiver.incoming_messages() {
let message = match message {
Ok(m) => m,
Err(e) => {
println!("Error: {:?}", e);
return;
}
};
match message {
websocket::Message::Close(_) => {
return;
}
websocket::Message::Text(message) => {
let json = Json::from_str(message.as_ref()).unwrap();
let mut decoder = Decoder::new(json);
let msg = Decodable::decode(&mut decoder).unwrap();
println!("Got {:?}", msg);
match msg {
Message::Nop => {}
Message::Push(Push::Mirror {
notification_id,
title,
body,
application_name,
..
}) => {
let _ = Command::new("notify-send")
.arg(title)
.arg(body)
.arg("-t")
.arg("8000")
.status().unwrap();
}
Message::Push(Push::Dismissal { notification_id }) => {
}
}
}
_ => {
println!("Got {:?}", message);
}
}
}
}
|
