### 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>                                   ###

import re, irclib

#####

# Module interface

##

services_settings = { 'channel_passwords': {} }

do_register_command = { 'description': "",

                        'long help': """\

""",

                        'arguments': [("nick [<email>]", "Register with nickserv."),

                                      ("channel <channel> <password> <description>", "Register channel with chanserv.")],

                        'public': False,

                        'level': 100}

owned_channels = []

getting_alist_from = None

waiting_for_pass_for = None

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

    if type(bot.settings.bot_channel_list) is dict:

        bot.settings.services_channel_passwords.extend(bot.settings.bot_channel_list)

        bot.settings.bot_channel_list = bot.settings.bot_channel_list.keys()

    identify_with_nickserv(bot, server)

    for channel in bot.channel_list:

        identify_with_chanserv(bot, server, channel)

        request_access_list(bot, server, channel)

    return True

def services_private_notice_handler(bot, server, sender, msg):

    sender_nick = irclib.nm_to_n(sender)

    if sender_nick.lower() == "nickserv":

        return handle_nickserv_notice(bot, server, msg)

    elif sender_nick.lower() == "chanserv":

        return handle_chanserv_notice(bot, server, msg)

def services_join_part_handler(bot, server, channel, join):

    if join:

        ### Handle join event ###

        identify_with_chanserv(bot, server, channel)

        request_access_list(bot, server, channel)

    else:

        ### Handle part event ###

        if channel in owned_channels:

            owned_channels.remove(channel)

def do_register(bot, server, sender, target, args):

    """Register either nick och channel with respective service."""

    if args:

        if args[0] == 'nick':

            email = bot.settings.bot_nick_email

            if args[1:]:

                email = args[1]

            register_with_nickserv(bot, server, email, irclib.nm_to_n(sender))

            return True

        elif args[0] == 'channel':

            if len(args) > 4: ### Thats 'channel', '<channel>', '<password>', '<Channel description>'

                channel = args[1]

                password = args[2]

                description = ' '.join(args[3:])

                if not channel in bot.channel_list:

                    bot.respond(server, sender, target, "Can not register channel without joining it.")

                    print " - Bot not in channel '%s'." % channel

                if password == '*':

                    if channel in bot.settings.services_channel_passwords:

                        password = bot.settings.services_channel_passwords[channel]

                        print " + Registering %s with the password '%s'." (channel, password)

                    else:

                        bot.respond(server, sender, target, "No password to use. Could not register.")

                        print " - No password used, could not register channel."

                        return False

                register_with_chanserv(bot, server, channel, password, description)

            return True

    else:

        bot.respond(server, sender, target, "Not enough arguments.")

        return False

##

# End of module interface

#####

def identify_with_nickserv(bot, server):

    print "Identifying with nickserv..."

    server.privmsg("nickserv", "identify %s" % bot.settings.bot_nick_password)

def identify_with_chanserv(bot, server, channel):

    print "Identifying with chanserv for channel '%s'..." % channel

    chanpass = bot.settings.services_channel_passwords.get(channel, "")

    if chanpass:

        server.privmsg("chanserv", "identify %s %s" % (channel, chanpass))

    else:

        print " - Has no password for '%s'." % channel

def register_with_nickserv(bot, server, email, sender_nick = None):

    print "Register %s with nickserv." % bot.settings.bot_nick_name

    if bot.settings.bot_nick_password:

        server.privmsg("nickserv", "register %s %s" % (bot.settings.bot_nick_password, email))

        if sender:

            bot.privmsg(sender_nick, "Register nickname '%s' with nickserv using e-mail '%s'." % (bot.settings.bot_nick_name, email))

        print " + Register nickname '%s' with nickserv using e-mail '%s'." % (bot.settings.bot_nick_name, email)

    else:

        if sender:

            bot.privmsg(sender_nick, "Nick password is not set. Could not register.")

        print " - No nick password, could not register."

def register_with_chanserv(bot, server, channel, password, description):

    print "Register %s with chanserv using description '%s'." % (channel, description)

    server.privmsg("chanserv", "register %s %s %s" % (channel, password, description))

def make_op(bot, server, channel, user):

    """Ask chanserv to op a user."""

    print "Make %s an operator on %s." % (user, channel)

    server.privmsg("chanserv", "op %s %s" % (channel, user))

