"""
MegaHAL - MegaHAL bot for IRC
Copyright (C) 2010  Brian S. Stephan

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""

import os
import re

from extlib import irclib
from extlib import megahal

from Module import Module

class MegaHAL(Module):
    """Have an annoying, dumb, and hilarious chatterbot on IRC."""

    def register_handlers(self, server):
        server.add_global_handler('welcome', self.on_connect)
        server.add_global_handler('pubmsg', self.on_pubmsg)
        server.add_global_handler('privmsg', self.on_privmsg)

    def unregister_handlers(self):
        self.server.remove_global_handler('welcome', self.on_connect)
        self.server.remove_global_handler('pubmsg', self.on_pubmsg)
        self.server.remove_global_handler('privmsg', self.on_privmsg)

    def on_connect(self, connection, event):
        """
        Upon connecting, open the MegaHAL brain and get ready for doing actual stuff.
        """

        self.megahal = megahal.MegaHAL()

    def reload(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
        """Broken. Don't do anything."""

    def on_pubmsg(self, connection, event):
        """
        Handle pubmsg events. Does some variable setup and initial sanity checking before
        calling Module.do, which should be implemented by subclasses and what can be
        ultimately responsible for the work.
        """

        nick = irclib.nm_to_n(event.source())
        userhost = irclib.nm_to_uh(event.source())
        replypath = event.target()
        what = event.arguments()[0]

        admin_unlocked = False

        try:
            if userhost == self.config.get('dr.botzo', 'admin_userhost'):
                admin_unlocked = True
        except NoOptionError: pass

        # only do commands if the bot has been addressed directly
        addressed_pattern = '^' + connection.get_nickname() + '[:,]?\s+'
        addressed_re = re.compile(addressed_pattern)

        need_prefix = True
        try:
            need_prefix = self.config.getboolean(self.__class__.__name__, 'meta.pubmsg_needs_bot_prefix')
        except NoOptionError: pass
        except NoSectionError: pass

        re_add_nick = False
        if not addressed_re.match(what) and need_prefix:
            return
        else:
            if addressed_re.match(what):
                re_add_nick = True
                what = addressed_re.sub('', what)

        if replypath is not None:
            what = self.try_recursion(connection, event, nick, userhost, replypath, what, admin_unlocked)

        # try reloading
        self.reload(connection, event, nick, userhost, replypath, what, admin_unlocked)

        if re_add_nick:
            self.do(connection, event, nick, userhost, replypath, connection.get_nickname() + ": " + what, admin_unlocked)
        else:
            self.do(connection, event, nick, userhost, replypath, what, admin_unlocked)

    def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
        """
        Say something on IRC if the bot is being addressed, or at least learn what
        was said.
        """

        line = what
        reply = ""
        append_nick = False

        if re.search(connection.get_nickname(), line, re.IGNORECASE) is not None:
            addressed_pattern = '^' + connection.get_nickname() + '[:,]\s+'
            addressed_re = re.compile(addressed_pattern)
            if addressed_re.match(line):
                line = addressed_re.sub('', line)
                append_nick = True

            reply = self.megahal.get_reply(line)
        else:
            self.megahal.learn(line)

        self.megahal.sync()
        if reply is not "":
            if append_nick:
                reply = nick + ": " + reply
            return self.reply(connection, replypath, reply)

# vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on;