root/adventure/movable.py

# -*- coding: utf-8 -*-

#  MUDMUC
#  movable.py
#  Copyright (c) 2008 Thibaut Girka, Anaël Verrier

#  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; version 3 only.

#  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, write to the Free Software
#  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

from os.path import isfile, isabs, join as path_join
from base64 import encodestring
from mimetypes import guess_type as guess_mimetype
from sha import sha

from xmpp import Message, Presence, NS_MUC_USER, NS_VCARD, Node

class Movable(object):
    def __init__(self, world, name, jid):
        self.world = world
        self.name = name
        self.jid = unicode(jid)
        self.place = None
        self.presence = None
        self.image_path = None
        
        self.vcard = None
    
    def set_vcard(self, image=None, birthday=None):
        vcard = Node(NS_VCARD + ' vCard')
        vcard.addChild(node=Node('NICKNAME', {}, self.name))
        if image:
            if not isabs(image):
                image_path = path_join(self.world.data_dir, image)
            if isfile(image_path):
                self.image_path = image
                
                file = open(image_path, 'rb')
                data = file.read()
                encoded_data = encodestring(data)
                mime = guess_mimetype(image)[0]
                file.close()
                photo = vcard.setTag('PHOTO')
                photo.addChild(node=Node('TYPE', {}, mime))
                photo.addChild(node=Node('BINVAL', {}, encoded_data))
                
                vcard_pres = self.presence.getTag('x', namespace=NS_VCARD + ':x:update')
                if not vcard_pres:
                    vcard_pres = self.presence.setTag('x', namespace=NS_VCARD + ':x:update')
                vcard_pres.addChild(node=Node('photo', {},
                                              sha(data).hexdigest()))
            
        if birthday:
            vcard.addChild(node=Node('BDAY', {}, birthday))
        #TODO: DESC
        self.vcard = vcard
    
    def say(self, text, target=None):
        """ Make the player say something. If target is None, broadcast """
        if target is None:
            self.place.broadcast_message(self.name, text)
        else:
            target.send_message(self.name, text)
    
    def move_to(self, to_, warning=True):
        """ Move the player to its new place """
        oldplace = self.place
        self.place = to_
        if self.place:
            self.place.player_entered(self, oldplace)
        elif warning:
            self.send_message(None,
                              'Sorry! The room isn\'t available!\n'
                              'It\'s a bug for sure '
                              '(either in the data '
                              'either in the engine) :/')
        if oldplace is not None:
            oldplace.player_leaved(self)
                              
    def fix_presence(self, role=None, affiliation=None):
        """
        Fix presence: Remove double presence tag and set
        the specified role and affiliation
        """
        pres = Presence(node=self.presence)
        
        tag_x = pres.getTag('x', namespace=NS_MUC_USER)
        if not tag_x:
            tag_x = pres.setTag('x', namespace=NS_MUC_USER)
        
        tag_item = tag_x.getTag('item', namespace=NS_MUC_USER)
        if not tag_item:
            tag_item = tag_x.setTag('item', namespace=NS_MUC_USER)
        
        tag_item.setAttr('affiliation', affiliation)
        tag_item.setAttr('role', role)
        
        return pres