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
|
use crate::agenda::{
parse_message,
AgendaPoint
};
use discord::{
model::{
ChannelId,
Event,
},
Discord,
Error,
};
use futures::join;
use std::sync::{
Arc,
Mutex,
};
use tokio::{
sync::mpsc,
task::{
spawn,
spawn_blocking,
},
};
const TOKEN: Option<&str> = None;
const CHANNEL: Option<u64> = None;
pub async fn handle(
sender: mpsc::UnboundedSender<AgendaPoint>,
receiver: mpsc::UnboundedReceiver<AgendaPoint>,
) {
println!("Setting up Discord");
let token = std::env::var("DISCORD_API_TOKEN").unwrap_or_else(|_| TOKEN.expect("Missing Discord token").to_string());
let channel_id = ChannelId(
match std::env::var("DISCORD_CHANNEL") {
Ok(var) => var.parse().unwrap(),
Err(_) => CHANNEL.expect("Missing Discord channel"),
});
let client = Discord::from_bot_token(&token);
if let Ok(client) = client {
let (connection, _) = client.connect().expect("Discord connect failed"); //TODO
let our_id = client.get_current_user().unwrap().id;
let client = Arc::new(Mutex::new(client));
println!("Discord ready");
let (_, _) = join!( //TODO?
spawn(receive_from_slack(receiver, Arc::clone(&client), channel_id)),
spawn_blocking(move || receive_events(our_id, connection, sender, client, channel_id)),
);
}
}
fn receive_events(
our_id: discord::model::UserId,
mut connection: discord::Connection,
sender: mpsc::UnboundedSender<AgendaPoint>,
client: Arc<Mutex<discord::Discord>>,
channel_id: ChannelId,
) {
loop {
match connection.recv_event() {
Ok(Event::MessageCreate(message)) => {
if let Ok(Some(s)) = parse_message(&message.content, &message.author.name) {
client.lock().unwrap().send_message(channel_id,
&s,
"",
false).unwrap();
}
//if message.author.id != our_id {
// sender.send(AgendaPoint{
// title: message.content,
// adder: message.author.name,
// }).unwrap();
//}
}
Ok(_) => {}
Err(Error::Closed(code, body)) => {
println!("Discord closed with code {:?}: {}", code, body);
break;
}
Err(e) => {
println!("Discord error: {:?}", e);
}
}
}
}
async fn receive_from_slack(
mut receiver: mpsc::UnboundedReceiver<AgendaPoint>,
client: Arc<Mutex<discord::Discord>>,
channel_id: ChannelId
) {
while let Some(point) = receiver.recv().await {
println!("Discord received '{}'", point);
client.lock().unwrap().send_message(channel_id,
&point.to_add_message(),
"",
false).unwrap();
}
}
|
