""" Contained commands and views for record commands. """ import nextcord from discord import TextInputStyle from nextcord.ext import commands from config import DakinyConfig class RecordCog(commands.Cog, name="Record Commands"): """ Class holds record commands. """ def __init__(self, bot) -> None: """ Default constructor for RecordCog class. Does nothing. """ super().__init__() self.bot: commands.Bot = bot @nextcord.slash_command() async def record(self, interaction: nextcord.Interaction): """ Parent slash command for record. Never called. """ @record.subcommand(description="Record an event and its participants.") async def event( self, interaction: nextcord.Interaction, name: str = nextcord.SlashOption( required=True, description="Name of the event that was hosted." ), host: nextcord.Member = nextcord.SlashOption( required=False, description="An alternative host if you are recording event for someone else.", ), ): # check role if ( interaction.guild.get_role(self.bot.config.params.event_host_role_id) not in interaction.user.roles ): await interaction.response.send_message( f"You must have the <@&{self.bot.config.params.event_host_role_id}> role to record an event.", ephemeral=True, ) return if host is None: host = interaction.user embed: nextcord.Embed = nextcord.Embed(title=name, description="") embed.add_field(name="Participants", value="") embed.add_field(name="Non-Discord Participants", value="") await interaction.response.send_message( "Select all members who participated!", view=EventView(host, name, self.bot.config), ephemeral=True, embed=embed, ) # add points to table class EventView(nextcord.ui.View): def __init__(self, host: nextcord.Member, name: str, config: DakinyConfig): super().__init__() self.host: nextcord.Member = host self.name: str = name self.config: DakinyConfig = config self.event_participants = ParticipantSelect() self.add_item(self.event_participants) @nextcord.ui.button(label="Cancel", style=nextcord.ButtonStyle.red, row=1) async def cancel( self, button: nextcord.ui.Button, interaction: nextcord.Interaction ): await interaction.response.send_message("Cancelling", ephemeral=True) self.stop() @nextcord.ui.button(label="Edit", style=nextcord.ButtonStyle.blurple, row=1) async def edit(self, button: nextcord.ui.Button, interaction: nextcord.Interaction): await interaction.response.send_modal( ParticipantModal(interaction.message.embeds[0]) ) @nextcord.ui.button(label="Confirm", style=nextcord.ButtonStyle.green, row=1) async def confirm( self, button: nextcord.ui.Button, interaction: nextcord.Interaction ): msg = f"Event: {self.name}\n" msg += f"Hosted by: <@{self.host.id}>\n\n" msg += "Participants:\n" for participant in self.event_participants.values.members: msg += f"<@{participant.id}>\n" event_channel: nextcord.TextChannel = interaction.guild.get_channel( self.config.params.event_post_channel_id ) await event_channel.send(msg) self.stop() class ParticipantModal(nextcord.ui.Modal): def __init__(self, embed): super().__init__( "Add non-discord members as participats!", timeout=5 * 60, ) self.embed: nextcord.Embed = embed self.name: nextcord.ui.TextInput = nextcord.ui.TextInput( required=True, label="Event Name", default_value=self.embed.title, ) self.description: nextcord.ui.TextInput = nextcord.ui.TextInput( required=False, label="Event Description", default_value=self.embed.description, style=TextInputStyle.paragraph, placeholder="Add a description if desired.", max_length=1000, ) self.participants: nextcord.ui.TextInput = nextcord.ui.TextInput( required=False, label="Usernames", style=TextInputStyle.paragraph, placeholder="Please separate usernames with a new line.", default_value=self.embed.fields[1].value, ) self.add_item(self.name) self.add_item(self.description) self.add_item(self.participants) async def callback(self, interaction: nextcord.Interaction): self.embed.title = self.name.value self.embed.description = self.description.value self.embed.set_field_at( index=1, name="Non-Discord Participants", value=self.participants.value ) await interaction.response.edit_message(embed=self.embed) class ParticipantSelect(nextcord.ui.UserSelect): def __init__(self) -> None: super().__init__( min_values=1, max_values=25, placeholder="Select all participants", row=0, ) async def callback(self, interaction: nextcord.Interaction): embed: nextcord.Embed = interaction.message.embeds[0] participants: str = "\n".join( [f"<@{participant.id}>" for participant in self.values.members] ) embed.set_field_at(0, name="Participants", value=participants) await interaction.response.edit_message(embed=embed)