76 lines
2.1 KiB
Rust
76 lines
2.1 KiB
Rust
use phoebe::{mid_chat::*, prelude::*};
|
|
use serenity::{
|
|
client::{Context, EventHandler},
|
|
model::prelude::*,
|
|
};
|
|
use tracing::debug;
|
|
|
|
use crate::discord_reference;
|
|
|
|
pub struct DiscordHandler {
|
|
pub core_db: SqlitePool,
|
|
pub discord_media_db: SqlitePool,
|
|
pub chat_event_tx: ChatEventSender,
|
|
pub ctx_tx: tokio::sync::mpsc::UnboundedSender<Context>,
|
|
}
|
|
|
|
impl DiscordHandler {
|
|
async fn get_author(&self, ctx: &Context, message: &Message) -> ChatAuthor {
|
|
let display_name = message
|
|
.author_nick(ctx)
|
|
.await
|
|
.unwrap_or_else(|| message.author.name.clone());
|
|
|
|
async fn tag_color(ctx: &Context, message: &Message) -> Option<[u8; 3]> {
|
|
let color = message.member(ctx).await.ok()?.colour(ctx).await?;
|
|
Some([color.r(), color.b(), color.g()])
|
|
}
|
|
|
|
let display_color = tag_color(ctx, message).await;
|
|
|
|
ChatAuthor {
|
|
reference: discord_reference(message.author.id),
|
|
display_name,
|
|
display_color,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl EventHandler for DiscordHandler {
|
|
async fn ready(&self, ctx: Context, _ready: Ready) {
|
|
debug!("Discord signalled ready!");
|
|
let _ = self.ctx_tx.send(ctx);
|
|
}
|
|
|
|
async fn message(&self, ctx: Context, message: Message) {
|
|
let origin = ChatReference {
|
|
service: "discord",
|
|
id: message.id.to_string(),
|
|
};
|
|
|
|
let author = self.get_author(&ctx, &message).await;
|
|
|
|
let content = discord_message_format::parse(&message.content);
|
|
let content = super::chat_conv::convert(&content);
|
|
|
|
let replies_to = message.referenced_message.as_ref().map(|m| ChatReference {
|
|
service: "discord",
|
|
id: m.id.to_string(),
|
|
});
|
|
|
|
let chat_message = ChatMessage {
|
|
origin,
|
|
author,
|
|
content,
|
|
attachments: vec![],
|
|
replying: replies_to,
|
|
};
|
|
|
|
let _ = self
|
|
.chat_event_tx
|
|
.send(ChatEvent::NewMessage(chat_message))
|
|
.expect("Failed to dispatch incoming Discord chat message");
|
|
}
|
|
}
|