The begining
This commit is contained in:
commit
dee311e709
71 changed files with 8349 additions and 0 deletions
193
bot.py
Executable file
193
bot.py
Executable file
|
|
@ -0,0 +1,193 @@
|
|||
import discord
|
||||
from discord.ext import commands
|
||||
import os
|
||||
import asyncio
|
||||
from dotenv import load_dotenv
|
||||
from src.logger import kuby_logger, log_performance
|
||||
import logging
|
||||
from src.advanced_logger import AdvancedLogger
|
||||
|
||||
# Chargement des variables d'environnement
|
||||
load_dotenv()
|
||||
|
||||
# Récupération et validation de l'ID d'application
|
||||
application_id_raw = os.getenv("APPLICATION_ID")
|
||||
kuby_logger.debug(f"Raw APPLICATION_ID: {application_id_raw}")
|
||||
try:
|
||||
application_id = int(application_id_raw) if application_id_raw else None
|
||||
kuby_logger.info(f"✅ APPLICATION_ID loaded successfully: {application_id}")
|
||||
except ValueError:
|
||||
kuby_logger.error(f"❌ APPLICATION_ID invalide : {application_id_raw}")
|
||||
application_id = None
|
||||
|
||||
# Configuration des intents Discord
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
intents.members = True
|
||||
intents.voice_states = True
|
||||
kuby_logger.debug(f"Discord intents configured: {intents}")
|
||||
|
||||
class MyBot(commands.Bot):
|
||||
def __init__(self):
|
||||
kuby_logger.debug("Initializing MyBot class")
|
||||
super().__init__(
|
||||
command_prefix="!",
|
||||
intents=intents,
|
||||
application_id=application_id,
|
||||
)
|
||||
kuby_logger.info("MyBot instance created successfully")
|
||||
|
||||
async def setup_hook(self):
|
||||
kuby_logger.info("🚀 Démarrage du bot Kuby...")
|
||||
|
||||
# Initialize advanced logger
|
||||
self.advanced_logger = AdvancedLogger(self)
|
||||
kuby_logger.info("✅ Advanced logger initialized")
|
||||
|
||||
# Charger les extensions
|
||||
extensions = [
|
||||
"commandes.whitelist",
|
||||
"commandes.whitelist_monitor",
|
||||
"commandes.sendbotmessages",
|
||||
"commandes.goodbye",
|
||||
"commandes.welcome",
|
||||
"commandes.fun",
|
||||
"commandes.security",
|
||||
"commandes.snipe",
|
||||
"commandes.scan",
|
||||
"commandes.no_link",
|
||||
"commandes.blacklist",
|
||||
"commandes.ticket",
|
||||
"commandes.modules_security.antispam",
|
||||
"commandes.modules_security.antiraid",
|
||||
"commandes.modules_security.logs_manager",
|
||||
]
|
||||
|
||||
for extension in extensions:
|
||||
try:
|
||||
await self.load_extension(extension)
|
||||
kuby_logger.info(f"✅ Extension chargée : {extension}")
|
||||
except Exception as e:
|
||||
kuby_logger.error(f"❌ Erreur lors du chargement de {extension} : {e}", exc_info=True)
|
||||
|
||||
# --- SYSTÈME DE WHITELIST (CORRIGÉ POUR ÉVITER LES ERREURS) ---
|
||||
@self.check
|
||||
async def whitelist_check(ctx):
|
||||
try:
|
||||
# 1. Bypass pour le propriétaire (Owner)
|
||||
if await self.is_owner(ctx.author):
|
||||
return True
|
||||
|
||||
# 2. Récupération du Cog
|
||||
whitelist_cog = self.get_cog("WhitelistMonitor")
|
||||
if not whitelist_cog:
|
||||
kuby_logger.warning("WhitelistMonitor cog not found, allowing all commands")
|
||||
return True
|
||||
|
||||
is_whitelisted = whitelist_cog.is_whitelisted(ctx.author.id)
|
||||
|
||||
if not is_whitelisted:
|
||||
kuby_logger.info(f"User {ctx.author} (ID: {ctx.author.id}) attempted to use command but is not whitelisted")
|
||||
|
||||
# Réponse adaptée (Slash vs Message)
|
||||
msg = "❌ Vous n'êtes pas autorisé à utiliser cette commande. Veuillez demander à être ajouté à la whitelist."
|
||||
if ctx.interaction:
|
||||
await ctx.interaction.response.send_message(msg, ephemeral=True)
|
||||
else:
|
||||
await ctx.send(msg)
|
||||
return False
|
||||
|
||||
return True
|
||||
except Exception as e:
|
||||
kuby_logger.error(f"Error in whitelist check: {e}")
|
||||
return True
|
||||
|
||||
# Sync slash commands
|
||||
try:
|
||||
await self.tree.sync()
|
||||
kuby_logger.info("✅ Commandes slash synchronisées avec Discord")
|
||||
except Exception as e:
|
||||
kuby_logger.error(f"❌ Erreur lors de la synchronisation des commandes slash : {e}", exc_info=True)
|
||||
|
||||
# --- ÉVÉNEMENTS RESTAURÉS ---
|
||||
|
||||
async def on_ready(self):
|
||||
kuby_logger.info(f"🤖 Bot connecté : {self.user} (ID : {self.user.id})")
|
||||
kuby_logger.info(f"📊 Serveurs : {len(self.guilds)}")
|
||||
for guild in self.guilds:
|
||||
kuby_logger.debug(f"Guild: {guild.name} (ID: {guild.id}) - Members: {guild.member_count}")
|
||||
|
||||
total_members = sum(guild.member_count for guild in self.guilds)
|
||||
kuby_logger.info(f"📊 Total members across all guilds: {total_members}")
|
||||
kuby_logger.info("✅ Bot prêt et opérationnel !")
|
||||
|
||||
async def on_guild_join(self, guild):
|
||||
kuby_logger.info(f"🆕 Bot joined new guild: {guild.name} (ID: {guild.id})")
|
||||
kuby_logger.audit(f"Bot added to guild: {guild.name}")
|
||||
|
||||
async def on_guild_remove(self, guild):
|
||||
kuby_logger.info(f"🚪 Bot removed from guild: {guild.name}")
|
||||
kuby_logger.audit(f"Bot removed from guild: {guild.name}")
|
||||
|
||||
async def on_member_join(self, member):
|
||||
kuby_logger.info(f"👋 {member} joined {member.guild.name}")
|
||||
if hasattr(self, 'advanced_logger'):
|
||||
self.advanced_logger.log_member_join(member)
|
||||
previous_roles = self.advanced_logger.check_previous_roles(member.id)
|
||||
if previous_roles:
|
||||
guild = member.guild
|
||||
roles_to_add = [guild.get_role(rid) for rid in previous_roles.get("role_ids", [])]
|
||||
roles_to_add = [r for r in roles_to_add if r and r < guild.me.top_role]
|
||||
if roles_to_add:
|
||||
try:
|
||||
await member.add_roles(*roles_to_add)
|
||||
kuby_logger.info(f"Restored {len(roles_to_add)} roles for {member}")
|
||||
except Exception as e:
|
||||
kuby_logger.error(f"Failed to restore roles: {e}")
|
||||
|
||||
async def on_member_remove(self, member):
|
||||
kuby_logger.info(f"👋 {member} left {member.guild.name}")
|
||||
if hasattr(self, 'advanced_logger'):
|
||||
self.advanced_logger.log_member_leave(member)
|
||||
# Notification de départ
|
||||
try:
|
||||
from commandes.security import load_settings
|
||||
settings = load_settings(member.guild.id)
|
||||
log_channel_id = settings.get("log_channel_id")
|
||||
if log_channel_id:
|
||||
log_channel = member.guild.get_channel(log_channel_id)
|
||||
if log_channel:
|
||||
await self.advanced_logger.send_leave_notification(member, log_channel)
|
||||
except Exception as e:
|
||||
kuby_logger.error(f"Error in leave notification: {e}")
|
||||
|
||||
async def on_message_delete(self, message):
|
||||
if message.author.bot: return
|
||||
if hasattr(self, 'advanced_logger'):
|
||||
self.advanced_logger.log_message_event(message, "deleted")
|
||||
|
||||
async def on_message_edit(self, before, after):
|
||||
if before.author.bot or before.content == after.content: return
|
||||
if hasattr(self, 'advanced_logger'):
|
||||
self.advanced_logger.log_message_event(before, "edited", {
|
||||
"before_content": before.content, "after_content": after.content
|
||||
})
|
||||
|
||||
async def on_voice_state_update(self, member, before, after):
|
||||
if member.bot: return
|
||||
if before.channel != after.channel and hasattr(self, 'advanced_logger'):
|
||||
if before.channel is None:
|
||||
self.advanced_logger.log_voice_event(member, after.channel, "joined")
|
||||
elif after.channel is None:
|
||||
self.advanced_logger.log_voice_event(member, before.channel, "left")
|
||||
else:
|
||||
self.advanced_logger.log_voice_event(member, before.channel, "moved_from")
|
||||
self.advanced_logger.log_voice_event(member, after.channel, "moved_to")
|
||||
|
||||
async def on_command_error(self, ctx, error):
|
||||
if isinstance(error, commands.CheckFailure):
|
||||
return # Déjà géré dans le whitelist_check
|
||||
kuby_logger.error(f"Command error in {ctx.command}: {str(error)}", exc_info=True)
|
||||
|
||||
bot = MyBot()
|
||||
bot.run(os.getenv("DISCORD_TOKEN"))
|
||||
Loading…
Add table
Add a link
Reference in a new issue