101 lines
3.8 KiB
Rust
101 lines
3.8 KiB
Rust
use std::sync::atomic::AtomicBool;
|
|
|
|
use poise::{Event, FrameworkContext};
|
|
use poise::serenity_prelude::model::{
|
|
gateway::Ready,
|
|
};
|
|
use poise::serenity_prelude::{Context, Message};
|
|
use crate::ManifoldData;
|
|
use crate::error::{ManifoldError, ManifoldResult};
|
|
use crate::models::user::UserInfo;
|
|
|
|
pub struct Handler {
|
|
pub timer_running: AtomicBool,
|
|
}
|
|
|
|
impl Handler {
|
|
pub fn new() -> Self {
|
|
Handler {
|
|
timer_running: AtomicBool::from(false)
|
|
}
|
|
}
|
|
|
|
pub async fn listen(ctx: &Context, framework_ctx: FrameworkContext<'_, ManifoldData, ManifoldError>, event: &Event<'_>) -> ManifoldResult<()> {
|
|
match event {
|
|
Event::Ready { data_about_bot} => Handler::standard_startup(&ctx, &framework_ctx, data_about_bot).await,
|
|
Event::Message { new_message } => Handler::message(&ctx, &framework_ctx, &new_message).await,
|
|
Event::MessageUpdate { old_if_available, new, event: _event } => Handler::message_edited(&ctx, &framework_ctx, old_if_available, new).await,
|
|
_ => Ok(())
|
|
}
|
|
}
|
|
|
|
pub async fn standard_startup(ctx: &Context, framework_ctx: &FrameworkContext<'_, ManifoldData, ManifoldError>, data_about_bot: &Ready) -> ManifoldResult<()> {
|
|
let config = &framework_ctx.user_data().await.bot_config;
|
|
let responses = &framework_ctx.user_data().await.responses;
|
|
|
|
|
|
let greeting = match responses.get_response(&"bot startup".to_string()) {
|
|
Some(g) => g.to_owned(),
|
|
None => "Manifold bot connected to discord and ready to begin broadcast operations.".to_string(),
|
|
};
|
|
|
|
for guild in &data_about_bot.guilds {
|
|
match guild.id.edit_nickname(&ctx, Some(&*config.nickname)).await {
|
|
Ok(()) => (),
|
|
Err(e) => {
|
|
error!("Error setting bot nickname (lack permission?): {:?}", e);
|
|
}
|
|
}
|
|
}
|
|
|
|
config.channels.log.say(&ctx, greeting).await.expect("Couldn't message log channel!");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn message(_ctx: &Context, fctx: &FrameworkContext<'_, ManifoldData, ManifoldError>, msg: &Message) -> ManifoldResult<()> {
|
|
let userinfo = &mut fctx.user_data().await.user_info.lock().await;
|
|
|
|
if let Some(u) = userinfo.get_mut(&msg.author.id.as_u64()) {
|
|
u.last_seen = Some(chrono::Utc::now().timestamp());
|
|
} else {
|
|
let new_user = UserInfo {
|
|
user_id: msg.author.id.as_u64().clone() as i64,
|
|
username: msg.author.name.to_owned(),
|
|
weather_location: None,
|
|
weather_units: None,
|
|
timezone: None,
|
|
last_seen: Some(chrono::Utc::now().timestamp()),
|
|
};
|
|
|
|
userinfo.insert(msg.author.id.as_u64().clone(), new_user);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn message_edited(ctx: &Context, fctx: &FrameworkContext<'_, ManifoldData, ManifoldError>, original: &Option<Message>, new_message: &Option<Message>) -> ManifoldResult<()> {
|
|
let log_channel = fctx.user_data().await.bot_config.channels.log;
|
|
|
|
if let Some(new) = new_message.as_ref() {
|
|
log_channel.send_message(ctx, |f| {
|
|
f
|
|
.content("")
|
|
.embed(|e| {
|
|
e
|
|
.title("Message updated")
|
|
.author(|a| a.name(new.author.name.clone()))
|
|
.timestamp(new.timestamp)
|
|
.field("Original Content", match original.as_ref() {
|
|
Some(m) => m.content.clone(),
|
|
None => "Not available".to_string(),
|
|
}, false)
|
|
.field("New Content", new.content.clone(), false)
|
|
})
|
|
}).await?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|