bot

import os
import random

import zope.interface, logging
from twisted.internet import reactor

from Houdini.Plugins import Plugin
from Houdini.Handlers import Handlers

class Bot(object):
    zope.interface.implements(Plugin)

    author = "Houdini Team"
    version = 0.1
    description = "A bot plugin"

    id = 0
    name = None
    x, y = (0, 0)
    frame = 1
    membershipDays = 750
    nameglow = ''
    namecolor = ''
    bubblecolor = ''
    bubbletext = ''
    speed = 4
    mood = ''
    bubbleglow = ''
    moodglow = ''
    moodcolor = ''
    walls = 1
    size = 100

    # Set the attribute to None for random clothing
    clothing = None

    # If you set this to False, the bot will not automatically spawn in the room.
    # Instead, he will show up and disappear whenever he's needed.
    isStationary = True

    botString = None

    namesFile = "names.txt" # Path relative to this plugin
    namesFileUrl = "https://www.cs.cmu.edu/Groups/AI/areas/nlp/corpora/names/other/names.txt"
    namesList = None

    def __init__(self, server):
        self.logger = logging.getLogger("Houdini")

        self.server = server

        self.namesFile = os.path.join(os.path.dirname(os.path.abspath(__file__)), self.namesFile)

        if self.name is None: # Ooh~
            self.randomizeName()

        allItems = self.server.items.schemaObjects.keys()

        """
        This is intentionally placed outside of the if statement to allow randomization (via the command)
        even if the bot wasn't initially created randomly.
        """
        # TODO: Exclude bait items probably (LAZY)
        self.headIds = [itemId for itemId in allItems if self.server.items.isItemHead(itemId)]
        self.faceIds = [itemId for itemId in allItems if self.server.items.isItemFace(itemId)]
        self.neckIds = [itemId for itemId in allItems if self.server.items.isItemNeck(itemId)]
        self.bodyIds = [itemId for itemId in allItems if self.server.items.isItemBody(itemId)]
        self.handIds = [itemId for itemId in allItems if self.server.items.isItemHand(itemId)]
        self.feetIds = [itemId for itemId in allItems if self.server.items.isItemFeet(itemId)]
        self.flagIds = [itemId for itemId in allItems if self.server.items.isItemPin(itemId)]
        self.photoIds = [itemId for itemId in allItems if self.server.items.isItemPhoto(itemId)]

        if self.clothing is None: # Lewd~! o//o
            self.randomizeClothing()

        self.updateString()

        Handlers.GetInventory += self.handleJoinRoom # For the initial room join
        Handlers.JoinRoom += self.handleJoinRoom
        Handlers.JoinPlayerIgloo += self.handleJoinRoom

    def addToRoom(self, player):
        player.sendXt("ap", self.botString)

    def removeFromRoom(self, player):
        player.sendXt("rp", self.id)

    # TODO: Don't join game rooms (?)
    def handleJoinRoom(self, player, data):
        if self.isStationary:
            self.addToRoom(player)

    def sendMessage(self, player, message):
        if not self.isStationary:
            reactor.callFromThread(self.addToRoom, player)

        reactor.callFromThread(player.sendXt, "sm", self.id, message)

        if not self.isStationary:
            # May need to proxy this call through callFromThread
            reactor.callLater(2, self.removeFromRoom, player)

    def ready(self):
        self.logger.info("Bot plugin has been loaded")

    def randomizeName(self):
        if not os.path.isfile(self.namesFile):
            self.downloadNames()

        if self.namesList is None:
            with open(self.namesFile, "r") as namesFile:
                self.namesList = namesFile.readlines()

        self.name = random.choice(self.namesList).strip()

    def randomizeClothing(self):
        self.clothing = {
            "Color": random.randrange(1, 14),
            "Head": random.choice(self.headIds),
            "Face": random.choice(self.faceIds),
            "Neck": random.choice(self.neckIds),
            "Body": random.choice(self.bodyIds),
            "Hand": random.choice(self.handIds),
            "Feet": random.choice(self.feetIds),
            "Flag": random.choice(self.flagIds),
            "Photo": random.choice(self.photoIds)
        }

    def updateString(self):
        self.botString = ("{bot.id}|"
                          "{bot.name}|"
                          "1|"
                          "{Color}|"
                          "{Head}|"
                          "{Face}|"
                          "{Neck}|"
                          "{Body}|"
                          "{Hand}|"
                          "{Feet}|"
                          "{Flag}|"
                          "{Photo}|"
                          "{bot.x}|"
                          "{bot.y}|"
                          "{bot.frame}|"
                          "1|"
                          "{bot.membershipDays}|"
                          "{bot.nameglow}|"
                          "{bot.namecolor}|"
                          "{bot.bubblecolor}|"
                          "{bot.bubbletext}|"
                          "{bot.speed}|"
                          "{bot.mood}|"
                          "{bot.bubbleglow}|"
                          "{bot.moodglow}|"
                          "{bot.moodcolor}|"
                          "{bot.walls}|"
                          "{bot.size}".format(bot=self, **self.clothing))

    def downloadNames(self):
        import urllib2
        namesResponse = urllib2.urlopen(self.namesFileUrl)

        with open(self.namesFile, "w") as namesFile:
            namesFile.write(namesResponse.read())

        namesResponse.close()

        self.logger.info("[Bot] Names file has been downloaded.")