import discord from discord import app_commands from discord.ext import commands import json import os DATA_FILE = 'data/raid_data.json' RAID_GATE_CHANNEL = 'raid-gate' RAID_HOSTER_ROLE = 'Raid Hoster' WAVE_CHOICES = [app_commands.Choice(name=f'Wave {i}', value=i) for i in range(1, 21)] # ── Data helpers ────────────────────────────────────────────────────────────── def load_data() -> dict: if not os.path.exists(DATA_FILE): os.makedirs('data', exist_ok=True) return {'points': {}, 'active_raid': None} with open(DATA_FILE, 'r') as f: return json.load(f) def save_data(data: dict) -> None: os.makedirs('data', exist_ok=True) with open(DATA_FILE, 'w') as f: json.dump(data, f, indent=2) def get_points_for_wave(wave: int) -> int: if 1 <= wave <= 5: return 50 elif 6 <= wave <= 10: return 100 elif 11 <= wave <= 13: return 150 elif 14 <= wave <= 17: return 250 elif 18 <= wave <= 20: return 500 return 0 # ── Embeds ──────────────────────────────────────────────────────────────────── def active_raid_embed(hoster_name: str, player_count: int) -> discord.Embed: embed = discord.Embed( title='⚔️ Demonfall Raid — ACTIVE', description=( 'A raid is now live! Click **Get Code** below, enter your Roblox username, ' 'and you will receive the private server code privately.' ), color=discord.Color.red() ) embed.add_field(name='🎮 Hosted by', value=hoster_name, inline=True) embed.add_field(name='👥 Players Joined', value=str(player_count), inline=True) embed.set_footer(text='Demonfall Raid System • Click the button to get the code') return embed def ended_raid_embed(hoster_name: str, wave: int, points: int, player_count: int) -> discord.Embed: embed = discord.Embed( title='⚔️ Demonfall Raid — ENDED', color=discord.Color.dark_gray() ) embed.add_field(name='🎮 Hosted by', value=hoster_name, inline=True) embed.add_field(name='🌊 Wave Reached', value=f'Wave {wave}', inline=True) embed.add_field(name='⭐ Points Earned', value=str(points), inline=True) embed.add_field(name='👥 Total Players', value=str(player_count), inline=True) embed.set_footer(text='This raid has ended — the code is no longer available') return embed # ── Modal ───────────────────────────────────────────────────────────────────── class RobloxUsernameModal(discord.ui.Modal, title='Enter Your Roblox Username'): username = discord.ui.TextInput( label='Roblox Username', placeholder='e.g. Builderman', required=True, min_length=3, max_length=20 ) async def on_submit(self, interaction: discord.Interaction): data = load_data() active_raid = data.get('active_raid') if not active_raid or not active_raid.get('active'): await interaction.response.send_message( '❌ This raid has ended — the code is no longer available.', ephemeral=True ) return user_id = str(interaction.user.id) claimers: list = active_raid.setdefault('claimers', []) if user_id not in claimers: claimers.append(user_id) data['active_raid'] = active_raid save_data(data) # Update player count on the raid-gate message try: channel = discord.utils.get(interaction.guild.text_channels, name=RAID_GATE_CHANNEL) if channel and active_raid.get('message_id'): msg = await channel.fetch_message(int(active_raid['message_id'])) await msg.edit(embed=active_raid_embed(active_raid['hoster_name'], len(claimers))) except Exception: pass await interaction.response.send_message( f'🔑 **Your Demonfall Private Server Code**\n\n' f'Hey **{self.username.value}**! Here is your private server code:\n' f'```\n{active_raid["server_code"]}\n```' f'Jump in and good luck on the raid! ⚔️', ephemeral=True ) # ── Persistent button view ──────────────────────────────────────────────────── class GetCodeView(discord.ui.View): """Persistent view — survives bot restarts via custom_id.""" def __init__(self): super().__init__(timeout=None) @discord.ui.button( label='Get Code', style=discord.ButtonStyle.green, custom_id='demonfall:get_code', emoji='🔑' ) async def get_code_btn(self, interaction: discord.Interaction, button: discord.ui.Button): data = load_data() active_raid = data.get('active_raid') if not active_raid or not active_raid.get('active'): await interaction.response.send_message( '❌ There is no active raid right now!', ephemeral=True ) return await interaction.response.send_modal(RobloxUsernameModal()) class DisabledCodeView(discord.ui.View): """Shown after a raid ends — button is disabled.""" def __init__(self): super().__init__(timeout=None) @discord.ui.button( label='Get Code', style=discord.ButtonStyle.gray, custom_id='demonfall:get_code_disabled', emoji='🔒', disabled=True ) async def get_code_btn(self, interaction: discord.Interaction, button: discord.ui.Button): pass # ── Cog ─────────────────────────────────────────────────────────────────────── class RaidsCog(commands.Cog): def __init__(self, bot: commands.Bot): self.bot = bot async def cog_load(self): # Register persistent views so buttons survive restarts self.bot.add_view(GetCodeView()) self.bot.add_view(DisabledCodeView()) # ── Check helpers ────────────────────────────────────────────────────────── def _has_hoster_role(self, member: discord.Member) -> bool: role = discord.utils.get(member.guild.roles, name=RAID_HOSTER_ROLE) return role is None or role in member.roles # if role doesn't exist, allow all # ── /startraid ───────────────────────────────────────────────────────────── @app_commands.command(name='startraid', description='Start a Demonfall raid and post it to #raid-gate') @app_commands.describe(server_code='The Roblox private server code for this raid') async def startraid(self, interaction: discord.Interaction, server_code: str): if not self._has_hoster_role(interaction.user): await interaction.response.send_message( f'❌ You need the **{RAID_HOSTER_ROLE}** role to host a raid!', ephemeral=True ) return data = load_data() if data.get('active_raid') and data['active_raid'].get('active'): await interaction.response.send_message( '❌ A raid is already active! Use `/endraid` to close it first.', ephemeral=True ) return channel = discord.utils.get(interaction.guild.text_channels, name=RAID_GATE_CHANNEL) if not channel: await interaction.response.send_message( f'❌ Could not find the **#{RAID_GATE_CHANNEL}** channel. Please create it first!', ephemeral=True ) return embed = active_raid_embed(interaction.user.display_name, 0) view = GetCodeView() message = await channel.send(embed=embed, view=view) data['active_raid'] = { 'active': True, 'server_code': server_code, 'hoster_id': str(interaction.user.id), 'hoster_name': interaction.user.display_name, 'message_id': str(message.id), 'claimers': [] } save_data(data) await interaction.response.send_message( f'✅ Raid started! The post is live in {channel.mention}.', ephemeral=True ) # ── /endraid ─────────────────────────────────────────────────────────────── @app_commands.command(name='endraid', description='End the current raid and log your wave + points') @app_commands.describe(wave='The wave you reached during the raid') @app_commands.choices(wave=WAVE_CHOICES) async def endraid(self, interaction: discord.Interaction, wave: app_commands.Choice[int]): if not self._has_hoster_role(interaction.user): await interaction.response.send_message( f'❌ You need the **{RAID_HOSTER_ROLE}** role to end a raid!', ephemeral=True ) return data = load_data() active_raid = data.get('active_raid') if not active_raid or not active_raid.get('active'): await interaction.response.send_message( '❌ There is no active raid to end!', ephemeral=True ) return points = get_points_for_wave(wave.value) hoster_id = active_raid['hoster_id'] hoster_name = active_raid['hoster_name'] # Award points to the hoster if hoster_id not in data['points']: data['points'][hoster_id] = {'name': hoster_name, 'total': 0, 'raids': 0} data['points'][hoster_id]['total'] += points data['points'][hoster_id]['raids'] += 1 data['points'][hoster_id]['name'] = hoster_name # keep display name fresh new_total = data['points'][hoster_id]['total'] # Update raid-gate message try: channel = discord.utils.get(interaction.guild.text_channels, name=RAID_GATE_CHANNEL) if channel and active_raid.get('message_id'): msg = await channel.fetch_message(int(active_raid['message_id'])) player_count = len(active_raid.get('claimers', [])) await msg.edit( embed=ended_raid_embed(hoster_name, wave.value, points, player_count), view=DisabledCodeView() ) except Exception: pass data['active_raid'] = {'active': False} save_data(data) await interaction.response.send_message( f'✅ **Raid ended!**\n' f'🌊 Wave Reached: **Wave {wave.value}**\n' f'⭐ Points Earned: **{points}**\n' f'📊 Your Total Points: **{new_total}**', ephemeral=True ) # ── /raidleaderboard ─────────────────────────────────────────────────────── @app_commands.command(name='raidleaderboard', description='View the Demonfall raid points leaderboard') async def raidleaderboard(self, interaction: discord.Interaction): data = load_data() points_data: dict = data.get('points', {}) raid_role = discord.utils.get(interaction.guild.roles, name=RAID_HOSTER_ROLE) leaderboard: list[tuple[str, int, int]] = [] for user_id, info in points_data.items(): member = interaction.guild.get_member(int(user_id)) if member is None: continue # Only include members who currently hold the Raid Hoster role if raid_role and raid_role not in member.roles: continue leaderboard.append((member.display_name, info['total'], info.get('raids', 0))) leaderboard.sort(key=lambda x: x[1], reverse=True) embed = discord.Embed( title='⚔️ Demonfall Raid Leaderboard', color=discord.Color.gold() ) if not leaderboard: embed.description = 'No raid points have been recorded yet!' else: medals = {0: '🥇', 1: '🥈', 2: '🥉'} lines = [] for i, (name, pts, raids) in enumerate(leaderboard[:20]): rank = medals.get(i, f'**#{i + 1}**') lines.append(f'{rank} **{name}** — {pts:,} pts *({raids} raid{"s" if raids != 1 else ""})*') embed.description = '\n'.join(lines) embed.set_footer(text=f'Showing top {min(len(leaderboard), 20)} Raid Hosters') await interaction.response.send_message(embed=embed) # ── Setup ───────────────────────────────────────────────────────────────────── async def setup(bot: commands.Bot): await bot.add_cog(RaidsCog(bot))