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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
use std::sync::Arc;
use tokio::sync::RwLock;
use tokio::sync::Mutex;
use twilight_model::gateway::OpCode;
use twilight_model::gateway::payload::outgoing::{update_presence::UpdatePresencePayload, UpdatePresence};
use twilight_gateway::{
Intents, Shard, ShardId,
};
use super::{Message, Status, SystemEvent, BotConfig};
pub struct Gateway {
shard: Arc<Mutex<Shard>>,
bot_conf: Arc<RwLock<BotConfig>>,
}
impl Gateway {
pub fn new(discord_token: &String, bot_conf: &Arc<RwLock<BotConfig>>) -> Self {
let intents = Intents::GUILD_MEMBERS
| Intents::GUILD_PRESENCES
| Intents::GUILD_MESSAGES
| Intents::MESSAGE_CONTENT;
Self {
shard: Arc::new(Mutex::new(Shard::new(
ShardId::ONE,
discord_token.clone(),
intents,
))),
bot_conf: bot_conf.clone(),
}
}
pub async fn set_status(&self, status: Status) {
{
let last_status = { (*self.bot_conf.read().await).last_status };
if status == last_status {
return
}
}
{
let mut shard = self.shard.lock().await;
shard.command(&UpdatePresence {
d: UpdatePresencePayload {
activities: Vec::new(),
afk: false,
since: None,
status,
},
op: OpCode::PresenceUpdate,
}).await.expect("Could not send command to gateway");
}
self.bot_conf.write().await.last_status = status;
}
pub fn start_listening(&self) {
let bot_conf = self.bot_conf.clone();
let shard = self.shard.clone();
tokio::spawn(async move {
loop {
let bot_conf = { (*bot_conf.read().await).clone() };
let next_event = { shard.lock().await.next_event().await };
let system_channel = bot_conf.system_handler.as_ref().expect("No system channel");
let message_channel = bot_conf.message_handler.as_ref().expect("No message channel");
match next_event {
Err(source) => {
system_channel
.send(SystemEvent::GatewayError(bot_conf.member_id, source.to_string()))
.await;
if source.is_fatal() {
system_channel.send(SystemEvent::GatewayClosed(bot_conf.member_id)).await;
return;
}
}
Ok(event) => match event {
twilight_gateway::Event::Ready(_) => {
system_channel
.send(SystemEvent::GatewayConnected(bot_conf.member_id))
.await;
}
twilight_gateway::Event::MessageCreate(message_create) => {
let message = message_create.0;
if message.author.id != bot_conf.reference_user_id {
continue;
}
message_channel
.send((message.timestamp, Message::Complete(message)))
.await;
}
twilight_gateway::Event::MessageUpdate(message_update) => {
if message_update.author.is_none()
|| message_update.author.as_ref().unwrap().id != bot_conf.reference_user_id
{
continue;
}
if message_update.edited_timestamp.is_none() || message_update.content.is_none() {
continue;
}
message_channel
.send((message_update.edited_timestamp.unwrap(), Message::Partial(*message_update, bot_conf.member_id)))
.await;
}
_ => (),
},
};
}
});
}
}
|
