Lambda-Bot/LambdaBot.py at main · s0ca/Lambda-Bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import discord
from discord.ext import commands
from discord import app_commands
import settings

class MyBot(commands.Bot):
    def __init__(self):
        intents = discord.Intents.default()
        super().__init__(command_prefix='', intents=intents)

    async def setup_hook(self):
        # Charger les cogs
        await self.load_extension("cogs.music")
        await self.load_extension("cogs.administration")
        # Synchronisation des commandes slash avec Discord.
        self.tree.copy_global_to(guild=discord.Object(id=settings.GUILD))
        await self.tree.sync(guild=discord.Object(id=settings.GUILD))

bot = MyBot()

@bot.event
async def on_ready():
    print(f"{bot.user} est connecté à Discord!")

@bot.tree.command(name="source", description="Affiche le lien source du bot.")
async def source(interaction: discord.Interaction):
    await interaction.response.send_message('https://github.com/s0ca/Lambda-bot')

@bot.tree.command(name="request", description="Demande une nouvelle fonctionnalité.")
async def request(interaction: discord.Interaction, feature: str):
    creator = await bot.fetch_user(settings.OWNER)
    await creator.send(f''':pencil: {interaction.user}: {feature}''')
    await interaction.response.send_message(f''':pencil: Merci, "{feature}" a été demandé!''')

@bot.tree.command(name="report", description="Signale un bug ou une erreur.")
async def report(interaction: discord.Interaction, error_report: str):
    creator = await bot.fetch_user(settings.OWNER)
    await creator.send(f''':triangular_flag_on_post: {interaction.user}: {error_report}''')
    await interaction.response.send_message(f''':triangular_flag_on_post: Merci pour votre aide, "{error_report}" a été signalé!''')

@bot.event
async def on_message(message):
    # Ignorer les messages du bot
    if message.author == bot.user:
        return

    # Chercher des liens YouTube complets ou raccourcis dans le message
    youtube_regex = r'(https?://(?:www\.)?youtube\.com/watch\?[^ ]+|https?://(?:www\.)?youtu\.be/[^ ]+)'
    match = re.search(youtube_regex, message.content)

    if match:
        original_url = match.group(0)
        cleaned_url = clean_youtube_url(original_url)

        # Si le lien a été modifié, renvoyer la version propre
        if original_url != cleaned_url:
            await message.channel.send(f"Lien YouTube nettoyé : {cleaned_url}")
            await message.delete()  # Supprimer l'ancien message contenant le lien avec tracking

    await bot.process_commands(message)

def clean_youtube_url(url):
    # Parse l'URL
    parsed_url = urlparse(url)

    # Gestion des liens 'youtu.be'
    if 'youtu.be' in parsed_url.netloc:
        # Extraire l'ID de la vidéo et ignorer tout ce qui suit un '=' ou un '?'
        video_id = parsed_url.path.split('=')[0].split('?')[0]
        clean_url = f"https://youtu.be{video_id}"
    else:
        # Pour les liens 'youtube.com', ne conserver que le paramètre 'v'
        query_params = parse_qs(parsed_url.query)
        clean_params = {key: value for key, value in query_params.items() if key == 'v'}
        clean_query = urlencode(clean_params, doseq=True)
        clean_url = urlunparse((parsed_url.scheme, parsed_url.netloc, parsed_url.path, parsed_url.params, clean_query, parsed_url.fragment))

    return clean_url

@bot.event
async def on_app_command_error(interaction: discord.Interaction, error):
    if isinstance(error, app_commands.CommandInvokeError):
        await interaction.response.send_message("Une erreur est survenue lors de l'exécution de la commande.")

if __name__ == '__main__':
    bot.run(settings.TOKEN)