KassouBot/src/main.py

print("Chargement des extensions & librairie...", end = " ")

import discord, re, pytz, os
from discord.ext import commands
from random import choice
from datetime import datetime
from pytz import timezone
customPrefix = os.environ['PREFIX']
customTimezone = os.environ['TIMEZONE']

client = commands.Bot(command_prefix = customPrefix, case_insensitive = True, intents = discord.Intents.all())

client.load_extension("cogs.help")
client.load_extension("cogs.utils")
client.load_extension("cogs.internet")
client.load_extension("cogs.music")
client.load_extension("cogs.games")
client.load_extension("cogs.fun")
client.load_extension("cogs.autopublish")
client.load_extension("cogs.school")
print("Terminé !")

@client.event
async def on_connect():
    print(f"Connecté !")

@client.event
async def on_ready():
    await client.change_presence(status = discord.Status.online, activity = discord.Activity(name = f"{customPrefix}help", type = discord.ActivityType.playing))
    print("Bot prêt.")
    channel = client.get_channel(742564480050790512)
    await channel.send("Le bot a bien démarré.")

@client.event
async def on_member_join(member):
    if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
        if member.bot == True:
            role = discord.utils.get(member.guild.roles, name = "Bot")
        else:
            role = discord.utils.get(member.guild.roles, name = "Copain")
        await member.add_roles(role)
        try:
            await member.send(f"Coucou **{member.name}** sur {member.guild.name} ! 🥰\n\nTu as le rôle **{role}** 💖!")
        except:
            pass
        channel = client.get_channel(741639570172674120) # salons des arrivées
        switch = [
            f"Bienvenue, {member.mention}. On espère que tu as apporté de la pizza.",
            f"C'est un plaisir de te voir, {member.mention}.",
            f"{member.mention} vient juste d'arriver !",
            f"{member.mention} vient juste d'atterrir.",
            f"{member.mention} vient de se glisser dans le serveur.",
            f"{member.mention} a bondi dans le serveur.",
            f"Contents de te voir, {member.mention}.",
            f"{member.mention} est arrivé(e).",
            f"Tout le monde, accueillez comme il se doit {member.mention} !",
            f"Youhou, tu as réussi, {member.mention} !",
            f"{member.mention} a rejoint le groupe."
        ]
        message = await channel.send("...") # évite d'envoyer une notification
        await message.edit(content = choice(switch))

@client.event
async def on_member_remove(member):
    if member.guild.id == 441208120644075520: # Confrérie du Kassoulait
        channel = client.get_channel(741639570172674120) # salons des arrivées
        await channel.send(f"{member.mention} vient de quitter le serveur.")

@client.event
async def on_raw_reaction_add(payload):
    if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
        if payload.emoji.name == '✅':
            role = discord.utils.get(payload.member.guild.roles, name="règles-acceptés")
            await payload.member.add_roles(role)

@client.event
async def on_raw_reaction_remove(payload):
    if payload.message_id == 644922358745792512: # Règles de la Confrérie du Kassoulait
        if payload.emoji.name == '✅':
            guild = discord.utils.find(lambda g : g.id == payload.guild_id, client.guilds)
            member = discord.utils.find(lambda m: m.id == payload.user_id, guild.members)
            role = discord.utils.get(guild.roles, name="règles-acceptés")
            await member.remove_roles(role)

@client.event
async def on_command_error(ctx, error):
    if not ctx.invoked_with.startswith('.'):
        print(error)
        await ctx.message.add_reaction(emoji = '❓')

