From 23e9529073c1cb59930e0b5d3fba8b358c5155bf Mon Sep 17 00:00:00 2001 From: diamante0018 Date: Tue, 10 Jun 2025 22:00:01 +0200 Subject: [PATCH] feat: lock AI --- bot/events_handlers/message_events.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/bot/events_handlers/message_events.py b/bot/events_handlers/message_events.py index c42290e..be70395 100644 --- a/bot/events_handlers/message_events.py +++ b/bot/events_handlers/message_events.py @@ -23,6 +23,11 @@ SPAM_ROLE_ID = 1350511935677927514 ADMIN_ROLE_ID = 1112364483915042908 GROK_ROLE_ID = 1362837967919386916 +ALLOWED_CHANNELS = [ + 1112048063448617142, # off-topic + 1119371841711112314, # vip-channel +] + # Cooldown: user_id -> [timestamps] MENTION_COOLDOWNS = {} @@ -45,6 +50,14 @@ async def handle_bot_mention(message, bot, no_context=False): staff_role = message.guild.get_role(ADMIN_ROLE_ID) member = message.guild.get_member(message.author.id) + # Check if the message is in an allowed channel + if message.channel.id not in ALLOWED_CHANNELS: + await message.reply( + "The AI cannot used in this channel.", + mention_author=True, + ) + return True + # Cooldown logic: max 1 use per minute per user now = time.time() user_id = message.author.id