dakiny-bot/cogs/commands/record.py

"""
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)