#!/usr/bin/env python3

from argparse import ArgumentParser
from os.path import dirname, realpath

from bot import Bot, Tasks, Responses
from actions import actions
from coroutines import coroutines

parser = ArgumentParser(description="A meta bot for ~team")
parser.add_argument(
    "-c", 
    "--config", 
    dest="config",
    default="settings.json", 
    help="Load config file"
)
arguments = parser.parse_args()

bot = Bot("127.0.0.1", 6667)
responses = Responses(bot)
tasks = Tasks(bot)

# for coro in coroutines:
#     worker = coro["worker"]
#     interval = coro["interval"]
#     state = coro.get("state", None)
#     coro_state = state if state is not None else (bot,)
#     tasks.add_coroutine(worker, interval, coro_state)
tasks.coroutines = coroutines

for action in actions:
    if "type" in action and "pattern" in action and "callback" in action:
        responses.add_trigger(
            action["type"], 
            action["pattern"], 
            action["callback"]
        )

def try_to_king_me(channel):
    bot.send_message("ChanServ", "REGISTER {}", channel)
    bot.send_message("ChanServ", "SET Successor {} {}", channel, bot.botnick)
    bot.send_message("ChanServ", "SET Founder {} {}", channel, bot.author)

def handle_pm(name, response):
    print("PM: {} - {}".format(name, response))

def handle_mode(channel, mode):
    if mode == "-r":
        try_to_king_me(channel)

def handle_invite(channel, name):
    changed = False
    kingme = bot.settings.get("extras", dict()).get("kingme", [])

    if channel in kingme:
        try_to_king_me(channel)

    users = bot.memories["users"]
    if name not in users:
        bot.memories["users"][name] = dict()
        changed = True

    if "invites" not in users[name]:
        bot.memories["users"][name]["invites"] = list()
        changed = True

    if channel not in bot.memories["users"][name]["invites"]:
        bot.memories["users"][name]["invites"].append(channel)
        changed = True

    if changed:
        bot.thread(bot.save_memories)

def handle_kick(name):
    users = bot.memories["users"]
    if name not in users:
        bot.memories["users"][name] = dict()

    bot.memories["users"][name]["kicker"] = True
    bot.thread(bot.save_memories)

def handle_message(name, source, response):
    responses.parse(name, source, response)
    if response == "!debug":
        bot.logger.debug(":: {}".format(bot.memories))

def handle_crashed():
    bot.logger.debug("Rebooting")
    bot.crashed = True
    bot.tasks.stop()

    tasks = Tasks(bot)
    tasks.coroutines = coroutines

    bot.tasks = tasks
    bot.start(arguments.config, dirname(realpath(__file__)), {
        "pm": handle_pm,
        "mode": handle_mode,
        "invite": handle_invite,
        "kick": handle_kick,
        "crashed": handle_crashed,
        "message": handle_message
    })

if __name__ == "__main__":
    bot.tasks = tasks
    bot.start(arguments.config, dirname(realpath(__file__)), {
        "pm": handle_pm,
        "mode": handle_mode,
        "invite": handle_invite,
        "kick": handle_kick,
        "crashed": handle_crashed,
        "message": handle_message
    })