v4: Return api.InteractionResponse for CommandHandler

[diamondburned]

Currently, commands can only respond with a new message. Changing this to return an api.InteractionResponse instead allows commands to show modal dialogs as well.

[ljluestc]

import discord
from discord import app_commands
from enum import Enum
import typing

# Custom response type enum (mimicking Discord's InteractionResponseType)
class InteractionResponseType(Enum):
    CHANNEL_MESSAGE_WITH_SOURCE = 4  # Regular message response
    MODAL = 9  # Modal dialog response

# Custom InteractionResponse class to structure responses
class InteractionResponse:
    def __init__(self, response_type: InteractionResponseType, data: dict):
        self.type = response_type
        self.data = data

    def to_json(self):
        return {
            "type": self.type.value,
            "data": self.data
        }

# Command interface
class Command:
    async def execute(self, interaction: discord.Interaction) -> InteractionResponse:
        raise NotImplementedError

# Example command: Sends a message
class HelloCommand(Command):
    async def execute(self, interaction: discord.Interaction) -> InteractionResponse:
        return InteractionResponse(
            response_type=InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE,
            data={
                "content": "Hello, world!",
                "embeds": [
                    {"title": "Greeting", "description": "This is a rich embed!"}
                ]
            }
        )

# Example command: Shows a modal
class ModalCommand(Command):
    async def execute(self, interaction: discord.Interaction) -> InteractionResponse:
        return InteractionResponse(
            response_type=InteractionResponseType.MODAL,
            data={
                "title": "User Input Modal",
                "custom_id": "user_input_modal",
                "components": [
                    {
                        "type": 1,  # Action row
                        "components": [
                            {
                                "type": 4,  # Text input
                                "custom_id": "name_input",
                                "style": 1,  # Short text
                                "label": "Your Name",
                                "placeholder": "Enter your name",
                                "required": True
                            }
                        ]
                    },
                    {
                        "type": 1,  # Action row
                        "components": [
                            {
                                "type": 4,  # Text input
                                "custom_id": "feedback_input",
                                "style": 2,  # Paragraph text
                                "label": "Feedback",
                                "placeholder": "Tell us what you think",
                                "required": False
                            }
                        ]
                    }
                ]
            }
        )

# Command handler
class CommandHandler:
    def __init__(self):
        self.commands = {
            "hello": HelloCommand(),
            "modal": ModalCommand()
        }

    async def handle(self, interaction: discord.Interaction, command_name: str):
        command = self.commands.get(command_name)
        if not command:
            await interaction.response.send_message("Unknown command!", ephemeral=True)
            return

        response = await command.execute(interaction)
        if response.type == InteractionResponseType.CHANNEL_MESSAGE_WITH_SOURCE:
            await interaction.response.send_message(
                content=response.data.get("content"),
                embeds=[discord.Embed(**embed) for embed in response.data.get("embeds", [])],
                ephemeral=False
            )
        elif response.type == InteractionResponseType.MODAL:
            await interaction.response.send_modal(
                discord.ui.Modal(
                    title=response.data["title"],
                    custom_id=response.data["custom_id"],
                    components=[
                        discord.ui.TextInput(
                            label=comp["components"][0]["label"],
                            custom_id=comp["components"][0]["custom_id"],
                            style=discord.TextStyle.short if comp["components"][0]["style"] == 1 else discord.TextStyle.paragraph,
                            placeholder=comp["components"][0].get("placeholder"),
                            required=comp["components"][0].get("required", False)
                        ) for comp in response.data["components"]
                    ]
                )
            )

# Discord bot
class MyBot(discord.Client):
    def __init__(self):
        super().__init__(intents=discord.Intents.default())
        self.tree = app_commands.CommandTree(self)
        self.command_handler = CommandHandler()

    async def setup_hook(self):
        # Sync slash commands
        await self.tree.sync()

    async def on_ready(self):
        print(f"Logged in as {self.user}")

# Create bot instance
bot = MyBot()

# Slash command: /hello
@bot.tree.command(name="hello", description="Say hello")
async def hello(interaction: discord.Interaction):
    await bot.command_handler.handle(interaction, "hello")

# Slash command: /modal
@bot.tree.command(name="modal", description="Show a modal")
async def modal(interaction: discord.Interaction):
    await bot.command_handler.handle(interaction, "modal")

# Modal submission handler
@bot.event
async def on_interaction(interaction: discord.Interaction):
    if interaction.type == discord.InteractionType.modal_submit:
        name = interaction.data.get("components", [{}])[0]["components"][0]["value"]
        feedback = interaction.data.get("components", [{}])[1]["components"][0]["value"] or "No feedback"
        await interaction.response.send_message(
            f"Thanks, {name}! Your feedback: {feedback}",
            ephemeral=True
        )

# Run the bot
bot.run("YOUR_BOT_TOKEN")  # Replace with your Discord bot token