from discord.ext import commands
from discord import option
import discord
import markovify
import random


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

    @commands.slash_command(
        guild_ids=None,
        name="markov",
        description="Generate a markov chain based on a user in the server",
    )
    @option("User", description="Choose a member", input_type="user")
    async def markov(self, ctx: discord.ApplicationContext, user: discord.Member):
        await ctx.defer()
        temp_message = await ctx.respond(
            f"Generating a markov chain for {user.mention}..."
        )

        # Get messages from passed in user
        authors_mesages = []
        for message in await ctx.history(limit=20000).flatten():
            if message.author.id == user.id and "https" not in message.content:
                authors_mesages.append(message.content)

        # Make the model
        user_model = markovify.Text(". ".join(authors_mesages))
        user_model.to_json()

        dummy = []
        for i in range(random.randint(3, 9)):
            dummy.append(str(user_model.make_sentence(max_words=100, tries=900)))
        await temp_message.edit(
            f"Heres a markov chain based on {user.mention}'s shitposts"
            + "\n\n"
            + " ".join(dummy)
        )


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