Initial async update commit

+ Naive port of all code to async serenity
+ Limited testing has been done, but the bot seems to work
+ Removed WhiteRabbit dependency (does not work in async)
+ Further cleanup needed
async_update
Joey Hines 2021-01-09 15:45:16 -06:00
parent 47a1bc9a2a
commit 30177474e1
No known key found for this signature in database
GPG Key ID: 80F567B5C968F91B
7 changed files with 550 additions and 1014 deletions

4
.gitignore vendored
View File

@ -1,4 +1,6 @@
/target /target
**/*.rs.bk **/*.rs.bk
.env .env
log/* log/
config.toml
.idea

1173
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@ -1,6 +1,6 @@
[package] [package]
name = "hype_bot" name = "hype_bot"
version = "0.1.0" version = "0.2.0"
authors = ["Joey Hines <joey@ahines.net>"] authors = ["Joey Hines <joey@ahines.net>"]
edition = "2018" edition = "2018"
@ -19,8 +19,8 @@ log = "0.4.8"
log4rs = "0.11.0" log4rs = "0.11.0"
strfmt = "0.1.6" strfmt = "0.1.6"
url = "2.1.1" url = "2.1.1"
white_rabbit = "0.1.1" tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] }
[dependencies.serenity] [dependencies.serenity]
version = "0.8.6" version = "0.10.2"
features = ["framework", "standard_framework"] features = ["framework", "standard_framework", "utils", "builder"]

View File

@ -1,6 +1,6 @@
MIT License MIT License
Copyright (c) 2020 ZeroHD Copyright (c) 2020-2021 ZeroHD
Permission is hereby granted, free of charge, to any person obtaining a copy Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal of this software and associated documentation files (the "Software"), to deal

View File

@ -20,25 +20,25 @@ use url::Url;
/// ///
/// **Note** /// **Note**
/// You can only post events you have created. Only one preview event can exist at a time. /// You can only post events you have created. Only one preview event can exist at a time.
fn confirm(ctx: &mut Context, msg: &Message, _args: Args) -> CommandResult { async fn confirm(ctx: &Context, msg: &Message, _args: Args) -> CommandResult {
let config = get_config(&ctx.data)?; let config = get_config(&ctx.data).await?;
let draft_event = get_draft_event(&ctx.data)?; let draft_event = get_draft_event(&ctx.data).await?;
let mut new_event = draft_event.event.clone(); let mut new_event = draft_event.event.clone();
// Check to to see if message author is the owner of the pending event // Check to to see if message author is the owner of the pending event
if draft_event.creator_id == msg.author.id.0 { if draft_event.creator_id == msg.author.id.0 {
// Send event message // Send event message
let event_msg = send_event_msg(&ctx.http, &config, config.event_channel, &new_event, true)?; let event_msg = send_event_msg(&ctx, &config, config.event_channel, &new_event, true).await?;
msg.reply(&ctx, "Event posted!")?; msg.reply(&ctx, "Event posted!").await?;
new_event.message_id = event_msg.id.0.to_string(); new_event.message_id = event_msg.id.0.to_string();
let event = insert_event(config.db_url.clone(), &new_event)?; let event = insert_event(config.db_url.clone(), &new_event)?;
schedule_event(&ctx.http, &ctx.data, &event); schedule_event(&ctx, &event).await;
} else { } else {
msg.reply(&ctx, format!("You do not have a pending event!"))?; msg.reply(&ctx, format!("You do not have a pending event!")).await?;
} }
Ok(()) Ok(())
@ -57,29 +57,29 @@ fn confirm(ctx: &mut Context, msg: &Message, _args: Args) -> CommandResult {
/// ///
/// **Organizer** /// **Organizer**
/// The user or group that is organizing the event, defaults to the user creating the event /// The user or group that is organizing the event, defaults to the user creating the event
fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult { async fn create(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
// Get config // Get config
let config = get_config(&ctx.data)?; let config = get_config(&ctx.data).await?;
// Parse args // Parse args
let event_name = match args.find::<String>() { let event_name = match args.find::<String>() {
Ok(event_name) => event_name.replace("\"", ""), Ok(event_name) => event_name.replace("\"", ""),
Err(_) => { Err(_) => {
msg.reply(&ctx, "No event name provided.".to_string())?; msg.reply(&ctx, "No event name provided.".to_string()).await?;
return Ok(()); return Ok(());
} }
}; };
let date_string = match args.find::<String>() { let date_string = match args.find::<String>() {
Ok(date_string) => date_string.replace("\"", ""), Ok(date_string) => date_string.replace("\"", ""),
Err(_) => { Err(_) => {
msg.reply(&ctx, "No date provided.".to_string())?; msg.reply(&ctx, "No date provided.".to_string()).await?;
return Ok(()); return Ok(());
} }
}; };
let description = match args.find::<String>() { let description = match args.find::<String>() {
Ok(desc) => desc.replace("\"", ""), Ok(desc) => desc.replace("\"", ""),
Err(_) => { Err(_) => {
msg.reply(&ctx, "No description provided.".to_string())?; msg.reply(&ctx, "No description provided.".to_string()).await?;
return Ok(()); return Ok(());
} }
}; };
@ -87,7 +87,7 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
let location = match args.find::<String>() { let location = match args.find::<String>() {
Ok(desc) => desc.replace("\"", ""), Ok(desc) => desc.replace("\"", ""),
Err(_) => { Err(_) => {
msg.reply(&ctx, "No location provided.".to_string())?; msg.reply(&ctx, "No location provided.".to_string()).await?;
return Ok(()); return Ok(());
} }
}; };
@ -99,7 +99,7 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
let organizer = match args.find::<String>() { let organizer = match args.find::<String>() {
Ok(link) => link.replace("\"", ""), Ok(link) => link.replace("\"", ""),
Err(_) => msg.author.mention(), Err(_) => msg.author.mention().to_string(),
}; };
// Parse date // Parse date
@ -110,7 +110,7 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
msg.reply( msg.reply(
&ctx, &ctx,
"Invalid date format. Format is HH:MMam YYYY-MM-DD".to_string(), "Invalid date format. Format is HH:MMam YYYY-MM-DD".to_string(),
)?; ).await?;
return Ok(()); return Ok(());
} }
}; };
@ -130,7 +130,7 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
let event_time = input_date.with_timezone(&Utc).naive_utc(); let event_time = input_date.with_timezone(&Utc).naive_utc();
if Utc::now().naive_utc() > event_time { if Utc::now().naive_utc() > event_time {
msg.reply(&ctx, "The scheduled time has already passed!")?; msg.reply(&ctx, "The scheduled time has already passed!").await?;
return Ok(()); return Ok(());
} }
@ -141,10 +141,10 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
.clean_user(false) .clean_user(false)
.clean_everyone(true); .clean_everyone(true);
let description = content_safe(&ctx.cache, description, &settings); let description = content_safe(&ctx.cache, description, &settings).await;
let event_name = content_safe(&ctx.cache, event_name, &settings); let event_name = content_safe(&ctx.cache, event_name, &settings).await;
let location = content_safe(&ctx.cache, location, &settings); let location = content_safe(&ctx.cache, location, &settings).await;
let organizer = content_safe(&ctx.cache, organizer, &settings); let organizer = content_safe(&ctx.cache, organizer, &settings).await;
update_draft_event( update_draft_event(
&ctx, &ctx,
@ -155,8 +155,8 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
thumbnail_link, thumbnail_link,
event_time, event_time,
msg.author.id.0, msg.author.id.0,
)?; ).await?;
send_draft_event(&ctx, msg.channel_id)?; send_draft_event(&ctx, msg.channel_id).await?;
Ok(()) Ok(())
} }
@ -165,31 +165,32 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
/// Cancels an already scheduled event /// Cancels an already scheduled event
/// ///
/// `~cancel "event name"` /// `~cancel "event name"`
fn cancel(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult { async fn cancel(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
let config = get_config(&ctx.data)?; let config = get_config(&ctx.data).await?;
// Parse args // Parse args
let event_name = args.single::<String>()?.replace("\"", ""); let event_name = args.single::<String>()?.replace("\"", "");
let event = get_event_by_name(config.db_url.clone(), event_name)?; let event = get_event_by_name(config.db_url.clone(), event_name)?;
let message_id = event.message_id.parse::<u64>()?; let message_id = event.message_id.parse::<u64>()?;
let message = ctx.http.get_message(config.event_channel, message_id)?; let message = ctx.http.get_message(config.event_channel, message_id).await?;
let reaction_users = message let reaction_users = message
.reaction_users(&ctx.http, INTERESTED_EMOJI, None, None) .reaction_users(&ctx.http, INTERESTED_EMOJI, None, None)
.await
.unwrap_or(Vec::<User>::new()); .unwrap_or(Vec::<User>::new());
let cancel_msg = format!("**{}** has been canceled!", event.event_name.clone()); let cancel_msg = format!("**{}** has been canceled!", event.event_name.clone());
for user in reaction_users { for user in reaction_users {
send_dm_message(&ctx.http, user, &cancel_msg); send_dm_message(ctx, user, &cancel_msg).await;
} }
remove_event(config.db_url.clone(), event.id)?; remove_event(config.db_url.clone(), event.id)?;
message.delete(&ctx)?; message.delete(&ctx).await?;
msg.reply(&ctx, &cancel_msg)?; msg.reply(&ctx, &cancel_msg).await?;
Ok(()) Ok(())
} }

View File

@ -1,19 +1,20 @@
use crate::database::models::{Event, NewEvent}; use crate::database::models::{Event, NewEvent};
use crate::database::{get_event_by_msg_id, remove_event, set_reminder}; use crate::database::{get_event_by_msg_id, remove_event};
use crate::hypebot_config::HypeBotConfig; use crate::hypebot_config::HypeBotConfig;
use crate::{INTERESTED_EMOJI, UNINTERESTED_EMOJI}; use crate::{INTERESTED_EMOJI, UNINTERESTED_EMOJI};
use chrono::{DateTime, NaiveDateTime, Utc}; use chrono::{DateTime, NaiveDateTime, Utc};
use serenity::framework::standard::{CommandError, CommandResult}; use serenity::framework::standard::{CommandError, CommandResult};
use serenity::http::Http; use serenity::http::Http;
use serenity::model::prelude::{ChannelId, Message, Reaction, User}; use serenity::model::prelude::{ChannelId, Message, Reaction, User};
use serenity::prelude::TypeMapKey; use serenity::prelude::{TypeMapKey, TypeMap};
use serenity::prelude::{Context, RwLock, ShareMap}; use serenity::prelude::{Context, RwLock};
use serenity::utils::Colour; use serenity::utils::Colour;
use serenity::Result; use serenity::Result;
use serenity::framework::standard::macros::hook;
use std::collections::HashMap; use std::collections::HashMap;
use std::sync::Arc; use std::sync::Arc;
use strfmt::strfmt; use strfmt::strfmt;
use white_rabbit::{DateResult, Scheduler}; use diesel::result::Error;
pub mod events; pub mod events;
@ -28,22 +29,18 @@ impl TypeMapKey for DraftEvent {
type Value = DraftEvent; type Value = DraftEvent;
} }
pub struct SchedulerKey;
impl TypeMapKey for SchedulerKey {
type Value = Arc<RwLock<Scheduler>>;
}
/// Send a message to a reaction user /// Send a message to a reaction user
pub fn send_message_to_reaction_users(ctx: &Context, reaction: &Reaction, msg_text: &str) { pub async fn send_message_to_reaction_users(ctx: &Context, reaction: &Reaction, msg_text: &str) {
if let Ok(config) = get_config(&ctx.data) { if let Ok(config) = get_config(&ctx.data).await {
let db_link = config.db_url.clone(); let db_link = config.db_url.clone();
let message_id = reaction.message_id.0.to_string(); let message_id = reaction.message_id.0.to_string();
let event = match get_event_by_msg_id(db_link, message_id) { let event = match get_event_by_msg_id(db_link, message_id) {
Ok(event) => event, Ok(event) => event,
Err(e) => { Err(e) => {
error!("Error getting event from reaction {}", e); if !matches!(e, diesel::result::Error::NotFound) {
error!("Error getting event from reaction {}", e);
}
return; return;
} }
}; };
@ -62,16 +59,16 @@ pub fn send_message_to_reaction_users(ctx: &Context, reaction: &Reaction, msg_te
msg = format!("**{}** has already started!", &event.event_name) msg = format!("**{}** has already started!", &event.event_name)
} }
if let Ok(user) = reaction.user(&ctx.http) { if let Ok(user) = reaction.user(&ctx).await {
send_dm_message(&ctx.http, user, &msg); send_dm_message(&ctx, user, &msg).await;
} }
} }
} }
/// Send a DM message to a user /// Send a DM message to a user
pub fn send_dm_message(http: &Http, user: User, message: &String) { pub async fn send_dm_message(ctx: &Context, user: User, message: &String) {
if let Ok(dm_channel) = user.create_dm_channel(&http) { if let Ok(dm_channel) = user.create_dm_channel(ctx).await {
dm_channel.send_message(&http, |m| m.content(message)).ok(); dm_channel.send_message(ctx, |m| m.content(message)).await.ok();
} }
} }
@ -87,14 +84,14 @@ pub fn get_countdown_link(event_name: &String, utc: &DateTime<Utc>) -> String {
} }
/// Sends the event message to the event channel /// Sends the event message to the event channel
pub fn send_event_msg( pub async fn send_event_msg(
http: &Http, ctx: &Context,
config: &HypeBotConfig, config: &HypeBotConfig,
channel_id: u64, channel_id: u64,
event: &NewEvent, event: &NewEvent,
react: bool, react: bool,
) -> Result<Message> { ) -> Result<Message> {
let channel = http.get_channel(channel_id)?; let channel = ctx.http.get_channel(channel_id).await?;
let utc_time = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc); let utc_time = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
@ -109,7 +106,7 @@ pub fn send_event_msg(
.join(" "); .join(" ");
// Send message // Send message
let msg = channel.id().send_message(&http, |m| { let msg = channel.id().send_message(&ctx, |m| {
m.embed(|e| { m.embed(|e| {
e.title(event.event_name.clone()) e.title(event.event_name.clone())
.color(Colour::PURPLE) .color(Colour::PURPLE)
@ -126,19 +123,19 @@ pub fn send_event_msg(
.field("Location", &event.event_loc, true) .field("Location", &event.event_loc, true)
.field("Organizer", &event.organizer, true) .field("Organizer", &event.organizer, true)
}).content(ping_roles) }).content(ping_roles)
})?; }).await?;
if react { if react {
// Add reacts // Add reacts
msg.react(http, INTERESTED_EMOJI)?; msg.react(ctx, INTERESTED_EMOJI).await?;
msg.react(http, UNINTERESTED_EMOJI)?; msg.react(ctx, UNINTERESTED_EMOJI).await?;
} }
Ok(msg) Ok(msg)
} }
/// Updates the draft event stored in the context data /// Updates the draft event stored in the context data
pub fn update_draft_event( pub async fn update_draft_event(
ctx: &Context, ctx: &Context,
event_name: String, event_name: String,
event_desc: String, event_desc: String,
@ -148,10 +145,10 @@ pub fn update_draft_event(
event_time: NaiveDateTime, event_time: NaiveDateTime,
creator_id: u64, creator_id: u64,
) -> CommandResult { ) -> CommandResult {
let mut data = ctx.data.write(); let mut data = ctx.data.write().await;
let mut draft_event = data let mut draft_event = data
.get_mut::<DraftEvent>() .get_mut::<DraftEvent>()
.ok_or(CommandError("Unable get draft event!".to_string()))?; .ok_or(CommandError::from("Unable get draft event!".to_string()))?;
draft_event.event.event_name = event_name; draft_event.event.event_name = event_name;
draft_event.event.event_desc = event_desc; draft_event.event.event_desc = event_desc;
@ -165,62 +162,52 @@ pub fn update_draft_event(
} }
/// Sends the draft event stored in the context data /// Sends the draft event stored in the context data
pub fn send_draft_event(ctx: &Context, channel: ChannelId) -> CommandResult { pub async fn send_draft_event(ctx: &Context, channel: ChannelId) -> CommandResult {
let data = ctx.data.read(); let data = ctx.data.read().await;
let config = data let config = data
.get::<HypeBotConfig>() .get::<HypeBotConfig>()
.ok_or(CommandError("Config not found!".to_string()))?; .ok_or(CommandError::from("Config not found!".to_string()))?;
let draft_event = data let draft_event = data
.get::<DraftEvent>() .get::<DraftEvent>()
.ok_or(CommandError("Draft event not found!".to_string()))?; .ok_or(CommandError::from("Draft event not found!".to_string()))?;
channel.send_message(&ctx, |m| { channel.send_message(&ctx, |m| {
m.content(format!( m.content(format!(
"Draft message, use the `confirm` command to post it." "Draft message, use the `confirm` command to post it."
)) ))
})?; }).await?;
send_event_msg(&ctx.http, config, channel.0, &draft_event.event, false)?; send_event_msg(ctx, config, channel.0, &draft_event.event, false).await?;
Ok(()) Ok(())
} }
/// Gets the config from context data /// Gets the config from context data
pub fn get_config( pub async fn get_config(
data: &Arc<RwLock<ShareMap>>, data: &Arc<RwLock<TypeMap>>,
) -> std::result::Result<HypeBotConfig, CommandError> { ) -> std::result::Result<HypeBotConfig, CommandError> {
let data_read = data.read(); let data_read = data.read().await;
let config = data_read let config = data_read
.get::<HypeBotConfig>() .get::<HypeBotConfig>()
.ok_or(CommandError("Unable to get config".to_string()))?; .ok_or(CommandError::from("Unable to get config".to_string()))?;
Ok(config.clone()) Ok(config.clone())
} }
/// Gets the draft event from context data /// Gets the draft event from context data
pub fn get_draft_event( pub async fn get_draft_event(
data: &Arc<RwLock<ShareMap>>, data: &Arc<RwLock<TypeMap>>,
) -> std::result::Result<DraftEvent, CommandError> { ) -> std::result::Result<DraftEvent, CommandError> {
let data_read = data.read(); let data_read = data.read().await;
let draft_event = data_read let draft_event = data_read
.get::<DraftEvent>() .get::<DraftEvent>()
.ok_or(CommandError("Unable to queued event".to_string()))?; .ok_or(CommandError::from("Unable to queued event".to_string()))?;
Ok(draft_event.clone()) Ok(draft_event.clone())
} }
/// Get the scheduler
pub fn get_scheduler(
data: &Arc<RwLock<ShareMap>>,
) -> std::result::Result<Arc<RwLock<Scheduler>>, CommandError> {
let mut context = data.write();
Ok(context
.get_mut::<SchedulerKey>()
.ok_or(CommandError("Unable to get scheduler".to_string()))?
.clone())
}
/// Logs command errors to the logger /// Logs command errors to the logger
pub fn log_error( #[hook]
_ctx: &mut Context, pub async fn log_error(
_ctx: &Context,
_msg: &Message, _msg: &Message,
command_name: &str, command_name: &str,
result: std::result::Result<(), CommandError>, result: std::result::Result<(), CommandError>,
@ -232,19 +219,17 @@ pub fn log_error(
} }
/// Checks if the user has permission to use this bot /// Checks if the user has permission to use this bot
pub fn permission_check(ctx: &mut Context, msg: &Message, _command_name: &str) -> bool { #[hook]
pub async fn permission_check(ctx: &Context, msg: &Message, _command_name: &str) -> bool {
if let Some(guild_id) = msg.guild_id { if let Some(guild_id) = msg.guild_id {
if let Ok(config) = get_config(&ctx.data) { if let Ok(config) = get_config(&ctx.data).await {
if let Ok(roles) = ctx.http.get_guild_roles(guild_id.0) { if let Ok(roles) = ctx.http.get_guild_roles(guild_id.0).await {
for role in roles { for role in roles {
if config.event_roles.contains(&role.id.0) { if config.event_roles.contains(&role.id.0) {
let has_role = match msg.author.has_role(&ctx, guild_id, role) { return match msg.author.has_role(&ctx, guild_id, role).await {
Ok(has_role) => has_role, Ok(has_role) => has_role,
Err(_) => false, Err(_) => false,
}; };
if has_role {
return true;
}
} }
} }
} }
@ -255,63 +240,48 @@ pub fn permission_check(ctx: &mut Context, msg: &Message, _command_name: &str) -
} }
/// Schedule event reminders /// Schedule event reminders
pub fn schedule_event(http: &Arc<Http>, data: &Arc<RwLock<ShareMap>>, event: &Event) { pub async fn schedule_event(ctx: &Context, event: &Event) {
let scheduler = { let config = get_config(&ctx.data).await.expect("Unable to get config");
let mut context = data.write();
context
.get_mut::<SchedulerKey>()
.expect("Expected Scheduler.")
.clone()
};
let config = get_config(&data).unwrap();
if let Some(reminders) = config.reminders { if let Some(reminders) = config.reminders {
let event_time: DateTime<Utc> = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc); let event_time: DateTime<Utc> = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
let delete_time = event_time + chrono::Duration::minutes(1);
let mut scheduler = scheduler.write();
for reminder in reminders { for reminder in reminders {
let reminder_time = let reminder_time =
event_time - chrono::Duration::minutes(reminder.reminder_time as i64); event_time - chrono::Duration::minutes(reminder.reminder_time as i64);
if reminder_time > chrono::offset::Utc::now() { if reminder_time > chrono::offset::Utc::now() {
let ctx = ctx.clone();
let reminder_msg = reminder.msg.clone(); let reminder_msg = reminder.msg.clone();
let http = http.clone();
let data = data.clone();
let event = event.clone(); let event = event.clone();
scheduler.add_task_datetime(reminder_time, move |_| { tokio::task::spawn(async move {
send_reminders(&http, &data, &event, &reminder_msg) send_reminders_task(&ctx, reminder_time, &event, &reminder_msg).await;
}); }
);
} }
} }
let http = http.clone();
let data = data.clone();
let event = event.clone();
scheduler.add_task_datetime(delete_time, move |_| {
delete_event(&http, &data, &event);
DateResult::Done
});
} }
} }
/// Send reminders /// Send reminders
pub fn send_reminders( pub async fn send_reminders_task(
http: &Arc<Http>, ctx: &Context,
data: &Arc<RwLock<ShareMap>>, datetime: DateTime<Utc>,
event: &Event, event: &Event,
reminder_msg: &String, reminder_msg: &String,
) -> DateResult { ) {
let config = get_config(&data).unwrap(); let duration = datetime - chrono::offset::Utc::now();
tokio::time::sleep(tokio::time::Duration::from_millis(duration.num_milliseconds() as u64)).await;
let config = get_config(&ctx.data).await.expect("Unable to get config");
let event_channel_id = config.event_channel; let event_channel_id = config.event_channel;
if let Ok(message_id) = event.message_id.parse::<u64>() { if let Ok(message_id) = event.message_id.parse::<u64>() {
// Get message id // Get message id
if let Ok(message) = http.get_message(event_channel_id, message_id) { if let Ok(message) = ctx.http.get_message(event_channel_id, message_id).await {
let reaction_users = message let reaction_users = message
.reaction_users(&http, INTERESTED_EMOJI, None, None) .reaction_users(ctx, INTERESTED_EMOJI, None, None)
.await
.unwrap_or(Vec::<User>::new()); .unwrap_or(Vec::<User>::new());
// Build reminder message // Build reminder message
@ -319,20 +289,18 @@ pub fn send_reminders(
// Send reminder to each reacted user // Send reminder to each reacted user
for user in reaction_users { for user in reaction_users {
send_dm_message(&http, user, &msg); send_dm_message(ctx, user, &msg).await;
} }
} }
} }
DateResult::Done
} }
/// Delete event /// Delete event
pub fn delete_event(http: &Arc<Http>, data: &Arc<RwLock<ShareMap>>, event: &Event) { pub async fn delete_event(http: &Arc<Http>, data: &Arc<RwLock<TypeMap>>, event: &Event) {
let config = get_config(&data).unwrap(); let config = get_config(&data).await.expect("Unable to get config");
remove_event(config.db_url.clone(), event.id).ok(); remove_event(config.db_url.clone(), event.id).ok();
if let Ok(message_id) = event.message_id.parse::<u64>() { if let Ok(message_id) = event.message_id.parse::<u64>() {
http.delete_message(config.event_channel, message_id).ok(); http.delete_message(config.event_channel, message_id).await.ok();
} }
} }

View File

@ -2,52 +2,52 @@
extern crate diesel; extern crate diesel;
#[macro_use] #[macro_use]
extern crate diesel_migrations; extern crate diesel_migrations;
extern crate serde;
#[macro_use] #[macro_use]
extern crate log; extern crate log;
extern crate log4rs; extern crate log4rs;
extern crate serde;
use std::collections::HashSet;
use std::path::Path;
use std::process::exit;
use chrono::{DateTime, Utc}; use chrono::{DateTime, Utc};
use clap::{App, Arg}; use clap::{App, Arg};
use log::LevelFilter;
use log4rs::append::console::ConsoleAppender; use log4rs::append::console::ConsoleAppender;
use log4rs::append::rolling_file::policy::compound::CompoundPolicy;
use log4rs::append::rolling_file::policy::compound::roll::fixed_window::FixedWindowRoller; use log4rs::append::rolling_file::policy::compound::roll::fixed_window::FixedWindowRoller;
use log4rs::append::rolling_file::policy::compound::trigger::size::SizeTrigger; use log4rs::append::rolling_file::policy::compound::trigger::size::SizeTrigger;
use log4rs::append::rolling_file::policy::compound::CompoundPolicy;
use log4rs::append::rolling_file::RollingFileAppender; use log4rs::append::rolling_file::RollingFileAppender;
use log4rs::config::{Appender, Config, Root}; use log4rs::config::{Appender, Config, Root};
use log4rs::encode::pattern::PatternEncoder; use log4rs::encode::pattern::PatternEncoder;
use log4rs::filter::threshold::ThresholdFilter; use log4rs::filter::threshold::ThresholdFilter;
use log4rs::init_config; use log4rs::init_config;
use log::LevelFilter;
use serenity::client::Client; use serenity::client::Client;
use serenity::framework::standard::macros::{group, help}; use serenity::async_trait;
use serenity::framework::standard::{help_commands, Args, CommandGroup, HelpOptions}; use serenity::framework::standard::{Args, CommandGroup, help_commands, HelpOptions};
use serenity::framework::standard::{CommandResult, StandardFramework}; use serenity::framework::standard::{CommandResult, StandardFramework};
use serenity::framework::standard::macros::{group, help};
use serenity::model::channel::{Message, Reaction}; use serenity::model::channel::{Message, Reaction};
use serenity::model::id::UserId; use serenity::model::id::UserId;
use serenity::model::prelude::Ready; use serenity::model::prelude::Ready;
use serenity::prelude::{Context, EventHandler, RwLock}; use serenity::prelude::{Context, EventHandler};
use std::collections::HashSet;
use std::path::Path; use database::*;
use std::process::exit; use database::models::NewEvent;
use std::sync::Arc; use discord::{
use white_rabbit::{DateResult, Scheduler}; delete_event, DraftEvent, get_config, log_error, permission_check,
schedule_event, send_message_to_reaction_users,
};
use discord::events::{CANCEL_COMMAND, CONFIRM_COMMAND, CREATE_COMMAND};
use hypebot_config::HypeBotConfig;
mod database; mod database;
mod discord; mod discord;
mod hypebot_config; mod hypebot_config;
use database::models::NewEvent; const INTERESTED_EMOJI: char = '\u{2705}';
use database::*; const UNINTERESTED_EMOJI: char = '\u{274C}';
use discord::events::{CANCEL_COMMAND, CONFIRM_COMMAND, CREATE_COMMAND};
use discord::{
delete_event, get_config, get_scheduler, log_error, permission_check, schedule_event,
send_message_to_reaction_users, DraftEvent, SchedulerKey,
};
use hypebot_config::HypeBotConfig;
const INTERESTED_EMOJI: &str = "\u{2705}";
const UNINTERESTED_EMOJI: &str = "\u{274C}";
type HypeBotResult<T> = std::result::Result<T, Box<dyn std::error::Error>>; type HypeBotResult<T> = std::result::Result<T, Box<dyn std::error::Error>>;
@ -61,65 +61,78 @@ struct EventCommands;
/// Handler for Discord events /// Handler for Discord events
struct Handler; struct Handler;
#[async_trait]
impl EventHandler for Handler { impl EventHandler for Handler {
/// On reaction add /// On reaction add
fn reaction_add(&self, ctx: Context, reaction: Reaction) { async fn reaction_add(&self, ctx: Context, reaction: Reaction) {
let config = match get_config(&ctx.data) { let config = match get_config(&ctx.data).await {
Ok(config) => config, Ok(config) => config,
Err(e) => { Err(e) => {
error!("Unable to get config: {}", e.0); error!("Unable to get config: {}", e);
return; return;
} }
}; };
if reaction.channel_id.0 == config.event_channel if reaction.channel_id.0 == config.event_channel
&& reaction.emoji.as_data() == INTERESTED_EMOJI && reaction.emoji.as_data().chars().next().unwrap() == INTERESTED_EMOJI
{ {
send_message_to_reaction_users( send_message_to_reaction_users(
&ctx, &ctx,
&reaction, &reaction,
"Hello, you are now receiving reminders for **{event}**", "Hello, you are now receiving reminders for **{event}**",
); ).await;
} }
} }
/// On reaction remove /// On reaction remove
fn reaction_remove(&self, ctx: Context, reaction: Reaction) { async fn reaction_remove(&self, ctx: Context, reaction: Reaction) {
let config = match get_config(&ctx.data) { let config = match get_config(&ctx.data).await {
Ok(config) => config, Ok(config) => config,
Err(e) => { Err(e) => {
error!("Unable to get config: {}", e.0); error!("Unable to get config: {}", e);
return; return;
} }
}; };
if reaction.channel_id.0 == config.event_channel if reaction.channel_id.0 == config.event_channel
&& reaction.emoji.as_data() == INTERESTED_EMOJI && reaction.emoji.as_data().chars().next().unwrap() == INTERESTED_EMOJI
{ {
send_message_to_reaction_users( send_message_to_reaction_users(
&ctx, &ctx,
&reaction, &reaction,
"Hello, you are no longer receiving reminders for **{event}**", "Hello, you are no longer receiving reminders for **{event}**",
); ).await;
} }
} }
/// On bot ready /// On bot ready
fn ready(&self, _: Context, ready: Ready) { async fn ready(&self, ctx: Context, ready: Ready) {
info!("Connected to Discord as {}", ready.user.name); info!("Connected to Discord as {}", ready.user.name);
// Schedule current events
let config = get_config(&ctx.data).await.expect("Unable to get config");
for event in get_all_events(config.db_url.clone()).unwrap() {
let event_time: DateTime<Utc> =
DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
if event.reminder_sent == 0 {
schedule_event(&ctx, &event).await;
}
}
} }
} }
#[help] #[help]
#[command_not_found_text = "Could not find: `{}`."] #[command_not_found_text = "Could not find: `{}`."]
#[strikethrough_commands_tip_in_guild("HypeBot")] #[strikethrough_commands_tip_in_guild("")]
fn bot_help( async fn bot_help(
context: &mut Context, context: &Context,
msg: &Message, msg: &Message,
args: Args, args: Args,
help_options: &'static HelpOptions, help_options: &'static HelpOptions,
groups: &[&'static CommandGroup], groups: &[&'static CommandGroup],
owners: HashSet<UserId>, owners: HashSet<UserId>,
) -> CommandResult { ) -> CommandResult {
help_commands::with_embeds(context, msg, args, help_options, groups, owners) help_commands::with_embeds(context, msg, args, help_options, groups, owners).await;
Ok(())
} }
/// Does the setup for logging /// Does the setup for logging
@ -181,7 +194,8 @@ fn setup_logging(config: &HypeBotConfig) -> HypeBotResult<()> {
} }
embed_migrations!("migrations/"); embed_migrations!("migrations/");
fn main() -> HypeBotResult<()> { #[tokio::main]
async fn main() -> HypeBotResult<()> {
// Initialize arg parser // Initialize arg parser
let mut app = App::new("Hype Bot") let mut app = App::new("Hype Bot")
.about("Hype Bot: Hype Up Your Discord Events!") .about("Hype Bot: Hype Up Your Discord Events!")
@ -217,26 +231,23 @@ fn main() -> HypeBotResult<()> {
// New client // New client
let mut client = let mut client =
Client::new(cfg.discord_key.clone(), Handler).expect("Error creating client"); Client::builder(cfg.discord_key.clone()).event_handler(Handler).framework(
StandardFramework::new()
// Configure client .configure(|c| {
client.with_framework( c.prefix(cfg.prefix.as_str().clone())
StandardFramework::new() .allow_dm(false)
.configure(|c| { .ignore_bots(true)
c.prefix(cfg.prefix.as_str().clone()) .ignore_webhooks(true)
.allow_dm(false) })
.ignore_bots(true) .before(permission_check)
.ignore_webhooks(true) .after(log_error)
}) .group(&EVENTCOMMANDS_GROUP)
.before(permission_check) .help(&BOT_HELP),
.after(log_error) ).await?;
.group(&EVENTCOMMANDS_GROUP)
.help(&BOT_HELP),
);
// Copy config data to client data and setup scheduler // Copy config data to client data and setup scheduler
{ {
let mut data = client.data.write(); let mut data = client.data.write().await;
data.insert::<HypeBotConfig>(cfg); data.insert::<HypeBotConfig>(cfg);
data.insert::<DraftEvent>(DraftEvent { data.insert::<DraftEvent>(DraftEvent {
event: NewEvent { event: NewEvent {
@ -251,41 +262,12 @@ fn main() -> HypeBotResult<()> {
}, },
creator_id: 0, creator_id: 0,
}); });
// Create scheduler
let scheduler = Scheduler::new(2);
let scheduler = Arc::new(RwLock::new(scheduler));
data.insert::<SchedulerKey>(scheduler);
} }
// Schedule current events
let config = get_config(&client.data).expect("Unable to find get config");
let duration = chrono::Duration::minutes(60);
for event in get_all_events(config.db_url.clone()).unwrap() {
let event_time: DateTime<Utc> =
DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
if Utc::now() > event_time + duration {
delete_event(&client.cache_and_http.http, &client.data, &event);
} else if Utc::now() > event_time {
let scheduler = get_scheduler(&client.data).unwrap();
let mut scheduler = scheduler.write();
let cancel_time = event_time + duration;
let http = client.cache_and_http.http.clone();
let data = client.data.clone();
scheduler.add_task_datetime(cancel_time, move |_| {
delete_event(&http, &data, &event);
DateResult::Done
});
} else if event.reminder_sent == 0 {
schedule_event(&client.cache_and_http.http, &client.data, &event);
}
}
// Start bot // Start bot
info!("Starting HypeBot!"); info!("Starting HypeBot!");
if let Err(why) = client.start() { if let Err(why) = client.start().await {
error!("An error occurred while running the client: {:?}", why); error!("An error occurred while running the client: {:?}", why);
} }
} else { } else {