[Def] Parse message replies too

Twentysix26 · Jun 9, 2023 · de8f67e · de8f67e
1 parent 095e621
commit de8f67e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/defender/core/events.py b/defender/core/events.py
@@ -29,6 +29,8 @@
 import logging
 import asyncio
 
+ALLOWED_MESSAGE_TYPES = (MessageType.default, MessageType.reply)
+
 log = logging.getLogger("red.x26cogs.defender")
 
 class Events(MixinMeta, metaclass=CompositeMetaClass): # type: ignore
@@ -42,7 +44,7 @@ async def on_message(self, message: discord.Message):
             return
         if author.bot:
             return
-        if message.type != MessageType.default:
+        if message.type not in ALLOWED_MESSAGE_TYPES:
             return
         if not await self.config.guild(guild).enabled():
             return
@@ -153,7 +155,7 @@ async def on_message_edit(self, message_before: discord.Message, message: discor
             return
         if author.bot:
             return
-        if message.type != MessageType.default:
+        if message.type not in ALLOWED_MESSAGE_TYPES:
             return
         if message_before.content == message.content:
             return
@@ -233,7 +235,7 @@ async def on_message_delete(self, message: discord.Message):
             return
         if author.bot:
             return
-        if message.type != MessageType.default:
+        if message.type not in ALLOWED_MESSAGE_TYPES:
             return
 
         if not await self.config.guild(guild).enabled():