# coding=utf8
from __future__ import unicode_literals, absolute_import, division, print_function
"""
This is the SpiceBot Prerun system.
"""

from sopel.tools import Identifier

import functools
import copy
import datetime

import spicemanip

from .Tools import command_permissions_check
from .Commands import commands as botcommands
from .Database import db as botdb
from .Channels import channels as botchannels
from .MessageLog import messagelog as botmessagelog


def prerun(t_command_type='module', t_command_subtype=None):

    def actual_decorator(function):

        @functools.wraps(function)
        def internal_prerun(bot, trigger, *args, **kwargs):

            trigger_command_type = str(t_command_type)

            # Primary command used for trigger, and a list of all words
            trigger_args, trigger_command = make_trigger_args(trigger.args[1], trigger_command_type)

            trigger_command_type = botcommands.find_command_type(trigger_command)
            if not trigger_command_type:
                return

            if t_command_subtype:
                if trigger_command_type != t_command_subtype:
                    return

            # Argsdict Defaults
            argsdict_default = {}
            argsdict_default["type"] = trigger_command_type
            argsdict_default["com"] = trigger_command

            # messagelog ID
            argsdict_default["log_id"] = botmessagelog.messagelog_assign()

            argsdict_default["realcom"] = botcommands.get_realcom(argsdict_default["com"], trigger_command_type)

            if argsdict_default["type"] == 'nickname':
                argsdict_default["comtext"] = "'" + bot.nick + " " + str(argsdict_default["com"]) + "'"
                argsdict_default["realcomtext"] = "'" + bot.nick + " " + str(argsdict_default["realcom"]) + "'"
            else:
                argsdict_default["comtext"] = "'" + str(argsdict_default["com"]) + "'"
                argsdict_default["realcomtext"] = "'" + str(argsdict_default["realcom"]) + "'"

            argsdict_default["realcomref"] = argsdict_default["type"] + "_" + str(argsdict_default["realcom"])

            argsdict_default["dict"] = botcommands.get_command_dict(argsdict_default["realcom"], trigger_command_type)

            # split into && groupings
            and_split = trigger_and_split(trigger_args)

            # Create dict listings for trigger.sb
            argsdict_list = trigger_argsdict_list(argsdict_default, and_split)

            # Run the function for all splits
            botmessagelog.messagelog_start(bot, trigger, argsdict_default["log_id"])
            runcount = 0
            for argsdict in argsdict_list:
                runcount += 1
                trigger.sb = copy.deepcopy(argsdict)

                trigger.sb["runcount"] = runcount

                trigger.sb["args"], trigger.sb["hyphen_arg"] = trigger_hyphen_args(trigger.sb["args"])
                if not trigger.sb["hyphen_arg"]:
                    # check if anything prohibits the nick from running the command
                    if trigger_runstatus(bot, trigger):
                        function(bot, trigger, *args, **kwargs)
                else:
                    trigger_hyphen_arg_handler(bot, trigger)
            botmessagelog.messagelog_exit(bot, argsdict_default["log_id"])

        return internal_prerun
    return actual_decorator


def make_trigger_args(triggerargs_one, trigger_command_type='module'):
    trigger_args = spicemanip.main(triggerargs_one, 'create')
    if trigger_command_type in ['nickname']:
        trigger_command = spicemanip.main(trigger_args, 2).lower()
        trigger_args = spicemanip.main(trigger_args, '3+', 'list')
    else:
        trigger_command = spicemanip.main(trigger_args, 1).lower()[1:]
        trigger_args = spicemanip.main(trigger_args, '2+', 'list')
    return trigger_args, trigger_command


def trigger_and_split(trigger_args):
    trigger_args_list_split = spicemanip.main(trigger_args, "split_&&")
    if not len(trigger_args_list_split):
        trigger_args_list_split.append([])
    return trigger_args_list_split


