easterobot 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

easterobot/games/rock_paper_scissor.py

@@ -0,0 +1,206 @@
+"""TicTacToe."""
+
+from functools import partial
+from typing import Optional
+
+import discord
+from typing_extensions import override
+
+from easterobot.commands.base import Interaction
+from easterobot.games.game import Button, Game
+
+ROCK = b"\xf0\x9f\xaa\xa8".decode("utf-8")
+PAPER = b"\xf0\x9f\x93\x84".decode("utf-8")
+SCISSORS = b"\xe2\x9c\x82\xef\xb8\x8f".decode("utf-8")
+EMOJIS = [PAPER, ROCK, SCISSORS]
+
+
+class RockPaperScissor(Game):
+    view: discord.ui.View
+
+    def __init__(
+        self,
+        player1: discord.Member,
+        player2: discord.Member,
+        message: discord.Message,
+        win_count: int = 3,
+        max_turn: int = 10,
+    ) -> None:
+        """Initialize grid."""
+        self.timeout = False
+        self.max_turn = max_turn
+        self.win_count = win_count
+        self.player1 = player1
+        self.player2 = player2
+        self.play1: Optional[str] = None
+        self.play2: Optional[str] = None
+        self.history: list[tuple[str, str]] = []
+        super().__init__(message)
+
+    @override
+    async def on_start(self) -> None:
+        """Run."""
+        embed = discord.Embed(color=0xF2BC32)
+        embed.set_author(
+            name="Partie en cours", icon_url=self.bot.app_emojis["wait"].url
+        )
+        self.view = discord.ui.View()
+        rock_btn: Button = discord.ui.Button(
+            style=discord.ButtonStyle.gray,
+            emoji=ROCK,
+        )
+        paper_btn: Button = discord.ui.Button(
+            style=discord.ButtonStyle.gray,
+            emoji=PAPER,
+        )
+        scissor_btn: Button = discord.ui.Button(
+            style=discord.ButtonStyle.gray,
+            emoji=SCISSORS,
+        )
+
+        async def action_callback(
+            interaction: Interaction,
+            label: str,
+        ) -> None:
+            update = False
+
+            # Player 1 click on button
+            if interaction.user == self.player1 and self.play1 is None:
+                self.play1 = label
+                update = True
+
+            # Player 2 click on button
+            elif interaction.user == self.player2 and self.play2 is None:
+                self.play2 = label
+                update = True
+
+            # One button has been pressed
+            if update:
+                await interaction.response.defer()
+                await self.update()
+            else:
+                await interaction.response.send_message(
+                    "Vous ne pouvez pas intéragir dans cette partie !",
+                    ephemeral=True,
+                    delete_after=5,
+                )
+
+        rock_btn.callback = partial(action_callback, label=ROCK)  # type: ignore[method-assign]
+        paper_btn.callback = partial(action_callback, label=PAPER)  # type: ignore[method-assign]
+        scissor_btn.callback = partial(action_callback, label=SCISSORS)  # type: ignore[method-assign]
+        self.view.add_item(rock_btn)
+        self.view.add_item(paper_btn)
+        self.view.add_item(scissor_btn)
+        await self.update()
+
+    async def update(self) -> None:  # noqa: PLR0912, PLR0915
+        """Update the current display."""
+        embed = discord.Embed(color=0xF2BC32)
+        # Both player have played
+        header = "Partie en cours"
+        if self.play1 is None:
+            icon_url = self.player1.display_avatar.url
+            info = f"En attente de {self.player1.mention} ..."
+        elif self.play2 is None:
+            icon_url = self.player2.display_avatar.url
+            info = f"En attente de {self.player2.mention} ..."
+        else:
+            # Play and fight
+            self.history.append((self.play1, self.play2))
+            self.play1 = None
+            self.play2 = None
+            icon_url = self.bot.app_emojis["wait"].url
+            info = (
+                f"En attente de {self.player1.mention} "
+                f"et {self.player2.mention} ..."
+            )
+        pt1 = 0
+        pt2 = 0
+        morsels = []
+        for play1, play2 in self.history:
+            i1 = EMOJIS.index(play1)
+            i2 = EMOJIS.index(play2)
+            if i1 == (i2 + 1) % 3:
+                pt1 += 1
+                text = self.player1.mention
+            elif i1 == (i2 - 1) % 3:
+                pt2 += 1
+                text = self.player2.mention
+            else:
+                text = "**égalité**"
+            morsels.append(
+                f"### {play1} {self.bot.app_emojis['versus']} {play2} "
+                f"{self.bot.app_emojis['arrow']} {text}"
+            )
+        embed.description = "\n".join(morsels)
+        if (
+            len(self.history) >= self.max_turn
+            or pt1 >= self.win_count
+            or pt2 >= self.win_count
+            or self.timeout
+        ):
+            header = "Partie terminée"
+            embed.description += "\n\n"
+            if self.timeout:
+                final_winner = self.winner
+            elif pt1 < pt2:
+                final_winner = self.player2
+            elif pt2 < pt1:
+                final_winner = self.player1
+            else:
+                final_winner = None
+            if final_winner:
+                forfait = "par forfait " if self.timeout else ""
+                embed.description += (
+                    f"## Gagnant {forfait}{final_winner.mention} 🎉"
+                )
+                icon_url = final_winner.display_avatar.url
+            else:
+                embed.description += (
+                    f"## Égalité entre {self.player1.mention} "
+                    f"et {self.player2.mention} 🤝"
+                )
+                icon_url = self.bot.app_emojis["end"].url
+            self.view.stop()
+            self.view.clear_items()
+            if not self.timeout:
+                await self.set_winner(final_winner)
+        else:
+            dt = self.start_timer(63)
+            embed.description += f"\n\n{info}\n\nFin du tour {dt}"
+        embed.set_author(name=header, icon_url=icon_url)
+        await self.message.edit(embed=embed, view=self.view, content="")
+
+    def compute_winner(
+        self, play1: str, play2: str
+    ) -> Optional[discord.Member]:
+        """Get the winner."""
+        i1 = EMOJIS.index(play1)
+        i2 = EMOJIS.index(play2)
+        if i1 == (i2 + 1) % 3:
+            return self.player1
+        if i1 == (i2 - 1) % 3:
+            return self.player2
+        return None  # Draw
+
+    def color(
+        self, member: Optional[discord.Member]
+    ) -> Optional[discord.Colour]:
+        """Color of the embed."""
+        if member is None:
+            return discord.Colour.from_str("#d4d5d6")
+        if member == self.player1:
+            return discord.Colour.from_str("#ca2a3e")
+        if member == self.player2:
+            return discord.Colour.from_str("#5865F2")
+        error_message = f"Invalid member: {member!r}"
+        raise ValueError(error_message)
+
+    @override
+    async def on_timeout(self) -> None:
+        self.timeout = True
+        if self.play1 is None:
+            await self.set_winner(self.player2)
+        if self.play2 is None:
+            await self.set_winner(self.player1)
+        await self.update()

