repo: push main from local to remote

1 files changed, 179 insertions, 0 deletions

diff --git a/src/modules/commands/mods/mute.rs b/src/modules/commands/mods/mute.rs
new file mode 100644
index 0000000..12ee36d
--- /dev/null
+++ b/src/modules/commands/mods/mute.rs
@@ -0,0 +1,179 @@
+use chrono::Utc;
+use crate::core::colours;
+use crate::core::consts::*;
+use crate::core::consts::DB as db;
+use crate::core::model::TC;
+use crate::core::utils::*;
+use serenity::builder::CreateMessage;
+use serenity::framework::standard::{
+    Args,
+    Command,
+    CommandError,
+    CommandOptions
+};
+use serenity::model::channel::Message;
+use serenity::model::id::ChannelId;
+use serenity::model::Permissions;
+use serenity::prelude::Context;
+use std::sync::Arc;
+
+pub struct Mute;
+impl Command for Mute {
+    fn options(&self) -> Arc<CommandOptions> {
+        let default = CommandOptions::default();
+        let options = CommandOptions {
+            desc: Some("Someone being annoyin ? Mute em' ! (Optionally, include a reason and/ or time for the mute to expire).".to_string()),
+            usage: Some("<user_resolvable> [/t time] [/r reason]".to_string()),
+            example: Some("@fun /t 1day /r spam".to_string()),
+            required_permissions: Permissions::MANAGE_ROLES | Permissions::MUTE_MEMBERS,
+            ..default
+        };
+        Arc::new(options)
+    }
+
+    fn execute(&self, ctx: &mut Context, message: &Message, mut args: Args) -> Result<(), CommandError> {
+        if let Some(guild_lock) = message.guild() {
+            let guild = {
+                guild_lock.read().clone()
+            };
+            if let Some((_, mut member)) = parse_user(args.single::<String>().unwrap_or(String::new()), guild.id) {
+                let guild_data = db.get_guild(guild.id.0 as i64)?;
+                if guild_data.mute_setup {
+                    let switches = get_switches(args.rest().to_string());
+                    let time = match switches.get("t") {
+                        Some(s) => hrtime_to_seconds(s.clone()),
+                        None => 0,
+                    };
+                    let reason = match switches.get("r") {
+                        Some(s) => s.clone(),
+                        None => String::new(),
+                    };
+                    if let Some(mute_role) = guild.roles.values().find(|e| e.name.to_lowercase() == "muted") {
+                        if member.roles.contains(&mute_role.id) {
+                            message.channel_id.say("Member already muted.")?;
+                        } else {
+                            member.add_role(mute_role)?;
+                            let user = {
+                                member.user.read().clone()
+                            };
+                            let case = db.new_case(user.id.0 as i64, guild.id.0 as i64, "Mute".to_string(), Some(reason.clone()), message.author.id.0 as i64)?;
+                            let mut description = format!(
+                                "**User:** {} ({})\n**Moderator:** {} ({})"
+                                ,user.tag()
+                                ,user.id.0
+                                ,message.author.tag()
+                                ,message.author.id.0);
+                            if time != 0 {
+                                let data = ctx.data.lock();
+                                if let Some(tc_lock) = data.get::<TC>() {
+                                    let tc = tc_lock.lock();
+                                    let data = format!("UNMUTE||{}||{}||{}||{}||{}||{}",
+                                        user.id.0,
+                                        guild.id.0,
+                                        mute_role.id.0,
+                                        if guild_data.modlog && guild_data.modlog_channel > 0 {
+                                            guild_data.modlog_channel
+                                        } else { message.channel_id.0 as i64 },
+                                        time,
+                                        case.id);
+                                    let start_time = Utc::now().timestamp();
+                                    let end_time = start_time + time;
+                                    db.new_timer(start_time, end_time, data)?;
+                                    tc.request();
+                                    description = format!(
+                                        "{}\n**Duration:** {}"
+                                        ,description
+                                        ,seconds_to_hrtime(time as usize));
+                                } else {
+                                    message.channel_id.say("Something went wrong with the timer.")?;
+                                }
+                            }
+                            if !reason.is_empty() {
+                                description = format!(
+                                    "{}\n**Reason:** {}"
+                                    ,description
+                                    ,reason.to_string());
+                            }
+                            let response = CreateMessage::default()
+                                .embed(|e| e
+                                    .title("Member Muted")
+                                    .colour(*colours::RED)
+                                    .description(description)
+                                    .timestamp(now!()));
+
+                            if guild_data.modlog && guild_data.modlog_channel > 0 {
+                                let channel = ChannelId(guild_data.modlog_channel as u64);
+                                channel.send_message(|_| response)?;
+                            } else {
+                                message.channel_id.send_message(|_| response)?;
+                            }
+                        }
+                    } else { message.channel_id.say("No mute role")?; }
+                } else {
+                    message.channel_id.say("Please run `setup` before using this command. Without it, muting may not work right.")?;
+                }
+            } else { message.channel_id.say("I couldn't find that user.")?; }
+        } else { failed!(GUILD_FAIL); }
+        Ok(())
+    }
+}
+
+pub struct Unmute;
+impl Command for Unmute {
+    fn options(&self) -> Arc<CommandOptions> {
+        let default = CommandOptions::default();
+        let options = CommandOptions {
+            desc: Some("How nice, you'll unmute them !".to_string()),
+            usage: Some("<user_resolvable>".to_string()),
+            example: Some("@fun".to_string()),
+            required_permissions: Permissions::MANAGE_ROLES | Permissions::MUTE_MEMBERS,
+            ..default
+        };
+        Arc::new(options)
+    }
+
+    fn execute(&self, _: &mut Context, message: &Message, mut args: Args) -> Result<(), CommandError> {
+        if let Some(guild_lock) = message.guild() {
+            let guild = {
+                guild_lock.read().clone()
+            };
+            if let Some((_, mut member)) = parse_user(args.single::<String>().unwrap_or(String::new()), guild.id) {
+                let guild_data = db.get_guild(guild.id.0 as i64)?;
+                if guild_data.mute_setup {
+                    if let Some(mute_role) = guild.roles.values().find(|e| e.name.to_lowercase() == "muted") {
+                        let user = {
+                            member.user.read().clone()
+                        };
+                        let mut description = format!(
+                                "**User:** {} ({})\n**Moderator:** {} ({})"
+                                ,user.tag()
+                                ,user.id.0
+                                ,message.author.tag()
+                                ,message.author.id.0);
+                        let response = CreateMessage::default()
+                            .embed(|e| e
+                                .title("Member Unmuted")
+                                .colour(*colours::GREEN)
+                                .description(description)
+                                .timestamp(now!()));
+
+                        if member.roles.contains(&mute_role.id) {
+                            member.remove_role(mute_role)?;
+                            if guild_data.modlog && guild_data.modlog_channel > 0 {
+                                let channel = ChannelId(guild_data.modlog_channel as u64);
+                                channel.send_message(|_| response)?;
+                            } else {
+                                message.channel_id.send_message(|_| response)?;
+                            }
+                        } else {
+                            message.channel_id.say("Member was not muted.")?;
+                        }
+                    } else { message.channel_id.say("No mute role")?; }
+                } else {
+                    message.channel_id.say("Please run `setup` before using this command. Without it, muting may not work right.")?;
+                }
+            } else { message.channel_id.say("I couldn't find that user.")?; }
+        } else { failed!(GUILD_FAIL); }
+        Ok(())
+    }
+}