600 lines
No EOL
28 KiB
Python
600 lines
No EOL
28 KiB
Python
import discord
|
||
from discord import app_commands
|
||
from discord.ext import commands, tasks
|
||
import asyncio
|
||
from datetime import datetime, timedelta
|
||
import logging
|
||
import feedparser
|
||
|
||
logger = logging.getLogger('bot')
|
||
|
||
class System(commands.Cog):
|
||
def __init__(self, bot):
|
||
self.bot = bot
|
||
self.check_youtube.start()
|
||
|
||
def cog_unload(self):
|
||
self.check_youtube.cancel()
|
||
|
||
@tasks.loop(seconds=300)
|
||
async def check_youtube(self):
|
||
if 'youtube' not in self.bot.db.data:
|
||
return
|
||
|
||
for guild_id, settings in self.bot.db.data['youtube'].items():
|
||
if not settings.get('enabled') or not settings.get('channel_id') or not settings.get('youtube_channel_id'):
|
||
continue
|
||
|
||
try:
|
||
feed_url = f'https://www.youtube.com/feeds/videos.xml?channel_id={settings["youtube_channel_id"]}'
|
||
feed = await asyncio.to_thread(feedparser.parse, feed_url)
|
||
|
||
if not feed.entries:
|
||
continue
|
||
|
||
latest = feed.entries[0]
|
||
video_id = latest.yt_videoid if hasattr(latest, 'yt_videoid') else latest.id.split(':')[-1]
|
||
|
||
if video_id != settings.get('last_video_id') and settings.get('last_video_id'):
|
||
guild = self.bot.get_guild(int(guild_id))
|
||
if not guild:
|
||
continue
|
||
|
||
channel = guild.get_channel(int(settings['channel_id']))
|
||
if not channel:
|
||
continue
|
||
|
||
embed = discord.Embed(
|
||
title='🎬 New YouTube Video!',
|
||
description=f'**{latest.title}**',
|
||
url=latest.link,
|
||
color=0xFF0000,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
|
||
if hasattr(latest, 'media_thumbnail') and latest.media_thumbnail:
|
||
embed.set_thumbnail(url=latest.media_thumbnail[0]['url'])
|
||
|
||
embed.add_field(name='Channel', value=latest.author, inline=True)
|
||
|
||
if hasattr(latest, 'published'):
|
||
embed.add_field(name='Published', value=latest.published, inline=True)
|
||
|
||
await channel.send('📺 New video alert! @everyone', embed=embed)
|
||
logger.info(f'📺 Sent notification for: {latest.title}')
|
||
|
||
settings['last_video_id'] = video_id
|
||
self.bot.db.save()
|
||
|
||
except Exception as e:
|
||
logger.error(f'❌ Error checking YouTube: {e}')
|
||
|
||
@check_youtube.before_loop
|
||
async def before_check_youtube(self):
|
||
await self.bot.wait_until_ready()
|
||
|
||
@app_commands.command(name='ping', description='Check bot latency')
|
||
async def ping(self, interaction: discord.Interaction):
|
||
latency = round(self.bot.latency * 1000)
|
||
await interaction.response.send_message(f'🏓 Pong! Latency: `{latency}ms`')
|
||
|
||
@app_commands.command(name='serverinfo', description='Display server information')
|
||
async def serverinfo(self, interaction: discord.Interaction):
|
||
guild = interaction.guild
|
||
|
||
embed = discord.Embed(
|
||
title=f'📊 {guild.name}',
|
||
color=0x5865F2,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
|
||
if guild.icon:
|
||
embed.set_thumbnail(url=guild.icon.url)
|
||
|
||
embed.add_field(name='👑 Owner', value=guild.owner.mention, inline=True)
|
||
embed.add_field(name='👥 Members', value=guild.member_count, inline=True)
|
||
embed.add_field(name='📅 Created', value=guild.created_at.strftime('%Y-%m-%d'), inline=True)
|
||
embed.add_field(name='💬 Channels', value=len(guild.channels), inline=True)
|
||
embed.add_field(name='😀 Emojis', value=len(guild.emojis), inline=True)
|
||
embed.add_field(name='🎭 Roles', value=len(guild.roles), inline=True)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
|
||
@app_commands.command(name='userinfo', description='Display user information')
|
||
async def userinfo(self, interaction: discord.Interaction, member: discord.Member = None):
|
||
target = member or interaction.user
|
||
|
||
embed = discord.Embed(
|
||
title=f'👤 {target.name}',
|
||
color=target.color if target.color != discord.Color.default() else 0x5865F2,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
|
||
embed.set_thumbnail(url=target.display_avatar.url)
|
||
|
||
embed.add_field(name='🆔 ID', value=target.id, inline=True)
|
||
embed.add_field(name='📛 Nickname', value=target.nick or 'None', inline=True)
|
||
embed.add_field(name='🤖 Bot', value='Yes' if target.bot else 'No', inline=True)
|
||
embed.add_field(name='📅 Joined Server', value=target.joined_at.strftime('%Y-%m-%d %H:%M') if target.joined_at else 'Unknown', inline=True)
|
||
embed.add_field(name='📅 Account Created', value=target.created_at.strftime('%Y-%m-%d %H:%M'), inline=True)
|
||
|
||
roles = [role.mention for role in target.roles if role.name != '@everyone']
|
||
embed.add_field(name=f'🎭 Roles ({len(roles)})', value=' '.join(roles) if roles else 'None', inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
|
||
@app_commands.command(name='kick', description='[MOD] Kick a member from the server')
|
||
@app_commands.describe(member='Member to kick', reason='Reason for kick')
|
||
@app_commands.default_permissions(kick_members=True)
|
||
async def kick(self, interaction: discord.Interaction, member: discord.Member, reason: str = 'No reason provided'):
|
||
if member.top_role >= interaction.user.top_role and interaction.user != interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot kick this member!', ephemeral=True)
|
||
return
|
||
|
||
if member == interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot kick the server owner!', ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
await member.kick(reason=f'{reason} | Kicked by {interaction.user}')
|
||
|
||
embed = discord.Embed(
|
||
title='👢 Member Kicked',
|
||
description=f'{member.mention} has been kicked from the server',
|
||
color=0xFF9500,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
embed.add_field(name='Reason', value=reason, inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'👢 {member} kicked by {interaction.user} - Reason: {reason}')
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to kick this member!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='ban', description='[MOD] Ban a member from the server')
|
||
@app_commands.describe(member='Member to ban', reason='Reason for ban', delete_days='Days of messages to delete (0-7)')
|
||
@app_commands.default_permissions(ban_members=True)
|
||
async def ban(self, interaction: discord.Interaction, member: discord.Member, reason: str = 'No reason provided', delete_days: int = 0):
|
||
if member.top_role >= interaction.user.top_role and interaction.user != interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot ban this member!', ephemeral=True)
|
||
return
|
||
|
||
if member == interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot ban the server owner!', ephemeral=True)
|
||
return
|
||
|
||
if delete_days < 0 or delete_days > 7:
|
||
await interaction.response.send_message('❌ Delete days must be between 0 and 7!', ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
await member.ban(reason=f'{reason} | Banned by {interaction.user}', delete_message_days=delete_days)
|
||
|
||
embed = discord.Embed(
|
||
title='🔨 Member Banned',
|
||
description=f'{member.mention} has been banned from the server',
|
||
color=0xFF0000,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
embed.add_field(name='Reason', value=reason, inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'🔨 {member} banned by {interaction.user} - Reason: {reason}')
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to ban this member!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='unban', description='[MOD] Unban a user from the server')
|
||
@app_commands.describe(user_id='User ID to unban', reason='Reason for unban')
|
||
@app_commands.default_permissions(ban_members=True)
|
||
async def unban(self, interaction: discord.Interaction, user_id: str, reason: str = 'No reason provided'):
|
||
try:
|
||
user = await self.bot.fetch_user(int(user_id))
|
||
except:
|
||
await interaction.response.send_message('❌ Invalid user ID!', ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
await interaction.guild.unban(user, reason=f'{reason} | Unbanned by {interaction.user}')
|
||
|
||
embed = discord.Embed(
|
||
title='✅ User Unbanned',
|
||
description=f'{user.mention} has been unbanned from the server',
|
||
color=0x00FF00,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
embed.add_field(name='Reason', value=reason, inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'✅ {user} unbanned by {interaction.user} - Reason: {reason}')
|
||
except discord.NotFound:
|
||
await interaction.response.send_message('❌ This user is not banned!', ephemeral=True)
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to unban users!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='timeout', description='[MOD] Timeout a member')
|
||
@app_commands.describe(member='Member to timeout', duration='Duration in minutes', reason='Reason for timeout')
|
||
@app_commands.default_permissions(moderate_members=True)
|
||
async def timeout(self, interaction: discord.Interaction, member: discord.Member, duration: int, reason: str = 'No reason provided'):
|
||
if member.top_role >= interaction.user.top_role and interaction.user != interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot timeout this member!', ephemeral=True)
|
||
return
|
||
|
||
if member == interaction.guild.owner:
|
||
await interaction.response.send_message('❌ You cannot timeout the server owner!', ephemeral=True)
|
||
return
|
||
|
||
if duration < 1 or duration > 40320:
|
||
await interaction.response.send_message('❌ Duration must be between 1 minute and 28 days!', ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
timeout_until = datetime.utcnow() + timedelta(minutes=duration)
|
||
await member.timeout(timeout_until, reason=f'{reason} | Timed out by {interaction.user}')
|
||
|
||
embed = discord.Embed(
|
||
title='⏰ Member Timed Out',
|
||
description=f'{member.mention} has been timed out',
|
||
color=0xFFA500,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
embed.add_field(name='Duration', value=f'{duration} minutes', inline=True)
|
||
embed.add_field(name='Reason', value=reason, inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'⏰ {member} timed out by {interaction.user} for {duration}m - Reason: {reason}')
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to timeout this member!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='warn', description='[MOD] Warn a member')
|
||
@app_commands.describe(member='Member to warn', reason='Reason for warning')
|
||
@app_commands.default_permissions(moderate_members=True)
|
||
async def warn(self, interaction: discord.Interaction, member: discord.Member, reason: str):
|
||
guild_id = str(interaction.guild.id)
|
||
user_id = str(member.id)
|
||
|
||
if 'warnings' not in self.bot.db.data:
|
||
self.bot.db.data['warnings'] = {}
|
||
|
||
if guild_id not in self.bot.db.data['warnings']:
|
||
self.bot.db.data['warnings'][guild_id] = {}
|
||
|
||
if user_id not in self.bot.db.data['warnings'][guild_id]:
|
||
self.bot.db.data['warnings'][guild_id][user_id] = []
|
||
|
||
warning = {
|
||
'reason': reason,
|
||
'moderator': str(interaction.user.id),
|
||
'timestamp': datetime.utcnow().isoformat()
|
||
}
|
||
|
||
self.bot.db.data['warnings'][guild_id][user_id].append(warning)
|
||
self.bot.db.save()
|
||
|
||
warning_count = len(self.bot.db.data['warnings'][guild_id][user_id])
|
||
|
||
embed = discord.Embed(
|
||
title='⚠️ Member Warned',
|
||
description=f'{member.mention} has been warned',
|
||
color=0xFFFF00,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
embed.add_field(name='Total Warnings', value=warning_count, inline=True)
|
||
embed.add_field(name='Reason', value=reason, inline=False)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
|
||
try:
|
||
await member.send(f'⚠️ You have been warned in **{interaction.guild.name}**\n**Reason:** {reason}\n**Total Warnings:** {warning_count}')
|
||
except:
|
||
pass
|
||
|
||
logger.info(f'⚠️ {member} warned by {interaction.user} - Reason: {reason}')
|
||
|
||
@app_commands.command(name='warnings', description='View warnings for a member')
|
||
@app_commands.describe(member='Member to check warnings for')
|
||
async def warnings(self, interaction: discord.Interaction, member: discord.Member = None):
|
||
target = member or interaction.user
|
||
guild_id = str(interaction.guild.id)
|
||
user_id = str(target.id)
|
||
|
||
if 'warnings' not in self.bot.db.data:
|
||
self.bot.db.data['warnings'] = {}
|
||
|
||
user_warnings = self.bot.db.data['warnings'].get(guild_id, {}).get(user_id, [])
|
||
|
||
if not user_warnings:
|
||
await interaction.response.send_message(f'{target.mention} has no warnings!', ephemeral=True)
|
||
return
|
||
|
||
embed = discord.Embed(
|
||
title=f'⚠️ Warnings for {target.display_name}',
|
||
color=0xFFFF00,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
|
||
for i, warning in enumerate(user_warnings[-10:], 1):
|
||
moderator = interaction.guild.get_member(int(warning['moderator']))
|
||
mod_name = moderator.mention if moderator else f"<@{warning['moderator']}>"
|
||
timestamp = datetime.fromisoformat(warning['timestamp']).strftime('%Y-%m-%d %H:%M')
|
||
|
||
embed.add_field(
|
||
name=f'Warning #{i}',
|
||
value=f'**Moderator:** {mod_name}\n**Reason:** {warning["reason"]}\n**Date:** {timestamp}',
|
||
inline=False
|
||
)
|
||
|
||
embed.set_footer(text=f'Total warnings: {len(user_warnings)} | Showing last 10')
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
|
||
@app_commands.command(name='clearwarnings', description='[MOD] Clear warnings for a member')
|
||
@app_commands.describe(member='Member to clear warnings for')
|
||
@app_commands.default_permissions(moderate_members=True)
|
||
async def clearwarnings(self, interaction: discord.Interaction, member: discord.Member):
|
||
guild_id = str(interaction.guild.id)
|
||
user_id = str(member.id)
|
||
|
||
if 'warnings' not in self.bot.db.data:
|
||
self.bot.db.data['warnings'] = {}
|
||
|
||
if guild_id not in self.bot.db.data['warnings'] or user_id not in self.bot.db.data['warnings'][guild_id]:
|
||
await interaction.response.send_message(f'{member.mention} has no warnings to clear!', ephemeral=True)
|
||
return
|
||
|
||
warning_count = len(self.bot.db.data['warnings'][guild_id][user_id])
|
||
self.bot.db.data['warnings'][guild_id][user_id] = []
|
||
self.bot.db.save()
|
||
|
||
embed = discord.Embed(
|
||
title='✅ Warnings Cleared',
|
||
description=f'Cleared {warning_count} warning(s) for {member.mention}',
|
||
color=0x00FF00,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'✅ Cleared {warning_count} warnings for {member} by {interaction.user}')
|
||
|
||
@app_commands.command(name='purge', description='[MOD] Delete multiple messages')
|
||
@app_commands.describe(amount='Number of messages to delete (1-100)', member='Only delete messages from this member (optional)')
|
||
@app_commands.default_permissions(manage_messages=True)
|
||
async def purge(self, interaction: discord.Interaction, amount: int, member: discord.Member = None):
|
||
if amount < 1 or amount > 100:
|
||
await interaction.response.send_message('❌ Amount must be between 1 and 100!', ephemeral=True)
|
||
return
|
||
|
||
await interaction.response.defer(ephemeral=True)
|
||
|
||
try:
|
||
if member:
|
||
deleted = await interaction.channel.purge(limit=amount, check=lambda m: m.author == member)
|
||
else:
|
||
deleted = await interaction.channel.purge(limit=amount)
|
||
|
||
await interaction.followup.send(f'✅ Deleted {len(deleted)} message(s)!', ephemeral=True)
|
||
logger.info(f'🗑️ {interaction.user} purged {len(deleted)} messages in #{interaction.channel.name}')
|
||
except discord.Forbidden:
|
||
await interaction.followup.send('❌ I do not have permission to delete messages!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.followup.send(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='lock', description='[MOD] Lock a channel')
|
||
@app_commands.describe(channel='Channel to lock (defaults to current channel)')
|
||
@app_commands.default_permissions(manage_channels=True)
|
||
async def lock(self, interaction: discord.Interaction, channel: discord.TextChannel = None):
|
||
target_channel = channel or interaction.channel
|
||
|
||
try:
|
||
await target_channel.set_permissions(interaction.guild.default_role, send_messages=False)
|
||
|
||
embed = discord.Embed(
|
||
title='🔒 Channel Locked',
|
||
description=f'{target_channel.mention} has been locked',
|
||
color=0xFF0000,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'🔒 {target_channel.name} locked by {interaction.user}')
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to lock this channel!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@app_commands.command(name='unlock', description='[MOD] Unlock a channel')
|
||
@app_commands.describe(channel='Channel to unlock (defaults to current channel)')
|
||
@app_commands.default_permissions(manage_channels=True)
|
||
async def unlock(self, interaction: discord.Interaction, channel: discord.TextChannel = None):
|
||
target_channel = channel or interaction.channel
|
||
|
||
try:
|
||
await target_channel.set_permissions(interaction.guild.default_role, send_messages=None)
|
||
|
||
embed = discord.Embed(
|
||
title='🔓 Channel Unlocked',
|
||
description=f'{target_channel.mention} has been unlocked',
|
||
color=0x00FF00,
|
||
timestamp=datetime.utcnow()
|
||
)
|
||
embed.add_field(name='Moderator', value=interaction.user.mention, inline=True)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'🔓 {target_channel.name} unlocked by {interaction.user}')
|
||
except discord.Forbidden:
|
||
await interaction.response.send_message('❌ I do not have permission to unlock this channel!', ephemeral=True)
|
||
except Exception as e:
|
||
await interaction.response.send_message(f'❌ An error occurred: {e}', ephemeral=True)
|
||
|
||
@commands.Cog.listener()
|
||
async def on_raw_reaction_add(self, payload):
|
||
if payload.user_id == self.bot.user.id:
|
||
return
|
||
|
||
guild_id = str(payload.guild_id)
|
||
message_id = str(payload.message_id)
|
||
|
||
if 'reaction_roles' not in self.bot.db.data:
|
||
self.bot.db.data['reaction_roles'] = {}
|
||
|
||
if guild_id not in self.bot.db.data['reaction_roles']:
|
||
return
|
||
|
||
if message_id not in self.bot.db.data['reaction_roles'][guild_id]:
|
||
return
|
||
|
||
emoji_str = str(payload.emoji)
|
||
role_id = self.bot.db.data['reaction_roles'][guild_id][message_id].get(emoji_str)
|
||
|
||
if not role_id:
|
||
return
|
||
|
||
guild = self.bot.get_guild(payload.guild_id)
|
||
if not guild:
|
||
return
|
||
|
||
role = guild.get_role(int(role_id))
|
||
if not role:
|
||
return
|
||
|
||
member = guild.get_member(payload.user_id)
|
||
if not member:
|
||
return
|
||
|
||
try:
|
||
await member.add_roles(role)
|
||
logger.info(f'✅ Added role {role.name} to {member.name}')
|
||
except Exception as e:
|
||
logger.error(f'❌ Error adding role: {e}')
|
||
|
||
@commands.Cog.listener()
|
||
async def on_raw_reaction_remove(self, payload):
|
||
if payload.user_id == self.bot.user.id:
|
||
return
|
||
|
||
guild_id = str(payload.guild_id)
|
||
message_id = str(payload.message_id)
|
||
|
||
if 'reaction_roles' not in self.bot.db.data:
|
||
return
|
||
|
||
if guild_id not in self.bot.db.data.get('reaction_roles', {}):
|
||
return
|
||
|
||
if message_id not in self.bot.db.data['reaction_roles'][guild_id]:
|
||
return
|
||
|
||
emoji_str = str(payload.emoji)
|
||
role_id = self.bot.db.data['reaction_roles'][guild_id][message_id].get(emoji_str)
|
||
|
||
if not role_id:
|
||
return
|
||
|
||
guild = self.bot.get_guild(payload.guild_id)
|
||
if not guild:
|
||
return
|
||
|
||
role = guild.get_role(int(role_id))
|
||
if not role:
|
||
return
|
||
|
||
member = guild.get_member(payload.user_id)
|
||
if not member:
|
||
return
|
||
|
||
try:
|
||
await member.remove_roles(role)
|
||
logger.info(f'➖ Removed role {role.name} from {member.name}')
|
||
except Exception as e:
|
||
logger.error(f'❌ Error removing role: {e}')
|
||
|
||
@app_commands.command(name='reactionrole', description='[ADMIN] Create a reaction role')
|
||
@app_commands.describe(message_id='Message ID to add reactions to', emoji='Emoji to use (e.g., 🎮)', role='Role to assign')
|
||
@app_commands.default_permissions(administrator=True)
|
||
async def reactionrole(self, interaction: discord.Interaction, message_id: str, emoji: str, role: discord.Role):
|
||
try:
|
||
message = await interaction.channel.fetch_message(int(message_id))
|
||
except discord.NotFound:
|
||
await interaction.response.send_message('❌ Message not found in this channel!', ephemeral=True)
|
||
return
|
||
except ValueError:
|
||
await interaction.response.send_message('❌ Invalid message ID!', ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
await message.add_reaction(emoji)
|
||
except discord.HTTPException:
|
||
await interaction.response.send_message('❌ Invalid emoji or unable to add reaction!', ephemeral=True)
|
||
return
|
||
|
||
guild_id = str(interaction.guild.id)
|
||
|
||
if 'reaction_roles' not in self.bot.db.data:
|
||
self.bot.db.data['reaction_roles'] = {}
|
||
|
||
if guild_id not in self.bot.db.data['reaction_roles']:
|
||
self.bot.db.data['reaction_roles'][guild_id] = {}
|
||
|
||
if message_id not in self.bot.db.data['reaction_roles'][guild_id]:
|
||
self.bot.db.data['reaction_roles'][guild_id][message_id] = {}
|
||
|
||
self.bot.db.data['reaction_roles'][guild_id][message_id][emoji] = str(role.id)
|
||
self.bot.db.save()
|
||
|
||
embed = discord.Embed(
|
||
title='✅ Reaction Role Created',
|
||
description=f'React with {emoji} on the message to get {role.mention}',
|
||
color=0x00FF00
|
||
)
|
||
embed.add_field(name='Message ID', value=message_id, inline=True)
|
||
embed.add_field(name='Emoji', value=emoji, inline=True)
|
||
embed.add_field(name='Role', value=role.mention, inline=True)
|
||
|
||
await interaction.response.send_message(embed=embed)
|
||
logger.info(f'✅ Created reaction role: {emoji} -> {role.name}')
|
||
|
||
@app_commands.command(name='removereactionrole', description='[ADMIN] Remove a reaction role')
|
||
@app_commands.describe(message_id='Message ID', emoji='Emoji to remove (leave empty to remove all)')
|
||
@app_commands.default_permissions(administrator=True)
|
||
async def removereactionrole(self, interaction: discord.Interaction, message_id: str, emoji: str = None):
|
||
guild_id = str(interaction.guild.id)
|
||
|
||
if 'reaction_roles' not in self.bot.db.data:
|
||
self.bot.db.data['reaction_roles'] = {}
|
||
|
||
if guild_id not in self.bot.db.data['reaction_roles'] or message_id not in self.bot.db.data['reaction_roles'].get(guild_id, {}):
|
||
await interaction.response.send_message('❌ No reaction roles found for that message!', ephemeral=True)
|
||
return
|
||
|
||
if emoji:
|
||
if emoji not in self.bot.db.data['reaction_roles'][guild_id][message_id]:
|
||
await interaction.response.send_message('❌ That emoji is not set up for reaction roles!', ephemeral=True)
|
||
return
|
||
|
||
del self.bot.db.data['reaction_roles'][guild_id][message_id][emoji]
|
||
self.bot.db.save()
|
||
await interaction.response.send_message(f'✅ Removed reaction role for {emoji}')
|
||
else:
|
||
del self.bot.db.data['reaction_roles'][guild_id][message_id]
|
||
self.bot.db.save()
|
||
await interaction.response.send_message(f'✅ Removed all reaction roles from message {message_id}')
|
||
|
||
@app_commands.command(name='listreactionroles', description='List all reaction roles')
|
||
async def listreactionroles(self, interaction: discord.Interaction):
|
||
guild_id = str(interaction.guild.id)
|
||
|
||
if 'reaction |