@client.event
async def on_message(message):
    await client.process_commands(message)


    if message.author == client.user:
        return
    
    if client.user.mention == message.content.replace("!",""):
        ctx = await client.get_context(message)
        prefix = await client.get_prefix(message)
        await ctx.send(f">>> Coucou !\nMon préfix est `{prefix}` et ma commande d'aide est `{prefix}help`")
    
    urls = re.findall(r'http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', message.content)
    for i in range(len(urls)):
        if urls[i].startswith("https://discordapp.com/channels/") or urls[i].startswith("https://discord.com/channels/") or urls[i].startswith("https://ptb.discordapp.com/"):
            link = urls[i]
            linkURL = link
            if link.startswith("https://discord.com/channels/"):
                link = f'000{link}'
            if link.startswith("https://ptb.discordapp.com/"):
                link = link[4:]
            if "@me" in urls[i]:
                return await message.channel.send("Je ne cite pas les messages privés.", delete_after = 5)
            try:
                if int(link[32:-38]) == message.guild.id:
                    msgID = await client.get_channel(int(link[51:-19])).fetch_message(int(link[70:]))
                    couleur = 0x2f3136
                    msgFiles = msgID.attachments
                    imageExtensions = ["jpg", "jpeg", "png", "webp", "gif"]
                    desc = msgID.content
                    if len(msgFiles) > 1:
                        listOfFiles = ""
                        for i in range(0, len(msgFiles)):
                            listOfFiles = f"{listOfFiles}, {msgFiles[i].filename}"
                        listOfFiles = listOfFiles[2:]
                        if len(msgID.content) > 0:
                            desc = f"{msgID.content}\n\nIl y a plusieurs fichiers dans ce message : {listOfFiles}"
                        else:
                            desc = f"Il y a plusieurs fichiers dans ce message : {listOfFiles}"
                    else:
                        if len(msgFiles) == 1:
                            if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
                                if not len(msgID.content) > 0:
                                    desc = f"Une image jointe : {msgFiles[0].filename}"
                            else:
                                linkFile = msgFiles[0].url
                                if not len(msgID.content) > 0:
                                    desc = f"Un fichier joint : {msgFiles[0].filename}"
                    embed = discord.Embed(description = desc, colour = couleur)
                    auteur = "Auteur"
                    if message.author == msgID.author:
                        auteur = "Auteur & Citateur"
                    embed.add_field(name = auteur, value = msgID.author.mention, inline=True)
                    try:
                        if len(msgFiles) == 1:
                            if msgFiles[0].filename[-4:].split('.')[1] in imageExtensions:
                                embed.set_image(url=msgFiles[0].url)
                            else:
                                embed.add_field(name = "Fichier", value = f"[Lien]({linkFile})", inline=True)
                    except:
                        pass
                    embed.add_field(name = "Message", value = f"{msgID.channel.mention} - [Lien Message]({linkURL})", inline=True)
                    embed.set_author(name = "Citation", icon_url = msgID.author.avatar_url)
                    icon_url = message.author.avatar_url

                    date_1 = goodTimezone(msgID.created_at, 0)
                    edit = ""
                    if msgID.edited_at:
                        date_edit = goodTimezone(msgID.edited_at, 0)
                        edit = f" et modifié le {date_edit[0][8:]}/{date_edit[0][5:-3]}/{date_edit[0][:4]} à {date_edit[1]}"
                    messageDuBas = f"Posté le {date_1[0][8:]}/{date_1[0][5:-3]}/{date_1[0][:4]} à {date_1[1]}{edit}"

                    date_2 = goodTimezone(message.created_at, 0)
                    date_2 = f"{date_2[0][8:]}/{date_2[0][5:-3]}/{date_2[0][:4]} à {date_2[1]}"
                    
                    if auteur == "Auteur":
                        messageDuBas = messageDuBas + f"\nCité par {user_or_nick(message.author)} le {date_2}"
                    embed.set_footer(icon_url = icon_url, text = messageDuBas)
                    if message.content == linkURL.replace(' ',''):
                        await message.channel.send(embed = embed)
                        await message.delete()
                    else:
                        await message.reply(embed = embed, mention_author = False)
            except Exception as e:
                e = str(e)
                if not "invalid literal for int() with base 10:" in e or not "404 Not Found (error code: 10008)" in e: # faute de frappe / message supprimé
                    print(e)

@client.event
async def on_message_delete(message):
    if message.author.guild.id == 441208120644075520: # Confrérie du Kassoulait
        prefix = await client.get_prefix(message)
        if not (
            message.content.startswith(f"{prefix}note") or
            message.content.startswith(f"{prefix}memo") or
            len(re.findall(".com/channels/", message.content)) != 0 or
            client.user.id is message.author.id
        ):
            user_suppressed = None

            async for entry in message.guild.audit_logs(limit=1):
                if (datetime.now() - entry.created_at).seconds < 5 and str(entry.action) == 'AuditLogAction.message_delete':
                    user_suppressed = entry.user
            
            channel = client.get_channel(742588187456831659)
            embed = discord.Embed(description = f"{message.content}")

            embed.set_author(name = user_or_nick(message.author), icon_url = message.author.avatar_url)

            if not user_suppressed:
                embed.set_footer(text = f"Channel: #{message.channel.name} | Date : {goodTimezone(message.created_at, 1)}\nSupprimé le {datetime.now(pytz.timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
            else:                
                embed.set_footer(icon_url = user_suppressed.avatar_url, text = f"Channel: #{message.channel.name} | Date : {goodTimezone(message.created_at, 1)}\nSupprimé par {user_or_nick(user_suppressed)} le {datetime.now(pytz.timezone(customTimezone)).strftime('%d/%m/%Y à %H:%M:%S')}")
            
            await channel.send(embed = embed)
            # ne fonctionne pas quand un message a été supprimé avant que le bot ai démarré
            # info sur la personne qui a supprimé ne fonctionne pas si il a supprimé un message auparavant (les logs se rajoute a un log deja existant)

def user_or_nick(user):
    if user.nick:
        return f"{user.nick} ({user.name}#{user.discriminator})"
    else:
        return f"{user.name}#{user.discriminator}"

def goodTimezone(date, type):
    if type == 0:
        return str(pytz.timezone(customTimezone).fromutc(date))[:-13].replace('-', '/').split()
    elif type == 1:
        return str(pytz.timezone(customTimezone).fromutc(date))[:-13].replace('-', '/').replace(' ', ' à ')

print("Connexion à Discord...", end = " ")
client.run(os.environ['TOKEN_DISCORD'])