174 lines
6.2 KiB
Rust
174 lines
6.2 KiB
Rust
use crate::db::reddit::update_data;
|
|
use crate::r#gen::set_status;
|
|
use crate::messages::{make_post_embed, make_removed_embed, EmbedOptions};
|
|
use crate::cmds::reddit::generic_fns::{is_bk_mod, is_bk_mod_serenity, serenity_edit_msg_embed, serenity_send_msg};
|
|
use crate::websocket::send_cmd_json;
|
|
use crate::{Data, Error, lang, rs_println};
|
|
use crate::db::reddit::POSTS_KEY;
|
|
|
|
use poise::serenity_prelude::{self as serenity, ChannelId, ComponentInteraction, Interaction, Member, MessageId, Ready};
|
|
use serde_json::{json, Value};
|
|
|
|
use std::future::Future;
|
|
use std::pin::Pin;
|
|
|
|
|
|
// TODO: clean up
|
|
|
|
|
|
pub fn event_handler<'a>(
|
|
ctx: &'a serenity::Context,
|
|
event: &'a serenity::FullEvent,
|
|
_framework: poise::FrameworkContext<'a, Data, Error>,
|
|
data: &'a Data,
|
|
) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'a>> {
|
|
Box::pin(async move {
|
|
match event {
|
|
serenity::FullEvent::Ready { data_about_bot } => on_ready(ctx, data_about_bot, data).await,
|
|
serenity::FullEvent::InteractionCreate { interaction } => { let _ = handle_buttons(ctx, data, interaction).await; },
|
|
_ => {}
|
|
}
|
|
return Ok(());
|
|
})
|
|
}
|
|
|
|
|
|
async fn on_ready(ctx: &serenity::Context, data_about_bot: &Ready, data: &Data) {
|
|
rs_println!(
|
|
"Bot started as user \"{}\" with id {}",
|
|
data_about_bot.user.name,
|
|
data_about_bot.user.id
|
|
);
|
|
|
|
set_status(data.cfg.clone(), ctx).await;
|
|
}
|
|
|
|
|
|
async fn handle_buttons(ctx: &serenity::Context, data: &Data, interaction: &Interaction) -> Result<(), Error> {
|
|
let component = match interaction {
|
|
Interaction::Component(component) => component,
|
|
_ => return Err(Error::from("Not a message component interaction"))
|
|
};
|
|
|
|
let i_msg = interaction.clone().message_component();
|
|
if i_msg.is_none() { return Err(Error::from("message_component is None!")); }
|
|
let i_embed = i_msg.unwrap().message.embeds[0].clone();
|
|
let url = i_embed.url.clone().unwrap();
|
|
|
|
return match component.data.custom_id.as_str() {
|
|
"approve_btn" => approve_btn(ctx, data, component.member.as_ref().unwrap(), component, url, true).await,
|
|
"remove_btn" => remove_btn (ctx, data, component.member.as_ref().unwrap(), component, url, true).await,
|
|
"unapprove_btn" => approve_btn(ctx, data, component.member.as_ref().unwrap(), component, url, false).await,
|
|
"unremove_btn" => remove_btn (ctx, data, component.member.as_ref().unwrap(), component, url, false).await,
|
|
"unvote_btn" => vote_btn (ctx, data, component.member.as_ref().unwrap(), component, url, false).await,
|
|
"vote_btn" => vote_btn (ctx, data, component.member.as_ref().unwrap(), component, url, true).await,
|
|
_ => Err("Message button with that ID isn't handled.".into())
|
|
}
|
|
}
|
|
|
|
|
|
async fn update_embed(ctx: &serenity::Context, url: &str, new_data: &Value, c_id: &ChannelId, m_id: &MessageId) {
|
|
let remove = new_data["removed"]["removed"].as_bool().unwrap();
|
|
let e: EmbedOptions =
|
|
if remove { make_removed_embed(new_data, url, true) }
|
|
else { make_post_embed (new_data, url, true) };
|
|
|
|
serenity_edit_msg_embed(ctx, c_id, m_id, e).await;
|
|
}
|
|
|
|
|
|
async fn approve_btn(ctx: &serenity::Context, data: &Data, c_member: &Member, component: &ComponentInteraction, url: String, approve: bool) -> Result<(), Error> {
|
|
if !is_bk_mod_serenity(ctx, data, c_member, component).await { return Ok(()); }
|
|
|
|
let r = send_cmd_json("set_approve_post", Some(json!([approve, url])), true).await.unwrap();
|
|
|
|
let c_id = component.channel_id;
|
|
let m_id = component.message.id;
|
|
|
|
update_data().await;
|
|
let new_data = &data.reddit_data
|
|
.lock().await[POSTS_KEY][&url];
|
|
|
|
update_embed(ctx, &url, new_data, &c_id, &m_id).await;
|
|
|
|
if r["value"].as_bool().unwrap() {
|
|
if approve {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_approve_success"), true).await;
|
|
}
|
|
else {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_disapprove_success"), true).await;
|
|
}
|
|
}
|
|
else {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_approve_remove"), true).await;
|
|
}
|
|
|
|
return Ok(());
|
|
}
|
|
|
|
|
|
async fn remove_btn(ctx: &serenity::Context, data: &Data, c_member: &Member, component: &ComponentInteraction, url: String, remove: bool) -> Result<(), Error> {
|
|
if !is_bk_mod_serenity(ctx, data, c_member, component).await { return Ok(()); }
|
|
|
|
let r: Value = if remove {
|
|
send_cmd_json("remove_post_url", Some(json!([&url, &c_member.user.name, None::<String>])), true).await.unwrap()
|
|
}
|
|
else {
|
|
send_cmd_json("add_post_url", Some(json!([&url, false, true])), true).await.unwrap()
|
|
};
|
|
|
|
let c_id = component.channel_id;
|
|
let m_id = component.message.id;
|
|
|
|
update_data().await;
|
|
let new_data = &data.reddit_data
|
|
.lock().await[POSTS_KEY][&url];
|
|
|
|
update_embed(ctx, &url, new_data, &c_id, &m_id).await;
|
|
|
|
if r["value"].as_bool().unwrap() {
|
|
if remove {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_remove_success", &url), true).await;
|
|
}
|
|
else {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_unremove_success", &url), true).await;
|
|
}
|
|
}
|
|
|
|
return Ok(());
|
|
}
|
|
|
|
|
|
async fn vote_btn(ctx: &serenity::Context, data: &Data, c_member: &Member, component: &ComponentInteraction, url: String, vote: bool) -> Result<(), Error> {
|
|
let uid: u64 = c_member.user.id.into();
|
|
let is_mod = is_bk_mod(data.env_vars.reddit_mod_discord_ids.clone(), uid);
|
|
|
|
let r = send_cmd_json("set_vote_post", Some(json!([&url, uid, is_mod, true, !vote])), true).await.unwrap();
|
|
|
|
let c_id = component.channel_id;
|
|
let m_id = component.message.id;
|
|
|
|
update_data().await;
|
|
let new_data = &data.reddit_data
|
|
.lock().await[POSTS_KEY][&url];
|
|
|
|
update_embed(ctx, &url, new_data, &c_id, &m_id).await;
|
|
|
|
if r["value"].as_bool().unwrap() {
|
|
if vote {
|
|
if is_mod { serenity_send_msg(ctx, component, lang!("dc_msg_re_vote_mod_success"), true).await; }
|
|
else { serenity_send_msg(ctx, component, lang!("dc_msg_re_vote_success"), true).await; }
|
|
}
|
|
else {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_vote_remove_success"), true).await;
|
|
}
|
|
}
|
|
else if new_data["removed"]["removed"].as_bool().unwrap() {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_post_vote_removed_post"), true).await;
|
|
}
|
|
else if !vote {
|
|
serenity_send_msg(ctx, component, lang!("dc_msg_re_vote_remove_havent"), true).await;
|
|
}
|
|
|
|
return Ok(());
|
|
} |