diff --git a/bot.py b/bot.py index f15813a..be096ec 100755 --- a/bot.py +++ b/bot.py @@ -1,8 +1,8 @@ #!/usr/bin/env python3 from dataclasses import dataclass -import discord -from discord.ext import commands +import disnake +from disnake.ext import commands import logging import yaml @@ -26,12 +26,30 @@ bot = commands.Bot(command_prefix='!') @bot.event -async def on_message(message): +async def on_message(message: disnake.Message): await bot.process_commands(message) @bot.command() -async def helloworld(ctx): - await ctx.send("plop") +async def vote(ctx: commands.Context): + view = Confirm() + await ctx.message.reply("plop", view=view) + await view.wait() + + +# Define a simple View that gives us a confirmation menu +class Confirm(disnake.ui.View): + @disnake.ui.button(label="S'allier", style=disnake.ButtonStyle.green) + async def confirm(self, button: disnake.ui.Button, interaction: disnake.MessageInteraction): + self.clear_items() + await interaction.response.edit_message(content="Vous vous êtes allié.", view=self) + self.stop() + + @disnake.ui.button(label="Trahir", style=disnake.ButtonStyle.red) + async def cancel(self, button: disnake.ui.Button, interaction: disnake.MessageInteraction): + self.clear_items() + await interaction.response.edit_message("Vous avez trahi.", view=self) + self.stop() + bot.run(config.discord_token)