HypeBot/src/discord/events.rs

196 lines
5.9 KiB
Rust
Raw Normal View History

use super::{get_config, send_event_msg};
use crate::database::{get_event_by_name, insert_event, remove_event};
use crate::discord::{
get_draft_event, schedule_event, send_dm_message, send_draft_event, update_draft_event,
};
use crate::INTERESTED_EMOJI;
use chrono::offset::TimeZone;
use chrono::{Datelike, NaiveDateTime, Timelike, Utc};
use chrono_tz::Tz;
use serenity::framework::standard::{macros::command, Args, CommandResult};
use serenity::model::prelude::{Mentionable, Message, User};
use serenity::prelude::Context;
use serenity::utils::{content_safe, ContentSafeOptions};
use url::Url;
#[command]
/// Posts a previewed event
///
/// `~confirm`
///
/// **Note**
/// 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 {
let config = get_config(&ctx.data)?;
let draft_event = get_draft_event(&ctx.data)?;
let mut new_event = draft_event.event.clone();
// Check to to see if message author is the owner of the pending event
if draft_event.creator_id == msg.author.id.0 {
// Send event message
let event_msg = send_event_msg(&ctx.http, &config, config.event_channel, &new_event, true)?;
msg.reply(&ctx, "Event posted!")?;
new_event.message_id = event_msg.id.0.to_string();
let event = insert_event(config.db_url.clone(), &new_event)?;
schedule_event(&ctx.http, &ctx.data, &event);
} else {
msg.reply(&ctx, format!("You do not have a pending event!"))?;
}
Ok(())
}
#[command]
/// Creates an event and previews the announcement
///
2020-05-02 18:49:20 +00:00
/// `~create "event name" "04:20pm 2069-04-20" "event description" "http://optional.thumbnail.link" "optional organizer"`
///
/// **Time format**
/// The time format is HH:MMam YYYY-MM-DD
///
/// **Thumbnail Link**
/// The thumbnail link is optional, if one is not provided, a default image is shown
///
/// **Organizer**
/// 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 {
// Get config
let config = get_config(&ctx.data)?;
// Parse args
let event_name = match args.find::<String>() {
Ok(event_name) => event_name.replace("\"", ""),
Err(_) => {
msg.reply(&ctx, "No event name provided.".to_string())?;
return Ok(());
}
};
let date_string = match args.find::<String>() {
Ok(date_string) => date_string.replace("\"", ""),
Err(_) => {
msg.reply(&ctx, "No date provided.".to_string())?;
return Ok(());
}
};
let description = match args.find::<String>() {
Ok(desc) => desc.replace("\"", ""),
Err(_) => {
msg.reply(&ctx, "No description provided.".to_string())?;
return Ok(());
}
};
let location = match args.find::<String>() {
Ok(desc) => desc.replace("\"", ""),
Err(_) => {
msg.reply(&ctx, "No location provided.".to_string())?;
return Ok(());
}
};
let thumbnail_link = match args.find::<Url>() {
Ok(link) => link.into_string(),
Err(_) => config.default_thumbnail_link.clone(),
};
let organizer = match args.find::<String>() {
Ok(link) => link.replace("\"", ""),
Err(_) => msg.author.mention(),
};
// Parse date
let tz: Tz = config.event_timezone;
let input_date = match NaiveDateTime::parse_from_str(date_string.as_str(), "%I:%M%P %Y-%m-%d") {
Ok(date) => date,
Err(_) => {
msg.reply(
&ctx,
"Invalid date format. Format is HH:MMam YYYY-MM-DD".to_string(),
)?;
return Ok(());
}
};
let input_date = tz
.ymd(
input_date.date().year(),
input_date.date().month(),
input_date.date().day(),
)
.and_hms(
input_date.time().hour(),
input_date.time().minute(),
input_date.time().second(),
);
let event_time = input_date.with_timezone(&Utc).naive_utc();
if Utc::now().naive_utc() > event_time {
msg.reply(&ctx, "The scheduled time has already passed!")?;
return Ok(());
}
// Clean channel, role, and everyone pings
let settings = ContentSafeOptions::default()
.clean_role(true)
.clean_here(true)
.clean_user(false)
.clean_everyone(true);
let description = content_safe(&ctx.cache, description, &settings);
let event_name = content_safe(&ctx.cache, event_name, &settings);
let location = content_safe(&ctx.cache, location, &settings);
let organizer = content_safe(&ctx.cache, organizer, &settings);
update_draft_event(
&ctx,
event_name,
description,
organizer,
location,
thumbnail_link,
event_time,
msg.author.id.0,
)?;
send_draft_event(&ctx, msg.channel_id)?;
Ok(())
}
#[command]
/// Cancels an already scheduled event
///
/// `~cancel "event name"`
fn cancel(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
let config = get_config(&ctx.data)?;
// Parse args
let event_name = args.single::<String>()?.replace("\"", "");
let event = get_event_by_name(config.db_url.clone(), event_name)?;
let message_id = event.message_id.parse::<u64>()?;
let message = ctx.http.get_message(config.event_channel, message_id)?;
let reaction_users = message
.reaction_users(&ctx.http, INTERESTED_EMOJI, None, None)
.unwrap_or(Vec::<User>::new());
let string = format!("**{}** has been canceled!", event.event_name.clone());
for user in reaction_users {
send_dm_message(&ctx.http, user, &string);
}
remove_event(config.db_url.clone(), event.id)?;
message.delete(&ctx)?;
msg.reply(&ctx, "**{}** has been canceled")?;
Ok(())
}