def trigger_argsdict_list(argsdict_default, and_split):
    prerun_split = []
    for trigger_args_part in and_split:
        argsdict_part = copy.deepcopy(argsdict_default)
        argsdict_part["args"] = spicemanip.main(trigger_args_part, 'create')
        if len(argsdict_part["args"]) and argsdict_part["args"][0] == "-a":
            argsdict_part["adminswitch"] = True
            argsdict_part["args"] = spicemanip.main(argsdict_part["args"], '2+', 'list')
        else:
            argsdict_part["adminswitch"] = False
        prerun_split.append(argsdict_part)
    return prerun_split


def trigger_argsdict_single(argsdict_default, trigger_args_part):
    argsdict_part = copy.deepcopy(argsdict_default)
    argsdict_part["args"] = spicemanip.main(trigger_args_part, 'create')
    if len(argsdict_part["args"]) and argsdict_part["args"][0] == "-a":
        argsdict_part["adminswitch"] = True
        argsdict_part["args"] = spicemanip.main(argsdict_part["args"], '2+', 'list')
    else:
        argsdict_part["adminswitch"] = False
    return argsdict_part


def trigger_runstatus(bot, trigger):

    # Bots can't run commands
    if Identifier(trigger.nick) == bot.nick:
        return False

    # Allow permissions for enabling and disabling commands via hyphenargs
    if trigger.sb["adminswitch"]:
        if command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
            return True
        else:
            botmessagelog.messagelog_error(trigger.sb["log_id"], "The admin switch (-a) is for use by authorized nicks ONLY.")
            return False

    if trigger.sb["hyphen_arg"]:
        return False

    if trigger.sb["runcount"] > 1:
        # check channel multirun blocks
        if not trigger.is_privmsg:
            channel_disabled_list = botcommands.get_commands_disabled(str(trigger.sender), "multirun")
            if trigger.sb["realcom"] in list(channel_disabled_list.keys()):
                reason = channel_disabled_list[trigger.sb["realcom"]]["reason"]
                timestamp = channel_disabled_list[trigger.sb["realcom"]]["timestamp"]
                bywhom = channel_disabled_list[trigger.sb["realcom"]]["disabledby"]
                message = "The " + str(trigger.sb["comtext"]) + " command multirun usage was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
                return trigger_cant_run(bot, trigger, message)

        # don't run commands that are disabled for specific users
        nick_disabled_list = botcommands.get_commands_disabled(str(trigger.nick), "multirun")
        if trigger.sb["realcom"] in list(nick_disabled_list.keys()):
            bywhom = nick_disabled_list[trigger.sb["realcom"]]["disabledby"]
            if bywhom != trigger.nick:
                reason = nick_disabled_list[trigger.sb["realcom"]]["reason"]
                timestamp = nick_disabled_list[trigger.sb["realcom"]]["timestamp"]
                message = "The " + str(trigger.sb["comtext"]) + " command was multirun unsage disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
                return trigger_cant_run(bot, trigger, message)
            else:
                botcommands.unset_command_disabled(trigger.sb["realcomref"], trigger.nick, "multirun")
                botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now enabled for " + str(trigger.nick))

    # don't run commands that are disabled in channels
    if not trigger.is_privmsg:
        channel_disabled_list = botcommands.get_commands_disabled(str(trigger.sender), "fully")
        if trigger.sb["realcom"] in list(channel_disabled_list.keys()):
            reason = channel_disabled_list[trigger.sb["realcom"]]["reason"]
            timestamp = channel_disabled_list[trigger.sb["realcom"]]["timestamp"]
            bywhom = channel_disabled_list[trigger.sb["realcom"]]["disabledby"]
            message = "The " + str(trigger.sb["comtext"]) + " command was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
            return trigger_cant_run(bot, trigger, message)

    # don't run commands that are disabled for specific users
    nick_disabled_list = botcommands.get_commands_disabled(str(trigger.nick), "fully")
    if trigger.sb["realcom"] in list(nick_disabled_list.keys()):
        bywhom = nick_disabled_list[trigger.sb["realcom"]]["disabledby"]
        if bywhom != trigger.nick:
            reason = nick_disabled_list[trigger.sb["realcom"]]["reason"]
            timestamp = nick_disabled_list[trigger.sb["realcom"]]["timestamp"]
            message = "The " + str(trigger.sb["comtext"]) + " command was disabled by " + bywhom + " for " + str(trigger.sender) + " at " + str(timestamp) + " for the following reason: " + str(reason)
            return trigger_cant_run(bot, trigger, message)
        else:
            botcommands.unset_command_disabled(trigger.sb["realcomref"], trigger.nick, "fully")
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now enabled for " + str(trigger.nick))

    return True


