Responding to chat pings
This commit is contained in:
parent
427afa921d
commit
2559ef4452
390
src/chatbot.rs
390
src/chatbot.rs
|
|
@ -6,9 +6,229 @@
|
||||||
use std::{any::Any, future};
|
use std::{any::Any, future};
|
||||||
|
|
||||||
use discord::{
|
use discord::{
|
||||||
model::{Event, Message},
|
model::{ChannelId, Event, Message, ReadyEvent},
|
||||||
Discord,
|
Discord as DDiscord,
|
||||||
};
|
};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
|
const COMMAND_PREFIX: &str = "!";
|
||||||
|
|
||||||
|
struct Discord {
|
||||||
|
discord: DDiscord,
|
||||||
|
connection: Option<Arc<Mutex<discord::Connection>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Discord {
|
||||||
|
pub fn try_new(conf: Arc<Config>) -> Result<Self> {
|
||||||
|
let discord = DDiscord::from_bot_token(
|
||||||
|
conf.discord
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.bot_token
|
||||||
|
.clone()
|
||||||
|
.expose_secret(),
|
||||||
|
)?;
|
||||||
|
Ok(Self {
|
||||||
|
discord,
|
||||||
|
connection: None,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn connect(&mut self) -> Result<ReadyEvent> {
|
||||||
|
if self.connection.is_some() {
|
||||||
|
return Err(anyhow::anyhow!("already connected"));
|
||||||
|
}
|
||||||
|
let (connection, ready_ev) = self.discord.connect()?;
|
||||||
|
self.connection = Some(Arc::new(Mutex::new(connection)));
|
||||||
|
Ok(ready_ev)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn handle_commands(self: Arc<Self>) -> Result<()> {
|
||||||
|
if let Some(conn) = self.connection.clone() {
|
||||||
|
loop {
|
||||||
|
// TODO: I'm suspicious this will just always be locked here and
|
||||||
|
// I'm doing something stupid.
|
||||||
|
let mut conn = conn.lock().await;
|
||||||
|
while let Ok(event) = conn.recv_event() {
|
||||||
|
let rself = self.clone();
|
||||||
|
tokio::spawn(async move {
|
||||||
|
match rself.handle_event(event).await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(err) => {
|
||||||
|
error!("Failed to handle Discord event: {err}")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn handle_event(&self, event: Event) -> Result<()> {
|
||||||
|
match event {
|
||||||
|
Event::Ready(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::Resumed { trace: _ } => return Ok(self.ignore_event(event)),
|
||||||
|
Event::UserUpdate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::UserNoteUpdate(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::UserSettingsUpdate {
|
||||||
|
detect_platform_accounts: _,
|
||||||
|
developer_mode: _,
|
||||||
|
enable_tts_command: _,
|
||||||
|
inline_attachment_media: _,
|
||||||
|
inline_embed_media: _,
|
||||||
|
locale: _,
|
||||||
|
message_display_compact: _,
|
||||||
|
render_embeds: _,
|
||||||
|
server_positions: _,
|
||||||
|
show_current_game: _,
|
||||||
|
status: _,
|
||||||
|
theme: _,
|
||||||
|
convert_emoticons: _,
|
||||||
|
friend_source_flags: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::UserServerSettingsUpdate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::VoiceStateUpdate(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::VoiceServerUpdate {
|
||||||
|
server_id: _,
|
||||||
|
channel_id: _,
|
||||||
|
endpoint: _,
|
||||||
|
token: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::CallCreate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::CallUpdate {
|
||||||
|
channel_id: _,
|
||||||
|
message_id: _,
|
||||||
|
region: _,
|
||||||
|
ringing: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::CallDelete(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelRecipientAdd(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelRecipientRemove(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::TypingStart {
|
||||||
|
channel_id: _,
|
||||||
|
user_id: _,
|
||||||
|
timestamp: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::PresenceUpdate {
|
||||||
|
presence: _,
|
||||||
|
server_id: _,
|
||||||
|
roles: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::PresencesReplace(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::RelationshipAdd(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::RelationshipRemove(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::MessageCreate(msg) => return Ok(self.handle_message(msg)?),
|
||||||
|
Event::MessageUpdate {
|
||||||
|
id: _,
|
||||||
|
channel_id: _,
|
||||||
|
kind: _,
|
||||||
|
content: _,
|
||||||
|
nonce: _,
|
||||||
|
tts: _,
|
||||||
|
pinned: _,
|
||||||
|
timestamp: _,
|
||||||
|
edited_timestamp: _,
|
||||||
|
author: _,
|
||||||
|
mention_everyone: _,
|
||||||
|
mentions: _,
|
||||||
|
mention_roles: _,
|
||||||
|
attachments: _,
|
||||||
|
embeds: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::MessageAck {
|
||||||
|
channel_id: _,
|
||||||
|
message_id: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::MessageDelete {
|
||||||
|
channel_id: _,
|
||||||
|
message_id: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::MessageDeleteBulk {
|
||||||
|
channel_id: _,
|
||||||
|
ids: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerCreate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerUpdate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerDelete(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerMemberAdd(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerMemberUpdate {
|
||||||
|
server_id: _,
|
||||||
|
roles: _,
|
||||||
|
user: _,
|
||||||
|
nick: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerMemberRemove(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerMembersChunk(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerSync {
|
||||||
|
server_id: _,
|
||||||
|
large: _,
|
||||||
|
members: _,
|
||||||
|
presences: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerRoleCreate(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerRoleUpdate(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerRoleDelete(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerBanAdd(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerBanRemove(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerIntegrationsUpdate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ServerEmojisUpdate(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelCreate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelUpdate(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelDelete(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelPinsAck {
|
||||||
|
channel_id: _,
|
||||||
|
timestamp: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ChannelPinsUpdate {
|
||||||
|
channel_id: _,
|
||||||
|
last_pin_timestamp: _,
|
||||||
|
} => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ReactionAdd(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::ReactionRemove(_) => return Ok(self.ignore_event(event)),
|
||||||
|
Event::Unknown(_, _) => return Ok(self.ignore_event(event)),
|
||||||
|
_ => return Ok(self.ignore_event(event)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn ignore_event(&self, event: Event) -> () {
|
||||||
|
let event_type_id = event.type_id();
|
||||||
|
info!("Ignoring Discord event of type: {event_type_id:?}")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn handle_message(&self, msg: Message) -> Result<()> {
|
||||||
|
info!(
|
||||||
|
"Recieved Discord message in channel {} with content: {}",
|
||||||
|
msg.channel_id, msg.content
|
||||||
|
);
|
||||||
|
|
||||||
|
let text = msg.content.clone();
|
||||||
|
let trimmed_text = text.trim();
|
||||||
|
if trimmed_text.starts_with(COMMAND_PREFIX) {
|
||||||
|
self.handle_text_command(trimmed_text, msg)?
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn send_message(&self, channel: ChannelId, text: &str) -> Result<Message> {
|
||||||
|
self.discord
|
||||||
|
.send_message(channel, text, "", false)
|
||||||
|
.map_err(|e| e.into())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_text_command(&self, trimmed_text: &str, msg: Message) -> Result<()> {
|
||||||
|
let trimmed_text = trimmed_text.trim_start_matches(COMMAND_PREFIX);
|
||||||
|
match trimmed_text {
|
||||||
|
s if s.starts_with("ping") => {
|
||||||
|
let _ = self.send_message(msg.channel_id, "`pong`")?;
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
use crate::prelude::*;
|
use crate::prelude::*;
|
||||||
|
|
||||||
|
|
@ -19,167 +239,9 @@ pub async fn start(conf: Arc<Config>) -> Result<()> {
|
||||||
future::pending::<()>().await;
|
future::pending::<()>().await;
|
||||||
}
|
}
|
||||||
|
|
||||||
let discord = Discord::from_bot_token(
|
let mut discord = Discord::try_new(conf.clone())?;
|
||||||
conf.discord
|
let ready_ev = discord.connect().await;
|
||||||
.as_ref()
|
|
||||||
.unwrap()
|
|
||||||
.bot_token
|
|
||||||
.clone()
|
|
||||||
.expose_secret(),
|
|
||||||
)?;
|
|
||||||
|
|
||||||
let (mut conn, ready_ev) = discord.connect()?;
|
|
||||||
info!("Discord connection ready: {ready_ev:?}");
|
info!("Discord connection ready: {ready_ev:?}");
|
||||||
while let Ok(event) = conn.recv_event() {
|
let adiscord = Arc::new(discord);
|
||||||
tokio::spawn(async move {
|
adiscord.handle_commands().await
|
||||||
match handle_event(event).await {
|
|
||||||
Ok(_) => {}
|
|
||||||
Err(err) => {
|
|
||||||
error!("Failed to handle Discord event: {err}")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn handle_event(event: Event) -> Result<()> {
|
|
||||||
match event {
|
|
||||||
Event::Ready(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::Resumed { trace: _ } => return Ok(ignore_event(event)),
|
|
||||||
Event::UserUpdate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::UserNoteUpdate(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::UserSettingsUpdate {
|
|
||||||
detect_platform_accounts: _,
|
|
||||||
developer_mode: _,
|
|
||||||
enable_tts_command: _,
|
|
||||||
inline_attachment_media: _,
|
|
||||||
inline_embed_media: _,
|
|
||||||
locale: _,
|
|
||||||
message_display_compact: _,
|
|
||||||
render_embeds: _,
|
|
||||||
server_positions: _,
|
|
||||||
show_current_game: _,
|
|
||||||
status: _,
|
|
||||||
theme: _,
|
|
||||||
convert_emoticons: _,
|
|
||||||
friend_source_flags: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::UserServerSettingsUpdate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::VoiceStateUpdate(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::VoiceServerUpdate {
|
|
||||||
server_id: _,
|
|
||||||
channel_id: _,
|
|
||||||
endpoint: _,
|
|
||||||
token: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::CallCreate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::CallUpdate {
|
|
||||||
channel_id: _,
|
|
||||||
message_id: _,
|
|
||||||
region: _,
|
|
||||||
ringing: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::CallDelete(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelRecipientAdd(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelRecipientRemove(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::TypingStart {
|
|
||||||
channel_id: _,
|
|
||||||
user_id: _,
|
|
||||||
timestamp: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::PresenceUpdate {
|
|
||||||
presence: _,
|
|
||||||
server_id: _,
|
|
||||||
roles: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::PresencesReplace(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::RelationshipAdd(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::RelationshipRemove(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::MessageCreate(msg) => return Ok(handle_message(msg)),
|
|
||||||
Event::MessageUpdate {
|
|
||||||
id: _,
|
|
||||||
channel_id: _,
|
|
||||||
kind: _,
|
|
||||||
content: _,
|
|
||||||
nonce: _,
|
|
||||||
tts: _,
|
|
||||||
pinned: _,
|
|
||||||
timestamp: _,
|
|
||||||
edited_timestamp: _,
|
|
||||||
author: _,
|
|
||||||
mention_everyone: _,
|
|
||||||
mentions: _,
|
|
||||||
mention_roles: _,
|
|
||||||
attachments: _,
|
|
||||||
embeds: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::MessageAck {
|
|
||||||
channel_id: _,
|
|
||||||
message_id: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::MessageDelete {
|
|
||||||
channel_id: _,
|
|
||||||
message_id: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::MessageDeleteBulk {
|
|
||||||
channel_id: _,
|
|
||||||
ids: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerCreate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerUpdate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerDelete(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerMemberAdd(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerMemberUpdate {
|
|
||||||
server_id: _,
|
|
||||||
roles: _,
|
|
||||||
user: _,
|
|
||||||
nick: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerMemberRemove(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerMembersChunk(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerSync {
|
|
||||||
server_id: _,
|
|
||||||
large: _,
|
|
||||||
members: _,
|
|
||||||
presences: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerRoleCreate(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerRoleUpdate(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerRoleDelete(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerBanAdd(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerBanRemove(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerIntegrationsUpdate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ServerEmojisUpdate(_, _) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelCreate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelUpdate(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelDelete(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelPinsAck {
|
|
||||||
channel_id: _,
|
|
||||||
timestamp: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::ChannelPinsUpdate {
|
|
||||||
channel_id: _,
|
|
||||||
last_pin_timestamp: _,
|
|
||||||
} => return Ok(ignore_event(event)),
|
|
||||||
Event::ReactionAdd(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::ReactionRemove(_) => return Ok(ignore_event(event)),
|
|
||||||
Event::Unknown(_, _) => return Ok(ignore_event(event)),
|
|
||||||
_ => return Ok(ignore_event(event)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn ignore_event(event: Event) -> () {
|
|
||||||
let event_type_id = event.type_id();
|
|
||||||
info!("Ignoring Discord event of type: {event_type_id:?}")
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn handle_message(msg: Message) -> () {
|
|
||||||
info!(
|
|
||||||
"Recieved Discord message in channel {} with content: {}",
|
|
||||||
msg.channel_id, msg.content
|
|
||||||
);
|
|
||||||
|
|
||||||
return ();
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue