blacklist stuff

src/commands/moderation_cmds.rs

@@ -18,20 +18,21 @@ use crate::{
 #[command]
 #[supports_dm(false)]
 #[permission_level(Restricted)]
+#[can_blacklist(false)]
 async fn blacklist(ctx: &Context, msg: &Message, args: String) -> CommandResult {
     let pool = ctx.data.read().await
         .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
 
-    let mut channel = ChannelData::from_id(msg.channel(&ctx).await.unwrap(), pool.clone()).await.unwrap();
+    let mut channel = ChannelData::from_channel(msg.channel(&ctx).await.unwrap(), pool.clone()).await.unwrap();
 
     channel.blacklisted = !channel.blacklisted;
     channel.commit_changes(pool).await;
 
     if channel.blacklisted {
-
+        let _ = msg.channel_id.say(&ctx, "Blacklisted").await;
     }
     else {
-
+        let _ = msg.channel_id.say(&ctx, "Unblacklisted").await;
     }
 
     Ok(())

src/framework.rs

@@ -34,7 +34,10 @@ use std::{
     fmt,
 };
 
-use crate::SQLPool;
+use crate::{
+    models::ChannelData,
+    SQLPool,
+};
 
 type CommandFn = for<'fut> fn(&'fut Context, &'fut Message, String) -> BoxFuture<'fut, CommandResult>;
 
@@ -311,7 +314,13 @@ impl Framework for RegexFramework {
                     match check_self_permissions(&ctx, &guild, &channel).await {
                         Ok(perms) => match perms {
                             PermissionCheck::All => {
+                                let pool = ctx.data.read().await
+                                    .get::<SQLPool>().cloned().expect("Could not get SQLPool from data");
+
                                 let command = self.commands.get(full_match.name("cmd").unwrap().as_str()).unwrap();
+                                let channel_data = ChannelData::from_channel(msg.channel(&ctx).await.unwrap(), pool).await;
+
+                                if !command.can_blacklist || channel_data.map(|c| c.blacklisted).unwrap_or(false) {
                                     let args = full_match.name("args")
                                         .map(|m| m.as_str())
                                         .unwrap_or("")
@@ -321,6 +330,7 @@ impl Framework for RegexFramework {
                                         (command.func)(&ctx, &msg, args).await.unwrap();
                                     }
                                 }
+                            }
 
                             PermissionCheck::Basic => {
                                 let _ = msg.channel_id.say(&ctx, "Not enough perms").await;

src/models.rs

@@ -1,4 +1,5 @@
 use serenity::model::{
+    id::ChannelId,
     guild::Guild,
     channel::Channel
 };
@@ -58,7 +59,7 @@ SELECT id, guild, name, prefix FROM guilds WHERE guild = ?
 }
 
 impl ChannelData {
-    pub async fn from_id(channel: Channel, pool: MySqlPool) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
+    pub async fn from_channel(channel: Channel, pool: MySqlPool) -> Result<Self, Box<dyn std::error::Error + Sync + Send>> {
         let channel_id = channel.id().as_u64().clone();
 
         if let Ok(c) = sqlx::query_as_unchecked!(Self,