[experimental] antispam on_message

app.py

@@ -54,6 +54,7 @@ ping alerts for list of keywords
 '''
 # stats stuff ---------------------------------------------------------------------------------------------------------------------------------------------------------
 number_of_messages = 0
+user_cooldowns = {}
 
 @bot.event
 async def on_message(message):
@@ -64,6 +65,58 @@ async def on_message(message):
             number_of_messages = number_of_messages + 1
             message_link = f"[#{message.channel.name}]({message.jump_url})"
             dm_message = await user.send(f"{number_of_messages}| {message_link} |{message.author}: {message.content}")
+
+
+
+
+            
+            ### antispam experiment
+            cooldown_duration = 1  # 1 message per second
+            if message.author.id not in user_cooldowns:
+                user_cooldowns[message.author.id] = {'count': 1, 'timestamp': message.created_at}
+            else:
+                if (message.created_at - user_cooldowns[message.author.id]['timestamp']).total_seconds() > cooldown_duration:
+                    var1 = message.created_at
+                    var2 = user_cooldowns[message.author.id]['timestamp']
+                    print(f"seconds since last message by {message.author}: ({var1} - {var2}).seconds = {(var1 - var2).total_seconds()}")
+
+                    # if we wait longer than cooldown_duration, count will reset 
+                    user_cooldowns[message.author.id] = {'count': 1, 'timestamp': message.created_at}
+                else:
+                    user_cooldowns[message.author.id]['count'] += 1
+
+                    # tldr; if we post 2 messages with less than 1s between them
+                    if user_cooldowns[message.author.id]['count'] > 1:
+                        var1 = message.created_at
+                        var2 = user_cooldowns[message.author.id]['timestamp']
+                        print(f"seconds since last message by {message.author}: {(var1 - var2).total_seconds()}")   
+                        spam_count = user_cooldowns[message.author.id]['count']
+                        print(f"count: {user_cooldowns[message.author.id]['count']}")
+
+                        test_server = os.environ.get('TEST_SERVER')
+                        if test_server == 'True':
+                            alert = "<@&1106995261487710411>" # test @alerts role
+                        if test_server == 'False':
+                            alert = "<@&1108342563628404747>" # normal @alerts role
+
+                        await bot.log_channel.send(
+                            f"[EXPERIMENTAL ALERT] {message.author.mention} is posting too quickly! \n"
+                            f"Spam count: {user_cooldowns[message.author.id]['count']}\n"
+                            f"Message content: {message.content}\n"
+                            f"[Jump to message!](https://discord.com/channels/{message.guild.id}/{message.channel.id}/{message.id})\n"
+                            f"{alert}"
+                        )
+                    """
+                    if user_cooldowns[message.author.id]['count']/5 > cooldown_duration:
+                        # ping admins
+                        # timeout for user
+                        # kick user
+                    """    
+            user_cooldowns[message.author.id]['timestamp'] = message.created_at  
+
+
+            
+            
     
         # Allow other event handlers to process the message
         await bot.process_commands(message)