Handlers and arguments¶

"""One handler behind several buttons, each carrying its own argument."""

from __future__ import annotations

import enum
import os

import hikari

import risa
from risa import ui


class Priority(enum.Enum):
    LOW = 1
    NORMAL = 2
    URGENT = 3


@risa.register(name="ticket")
class Ticket(risa.View):
    priority: Priority = Priority.NORMAL

    def render(self) -> ui.Layout:
        buttons = [
            ui.Button(
                risa.bind(self.pick, priority),
                label=priority.name.title(),
                disabled=priority is self.priority,
            )
            for priority in Priority
        ]
        return ui.Container(
            ui.TextDisplay(f"Priority: **{self.priority.name.lower()}**"),
            ui.Row(*buttons),
        )

    @risa.handler
    async def pick(self, ctx: risa.ComponentContext, priority: Priority) -> None:
        self.priority = priority
        await ctx.rerender()


def main() -> None:
    bot = hikari.GatewayBot(
        os.environ["DISCORD_TOKEN"],
        intents=hikari.Intents.ALL_UNPRIVILEGED | hikari.Intents.MESSAGE_CONTENT,
    )
    client = risa.client_from_app(bot)
    client.add_view(Ticket)

    @bot.listen()
    async def on_message(event: hikari.GuildMessageCreateEvent) -> None:
        if event.content == "!ticket":
            rendered = await client.build(Ticket())
            await rendered.send_to(bot.rest, event.channel_id)

    bot.run()


if __name__ == "__main__":
    main()

"""A hikari enum as an argument: a view that fails to build, and the fix."""

from __future__ import annotations

import os

import hikari

import risa
from risa import ui


@risa.register(name="channel-kind-wrong")
class ChannelKindWrong(risa.View):
    def render(self) -> ui.Layout:
        voice = hikari.ChannelType.GUILD_VOICE
        return ui.Container(
            ui.TextDisplay("Which kind of channel?"),
            ui.Row(ui.Button(risa.bind(self.pick, voice), label="Voice")),
        )

    @risa.handler
    async def pick(
        self,
        ctx: risa.ComponentContext,
        kind: hikari.ChannelType,
    ) -> None:
        await ctx.respond(f"You picked {kind.name}.", ephemeral=True)


@risa.register(name="channel-kind")
class ChannelKind(risa.View):
    def render(self) -> ui.Layout:
        text = hikari.ChannelType.GUILD_TEXT
        voice = hikari.ChannelType.GUILD_VOICE
        return ui.Container(
            ui.TextDisplay("Which kind of channel?"),
            ui.Row(
                ui.Button(risa.bind(self.pick, int(text)), label="Text"),
                ui.Button(risa.bind(self.pick, int(voice)), label="Voice"),
            ),
        )

    @risa.handler
    async def pick(self, ctx: risa.ComponentContext, kind: int) -> None:
        picked = hikari.ChannelType(kind)
        await ctx.respond(f"You picked {picked.name}.", ephemeral=True)


def main() -> None:
    bot = hikari.GatewayBot(
        os.environ["DISCORD_TOKEN"],
        intents=hikari.Intents.ALL_UNPRIVILEGED | hikari.Intents.MESSAGE_CONTENT,
    )
    client = risa.client_from_app(bot)
    client.add_view(ChannelKind)

    @bot.listen()
    async def on_message(event: hikari.GuildMessageCreateEvent) -> None:
        if event.content == "!channelkind":
            rendered = await client.build(ChannelKind())
            await rendered.send_to(bot.rest, event.channel_id)

    bot.run()


if __name__ == "__main__":
    main()