# coding=utf-8

from __future__ import unicode_literals, absolute_import, division, print_function

import sopel
import sopel.module

import os

from sopel_modules.SpiceBot_LoadOrder.LoadOrder import set_bot_event, startup_bot_event
from sopel_modules.SpiceBot_Events.System import bot_events_startup_register, bot_events_recieved, bot_events_trigger
from sopel_modules.SpiceBot_SBTools import bot_logging

import spicemanip


def setup(bot):
    bot_logging(bot, 'SpiceBot_CommandsQuery', "Starting setup procedure")
    bot_events_startup_register(bot, ['2002'])

    startup_bot_event(bot, "SpiceBot_CommandsQuery")

    if 'SpiceBot_CommandsQuery' not in bot.memory:
        bot.memory['SpiceBot_CommandsQuery'] = {"counts": {}, "commands": {}}


def shutdown(bot):
    if "SpiceBot_CommandsQuery" in bot.memory:
        del bot.memory["SpiceBot_CommandsQuery"]


@sopel.module.event('1003')
@sopel.module.rule('.*')
def trigger_command_list_initial(bot, trigger):
    bot_events_recieved(bot, trigger.event)

    for comtype in ['module', 'nickname', 'rule']:
        if comtype not in bot.memory['SpiceBot_CommandsQuery']['counts'].keys():
            bot.memory['SpiceBot_CommandsQuery']['counts'][comtype] = 0
        if comtype not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
            bot.memory['SpiceBot_CommandsQuery']['commands'][comtype] = dict()

    filepathlisting = []

    # main Modules directory
    main_dir = os.path.dirname(os.path.abspath(sopel.__file__))
    modules_dir = os.path.join(main_dir, 'modules')
    filepathlisting.append(modules_dir)

    # Home Directory
    home_modules_dir = os.path.join(bot.config.homedir, 'modules')
    if os.path.isdir(home_modules_dir):
        filepathlisting.append(home_modules_dir)

    # pypi installed
    try:
        import sopel_modules
        for plugin_dir in set(sopel_modules.__path__):
            for pathname in os.listdir(plugin_dir):
                pypi_modules_dir = os.path.join(plugin_dir, pathname)
                filepathlisting.append(pypi_modules_dir)
    except Exception as e:
        bot_logging(bot, 'SpiceBot_CommandsQuery', "sopel_modules not loaded :" + str(e))

    # Extra directories
    filepathlist = []
    for directory in bot.config.core.extra:
        filepathlisting.append(directory)

    for directory in filepathlisting:
        for pathname in os.listdir(directory):
            path = os.path.join(directory, pathname)
            if (os.path.isfile(path) and path.endswith('.py') and not path.startswith('_')):
                filepathlist.append(str(path))

    # CoreTasks
    ct_path = os.path.join(main_dir, 'coretasks.py')
    filepathlist.append(ct_path)

    for modulefile in filepathlist:
        module_file_lines = []
        module_file = open(modulefile, 'r')
        lines = module_file.readlines()
        for line in lines:
            module_file_lines.append(line)
        module_file.close()

        dict_from_file = dict()
        filelinelist = []

        for line in module_file_lines:

            if str(line).startswith("@"):
                line = str(line)[1:]

                # Commands
                if str(line).startswith(tuple(["commands", "module.commands", "sopel.module.commands"])):
                    comtype = "module"
                    line = str(line).split("commands(")[-1]
                    line = str("(" + line)
                    validcoms = eval(str(line))
                    if isinstance(validcoms, tuple):
                        validcoms = list(validcoms)
                    else:
                        validcoms = [validcoms]
                    validcomdict = {"comtype": comtype, "validcoms": validcoms}
                    filelinelist.append(validcomdict)
                elif str(line).startswith(tuple(["nickname_commands", "module.nickname_commands", "sopel.module.nickname_commands"])):
                    comtype = "nickname"
                    line = str(line).split("commands(")[-1]
                    line = str("(" + line)
                    validcoms = eval(str(line))
                    if isinstance(validcoms, tuple):
                        validcoms = list(validcoms)
                    else:
                        validcoms = [validcoms]
                    nickified = []
                    for nickcom in validcoms:
                        nickified.append(str(bot.nick) + " " + nickcom)
                    validcomdict = {"comtype": comtype, "validcoms": nickified}
                    filelinelist.append(validcomdict)
                elif str(line).startswith(tuple(["rule", "module.rule", "sopel.module.rule"])):
                    comtype = "rule"
                    line = str(line).split("rule(")[-1]
                    validcoms = [str("(" + line)]
                    validcomdict = {"comtype": comtype, "validcoms": validcoms}
                    filelinelist.append(validcomdict)

        for atlinefound in filelinelist:

            comtype = atlinefound["comtype"]
            validcoms = atlinefound["validcoms"]

            bot.memory['SpiceBot_CommandsQuery']['counts'][comtype] += 1

            # default command to filename
            if "validcoms" not in dict_from_file.keys():
                dict_from_file["validcoms"] = validcoms

            maincom = dict_from_file["validcoms"][0]
            if len(dict_from_file["validcoms"]) > 1:
                comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
            else:
                comaliases = []

            bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][maincom] = dict_from_file
            for comalias in comaliases:
                if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys():
                    bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][comalias] = {"aliasfor": maincom}

    for comtype in ['module', 'nickname', 'rule']:
        bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.")

    set_bot_event(bot, "SpiceBot_CommandsQuery")
    bot_events_trigger(bot, 2002, "SpiceBot_CommandsQuery")


@sopel.module.event('1004')
@sopel.module.rule('.*')
def bot_events_complete(bot, trigger):
    bot_events_recieved(bot, trigger.event)

    for comtype in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
        if comtype not in ['module', 'nickname', 'rule']:
            bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.")


def commandsquery_register(bot, command_type, validcoms, aliasfor=None):

    if not isinstance(validcoms, list):
        validcoms = [validcoms]

    if command_type not in bot.memory['SpiceBot_CommandsQuery'].keys():
        if command_type not in bot.memory['SpiceBot_CommandsQuery']['counts'].keys():
            bot.memory['SpiceBot_CommandsQuery']['counts'][command_type] = 0
        if command_type not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
            bot.memory['SpiceBot_CommandsQuery']['commands'][command_type] = dict()
    bot.memory['SpiceBot_CommandsQuery']['counts'][command_type] += 1

    dict_from_file = dict()

    # default command to filename
    if "validcoms" not in dict_from_file.keys():
        dict_from_file["validcoms"] = validcoms

    if not aliasfor:

        maincom = dict_from_file["validcoms"][0]
        if len(dict_from_file["validcoms"]) > 1:
            comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
        else:
            comaliases = []
        bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][maincom] = dict_from_file
    else:
        comaliases = validcoms

    for comalias in comaliases:
        if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][command_type].keys():
            bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][comalias] = {"aliasfor": aliasfor}


@sopel.module.event('2002')
@sopel.module.rule('.*')
def bot_events_setup(bot, trigger):
    bot_events_recieved(bot, trigger.event)