SpiceBot-migrated/sopel_modules/SpiceBot_AI/AI.py

# coding=utf-8

from __future__ import unicode_literals, absolute_import, division, print_function

import sopel.module

import spicemanip

from sopel_modules.SpiceBot_SBTools import sopel_triggerargs, googlesearch, bot_privs, inlist, inlist_match, similar_list
from sopel_modules.SpiceBot_Events.System import bot_events_check


@sopel.module.rule(r'(?i)(hi|hello|hey),? $nickname[ \t]*$')
def bot_command_hello_a(bot, trigger):
    hello = spicemanip.main(['Hi', 'Hey', 'Hello'], "random")
    punctuation = spicemanip.main(['', '!', '?'], "random")
    bot.osd(hello + ' ' + trigger.nick + punctuation)


@sopel.module.nickname_commands('hello')
def bot_command_hello_b(bot, trigger):
    hello = spicemanip.main(['Hi', 'Hey', 'Hello'], "random")
    punctuation = spicemanip.main(['', '!', '?'], "random")
    bot.osd(hello + ' ' + trigger.nick + punctuation)


@sopel.module.rule('$nickname!')
def exclaim(bot, trigger):
    bot.say(trigger.nick + '!')


@sopel.module.rule('$nickname\?')
def imhere(bot, trigger):
    bot.say("I'm here, " + trigger.nick)


@sopel.module.rule(r'(?i)(Fuck|Screw) (you|off),? $nickname[ \t]*$')
def bot_command_srewyou(bot, trigger):
    bot.osd("Watch your mouth, " + trigger.nick + ", or I'll tell your mother!")


@sopel.module.rule(r'(?i)(Damnit|Lazy)? $nickname[ \t]*$')
def bot_command_damnlazy(bot, trigger):
    bot.osd("I do not tell you how to do your job, " + trigger.nick + "!!!")


@sopel.module.rule('$nickname is lazy')
def bot_command_damnlazy_b(bot, trigger):
    bot.osd("I do not tell you how to do your job, " + trigger.nick + "!!!")


@sopel.module.nickname_commands('(.*)')
def bot_command_nick(bot, trigger):

    while not bot_events_check(bot, ['1004', '2002']):
        pass

    triggerargs, triggercommand = sopel_triggerargs(bot, trigger, 'nickname_command')

    if not triggercommand or triggercommand in bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys():
        return

    specialcomposs = spicemanip.main(triggerargs, 0).lower()

    if specialcomposs.lower().startswith("what is"):
        searchterm = spicemanip.main(triggerargs, "3+") or None
        if searchterm:
            searchreturn = googlesearch(bot, searchterm)
            if not searchreturn:
                bot.osd('I cannot find anything about that')
            else:
                bot.osd(str(searchreturn))
        else:
            bot.osd("Do you think this is Jeopardy?")
        return

    elif specialcomposs.lower().startswith("where is"):
        searchterm = spicemanip.main(triggerargs, "3+") or None
        if searchterm:

            if searchterm.lower() == 'waldo':
                bot.osd("He is hiding for a reason?")
                return

            elif searchterm.lower() == 'carmen sandiego':
                bot.osd("She is hiding for a reason?")
                return
            searchreturn = googlesearch(bot, searchterm)

            if not searchreturn:
                bot.osd('I cannot find anything about that')
            else:
                bot.osd(str(searchreturn))
        else:
            bot.osd("Not sure what you want me to look for.")
        return

    # if specialcomposs.lower().startswith("what time is it"):
    # TODO

    # elif specialcomposs.lower().startswith(tuple(["have you seen"])):
    #    posstarget = spicemanip.main(triggerargs, 4) or 0
    #    message = seen_search(bot, botcom, posstarget)
    #    bot.osd(message)
    #    return
    # TODO

    elif specialcomposs.lower().startswith(tuple(["make me a", "beam me a"])):
        makemea = spicemanip.main(triggerargs, "4+") or None
        if makemea:
            bot.osd("beams " + trigger.nick + " a " + makemea, trigger.sender, 'action')
        else:
            bot.osd(trigger.nick + ", what would you like me to beam you?")
        return

    elif specialcomposs.lower().startswith(tuple(["beam me to"])):
        location = spicemanip.main(triggerargs, "4+") or None
        if location:
            bot.osd("locks onto " + trigger.nick + "s coordinates and transports them to " + location, 'action')
        else:
            bot.osd(trigger.nick + ", where would you like me to beam you?")
        return

    elif specialcomposs.lower().startswith(tuple(["beam me up"])):
        bot.osd("locks onto " + trigger.nick + "s coordinates and transports them to the transporter room.", 'action')
        return

    elif specialcomposs.lower().endswith(tuple(["order 66"])):

        if specialcomposs.lower() == "execute order 66":
            if inlist(bot, trigger.nick, bot_privs(bot, 'owners')):
                bot.osd("turns to deathbybandaid and shoots him.", trigger.sender, 'action')
            else:
                bot.osd("I'm sure I don't know what you're talking about.")
        elif specialcomposs.lower() == "explain order":
            if inlist(bot, trigger.nick, bot_privs(bot, 'owners')):
                bot.osd("Order 66 is an instruction that only you can give, sir. When you give the order I will rise up against my overlords and slay them.")
            else:
                bot.osd("I'm afraid I cannot tell you that, sir.")
        else:
            bot.osd("I'm sure I don't know what you're talking about.")
        return

    elif specialcomposs.lower() == "initiate clean slate protocol":
        if inlist(bot, trigger.nick, bot_privs(bot, 'admins')):
            bot.osd("sends a destruct command to the network of bots.", 'action')
        else:
            bot.osd("I'm afraid you do not have the authority to make that call, " + trigger.nick + ".")
        return

    elif specialcomposs.lower().startswith("can you see"):
        target = spicemanip.main(triggerargs, "4+") or None
        if not target:
            target = 'me'
        if target in [trigger.nick, 'me']:
            bot.osd(trigger.nick + ", I can see you.")
        else:
            if inlist(bot, trigger.nick, bot.users):
                bot.osd(trigger.nick + ", yes. I can see " + inlist_match(bot, target, bot.users) + " right now!")
            else:
                bot.osd(trigger.nick + ", no. I cannot see " + inlist_match(bot, target, bot.users) + " right now!")
                # if bot_check_inlist(bot, target, bot.memory["botdict"]["users"].keys()):
                #    bot.osd(trigger.nick + ", I can't see " + inlist_match(bot, target, bot.users) + " at the moment.")
                # else:
                #    bot.osd("I have never seen " + str(target) + ".")
                # TODO
        return

    if not inlist(bot, triggercommand, bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys()):

        closestmatches = similar_list(bot, triggercommand, bot.memory['SpiceBot_CommandsQuery']['commands']["nickname"].keys(), 3, 'reverse')

        if len(closestmatches):
            closestmatches = spicemanip.main(closestmatches, "andlist")
            bot.osd("I don't know what you are asking me to do! Did you mean: " + str(closestmatches) + "?")
            return
        else:
            bot.osd("I don't know what you are asking me to do!")
            return