### This file is part of KoFooBot and is licensed under BSD-license according to the   ###

### LICENSE file in the base directory.                                                ###

### Code in this file is contributed by:                                               ###

###    Krister Svanlund <krister.svanlund gmail.com>                                   ###

### MegaHAL implementation from http://megahal.alioth.debian.org/                      ###

import mh_python as megahal

import re

update_megahal_delay = 240

update_megahal_time_left = 0

def init_megahal(bot, server, sender = None):

    print " + Initiate MegaHAL module."

    megahal.initbrain()

    return True

def unload_megahal(bot, server, sender):

    print " + Unload MegaHAL module."

    megahal.cleanup()

    return True

def update_megahal(bot, server):

    print " + Clean up MegaHAL brain."

    megahal.cleanup()

def megahal_public_msg_handler(bot, server, sender, target, msg, action = False):

    if not action:

        res = re.match("(\S+)[:;,] (.*)", msg)

        if res:

            nick = res.group(1)

            message = res.group(2)

            if nick == bot.settings.bot_nick_name:

                reply = megahal.doreply(message)

                bot.respond(server, sender, target, reply)

                return False

            else:

                megahal.learn(message)

        megahal.learn(msg)

        return False

def megahal_private_msg_handler(bot, server, sender, msg, action = False):

    if not action:

        megahal.learn(msg)

    return False