#!/usr/bin/python
# -*- coding: utf-8 -*-

from discord.ext import commands
import discord
import traceback
import aiofiles

class Upload:
    """
    CogName should be the name of the cog
    """
    def __init__(self, bot):
        self.bot = bot
        print('upload loaded')

    @commands.command()
    async def reload(self, ctx, *, extension: str):
        """Reload an extension."""
        await ctx.trigger_typing()
        if ctx.author.id not in self.bot.ownerlist:
            return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)

        extension = extension.lower()
        try:
            self.bot.unload_extension("src.cogs.{}".format(extension))
            self.bot.load_extension("src.cogs.{}".format(extension))
        except Exception as e:
            traceback.print_exc()
            await ctx.send(f'Could not reload `{extension}` -> `{e}`')
        else:
            await ctx.send(f'Reloaded `{extension}`.')

    @commands.command()
    async def reloadall(self, ctx):
        """Reload all extensions."""
        await ctx.trigger_typing()
        if ctx.author.id not in self.bot.ownerlist:
            return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)

        try:
            for extension in self.bot.extensions:
                self.bot.unload_extension(extension)
                self.bot.load_extension(extension)
            await ctx.send(f"Reload success! :thumbsup:\n")
        except Exception as e:
            await ctx.send(f"Could not reload `{extension}` -> `{e}`.\n")

    @commands.command()
    async def unload(self, ctx, *, extension: str):
        """Unload an extension."""
        await ctx.trigger_typing()
        if ctx.author.id not in self.bot.ownerlist:
            return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)

        extension = extension.lower()
        try:
            self.bot.unload_extension("src.cogs.{}".format(extension))

        except Exception as e:
            traceback.print_exc()
            if ctx.message.author.id not in self.bot.owner_list:
                await ctx.send(f'Could not unload `{extension}` -> `{e}`')

        else:
            await ctx.send(f'Unloaded `{extension}`.')

    @commands.command()
    async def load(self, ctx,  *, extension: str):
        """Load an extension."""
        await ctx.trigger_typing()
        if ctx.author.id not in self.bot.ownerlist:
            return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)

        extension = extension.lower()
        try:
            self.bot.load_extension("src.cogs.{}".format(extension))
        except Exception as e:
            traceback.print_exc()
            await ctx.send(f'Could not load `{extension}` -> `{e}`')
        else:
            await ctx.send(f'Loaded `{extension}`.')
      
    @commands.command()
    async def permunload(self, ctx, extension=None):
        """Disables permanently a cog."""
        await ctx.trigger_typing()
        if ctx.author.id not in self.bot.ownerlist:
            return await ctx.send('Only my contributors can use me like this :blush:', delete_after=10)
        
        if cog is None:
            return await ctx.send("Please provide a extension. Do `help permunload` for more info")
        
        extension = extension.lower()
        
        async with aiofiles.open("extension.txt") as fp:
            lines=fp.readlines()
            
        removed = False
        async with aiofiles.open("extension.txt", "w") as fp:
            for i in lines:
                if i.replace("\n", "") != extension:
                    fp.write(i)
                else:
                    removed = True
                    break
                    
        if removed is True:
            try:
                self.bot.unload_extension(extension)
            except:
                pass
            return await ctx.send("Extension removed successfully")
        
        await ctx.send("Extension not found")

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