def request_access_list(bot, server, channel):

    """Ask chanserv for access list for channel."""

    print "Request access list for '%s'." % channel

    server.privmsg("chanserv", "access %s list" % channel)

def handle_nickserv_notice(bot, server, msg):

    print "Notice from nickserv."

    if msg == "Your nickname isn't registered.":

        print " - Nickname not registered."

        bot.tell_users(server, "Nickname is not registered.", level=100, notice=True)

        register_with_nickserv(bot, server, bot.settings.bot_nick_email)

        return True

    if msg[:17] == "Password accepted":

        print " + Successfully identified with nickserv."

        bot.tell_users(server, "Successfully identified with nickserv.", level=100, notice=True)

        return True

    return False

def handle_chanserv_notice(bot, server, msg):

    """On private notice from chanserv, similar to nickserv."""

    print "Notice from chanserv."

    global waiting_for_pass_for

    global getting_alist_from

    res = re.match('\s([0-9]{1,3})\s{2}(\S+)', msg)

    if res is not None:

        try:

            level = int(res.group(1))

            user = res.group(2)

            users = bot.settings.bot_user_list

            old_level = users.get(user, level)

            level = max(level, old_level)

            users[user] = level

            print " + '%s' has been given the level '%d'." % (user, level)

            return True

        except Exception, e:

            print " - Could not parse string '%s': %s" % (msg, str(e))

            return False

    res = re.match("Channel \\x02(\S*)\\x02 isn't registered.", msg)

    if res is not None:

        channel = res.group(1)

        print " + Channel %s is not registered." % channel

        bot.tell_users(server, "Channel %s is not registered." % channel, level=100, notice=True)

        return True

    res = re.match("Password accepted -- .* \\x02(.*)\\x02.", msg)

    if res is not None:

        channel = res.group(1)

        print " + Is taking control over %s." % channel

        owned_channels.append(channel)

        make_op(bot, server, channel, bot.settings.bot_nick_name)

        bot.tell_users(server, "Is taking control over %s." % channel, level=100, notice=True)

        return True

    res = re.match("Opped \\x02(\S*)\\x02 on channel \\x02(\S*)\\x02.", msg)

    if res is not None:

        user = res.group(1)

        channel = res.group(2)

        if user == bot.settings.bot_nick_name:

            print " + Control over %s established." % channel

        else:

            print " + %s has been opped on %s." % (user, channel)

        return True

    res = re.match("\\x02(\S*)\\x02 is already opped on channel \\x02(\S*)\\x02.", msg)

    if res is not None:

        user = res.group(1)

        channel = res.group(2)

        print " + %s is already opped in %s." % (user, channel)

        return True

    if waiting_for_pass_for:

        res = re.match("Your channel password is \\x02(\S*)\\x02 -- remember it for later use.", msg)

        if res is not None:

            channel = waiting_for_pass_for

            password = res.group(1)

            waiting_for_pass_for = None

            bot.settings.services_channel_passwords[channel] = password

            print " + Got password for %s: %s" % (channel, password)

            identify_with_chanserv(bot, server, channel)

            return True

    res = re.match("Channel \\x02(\S*)\\x02 registered under your nickname: (\S*)", msg)

    if res is not None:

        channel = res.group(1)

        user = res.group(2)

        if user == bot.settings.bot_nick_name:

            print " + Channel '%s' is registered." % channel

        else:

            print " + Channel '%s' is registered for '%s'." % (channel, user)

        bot.tell_users(server, "Registered channel '%s'." % channel)

        waiting_for_pass_for = channel

        return True

    res = re.match("Channel \\x02(\S*)\\x02 is already registerd!", msg)

    if res is not None:

        channel = res.group(1)

        print " - %s is already registered." % channel

        return True

    res = re.match('(\S*) access list is empty.', msg)

    if res is not None:

        channel = res.group(1)

        print " + Got empty access list from '%s'." % channel

        getting_alist_from = None

        return True

    res = re.match('Access list for (\S*):', msg)

    if res is not None:

        channel = res.group(1)

        if channel in owned_channels:

            getting_alist_from = channel

        else:

            print " - Got accesslist from unknown channel '%s'." % channel

        return True

    res = re.match('End of list; ([0-9]+)/([0-9]+) matches shown.', msg)

    if res is not None:

        print " + Got complete access list from '%s'." % str(getting_alist_from)

        getting_alist_from = None

        return True

    return False