Reformatted code to organize it better
+ Moved commands and discord helper functions to `discord` modulemaster
parent
01c986d708
commit
0f8fe04657
|
@ -0,0 +1,170 @@
|
||||||
|
use super::{get_config, send_event_msg};
|
||||||
|
use crate::database::{get_event_by_name, insert_event, remove_event};
|
||||||
|
use crate::discord::{send_dm_message, send_draft_event, update_draft_event};
|
||||||
|
use crate::{DraftEvent, INTERESTED_EMOJI};
|
||||||
|
use chrono::offset::TimeZone;
|
||||||
|
use chrono::{Datelike, NaiveDateTime, Timelike, Utc};
|
||||||
|
use chrono_tz::Tz;
|
||||||
|
use serenity::framework::standard::{macros::command, Args, CommandError, CommandResult};
|
||||||
|
use serenity::model::prelude::{Message, User};
|
||||||
|
use serenity::prelude::Context;
|
||||||
|
use serenity::utils::{content_safe, ContentSafeOptions};
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
/// Posts a previewed event
|
||||||
|
///
|
||||||
|
/// **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 data = ctx.data.read();
|
||||||
|
|
||||||
|
// Get draft event
|
||||||
|
if let Some(draft_event) = data.get::<DraftEvent>() {
|
||||||
|
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();
|
||||||
|
|
||||||
|
insert_event(config.db_url.clone(), &new_event)?;
|
||||||
|
} else {
|
||||||
|
msg.reply(&ctx, format!("You do not have a pending event!"))?;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
msg.reply(&ctx, format!("There are no pending events!!"))?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
/// Creates an event and previews the announcement
|
||||||
|
///
|
||||||
|
/// `~create "event name" "04:20pm 2069-04-20" "event description" <http://optional.thumbnail.link>`
|
||||||
|
///
|
||||||
|
/// **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
|
||||||
|
fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
||||||
|
// Get config
|
||||||
|
let config = get_config(&ctx.data)?;
|
||||||
|
let guild_id = msg
|
||||||
|
.guild_id
|
||||||
|
.ok_or(CommandError("Unable to get guild ID".to_string()))?;
|
||||||
|
|
||||||
|
// Parse args
|
||||||
|
let event_name = match args.single::<String>() {
|
||||||
|
Ok(event_name) => event_name.replace("\"", ""),
|
||||||
|
Err(_) => {
|
||||||
|
msg.reply(&ctx, "No event name provided.".to_string())?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let date_string = match args.single::<String>() {
|
||||||
|
Ok(date_string) => date_string.replace("\"", ""),
|
||||||
|
Err(_) => {
|
||||||
|
msg.reply(&ctx, "No date provided.".to_string())?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let description = match args.single::<String>() {
|
||||||
|
Ok(desc) => desc.replace("\"", ""),
|
||||||
|
Err(_) => {
|
||||||
|
msg.reply(&ctx, "No description provided.".to_string())?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let thumbnail_link = match args.single::<String>() {
|
||||||
|
Ok(link) => link.replace("<", "").replace(">", ""),
|
||||||
|
Err(_) => config.default_thumbnail_link.clone(),
|
||||||
|
};
|
||||||
|
|
||||||
|
// 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();
|
||||||
|
|
||||||
|
// Clean channel, role, and everyone pings
|
||||||
|
let settings = ContentSafeOptions::default()
|
||||||
|
.clean_role(true)
|
||||||
|
.clean_here(true)
|
||||||
|
.clean_user(true)
|
||||||
|
.clean_everyone(true)
|
||||||
|
.display_as_member_from(guild_id);
|
||||||
|
|
||||||
|
let description = content_safe(&ctx.cache, description, &settings);
|
||||||
|
let event_name = content_safe(&ctx.cache, event_name, &settings);
|
||||||
|
|
||||||
|
update_draft_event(
|
||||||
|
&ctx,
|
||||||
|
event_name,
|
||||||
|
description,
|
||||||
|
thumbnail_link,
|
||||||
|
event_time,
|
||||||
|
msg.author.id.0,
|
||||||
|
)?;
|
||||||
|
send_draft_event(&ctx, msg.channel_id)?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
/// Cancels an already scheduled event
|
||||||
|
///
|
||||||
|
/// `~create_event "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)?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
|
@ -0,0 +1,175 @@
|
||||||
|
use crate::database::get_event_by_msg_id;
|
||||||
|
use crate::database::models::NewEvent;
|
||||||
|
use crate::hypebot_config::HypeBotConfig;
|
||||||
|
use crate::{DraftEvent, INTERESTED_EMOJI, UNINTERESTED_EMOJI};
|
||||||
|
use chrono::{DateTime, NaiveDateTime, Utc};
|
||||||
|
use serenity::framework::standard::{CommandError, CommandResult};
|
||||||
|
use serenity::http::Http;
|
||||||
|
use serenity::model::prelude::{ChannelId, Message, Reaction, User};
|
||||||
|
use serenity::prelude::{Context, RwLock, ShareMap};
|
||||||
|
use serenity::utils::Colour;
|
||||||
|
use serenity::Result;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
pub mod events;
|
||||||
|
|
||||||
|
/// Send a message to a reaction user
|
||||||
|
///
|
||||||
|
/// Message will be sent in the format
|
||||||
|
/// ```
|
||||||
|
/// "{msg_text} **event_name**"
|
||||||
|
/// ```
|
||||||
|
pub fn send_message_to_reaction_users(ctx: &Context, reaction: &Reaction, msg_text: &str) {
|
||||||
|
if let Ok(config) = get_config(&ctx.data) {
|
||||||
|
let db_link = config.db_url.clone();
|
||||||
|
let message_id = reaction.message_id.0.to_string();
|
||||||
|
|
||||||
|
let event = match get_event_by_msg_id(db_link, message_id) {
|
||||||
|
Ok(event) => event,
|
||||||
|
Err(_) => {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// Format message
|
||||||
|
let msg: String = format!("{} **{}**", msg_text, event.event_name);
|
||||||
|
|
||||||
|
if let Ok(user) = reaction.user(&ctx.http) {
|
||||||
|
send_dm_message(&ctx.http, user, &msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Send a DM message to a user
|
||||||
|
pub fn send_dm_message(http: &Http, user: User, message: &String) {
|
||||||
|
if let Ok(dm_channel) = user.create_dm_channel(&http) {
|
||||||
|
dm_channel.send_message(&http, |m| m.content(message)).ok();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sends the event message to the event channel
|
||||||
|
pub fn send_event_msg(
|
||||||
|
http: &Http,
|
||||||
|
config: &HypeBotConfig,
|
||||||
|
channel_id: u64,
|
||||||
|
event: &NewEvent,
|
||||||
|
react: bool,
|
||||||
|
) -> Result<Message> {
|
||||||
|
let channel = http.get_channel(channel_id)?;
|
||||||
|
|
||||||
|
let utc_time = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
|
||||||
|
|
||||||
|
let native_time = utc_time.with_timezone(&config.event_timezone);
|
||||||
|
|
||||||
|
// Send message
|
||||||
|
let msg = channel.id().send_message(&http, |m| {
|
||||||
|
m.embed(|e| {
|
||||||
|
e.title(event.event_name.clone())
|
||||||
|
.color(Colour::PURPLE)
|
||||||
|
.description(format!(
|
||||||
|
"**{}**\n{}\n\nReact with {} below to receive event reminders!",
|
||||||
|
native_time.format("%A, %B %d @ %I:%M %P %t %Z"),
|
||||||
|
event.event_desc,
|
||||||
|
INTERESTED_EMOJI
|
||||||
|
))
|
||||||
|
.thumbnail(event.thumbnail_link.clone())
|
||||||
|
.footer(|f| f.text("Local Event Time"))
|
||||||
|
.timestamp(utc_time.to_rfc3339())
|
||||||
|
})
|
||||||
|
})?;
|
||||||
|
|
||||||
|
if react {
|
||||||
|
// Add reacts
|
||||||
|
msg.react(http, INTERESTED_EMOJI)?;
|
||||||
|
msg.react(http, UNINTERESTED_EMOJI)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Updates the draft event stored in the context data
|
||||||
|
pub fn update_draft_event(
|
||||||
|
ctx: &Context,
|
||||||
|
event_name: String,
|
||||||
|
event_desc: String,
|
||||||
|
thumbnail: String,
|
||||||
|
event_time: NaiveDateTime,
|
||||||
|
creator_id: u64,
|
||||||
|
) -> CommandResult {
|
||||||
|
let mut data = ctx.data.write();
|
||||||
|
let mut draft_event = data
|
||||||
|
.get_mut::<DraftEvent>()
|
||||||
|
.ok_or(CommandError("Unable get draft event!".to_string()))?;
|
||||||
|
|
||||||
|
draft_event.event.event_name = event_name;
|
||||||
|
draft_event.event.event_desc = event_desc;
|
||||||
|
draft_event.event.thumbnail_link = thumbnail;
|
||||||
|
draft_event.event.message_id = String::new();
|
||||||
|
draft_event.event.event_time = event_time;
|
||||||
|
draft_event.creator_id = creator_id;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sends the draft event stored in the context data
|
||||||
|
pub fn send_draft_event(ctx: &Context, channel: ChannelId) -> CommandResult {
|
||||||
|
let data = ctx.data.read();
|
||||||
|
let config = data
|
||||||
|
.get::<HypeBotConfig>()
|
||||||
|
.ok_or(CommandError("Config not found!".to_string()))?;
|
||||||
|
let draft_event = data
|
||||||
|
.get::<DraftEvent>()
|
||||||
|
.ok_or(CommandError("Draft event not found!".to_string()))?;
|
||||||
|
|
||||||
|
channel.send_message(&ctx, |m| {
|
||||||
|
m.content(format!(
|
||||||
|
"Draft message, use the `confirm` command to post it."
|
||||||
|
))
|
||||||
|
})?;
|
||||||
|
send_event_msg(&ctx.http, config, channel.0, &draft_event.event, false)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the config from context data
|
||||||
|
pub fn get_config(
|
||||||
|
data: &Arc<RwLock<ShareMap>>,
|
||||||
|
) -> std::result::Result<HypeBotConfig, CommandError> {
|
||||||
|
let data_read = data.read();
|
||||||
|
let config = data_read
|
||||||
|
.get::<HypeBotConfig>()
|
||||||
|
.ok_or(CommandError("Unable to get config".to_string()))?;
|
||||||
|
|
||||||
|
Ok(config.clone())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Logs command errors to the logger
|
||||||
|
pub fn log_error(
|
||||||
|
_ctx: &mut Context,
|
||||||
|
_msg: &Message,
|
||||||
|
command_name: &str,
|
||||||
|
result: std::result::Result<(), CommandError>,
|
||||||
|
) {
|
||||||
|
match result {
|
||||||
|
Ok(()) => (),
|
||||||
|
Err(why) => error!("Command '{}' returned error {:?}", command_name, why),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the user has permission to use this bot
|
||||||
|
pub fn permission_check(ctx: &mut Context, msg: &Message, _command_name: &str) -> bool {
|
||||||
|
if let Some(guild_id) = msg.guild_id {
|
||||||
|
if let Ok(config) = get_config(&ctx.data) {
|
||||||
|
if let Ok(roles) = ctx.http.get_guild_roles(guild_id.0) {
|
||||||
|
for role in roles {
|
||||||
|
if config.event_roles.contains(&role.id.0) {
|
||||||
|
return match msg.author.has_role(ctx, guild_id, role) {
|
||||||
|
Ok(has_role) => has_role,
|
||||||
|
Err(_) => false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
false
|
||||||
|
}
|
356
src/main.rs
356
src/main.rs
|
@ -7,27 +7,23 @@ extern crate serde;
|
||||||
extern crate log;
|
extern crate log;
|
||||||
extern crate log4rs;
|
extern crate log4rs;
|
||||||
|
|
||||||
use chrono::{DateTime, Datelike, NaiveDateTime, TimeZone, Timelike, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
use chrono_tz::Tz;
|
|
||||||
use clap::{App, Arg};
|
use clap::{App, Arg};
|
||||||
use log::LevelFilter;
|
use log::LevelFilter;
|
||||||
use log4rs::append::file::FileAppender;
|
use log4rs::append::file::FileAppender;
|
||||||
use log4rs::config::{Appender, Config, Root};
|
use log4rs::config::{Appender, Config, Root};
|
||||||
use log4rs::encode::pattern::PatternEncoder;
|
use log4rs::encode::pattern::PatternEncoder;
|
||||||
use serenity::client::Client;
|
use serenity::client::Client;
|
||||||
use serenity::framework::standard::macros::{command, group, help};
|
use serenity::framework::standard::macros::{group, help};
|
||||||
use serenity::framework::standard::{help_commands, Args, CommandGroup, HelpOptions};
|
use serenity::framework::standard::{help_commands, Args, CommandGroup, HelpOptions};
|
||||||
use serenity::framework::standard::{CommandError, CommandResult, StandardFramework};
|
use serenity::framework::standard::{CommandResult, StandardFramework};
|
||||||
use serenity::http::Http;
|
|
||||||
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::{ChannelId, Ready};
|
use serenity::model::prelude::Ready;
|
||||||
use serenity::model::user::User;
|
use serenity::model::user::User;
|
||||||
use serenity::prelude::TypeMapKey;
|
use serenity::prelude::TypeMapKey;
|
||||||
use serenity::prelude::{Context, EventHandler, RwLock, ShareMap};
|
use serenity::prelude::{Context, EventHandler, RwLock, ShareMap};
|
||||||
use serenity::utils::{content_safe, Colour, ContentSafeOptions};
|
|
||||||
use serenity::CacheAndHttp;
|
use serenity::CacheAndHttp;
|
||||||
use serenity::Result;
|
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
use std::process::exit;
|
use std::process::exit;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
@ -36,10 +32,15 @@ use std::thread::sleep;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
mod database;
|
mod database;
|
||||||
|
mod discord;
|
||||||
mod hypebot_config;
|
mod hypebot_config;
|
||||||
|
|
||||||
use crate::database::models::NewEvent;
|
use database::models::NewEvent;
|
||||||
use database::*;
|
use database::*;
|
||||||
|
use discord::events::{CANCEL_COMMAND, CONFIRM_COMMAND, CREATE_COMMAND};
|
||||||
|
use discord::{
|
||||||
|
get_config, log_error, permission_check, send_dm_message, send_message_to_reaction_users,
|
||||||
|
};
|
||||||
use hypebot_config::HypeBotConfig;
|
use hypebot_config::HypeBotConfig;
|
||||||
|
|
||||||
const INTERESTED_EMOJI: &str = "\u{2705}";
|
const INTERESTED_EMOJI: &str = "\u{2705}";
|
||||||
|
@ -47,7 +48,7 @@ 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>>;
|
||||||
|
|
||||||
/// Event commands group
|
/// Event discord group
|
||||||
#[group]
|
#[group]
|
||||||
#[only_in(guilds)]
|
#[only_in(guilds)]
|
||||||
#[description("Commands for Creating Events")]
|
#[description("Commands for Creating Events")]
|
||||||
|
@ -158,319 +159,20 @@ fn send_reminders(cache_and_http: &Arc<CacheAndHttp>, data: &Arc<RwLock<ShareMap
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Send a message to a reaction user
|
/// Does the setup for logging
|
||||||
///
|
fn setup_logging(config: &HypeBotConfig) -> HypeBotResult<()> {
|
||||||
/// Message will be sent in the format
|
// Setup logging
|
||||||
/// ```
|
let log_encode = Box::new(PatternEncoder::new("{d}:{l}-{m}\n"));
|
||||||
/// "{msg_text} **event_name**"
|
|
||||||
/// ```
|
|
||||||
fn send_message_to_reaction_users(ctx: &Context, reaction: &Reaction, msg_text: &str) {
|
|
||||||
if let Ok(config) = get_config(&ctx.data) {
|
|
||||||
let db_link = config.db_url.clone();
|
|
||||||
let message_id = reaction.message_id.0.to_string();
|
|
||||||
|
|
||||||
let event = match get_event_by_msg_id(db_link, message_id) {
|
let logfile = FileAppender::builder()
|
||||||
Ok(event) => event,
|
.encoder(log_encode)
|
||||||
Err(_) => {
|
.build(&config.log_path)?;
|
||||||
return;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Format message
|
let log_config = Config::builder()
|
||||||
let msg: String = format!("{} **{}**", msg_text, event.event_name);
|
.appender(Appender::builder().build("logfile", Box::new(logfile)))
|
||||||
|
.build(Root::builder().appender("logfile").build(LevelFilter::Info))?;
|
||||||
|
|
||||||
if let Ok(user) = reaction.user(&ctx.http) {
|
log4rs::init_config(log_config)?;
|
||||||
send_dm_message(&ctx.http, user, &msg);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Send a DM message to a user
|
|
||||||
fn send_dm_message(http: &Http, user: User, message: &String) {
|
|
||||||
if let Ok(dm_channel) = user.create_dm_channel(&http) {
|
|
||||||
dm_channel.send_message(&http, |m| m.content(message)).ok();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sends the event message to the event channel
|
|
||||||
fn send_event_msg(
|
|
||||||
http: &Http,
|
|
||||||
config: &HypeBotConfig,
|
|
||||||
channel_id: u64,
|
|
||||||
event: &NewEvent,
|
|
||||||
react: bool,
|
|
||||||
) -> Result<Message> {
|
|
||||||
let channel = http.get_channel(channel_id)?;
|
|
||||||
|
|
||||||
let utc_time = DateTime::<Utc>::from_utc(event.event_time.clone(), Utc);
|
|
||||||
|
|
||||||
let native_time = utc_time.with_timezone(&config.event_timezone);
|
|
||||||
|
|
||||||
// Send message
|
|
||||||
let msg = channel.id().send_message(&http, |m| {
|
|
||||||
m.embed(|e| {
|
|
||||||
e.title(event.event_name.clone())
|
|
||||||
.color(Colour::PURPLE)
|
|
||||||
.description(format!(
|
|
||||||
"**{}**\n{}\n\nReact with {} below to receive event reminders!",
|
|
||||||
native_time.format("%A, %B %d @ %I:%M %P %t %Z"),
|
|
||||||
event.event_desc,
|
|
||||||
INTERESTED_EMOJI
|
|
||||||
))
|
|
||||||
.thumbnail(event.thumbnail_link.clone())
|
|
||||||
.footer(|f| f.text("Local Event Time"))
|
|
||||||
.timestamp(utc_time.to_rfc3339())
|
|
||||||
})
|
|
||||||
})?;
|
|
||||||
|
|
||||||
if react {
|
|
||||||
// Add reacts
|
|
||||||
msg.react(http, INTERESTED_EMOJI)?;
|
|
||||||
msg.react(http, UNINTERESTED_EMOJI)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(msg)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Updates the draft event stored in the context data
|
|
||||||
fn update_draft_event(
|
|
||||||
ctx: &Context,
|
|
||||||
event_name: String,
|
|
||||||
event_desc: String,
|
|
||||||
thumbnail: String,
|
|
||||||
event_time: NaiveDateTime,
|
|
||||||
creator_id: u64,
|
|
||||||
) -> CommandResult {
|
|
||||||
let mut data = ctx.data.write();
|
|
||||||
let mut draft_event = data
|
|
||||||
.get_mut::<DraftEvent>()
|
|
||||||
.ok_or(CommandError("Unable get draft event!".to_string()))?;
|
|
||||||
|
|
||||||
draft_event.event.event_name = event_name;
|
|
||||||
draft_event.event.event_desc = event_desc;
|
|
||||||
draft_event.event.thumbnail_link = thumbnail;
|
|
||||||
draft_event.event.message_id = String::new();
|
|
||||||
draft_event.event.event_time = event_time;
|
|
||||||
draft_event.creator_id = creator_id;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Sends the draft event stored in the context data
|
|
||||||
fn send_draft_event(ctx: &Context, channel: ChannelId) -> CommandResult {
|
|
||||||
let data = ctx.data.read();
|
|
||||||
let config = data
|
|
||||||
.get::<HypeBotConfig>()
|
|
||||||
.ok_or(CommandError("Config not found!".to_string()))?;
|
|
||||||
let draft_event = data
|
|
||||||
.get::<DraftEvent>()
|
|
||||||
.ok_or(CommandError("Draft event not found!".to_string()))?;
|
|
||||||
|
|
||||||
channel.send_message(&ctx, |m| {
|
|
||||||
m.content(format!(
|
|
||||||
"Draft message, use the `confirm` command to post it."
|
|
||||||
))
|
|
||||||
})?;
|
|
||||||
send_event_msg(&ctx.http, config, channel.0, &draft_event.event, false)?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Gets the config from context data
|
|
||||||
fn get_config(data: &Arc<RwLock<ShareMap>>) -> std::result::Result<HypeBotConfig, CommandError> {
|
|
||||||
let data_read = data.read();
|
|
||||||
let config = data_read
|
|
||||||
.get::<HypeBotConfig>()
|
|
||||||
.ok_or(CommandError("Unable to get config".to_string()))?;
|
|
||||||
|
|
||||||
Ok(config.clone())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Checks if the user has permission to use this bot
|
|
||||||
fn permission_check(ctx: &mut Context, msg: &Message, _command_name: &str) -> bool {
|
|
||||||
if let Some(guild_id) = msg.guild_id {
|
|
||||||
if let Ok(config) = get_config(&ctx.data) {
|
|
||||||
if let Ok(roles) = ctx.http.get_guild_roles(guild_id.0) {
|
|
||||||
for role in roles {
|
|
||||||
if config.event_roles.contains(&role.id.0) {
|
|
||||||
return match msg.author.has_role(ctx, guild_id, role) {
|
|
||||||
Ok(has_role) => has_role,
|
|
||||||
Err(_) => false,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
false
|
|
||||||
}
|
|
||||||
|
|
||||||
fn log_error(
|
|
||||||
_ctx: &mut Context,
|
|
||||||
_msg: &Message,
|
|
||||||
command_name: &str,
|
|
||||||
result: std::result::Result<(), CommandError>,
|
|
||||||
) {
|
|
||||||
match result {
|
|
||||||
Ok(()) => (),
|
|
||||||
Err(why) => error!("Command '{}' returned error {:?}", command_name, why),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
|
||||||
/// Posts a previewed event
|
|
||||||
///
|
|
||||||
/// **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 data = ctx.data.read();
|
|
||||||
|
|
||||||
// Get draft event
|
|
||||||
if let Some(draft_event) = data.get::<DraftEvent>() {
|
|
||||||
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();
|
|
||||||
|
|
||||||
insert_event(config.db_url.clone(), &new_event)?;
|
|
||||||
} else {
|
|
||||||
msg.reply(&ctx, format!("You do not have a pending event!"))?;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
msg.reply(&ctx, format!("There are no pending events!!"))?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
|
||||||
/// Creates an event and previews the announcement
|
|
||||||
///
|
|
||||||
/// `~create "event name" "04:20pm 2069-04-20" "event description" \<http://optional.thumbnail.link>`
|
|
||||||
///
|
|
||||||
/// **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
|
|
||||||
fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
||||||
// Get config
|
|
||||||
let config = get_config(&ctx.data)?;
|
|
||||||
let guild_id = msg
|
|
||||||
.guild_id
|
|
||||||
.ok_or(CommandError("Unable to get guild ID".to_string()))?;
|
|
||||||
|
|
||||||
// Parse args
|
|
||||||
let event_name = match args.single::<String>() {
|
|
||||||
Ok(event_name) => event_name.replace("\"", ""),
|
|
||||||
Err(_) => {
|
|
||||||
msg.reply(&ctx, "No event name provided.".to_string())?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let date_string = match args.single::<String>() {
|
|
||||||
Ok(date_string) => date_string.replace("\"", ""),
|
|
||||||
Err(_) => {
|
|
||||||
msg.reply(&ctx, "No date provided.".to_string())?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let description = match args.single::<String>() {
|
|
||||||
Ok(desc) => desc.replace("\"", ""),
|
|
||||||
Err(_) => {
|
|
||||||
msg.reply(&ctx, "No description provided.".to_string())?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let thumbnail_link = match args.single::<String>() {
|
|
||||||
Ok(link) => link.replace("<", "").replace(">", ""),
|
|
||||||
Err(_) => config.default_thumbnail_link.clone(),
|
|
||||||
};
|
|
||||||
|
|
||||||
// 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();
|
|
||||||
|
|
||||||
// Clean channel, role, and everyone pings
|
|
||||||
let settings = ContentSafeOptions::default()
|
|
||||||
.clean_role(true)
|
|
||||||
.clean_here(true)
|
|
||||||
.clean_user(true)
|
|
||||||
.clean_everyone(true)
|
|
||||||
.display_as_member_from(guild_id);
|
|
||||||
|
|
||||||
let description = content_safe(&ctx.cache, description, &settings);
|
|
||||||
let event_name = content_safe(&ctx.cache, event_name, &settings);
|
|
||||||
|
|
||||||
update_draft_event(
|
|
||||||
&ctx,
|
|
||||||
event_name,
|
|
||||||
description,
|
|
||||||
thumbnail_link,
|
|
||||||
event_time,
|
|
||||||
msg.author.id.0,
|
|
||||||
)?;
|
|
||||||
send_draft_event(&ctx, msg.channel_id)?;
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
|
||||||
/// Cancels an already scheduled event
|
|
||||||
///
|
|
||||||
/// `~create_event "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)?;
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
@ -504,19 +206,7 @@ fn main() -> HypeBotResult<()> {
|
||||||
};
|
};
|
||||||
|
|
||||||
// Setup logging
|
// Setup logging
|
||||||
let log_encode = Box::new(PatternEncoder::new("{d}:{l}-{m}\n"));
|
setup_logging(&cfg)?;
|
||||||
|
|
||||||
let logfile =
|
|
||||||
FileAppender::builder()
|
|
||||||
.encoder(log_encode)
|
|
||||||
.build(&cfg.log_path)?;
|
|
||||||
|
|
||||||
|
|
||||||
let config = Config::builder()
|
|
||||||
.appender(Appender::builder().build("logfile", Box::new(logfile)))
|
|
||||||
.build(Root::builder().appender("logfile").build(LevelFilter::Info))?;
|
|
||||||
|
|
||||||
log4rs::init_config(config)?;
|
|
||||||
|
|
||||||
// Run migrations
|
// Run migrations
|
||||||
let connection = establish_connection(cfg.db_url.clone());
|
let connection = establish_connection(cfg.db_url.clone());
|
||||||
|
|
Loading…
Reference in New Issue