import json
import codecs
import discord
import time
import sys
import traceback

from discord.ext import commands
from cogs.utils.settings import Settings

from cogs.utils.localizer import Localizer

initial_extension = [
    'cogs.cogs',
    'cogs.botsettings',
    'cogs.misc'
]


with codecs.open("config.json", 'r', encoding='utf8') as f:
    config = json.load(f)

default_prefix = config["default_prefix"]


def _get_prefix(bot, message):
    if not message.guild:
        return default_prefix
    prefixes = bot.settings.get(message.guild.id, 'prefixes', 'default_prefix')
    return commands.when_mentioned_or(*prefixes)(bot, message)


class Bot(commands.Bot):
    def __init__(self, debug: bool=False):
        super().__init__(command_prefix=_get_prefix,
                         description=config["description"])

        self.settings = Settings(**config)
        # Todo, maybe fix. Ability to set default bot lang in settings
        self.localizer = Localizer(self.settings.default_lang_path, "en_en")
        self.debug = debug

        for extension in initial_extension:
            try:
                self.load_extension(extension)
            except Exception as e:
                print(e)

    async def on_command_error(self, ctx, err):
        if not self.debug:
            if (isinstance(err, commands.MissingRequiredArgument) or
                    isinstance(err, commands.BadArgument)):
                formatter = ctx.bot.formatter
                if ctx.invoked_subcommand is None:
                    _help = await formatter.format_help_for(ctx, ctx.command)
                else:
                    _help = await formatter.format_help_for(ctx,
                                                            ctx.invoked_subcommand)

                for message in _help:
                    await ctx.send(message)

            if isinstance(err, commands.CommandInvokeError):
                pass

            elif isinstance(err, commands.NoPrivateMessage):
                # Todo, fix, use default bot lang.
                await ctx.send('That command is not available in DMs')
                #await ctx.send(self.localizer.format_str("{commands.not_available_dm}", self.settings.get(ctx.guild.id, 'locale', 'default_locale')))

            elif isinstance(err, commands.CheckFailure):
                pass

            elif isinstance(err, commands.CommandNotFound):
                pass
        else:
            tb = err.__traceback__
            traceback.print_tb(tb)
            print(err)

    async def on_ready(self):
        if not hasattr(self, 'uptime'):
            self.uptime = time.time()
            if self.debug:
                print('\n\nDebug mode')

        print(f'\nLogged in as: {self.user.name}' +
              f' in {len(self.guilds)} servers.')
        print(f'Version: {discord.__version__}\n')

        self.load_extension('cogs.music')
        self.load_extension('cogs.musicevents')

        await self.change_presence(activity=discord.Game(type=0,
                                   name=config["playing"]),
                                   status=discord.Status.online)

    def run(self):
        try:
            super().run(config["token"], reconnect=True)
        except Exception as e:
            tb = e.__traceback__
            traceback.print_tb(tb)
            print(e)


def run_bot(debug: bool=False):
    bot = Bot(debug=debug)
    bot.run()

if __name__ == '__main__':
    if 'debug' in sys.argv:
        run_bot(debug=True)
    else:
        run_bot(debug=False)