Add permission checks to p.prefix command

src/groups/calls/lists/casual-insults-and-teasing.txt

@@ -3,4 +3,8 @@ idiot
 stupid
 dumb
 moron
-fool
+fool
+stfu
+syfm
+sybau
+gtfo

src/groups/calls/lists/explicit-sexual-language.txt

@@ -75,6 +75,8 @@ goonbeam
 goonbeaming
 gooncream
 gooncreaming
+goonbud
+goonbuds
 boty hole
 suck my pp
 ass cracker
@@ -97,4 +99,23 @@ whre
 tickle your pickle
 tickle yr pickle
 tickle his pickle
-tickle my pickle
+tickle my pickle
+peg
+peged
+peging
+butbang
+butbanged
+eat my ass
+eat my ahh
+eat mi ass
+eat my ahh
+eat my bum
+eat mi bum
+ima touch you
+ima touch u
+il touch you
+il touch u
+im going to touch you
+im going to touch u
+im gonna touch you
+im gonna touch u

src/groups/calls/lists/hate-speech-and-slurs.txt

@@ -32,15 +32,17 @@ dumb fck
 dumb fuk
 dumb fuh
 dumb fuhk
-stfu
+shut the fvck up
+shut the fck up
 shut the fuck up
-syfm
 shut your fucking mouth
-sybau
+shut your fcking mouth
+shut your fvcking mouth
 shut your bitch as up
 shut your bih as up
-gtfo
 get the fuck out
+get the fvck out
+get the fck out
+get tfo
 blackie
-motherfucker
-mf
+motherfucker

src/groups/servers/commands.rs

@@ -8,7 +8,9 @@ use crate::{
         Command, CommandError,
         arguments::{Argument, GreedyString, ParseArguments},
         context::CommandContext,
+        permissions::Permission,
     },
+    groups::servers::permissions::{GuildOnlyPermission, ServerModeratorPermission},
     state::State,
 };
 
@@ -37,6 +39,13 @@ impl Command<State> for PrefixCommand {
         }]
     }
 
+    fn permissions(&self) -> Vec<Box<dyn Permission<State> + Send + Sync + 'static>> {
+        vec![
+            Box::new(GuildOnlyPermission),
+            Box::new(ServerModeratorPermission),
+        ]
+    }
+
     async fn execute(
         &self,
         ctx: CommandContext<'_, State>,

src/groups/servers/permissions.rs

@@ -1,4 +1,5 @@
 use async_trait::async_trait;
+use twilight_model::guild::Permissions;
 
 use crate::{
     framework::commands::{CommandError, context::CommandContext, permissions::Permission},
@@ -11,7 +12,15 @@ pub struct GuildOnlyPermission;
 #[async_trait]
 impl Permission<State> for GuildOnlyPermission {
     async fn check(&self, ctx: &CommandContext<State>) -> Result<bool, CommandError> {
-        Ok(ctx.event.guild_id.is_none())
+        Ok(ctx.event.guild_id.is_some())
+    }
+
+    async fn on_handle_denied(&self, ctx: &CommandContext<State>) -> Result<(), CommandError> {
+        ctx.send()
+            .content("❌ This command can only be used in a server.")
+            .await?;
+
+        Ok(())
     }
 }
 
@@ -20,7 +29,37 @@ pub struct ServerModeratorPermission;
 
 #[async_trait]
 impl Permission<State> for ServerModeratorPermission {
-    async fn check(&self, _ctx: &CommandContext<State>) -> Result<bool, CommandError> {
-        Ok(true)
+    async fn check(&self, ctx: &CommandContext<State>) -> Result<bool, CommandError> {
+        if let Some(guild_id) = ctx.event.guild_id {
+            let permissions = ctx
+                .rt
+                .cache
+                .permissions()
+                .root(ctx.event.author.id, guild_id);
+
+            if let Ok(permissions) = permissions {
+                let guild = ctx
+                    .rt
+                    .cache
+                    .guild(guild_id)
+                    .ok_or(CommandError::CacheStateInconsistent)?;
+
+                Ok(permissions
+                    .contains(Permissions::MANAGE_CHANNELS | Permissions::MANAGE_MESSAGES)
+                    || guild.owner_id() == ctx.event.author.id)
+            } else {
+                Err(CommandError::CacheStateInconsistent)
+            }
+        } else {
+            Ok(true)
+        }
+    }
+
+    async fn on_handle_denied(&self, ctx: &CommandContext<State>) -> Result<(), CommandError> {
+        ctx.send()
+            .content("❌ This command can only be run by a server moderator.\n-# 🛈 Moderators must have permission to manage messages and to manage channels.")
+            .await?;
+
+        Ok(())
     }
 }