import io

import nerimity

import bot
import utils as u

# from catbox_async_uploader.catbox_async_uploader.enums import LitterboxDuration
# from pyzxz import ZeroXZero

def setup(bot: bot.Bot):
    @bot.command(name="chart", aliases=["c", "chartalbum", "albumchart", "collage"])
    @bot.slash_command(name="chartalbum", description="Generate an album collage.")
    async def chart_album(ctx: nerimity.Context, size: str = "3x3", timeframe: str = "7day", username: str = None):
        temp_msg = None

        if 'x' not in size:
            await ctx.send(u.error_msg("Please provide a valid size.\nie `/chartalbum 5x5`"))
            return

        if int(size.split('x')[0]) > 5 or int(size.split('x')[1]) > 5:
            await ctx.send(u.error_msg("Maximum size allowed is 5x5."))
            return
        
        if not username:
            try: username = await bot.get_lastfm(ctx.author.id)
            except Exception as e:
                print(e)
                await ctx.send(u.error_msg(f"Unknown database error:\n{e}"))

            if not username:
                await ctx.send(u.error_msg("Please provide a Last.fm username (or set yours with `/setfm`)."))
                print("returned")
                return

        if username.startswith("[@:"):
                mentioned = bot.get_user(username[3:-1])
                try: username = await bot.get_lastfm(mentioned.id)
                except Exception as e:
                    print(e)
                    await ctx.send(u.error_msg(f"Unknown database error:\n{e}"))

                if not username:
                    await ctx.send(u.error_msg(f"[@:{mentioned.id}] doesn't seem to have an account set. Do so with `/setfm`."))
                    print("returned")
                    return

        match timeframe:
            case "a" | "all" | "alltime" | "o" | "overall": timeframe = "overall"
            case "y" | "year" | "yearly": timeframe = "12month"
            case "m" | "month" | "monthly": timeframe = "1month"
            case "w" | "week" | "weekly": timeframe = "7day"
            case "d" | "day" | "daily": timeframe = "1day"
            case _: "7day"

        if timeframe == "1day":
            await ctx.send(u.error_msg("Daily charts are not yet supported. Sorry!!"))
            return

        try:
            temp_msg = await ctx.send(f"Generating album chart for **{username}**...")

            image = bot.collage_generator.generate(
                entity = "album", 
                username = username,
                rows = int(size.split(sep="x")[0]),
                cols = int(size.split(sep="x")[1]),
                period = timeframe
            )

            img_bytes = io.BytesIO()
            image.save(fp=img_bytes, format="png")

            # link = await bot.catbox_uploader.upload_to_litterbox(
            #     file_path_or_bytes = img_bytes,
            #     file_name = "collage.png",
            #     duration = LitterboxDuration.H12
            # )

            # attachment = await u.construct_attachment_from_bytes(
            #     filename = "chart",
            #     file_type = "png",
            #     bytes_arr = img_bytes,
            # ).upload()

            link = await bot.zxz.upload_from_bytes(img_bytes.getvalue(), "chart.png")

            await ctx.send(u.good_msg(f"**{size}** {timeframe} chart for [{username}](https://last.fm/user/{username}) successfully generated:\n{link}"))
            temp_msg.delete()

        except Exception as e:
            if temp_msg: temp_msg.delete()
            print(e)
            await ctx.send(u.error_msg(f"I crashed and burned while generating the chart:\n`{e}`"))