import discord
from discord.ext import commands
from discord import option


class QuickPoll(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.slash_command(
        guild_ids=None,
        pass_context=True,
        description="Create a poll with up to 10 options",
    )
    @option(name="option1", description="Poll option", required=True)
    @option(name="option2", description="Poll option", required=False)
    @option(name="option3", description="Poll option", required=False)
    @option(name="option4", description="Poll option", required=False)
    @option(name="option5", description="Poll option", required=False)
    @option(name="option6", description="Poll option", required=False)
    @option(name="option7", description="Poll option", required=False)
    @option(name="option8", description="Poll option", required=False)
    @option(name="option9", description="Poll option", required=False)
    @option(name="option10", description="Poll option", required=False)
    async def poll(
        self,
        ctx,
        question: str,
        option1: str,
        option2: str,
        option3: str,
        option4: str,
        option5: str,
        option6: str,
        option7: str,
        option8: str,
        option9: str,
        option10: str,
    ):
        # Filter out empty "options"
        options = [
            x
            for x in [
                option1,
                option2,
                option3,
                option4,
                option5,
                option6,
                option7,
                option8,
                option9,
                option10,
            ]
            if x
        ]
        if len(options) <= 1:
            await ctx.send("You need more than one option to make a poll!")
            return
        if len(options) > 10:
            await ctx.send("You cannot make a poll for more than 10 things!")
            return

        if len(options) == 2 and options[0] == "yes" and options[1] == "no":
            reactions = ["✅", "❌"]
        else:
            reactions = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣", "🔟"]

        description = []
        for x, option in enumerate(options):
            description += "\n {} {}".format(reactions[x], option)
        embed = discord.Embed(title=question, description="".join(description))

        await ctx.defer()
        react_message = await ctx.followup.send(embed=embed)

        for reaction in reactions[: len(options)]:
            await react_message.add_reaction(reaction)
        embed.set_footer(text="Poll ID: {}".format(react_message.id))
        await react_message.edit(embed=embed)

    @commands.command(pass_context=True)
    async def tally(self, ctx, id=None):
        poll_message = await ctx.channel.fetch_message(id)
        embed = poll_message.embeds[0]
        unformatted_options = [x.strip() for x in embed.description.split("\n")]
        print(f"unformatted{unformatted_options}")
        opt_dict = (
            {x[:2]: x[3:] for x in unformatted_options}
            if unformatted_options[0][0] == "1"
            else {x[:1]: x[2:] for x in unformatted_options}
        )
        # check if we're using numbers for the poll, or x/checkmark, parse accordingly
        voters = [
            self.bot.user.id
        ]  # add the bot's ID to the list of voters to exclude it's votes

        tally = {x: 0 for x in opt_dict.keys()}
        for reaction in poll_message.reactions:
            if reaction.emoji in opt_dict.keys():
                reactors = await reaction.users().flatten()
                for reactor in reactors:
                    if reactor.id not in voters:
                        tally[reaction.emoji] += 1
                        voters.append(reactor.id)
        output = f"Results of the poll for '{embed.title}':\n" + "\n".join(
            ["{}: {}".format(opt_dict[key], tally[key]) for key in tally.keys()]
        )
        await ctx.send(output)


def setup(bot):
    bot.add_cog(QuickPoll(bot))