def trigger_cant_run(bot, trigger, message=None):
    if message:
        botmessagelog.messagelog_error(trigger.sb["log_id"], message)
    if command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
        botmessagelog.messagelog_error(trigger.sb["log_id"], "You however are authorized to bypass this warning with the (-a) admin switch.")
    return False


def trigger_hyphen_args(trigger_args_part):
    valid_hyphen_args = [
                        'check',
                        'enable', 'disable',
                        'block', 'unblock',
                        "activate", "deactivate",
                        "on", "off",
                        'multirun', 'multiruns',
                        'example', 'usage',
                        'filename', 'filepath',
                        'foldername', 'folderpath',
                        "author",
                        'contribs', 'contrib', "contributors",
                        'alias', 'aliases'
                        ]
    hyphen_args = []
    trigger_args_unhyphend = []
    for worditem in trigger_args_part:
        if str(worditem).startswith("--") and worditem[2:] in valid_hyphen_args:
            hyphen_args.append(worditem[2:])
        else:
            trigger_args_unhyphend.append(worditem)
    if len(hyphen_args):
        hyphenarg = hyphen_args[0]
    else:
        hyphenarg = None
    return trigger_args_unhyphend, hyphenarg


def trigger_hyphen_arg_handler(bot, trigger):

    # Commands that cannot run via privmsg
    if trigger.sb["hyphen_arg"] in ['check']:
        if trigger.sb["com"].lower() != trigger.sb["realcom"]:
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is a valid alias command for " + trigger.sb["realcomtext"])
        else:
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is a valid command")
        return

    elif trigger.sb["hyphen_arg"] in [
                                        'enable', 'disable',
                                        'block', 'unblock',
                                        "activate", "deactivate",
                                        "on", "off"
                                        ]:

        target = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
        if not target:
            if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:
                botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I enabling " + str(trigger.sb["comtext"]) + " usage for?")
            else:
                botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I disabling " + str(trigger.sb["comtext"]) + " usage for?")
            return

        if not botdb.check_nick_id(target) and not botchannels.check_channel_bot(target, True):
            botmessagelog.messagelog_error(trigger.sb["log_id"], "I don't know who/what " + str(target) + " is.")
            return

        if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
            if target != trigger.nick:
                if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:
                    botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable this command for " + str(target) + " due to privilege issues.")
                else:
                    botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to disable this command for " + str(target) + " due to privilege issues.")
                return

        if trigger.sb["hyphen_arg"] in ['enable', 'unblock', "activate", "on"]:

            if not botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "fully"):
                botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " is already enabled for " + str(target))
                return

            botcommands.unset_command_disabled(trigger.sb["realcomref"], target, "fully")
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now enabled for " + str(target))
            return

        else:

            if botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "fully"):
                botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " is already disabled for " + str(target))
                return

            trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
            timestamp = str(datetime.datetime.utcnow())

            botcommands.set_command_disabled(trigger.sb["realcomref"], target, timestamp, trailingmessage, trigger.nick, "fully")
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " is now disabled for " + str(target))
            return

    if trigger.sb["hyphen_arg"] in ['multirun', 'multiruns']:

        onoff = spicemanip.main(trigger.sb["args"], 1) or None
        if not onoff or onoff not in ['enable', 'unblock', "activate", "on"]:
            botmessagelog.messagelog_error(trigger.sb["log_id"], "Do you want to enable or disable " + str(trigger.sb["comtext"]) + " multirun usage?")
            return

        target = spicemanip.main(trigger.sb["args"], 1) or trigger.nick
        if not target:
            if onoff in ['enable', 'unblock', "activate", "on"]:
                botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I enabling " + str(trigger.sb["comtext"]) + " multirun usage for?")
            else:
                botmessagelog.messagelog_error(trigger.sb["log_id"], "Who/Where am I disabling " + str(trigger.sb["comtext"]) + " multirun usage for?")
            return

        if not botdb.check_nick_id(target) and not botchannels.check_channel_bot(target, True):
            botmessagelog.messagelog_error(trigger.sb["log_id"], "I don't know who/what " + str(target) + " is.")
            return

        if not command_permissions_check(bot, trigger, ['admins', 'owner', 'OP', 'ADMIN', 'OWNER']):
            if target != trigger.nick:
                if onoff in ['enable', 'unblock', "activate", "on"]:
                    botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable multirun usage on " + str(trigger.sb["comtext"]) + " for " + str(target) + " due to privilege issues.")
                else:
                    botmessagelog.messagelog_error(trigger.sb["log_id"], "I was unable to enable multirun usage on " + str(trigger.sb["comtext"]) + " for " + str(target) + " due to privilege issues.")
                return

        if onoff in ['enable', 'unblock', "activate", "on"]:

            if not botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "multirun"):
                botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is already enabled for " + str(target))
                return

            botcommands.unset_command_disabled(trigger.sb["realcomref"], target, "multirun")
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now enabled for " + str(target))
            return

        else:

            if botcommands.check_commands_disabled(trigger.sb["realcomref"], target, "multirun"):
                botmessagelog.messagelog_error(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is already disabled for " + str(target))
                return

            trailingmessage = spicemanip.main(trigger.sb["args"], 0) or "No reason given."
            timestamp = str(datetime.datetime.utcnow())

            botcommands.set_command_disabled(trigger.sb["realcomref"], target, timestamp, trailingmessage, trigger.nick, "multirun")
            botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + " multirun is now disabled for " + str(target))
            return

    elif trigger.sb["hyphen_arg"] in ['example', 'usage']:
        botmessagelog.messagelog(trigger.sb["log_id"], trigger.sb["comtext"] + ": " + str(trigger.sb["dict"]["example"]))
        return

    elif trigger.sb["hyphen_arg"] in ['filename', 'filepath']:
        botmessagelog.messagelog(trigger.sb["log_id"], "The " + str(trigger.sb["comtext"]) + " file is located at " + str(trigger.sb["dict"][trigger.sb["hyphen_arg"]]))
        return

    elif trigger.sb["hyphen_arg"] in ['foldername', 'folderpath']:
        botmessagelog.messagelog(trigger.sb["log_id"], "The " + str(trigger.sb["comtext"]) + " folder is located at " + str(trigger.sb["dict"][trigger.sb["hyphen_arg"]]))
        return

    elif trigger.sb["hyphen_arg"] in ['author']:
        botmessagelog.messagelog(trigger.sb["log_id"], "The author of the " + str(trigger.sb["comtext"]) + " command is " + trigger.sb["dict"]["author"] + ".")
        return

    elif trigger.sb["hyphen_arg"] in ['contribs', 'contrib', "contributors"]:
        botmessagelog.messagelog(trigger.sb["log_id"], "The contributors of the " + str(trigger.sb["comtext"]) + " command are " + spicemanip.main(trigger.sb["dict"]["contributors"], "andlist") + ".")
        return

    elif trigger.sb["hyphen_arg"] in ['alias', 'aliases']:
        botmessagelog.messagelog(trigger.sb["log_id"], "The alaises of the " + str(trigger.sb["comtext"]) + " command are " + spicemanip.main(trigger.sb["dict"]["validcoms"], "andlist") + ".")
        return

    return