mirror of
https://github.com/alterware/aw-bot.git
synced 2025-10-26 14:15:54 +00:00
357 lines
12 KiB
Python
357 lines
12 KiB
Python
import re
|
|
from datetime import timedelta
|
|
|
|
import discord
|
|
|
|
from bot.config import message_patterns
|
|
from bot.utils import (
|
|
aware_utcnow,
|
|
generate_random_nickname,
|
|
is_numeric_name,
|
|
is_valid_username,
|
|
timeout_member,
|
|
)
|
|
|
|
BOT_LOG = 1112049391482703873
|
|
CRAZY_USER_ID = 1319364607487512658
|
|
CRAZY_URL = "https://cdn.discordapp.com/attachments/1119371841711112314/1329770453744746559/download.png"
|
|
crazy_last_response_time = None
|
|
|
|
ALLOWED_CHANNELS = [
|
|
1110531063744303138, # GENERAL_CHANNEL
|
|
1145458108190163014, # mw2 general
|
|
1145456435518525611, # mw2 mp
|
|
1112016681880014928, # mw2 sp
|
|
1200082418481250374, # mw2 dev
|
|
1145459504436220014, # iw5 support
|
|
1145469136919613551, # s1 general
|
|
1145459788151537804, # s1 support
|
|
1145469106133401682, # iw6 general
|
|
1145458770122649691, # iw6 support
|
|
1180796251529293844, # bo3 general
|
|
1180796301953212537, # bo3 support
|
|
BOT_LOG,
|
|
]
|
|
|
|
|
|
async def setup(bot):
|
|
async def detect_ghost_ping(message):
|
|
if not message.mentions:
|
|
return
|
|
|
|
channel = bot.get_channel(BOT_LOG)
|
|
if channel:
|
|
embed = discord.Embed(
|
|
title="Ghost Ping",
|
|
description="A ghost ping was detected.",
|
|
color=0xDD2E44,
|
|
)
|
|
embed.add_field(
|
|
name="Author", value=message.author.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(
|
|
name="Channel", value=message.channel.mention, inline=True
|
|
) # noqa
|
|
|
|
mentioned_users = ", ".join([user.name for user in message.mentions])
|
|
embed.add_field(
|
|
name="Mentions",
|
|
value=f"The message deleted by {message.author} mentioned: {mentioned_users}", # noqa
|
|
inline=False,
|
|
) # noqa
|
|
|
|
embed.set_footer(
|
|
text=f"Message ID: {message.id} | Author ID: {message.author.id}"
|
|
)
|
|
|
|
await channel.send(embed=embed)
|
|
|
|
async def detect_ghost_ping_in_edit(before, after):
|
|
before_mentions = set(before.mentions)
|
|
after_mentions = set(after.mentions)
|
|
|
|
if before_mentions == after_mentions:
|
|
return
|
|
|
|
added_mentions = after_mentions - before_mentions
|
|
removed_mentions = before_mentions - after_mentions
|
|
|
|
response = "The mentions in the message have been edited.\n"
|
|
if added_mentions:
|
|
response += f"Added mentions: {', '.join(user.name for user in added_mentions)}\n" # noqa
|
|
if removed_mentions:
|
|
response += f"Removed mentions: {', '.join(user.name for user in removed_mentions)}" # noqa
|
|
|
|
channel = bot.get_channel(BOT_LOG)
|
|
if channel:
|
|
embed = discord.Embed(
|
|
title="Ghost Ping",
|
|
description="A ghost ping was detected.",
|
|
color=0xDD2E44,
|
|
)
|
|
embed.add_field(
|
|
name="Author", value=before.author.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(
|
|
name="Channel", value=before.channel.mention, inline=True
|
|
) # noqa
|
|
|
|
embed.add_field(
|
|
name="Mentions",
|
|
value=response,
|
|
inline=False,
|
|
) # noqa
|
|
|
|
embed.set_footer(
|
|
text=f"Message ID: {before.id} | Author ID: {before.author.id}"
|
|
)
|
|
|
|
await channel.send(embed=embed)
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
global crazy_last_response_time
|
|
|
|
if message.author == bot.user:
|
|
return
|
|
|
|
# Too many mentions
|
|
if len(message.mentions) >= 3:
|
|
member = message.guild.get_member(message.author.id)
|
|
await timeout_member(member)
|
|
await message.delete()
|
|
return
|
|
|
|
# Auto delete torrent if post in chat.
|
|
for file in message.attachments:
|
|
if file.filename.endswith((".torrent", ".TORRENT")):
|
|
member = message.guild.get_member(message.author.id)
|
|
await timeout_member(member)
|
|
await message.delete()
|
|
|
|
if message.author.id == CRAZY_USER_ID:
|
|
now = aware_utcnow()
|
|
if (
|
|
crazy_last_response_time is None
|
|
or now - crazy_last_response_time >= timedelta(hours=8)
|
|
):
|
|
crazy_last_response_time = now
|
|
await message.channel.send(f"{CRAZY_URL}")
|
|
|
|
guild = message.guild
|
|
for channel in guild.text_channels:
|
|
if channel.id == message.channel.id:
|
|
continue
|
|
|
|
try:
|
|
async for msg in channel.history(limit=5):
|
|
# Too many false positives
|
|
if msg.embeds:
|
|
continue
|
|
# ^^
|
|
if message.attachments:
|
|
continue
|
|
# ^^
|
|
if not message.content.strip():
|
|
continue
|
|
|
|
if msg.author == message.author and msg.content == message.content:
|
|
current_time = aware_utcnow()
|
|
message_time = msg.created_at
|
|
|
|
time_difference = current_time - message_time
|
|
if time_difference >= timedelta(minutes=5):
|
|
continue
|
|
|
|
await message.channel.send(
|
|
f"Hey {message.author.name}, you've already sent this message in {channel.mention}!"
|
|
)
|
|
member = message.guild.get_member(message.author.id)
|
|
await timeout_member(member)
|
|
return
|
|
except discord.Forbidden:
|
|
print(
|
|
f"Bot does not have permission to read messages in {channel.name}."
|
|
)
|
|
except discord.HTTPException as e:
|
|
print(f"An error occurred: {e}")
|
|
|
|
# Check if the message is in an allowed channel
|
|
if message.channel.id not in ALLOWED_CHANNELS:
|
|
return
|
|
|
|
# Check if any of the patterns match the message
|
|
# print('Checking for patterns...')
|
|
for pattern in message_patterns:
|
|
if re.search(pattern["regex"], message.content, re.IGNORECASE):
|
|
response = pattern["response"]
|
|
reply_message = await message.reply(response, mention_author=True)
|
|
await reply_message.add_reaction("\U0000274C")
|
|
break
|
|
|
|
@bot.event
|
|
async def on_reaction_add(reaction, user):
|
|
# Ignore reactions from the bot itself
|
|
if user == bot.user:
|
|
return
|
|
|
|
if reaction.emoji != "\U0000274C":
|
|
return
|
|
|
|
if reaction.message.author != bot.user:
|
|
return
|
|
|
|
current_time = aware_utcnow()
|
|
time_difference = current_time - reaction.message.created_at
|
|
if time_difference >= timedelta(minutes=5):
|
|
return
|
|
|
|
if reaction.message.reference is None:
|
|
return
|
|
|
|
original_message = await reaction.message.channel.fetch_message(
|
|
reaction.message.reference.message_id
|
|
)
|
|
|
|
if original_message.author == user:
|
|
await reaction.message.delete()
|
|
else:
|
|
# If the user is not the original author, remove their reaction
|
|
await reaction.remove(user)
|
|
|
|
@bot.event
|
|
async def on_member_join(member):
|
|
name_to_check = member.name
|
|
|
|
if member.display_name:
|
|
name_to_check = member.display_name
|
|
|
|
if (
|
|
len(name_to_check) < 3
|
|
or not is_valid_username(name_to_check)
|
|
or is_numeric_name(name_to_check)
|
|
):
|
|
new_nick = generate_random_nickname()
|
|
await member.edit(nick=new_nick)
|
|
|
|
@bot.event
|
|
async def on_member_update(before, after):
|
|
name_to_check = after.name
|
|
|
|
if after.nick:
|
|
name_to_check = after.nick
|
|
|
|
if (
|
|
len(name_to_check) < 3
|
|
or not is_valid_username(name_to_check)
|
|
or is_numeric_name(name_to_check)
|
|
):
|
|
new_nick = generate_random_nickname()
|
|
await after.edit(nick=new_nick)
|
|
|
|
@bot.event
|
|
async def on_message_delete(message):
|
|
channel = bot.get_channel(BOT_LOG)
|
|
if not channel:
|
|
return
|
|
|
|
is_bot = message.author == bot.user
|
|
if is_bot and message.channel.id != BOT_LOG:
|
|
return
|
|
|
|
if is_bot:
|
|
await message.channel.send(
|
|
"You attempted to delete a message from a channel where messages are logged and stored indefinitely. Please refrain from doing so." # noqa
|
|
) # noqa
|
|
# It is impossible to recover the message at this point
|
|
return
|
|
|
|
embed = discord.Embed(
|
|
title="Deleted Message",
|
|
description="A message was deleted.",
|
|
color=0xDD2E44,
|
|
)
|
|
embed.add_field(
|
|
name="Author", value=message.author.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(
|
|
name="Channel", value=message.channel.mention, inline=True
|
|
) # noqa
|
|
if message.content:
|
|
embed.add_field(name="Content", value=message.content, inline=False) # noqa
|
|
|
|
if message.reference is not None:
|
|
original_message = await message.channel.fetch_message(
|
|
message.reference.message_id
|
|
)
|
|
|
|
embed.add_field(
|
|
name="Replied",
|
|
value=original_message.author.mention,
|
|
inline=False, # noqa
|
|
) # noqa
|
|
|
|
embed.set_footer(
|
|
text=f"Message ID: {message.id} | Author ID: {message.author.id}" # noqa
|
|
) # noqa
|
|
|
|
await detect_ghost_ping(message)
|
|
await channel.send(embed=embed)
|
|
|
|
@bot.event
|
|
async def on_bulk_message_delete(messages):
|
|
channel = bot.get_channel(BOT_LOG)
|
|
if channel:
|
|
for message in messages:
|
|
embed = discord.Embed(
|
|
title="Deleted Message",
|
|
description="A message was deleted.",
|
|
color=0xDD2E44,
|
|
)
|
|
embed.add_field(
|
|
name="Author", value=message.author.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(
|
|
name="Channel", value=message.channel.mention, inline=True
|
|
) # noqa
|
|
if message.content:
|
|
embed.add_field(
|
|
name="Content", value=message.content, inline=False
|
|
) # noqa
|
|
embed.set_footer(
|
|
text=f"Message ID: {message.id} | Author ID: {message.author.id}" # noqa
|
|
)
|
|
|
|
await channel.send(embed=embed)
|
|
|
|
@bot.event
|
|
async def on_message_edit(before, after):
|
|
channel = bot.get_channel(BOT_LOG)
|
|
if channel:
|
|
if not before.content:
|
|
return
|
|
|
|
if after.content and after.content == before.content:
|
|
return
|
|
|
|
embed = discord.Embed(
|
|
title="Edited Message",
|
|
description="A message was edited.",
|
|
color=0xDD2E44,
|
|
)
|
|
embed.add_field(
|
|
name="Author", value=before.author.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(
|
|
name="Channel", value=before.channel.mention, inline=True
|
|
) # noqa
|
|
embed.add_field(name="Content", value=before.content, inline=False) # noqa
|
|
embed.set_footer(
|
|
text=f"Message ID: {before.id} | Author ID: {before.author.id}"
|
|
)
|
|
|
|
await detect_ghost_ping_in_edit(before, after)
|
|
await channel.send(embed=embed)
|
|
|
|
print("Events extension loaded!")
|