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

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


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")

    @commands.command(hidden=True)
    async def reboot(self, ctx):
        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
            )
        await ctx.send("Sebi-Machine is restarting.")
        with open(f"src/config/reboot", "w") as f:
            f.write(f"1\n{ctx.channel.id}")
        # noinspection PyProtectedMember
        os._exit(1)


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