import pathlib

import aiohttp
import discord
from PIL import Image

from redbot.core import Config, checks, commands

from redbot.core.bot import Red

from MyQR import myqr
from redbot.core.data_manager import cog_data_path


class QRInvite:
    """
    V3 Cog Template
    """

    def __init__(self, bot: Red):
        self.bot = bot
        self.config = Config.get_conf(self, identifier=9811198108111121, force_registration=True)
        default_global = {}
        default_guild = {}

        self.config.register_global(**default_global)
        self.config.register_guild(**default_guild)

    @commands.command()
    async def qrinvite(self, ctx: commands.Context, invite: str=None, colorized: bool=False, image_url: str=None):
        """
        Create a custom QR code invite for this server
        """
        if invite is None:
            try:
                invite = await ctx.channel.create_invite()
            except discord.Forbidden:
                try:
                    invite = await ctx.channel.invites()
                    invite = invite[0]
                except discord.Forbidden:
                    await ctx.send("No permission to get an invite, please provide one")
                    return
            invite = invite.code

        if image_url is None:
            image_url = ctx.guild.icon_url

        extension = pathlib.Path(image_url).parts[-1].replace('.','?').split('?')[1]

        path: pathlib.Path = cog_data_path(self)
        image_path = path / (ctx.guild.icon+"."+extension)
        async with aiohttp.ClientSession() as session:
            async with session.get(image_url) as response:
                image = await response.read()

        with image_path.open("wb") as file:
            file.write(image)

        if extension == "webp":
            new_path = convert_png(str(image_path))
        else:
            new_path = str(image_path)

        myqr.run(invite,picture=new_path,save_name=ctx.guild.icon+"_qrcode.png",
                 save_dir=str(cog_data_path(self)),colorized=colorized,)

        png_path: pathlib.Path = path / (ctx.guild.icon+"_qrcode.png")
        with png_path.open("rb") as png_fp:
            await ctx.send(file=discord.File(png_fp.read(), "qrcode.png"))

def convert_png(path):
    im = Image.open(path)
    im.load()
    alpha = im.split()[-1]
    im = im.convert('RGB').convert('P', palette=Image.ADAPTIVE, colors=255)
    mask = Image.eval(alpha, lambda a: 255 if a <=128 else 0)
    im.paste(255, mask)
    newPath = path.replace(".webp",".png")
    im.save(newPath, transparency=255)
    return newPath