aboutsummaryrefslogtreecommitdiffstats
path: root/src/slack.rs
blob: c13fc47d6694df9fa7f097f03c00b83cc94d2018 (plain) (blame)
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
use futures::join;
use slack::{
    Event,
    Message,
};
use tokio::{
    sync::mpsc,
    task::{
        spawn,
        spawn_blocking,
    },
};

struct Handler {
    sender: mpsc::UnboundedSender<String>,
}

impl Handler {
    fn new(sender: mpsc::UnboundedSender<String>) -> Self {
        Self {
            sender
        }
    }

    fn sender(&self) -> &mpsc::UnboundedSender<String> {
        &self.sender
    }
}

impl slack::EventHandler for Handler {
    fn on_event(&mut self, _cli: &slack::RtmClient, event: slack::Event) {
        println!("on_event: {:#?}", event);
        match event {
            Event::Message(msg) => {
                match *msg {
                    Message::Standard(msg) => {
                        self.sender().send(format!("{}:{} says: {}",
                                                   msg.channel.unwrap_or("??".to_string()),
                                                   msg.user.unwrap_or("??".to_string()),
                                                   msg.text.unwrap_or("??".to_string()))
                                           .to_string()).unwrap();
                    }
                    _ => {}
                }
            }
            _ => {}
        }
    }

    fn on_close(&mut self, _cli: &slack::RtmClient) {
        println!("on_close")
    }

    fn on_connect(&mut self, _cli: &slack::RtmClient) {
        println!("on_connect");
    }
}

pub async fn handle(
    token: Option<String>,
    sender: mpsc::UnboundedSender<String>,
    receiver: mpsc::UnboundedReceiver<String>,
) {
    println!("Setting up Slack");

    let token = std::env::var("SLACK_API_TOKEN").unwrap_or(token.unwrap());
    let client = spawn_blocking(move || {
        slack::RtmClient::login(&token).unwrap()
    }).await.unwrap();

    let slack_sender = client.sender().clone();

    join!(
        spawn_blocking(move || {
            let mut handler = Handler::new(sender);
            match client.run(&mut handler) {
                Ok(_) => {}
                Err(e) => {
                    println!("Error: {}", e)
                }
            }
        }),
        spawn(receive_from_discord(receiver, slack_sender))
    );
}

async fn receive_from_discord(
    mut receiver: mpsc::UnboundedReceiver<String>,
    sender: slack::Sender,
) {
    while let Some(s) = receiver.recv().await {
        println!("Slack received '{}'", s);
        //TODO Sending messages is very slow sometimes. Have seen delays
        // from 5 up to 20(!) seconds.
        sender.send_typing("CPBAA5FA7").unwrap();
        println!("Typing");
        sender.send_message("CPBAA5FA7", &s).unwrap();
        println!("Sent");
    }
}