Fox-V3/hangman/hangman.py

from collections import defaultdict
from random import randint
from typing import Union

import discord
from redbot.core import Config, checks, commands
from redbot.core.commands import Cog
from redbot.core.data_manager import bundled_data_path


class Hangman(Cog):
    """Lets anyone play a game of hangman with custom phrases"""

    navigate = "🔼🔽"
    letters = "🇦🇧🇨🇩🇪🇫🇬🇭🇮🇯🇰🇱🇲🇳🇴🇵🇶🇷🇸🇹🇺🇻🇼🇽🇾🇿"

    def __init__(self, bot):
        super().__init__()
        self.bot = bot
        self.config = Config.get_conf(self, identifier=1049711010310997110)
        default_guild = {"theface": ":thinking:", "emojis": True}

        self.config.register_guild(**default_guild)

        self.the_data = defaultdict(
            lambda: {
                "running": False,
                "hangman": 0,
                "guesses": [],
                "trackmessage": False,
                "answer": "",
            }
        )
        #         self.path = str(cog_data_path(self)).replace("\\", "/")

        #         self.answer_path = self.path + "/bundled_data/hanganswers.txt"

        self.answer_path = bundled_data_path(self) / "hanganswers.txt"

        self.winbool = defaultdict(lambda: False)

        self.hanglist = {}

    async def red_delete_data_for_user(self, **kwargs):
        """Nothing to delete"""
        return

    async def _update_hanglist(self):
        for guild in self.bot.guilds:
            theface = await self.config.guild(guild).theface()
            self.hanglist[guild] = (
                """>
                   \\_________
                    |/        
                    |              
                    |                
                    |                 
                    |               
                    |                   
                    |\\___                 
                    """,
                """>
                   \\_________
                    |/   |      
                    |              
                    |                
                    |                 
                    |               
                    |                   
                    |\\___                 
                    H""",
                """>
                   \\_________       
                    |/   |              
                    |   """
                + theface
                + """
                    |                         
                    |                       
                    |                         
                    |                          
                    |\\___                       
                    HA""",
                """>
                   \\________               
                    |/   |                   
                    |   """
                + theface
                + """                   
                    |    |                     
                    |    |                    
                    |                           
                    |                            
                    |\\___                    
                    HAN""",
                """>
                   \\_________             
                    |/   |               
                    |   """
                + theface
                + """                    
                    |   /|                     
                    |     |                    
                    |                        
                    |                          
                    |\\___                          
                    HANG""",
                """>
                   \\_________              
                    |/   |                     
                    |   """
                + theface
                + """                      
                    |   /|\\                    
                    |     |                       
                    |                             
                    |                            
                    |\\___                          
                    HANGM""",
                """>
                   \\________                   
                    |/   |                         
                    |   """
                + theface
                + """                       
                    |   /|\\                             
                    |     |                          
                    |   /                            
                    |                                  
                    |\\___                              
                    HANGMA""",
                """>
                   \\________
                    |/   |     
                    |   """
                + theface
                + """     
                    |   /|\\           
                    |     |        
                    |   / \\        
                    |               
                    |\\___           
                    HANGMAN""",
            )

    @commands.group(aliases=["sethang"])
    @checks.mod_or_permissions(administrator=True)
    async def hangset(self, ctx):
        """Adjust hangman settings"""
        pass

    @hangset.command()
    async def face(self, ctx: commands.Context, theface):
        """Set the face of the hangman"""
        message = ctx.message
        # Borrowing FlapJack's emoji validation
        # (https://github.com/flapjax/FlapJack-Cogs/blob/master/smartreact/smartreact.py)
        if theface[:2] == "<:":
            theface = self.bot.get_emoji(int(theface.split(":")[2][:-1]))

        if theface is None:
            await ctx.maybe_send_embed("I could not find that emoji")
            return

        try:
            # Use the face as reaction to see if it's valid (THANKS FLAPJACK <3)
            await message.add_reaction(theface)
        except discord.errors.HTTPException:
            await ctx.maybe_send_embed("That's not an emoji I recognize.")
            return

        await self.config.guild(ctx.guild).theface.set(str(theface))
        await self._update_hanglist()
        await ctx.maybe_send_embed("Face has been updated!")

    @hangset.command()
    async def toggleemoji(self, ctx: commands.Context):
        """Toggles whether to automatically react with the alphabet"""

        current = await self.config.guild(ctx.guild).emojis()
        await self.config.guild(ctx.guild).emojis.set(not current)
        await ctx.maybe_send_embed(
            "Emoji Letter reactions have been set to {}".format(not current)
        )

    @commands.command(aliases=["hang"])
    async def hangman(self, ctx, guess: str = None):
        """Play a game of hangman against the bot!"""
        if guess is None:
            if self.the_data[ctx.guild]["running"]:
                await ctx.maybe_send_embed(
                    "Game of hangman is already running!\nEnter your guess!"
                )
                await self._printgame(ctx.channel)
                """await self.bot.send_cmd_help(ctx)"""
            else:
                await ctx.maybe_send_embed("Starting a game of hangman!")
                self._startgame(ctx.guild)
                await self._printgame(ctx.channel)
        elif not self.the_data[ctx.guild]["running"]:
            await ctx.maybe_send_embed(
                "Game of hangman is not yet running!\nStarting a game of hangman!"
            )
            self._startgame(ctx.guild)
            await self._printgame(ctx.channel)
        else:
            await ctx.maybe_send_embed("Guess by reacting to the message")
            # await self._guessletter(guess, ctx.channel)

    def _startgame(self, guild):
        """Starts a new game of hangman"""
        self.the_data[guild]["answer"] = self._getphrase().upper()
        self.the_data[guild]["hangman"] = 0
        self.the_data[guild]["guesses"] = []
        self.winbool[guild] = False
        self.the_data[guild]["running"] = True
        self.the_data[guild]["trackmessage"] = False

    def _stopgame(self, guild):
        """Stops the game in current state"""
        self.the_data[guild]["running"] = False
        self.the_data[guild]["trackmessage"] = False

    async def _checkdone(self, channel):
        if self.winbool[channel.guild]:
            await channel.send("You Win!")
            self._stopgame(channel.guild)
        elif self.the_data[channel.guild]["hangman"] >= 7:
            await channel.send(
                "You Lose!\nThe Answer was: **" + self.the_data[channel.guild]["answer"] + "**"
            )

            self._stopgame(channel.guild)

    def _getphrase(self):
        """Get a new phrase for the game and returns it"""

        with open(self.answer_path, "r") as phrasefile:
            phrases = phrasefile.readlines()

        outphrase = ""
        while outphrase == "":
            outphrase = phrases[randint(0, len(phrases) - 1)].partition(" (")[0]
        return outphrase

    def _hideanswer(self, guild):
        """Returns the obscured answer"""
        out_str = ""

        self.winbool[guild] = True
        for i in self.the_data[guild]["answer"]:
            if i in [" ", "-"]:
                out_str += i * 2
            elif i in self.the_data[guild]["guesses"] or i not in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
                out_str += "__" + i + "__ "
            else:
                out_str += "**\\_** "
                self.winbool[guild] = False

        return out_str

    def _guesslist(self, guild):
        """Returns the current letter list"""
        out_str = "".join(str(i) + "," for i in self.the_data[guild]["guesses"])
        out_str = out_str[:-1]

        return out_str

    async def _guessletter(self, guess, message):
        """Checks the guess on a letter and prints game if acceptable guess"""
        channel = message.channel
        if guess.upper() not in "ABCDEFGHIJKLMNOPQRSTUVWXYZ" or len(guess) != 1:
            await channel.send("Invalid guess. Only A-Z is accepted")
            return

        if guess.upper() in self.the_data[channel.guild]["guesses"]:
            await channel.send("Already guessed that! Try again")
            return
        if guess.upper() not in self.the_data[channel.guild]["answer"]:
            self.the_data[channel.guild]["hangman"] += 1

        self.the_data[channel.guild]["guesses"].append(guess.upper())

        await self._reprintgame(message)

    @commands.Cog.listener("on_reaction_add")
    async def on_react(self, reaction, user: Union[discord.User, discord.Member]):
        """Thanks to flapjack reactpoll for guidelines
        https://github.com/flapjax/FlapJack-Cogs/blob/master/reactpoll/reactpoll.py"""
        guild: discord.Guild = getattr(user, "guild", None)
        if guild is None:
            return

        if reaction.message.id != self.the_data[guild]["trackmessage"]:
            return

        if user.bot:
            return  # Don't react to bot reactions

        if await self.bot.cog_disabled_in_guild(self, guild):
            return

        message = reaction.message
        emoji = reaction.emoji

        if str(emoji) in self.letters:
            letter = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"[self.letters.index(str(emoji))]
            await self._guessletter(letter, message)
            await message.remove_reaction(emoji, user)
            await message.remove_reaction(emoji, self.bot.user)

        if str(emoji) in self.navigate:
            if str(emoji) == self.navigate[0]:
                await self._reactmessage_am(message)

            if str(emoji) == self.navigate[-1]:
                await self._reactmessage_nz(message)

    async def _try_clear_reactions(self, message):
        try:
            await message.clear_reactions()
        except discord.Forbidden:
            pass

    async def _reactmessage_menu(self, message):
        """React with menu options"""
        if not await self.config.guild(message.guild).emojis():
            return

        await self._try_clear_reactions(message)

        await message.add_reaction(self.navigate[0])
        await message.add_reaction(self.navigate[-1])

    async def _reactmessage_am(self, message):
        if not await self.config.guild(message.guild).emojis():
            return

        await self._try_clear_reactions(message)

        for x in range(len(self.letters)):
            if x in [
                i for i, b in enumerate("ABCDEFGHIJKLM") if b not in self._guesslist(message.guild)
            ]:
                await message.add_reaction(self.letters[x])

        await message.add_reaction(self.navigate[-1])

    async def _reactmessage_nz(self, message):
        if not await self.config.guild(message.guild).emojis():
            return

        await self._try_clear_reactions(message)

        for x in range(len(self.letters)):
            if x in [
                i for i, b in enumerate("NOPQRSTUVWXYZ") if b not in self._guesslist(message.guild)
            ]:
                await message.add_reaction(self.letters[x + 13])

        await message.add_reaction(self.navigate[0])

    async def _make_say(self, guild):
        c_say = "Guess this: " + str(self._hideanswer(guild)) + "\n"
        c_say += "Used Letters: " + str(self._guesslist(guild)) + "\n"
        c_say += self.hanglist[guild][self.the_data[guild]["hangman"]] + "\n"
        if await self.config.guild(guild).emojis():
            c_say += "{} for A-M, {} for N-Z".format(self.navigate[0], self.navigate[-1])
        else:
            c_say += "React with {} - {} to guess".format(self.letters[0], self.letters[-1])

        return c_say

    async def _reprintgame(self, message):
        if message.guild not in self.hanglist:
            await self._update_hanglist()

        c_say = await self._make_say(message.guild)

        await message.edit(content=c_say)
        self.the_data[message.guild]["trackmessage"] = message.id

        await self._checkdone(message.channel)

    async def _printgame(self, channel):
        """Print the current state of game"""
        if channel.guild not in self.hanglist:
            await self._update_hanglist()

        c_say = await self._make_say(channel.guild)

        message = await channel.send(c_say)

        self.the_data[channel.guild]["trackmessage"] = message.id

        await self._reactmessage_menu(message)
        await self._checkdone(channel)