easterobot/games/tic_tac_toe.py

@@ -0,0 +1,168 @@
+"""TicTacToe."""
+
+import datetime
+from typing import Optional
+
+import discord
+from discord.utils import format_dt
+from typing_extensions import override
+
+from easterobot.games.game import Game, Player
+
+EMOJIS_MAPPER = {
+    "1️⃣": 0,
+    "2️⃣": 1,
+    "3️⃣": 2,
+    "4️⃣": 3,
+    "5️⃣": 4,
+    "6️⃣": 5,
+    "7️⃣": 6,
+    "8️⃣": 7,
+    "9️⃣": 8,
+}
+EMOJIS = tuple(EMOJIS_MAPPER)
+
+
+class TicTacToe(Game):
+    def __init__(
+        self,
+        player1: discord.Member,
+        player2: discord.Member,
+        message: discord.Message,
+    ) -> None:
+        """Initialize grid."""
+        self.grid: list[Optional[Player]] = [None] * 9
+        self.timeout = False
+        self.player1 = Player(player1, 1)
+        self.player2 = Player(player2, 2)
+        self.turn = 0
+        super().__init__(message)
+
+    @override
+    async def on_start(self) -> None:
+        """Run."""
+        await self.update()
+        for emoji in EMOJIS:
+            await self.message.add_reaction(emoji)
+        self.start_timer(60)
+
+    async def update(self) -> None:
+        """Update the message."""
+        label = ""
+        footer = ""
+        if not self.terminate:
+            label = "⭕" if self.turn % 2 else "❌"
+            label += f" Joueur actuel : {self.current.member.mention}\n\n"
+            user: Optional[Player] = self.current
+        elif self.winner:
+            forfait = "par forfait " if self.timeout else ""
+            footer = f"\n## Gagnant {forfait}{self.winner.mention} 🎉"
+            user = self.current
+        else:
+            footer = (
+                "\n## Égalité entre "
+                f"{self.player1.member.mention} "
+                f"et {self.player2.member.mention} 🤝"
+            )
+            user = None
+
+        content = label
+        lines = []
+        for row in range(3):
+            pieces = []
+            for col in range(3):
+                index = row * 3 + col
+                player = self.grid[index]
+                if player == self.player1:
+                    piece = "❌"
+                elif player == self.player2:
+                    piece = "⭕"
+                else:
+                    piece = EMOJIS[index]
+                pieces.append(piece)
+            lines.append("│".join(pieces))
+        content += "\n".join(lines)
+        content += footer
+
+        if not self.terminate:
+            now = datetime.datetime.now() + datetime.timedelta(seconds=62)  # noqa: DTZ005
+            content += f"\n\nFin du tour {format_dt(now, style='R')}"
+
+        embed = discord.Embed(description=content, color=self.color(user))
+        embed.set_author(
+            name="Partie terminée" if self.terminate else "Partie en cours",
+            icon_url=(
+                user.member.display_avatar.url
+                if user
+                else self.bot.app_emojis["end"].url
+            ),
+        )
+        self.message = await self.message.edit(
+            embed=embed,
+            content="",
+            view=None,
+        )
+
+    @override
+    async def on_reaction(
+        self, member_id: int, reaction: discord.PartialEmoji
+    ) -> None:
+        if reaction.name not in EMOJIS or member_id != self.current.member.id:
+            return
+        index = EMOJIS_MAPPER[reaction.name]
+        await self.place(index, self.current)
+
+    @property
+    def current(self) -> Player:
+        """Get current member."""
+        return [self.player1, self.player2][self.turn % 2]
+
+    def color(self, player: Optional[Player]) -> Optional[discord.Colour]:
+        """Color of the embed."""
+        if player is None:
+            return discord.Colour.from_str("#d4d5d6")
+        if player == self.player1:
+            return discord.Colour.from_str("#F17720")
+        if player == self.player2:
+            return discord.Colour.from_str("#0474BA")
+        error_message = f"Invalid player: {player!r}"
+        raise ValueError(error_message)
+
+    async def place(self, index: int, player: Player) -> None:
+        """Place a piece."""
+        if self.terminate or self.grid[index] is not None:
+            return
+        async with self.lock:
+            await self.stop_timer()
+            self.grid[index] = player
+
+            if self._is_winner(player):
+                await self.set_winner(player.member)
+            elif all(cell is not None for cell in self.grid):
+                await self.set_winner(None)
+            else:
+                self.turn += 1
+                self.start_timer(60)
+            await self.update()
+
+    @override
+    async def on_timeout(self) -> None:
+        self.turn += 1
+        self.timeout = True
+        await self.set_winner(self.current.member)
+        await self.update()
+
+    def _is_winner(self, player: Player) -> bool:
+        wins = [
+            [0, 1, 2],
+            [3, 4, 5],
+            [6, 7, 8],  # rows
+            [0, 3, 6],
+            [1, 4, 7],
+            [2, 5, 8],  # cols
+            [0, 4, 8],
+            [2, 4, 6],  # diagonals
+        ]
+        return any(
+            all(self.grid[i] == player for i in combo) for combo in wins
+        )

easterobot/hunts/__init__.py

@@ -0,0 +1,14 @@
+"""Enable hunt."""
+
+from easterobot.bot import Easterobot
+from easterobot.hunts.hunt import HuntCog
+
+__all__ = [
+    "HuntCog",
+]
+
+
+async def setup(bot: Easterobot) -> None:
+    hunt_cog = HuntCog(bot)
+    bot.hunt = hunt_cog
+    await bot.add_cog(hunt_cog)