Switch to Sleekxmpp. fixed #1768

1 files changed, 76 insertions, 316 deletions

diff --git a/src/multiuserchat.py b/src/multiuserchat.py
index bd60641e..41b1ebd5 100644
--- a/src/multiuserchat.py
+++ b/src/multiuserchat.py
@@ -1,4 +1,3 @@
-# Copyright 2009, 2010 Erwan Briand
 # Copyright 2010, Florent Le Coz <louizatakk@fedoraproject.org>
 
 # This program is free software: you can redistribute it and/or modify
@@ -13,331 +12,92 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-# Implementation of the XEP-0045: Multi-User Chat.
+"""
+Implementation of the XEP-0045: Multi-User Chat.
+Add some facilities that are not available on the XEP_0045
+sleek plugin
+"""
 
-from xmpp import NS_MUC_ADMIN, NS_MUC
-from xmpp.protocol import Presence, Iq, Message, JID
-import xmpp
-import common
-import threading
-import os
+import sleekxmpp
 
-from time import (altzone, gmtime, localtime, strftime, timezone)
+from xml.etree import cElementTree as ET
 
-from handler import Handler
-from config import config
 
-from common import get_stripped_jid
-from common import is_jid
+from common import debug
 
-class VcardSender(threading.Thread):
+def send_private_message(xmpp, jid, line):
     """
-    avatar sending is really slow (don't know why...)
-    use a thread to send it...
+    Send a private message
     """
-    def __init__(self, connection):
-        threading.Thread.__init__(self)
-        self.connection = connection
-        self.handler = Handler()
+    msg = xmpp.makeMessage(jid)
+    msg['to'] = jid
+    msg['type'] = 'chat'
+    msg['body'] = line
+    msg.send()
 
-    def run(self):
-        self.send_vcard()
-
-    def send_vcard(self):
-        """
-        Method stolen from Gajim (thanks)
-        ## Copyright (C) 2006 Dimitur Kirov <dkirov AT gmail.com>
-        ##                    Junglecow J <junglecow AT gmail.com>
-        ## Copyright (C) 2006-2007 Tomasz Melcer <liori AT exroot.org>
-        ##                         Travis Shirk <travis AT pobox.com>
-        ##                         Nikos Kouremenos <kourem AT gmail.com>
-        ## Copyright (C) 2006-2008 Yann Leboulanger <asterix AT lagaule.org>
-        ## Copyright (C) 2007 Julien Pivotto <roidelapluie AT gmail.com>
-        ## Copyright (C) 2007-2008 Brendan Taylor <whateley AT gmail.com>
-        ##                         Jean-Marie Traissard <jim AT lapin.org>
-        ##                         Stephan Erb <steve-e AT h3c.de>
-        ## Copyright (C) 2008 Jonathan Schleifer <js-gajim AT webkeks.org>
-        (one of these people coded this method, probably)
-        """
-        if not self.connection:
-            return
-        vcard = {
-            "FN":config.get('full_name', ''),
-            "URL":config.get('website', ''),
-            "EMAIL":{
-                "USERID":config.get('email', '')
-                },
-            "DESC":config.get('comment', 'A proud Poezio user')
-            }
-        photo_file_path = config.get('photo', '../data/poezio_80.png')
-        (image, mime_type, sha1) = common.get_base64_from_file(photo_file_path)
-        if image:
-            vcard['PHOTO'] = {"TYPE":mime_type,"BINVAL":image}
-        iq = xmpp.Iq(typ = 'set')
-        iq2 = iq.addChild('vCard', namespace=xmpp.NS_VCARD)
-        for i in vcard:
-            if i == 'jid':
-                continue
-            if isinstance(vcard[i], dict):
-                iq3 = iq2.addChild(i)
-                for j in vcard[i]:
-                    iq3.addChild(j).setData(vcard[i][j])
-            elif isinstance(vcard[i], list):
-                for j in vcard[i]:
-                    iq3 = iq2.addChild(i)
-                    for k in j:
-                        iq3.addChild(k).setData(j[k])
-            else:
-                iq2.addChild(i).setData(vcard[i])
-        self.connection.send(iq)
-        iq = xmpp.Iq(typ = 'set')
-        iq2 = iq.addChild('vCard', namespace=xmpp.NS_VCARD_UPDATE)
-        iq2.addChild('PHOTO').setData(sha1)
-        self.connection.send(iq)
-
-class MultiUserChat(object):
-    def __init__(self, connection):
-        self.connection = connection
-        self.vcard_sender = VcardSender(self.connection)
-
-        self.rooms = []
-        self.rn = {}
-
-        self.own_jid = None
-
-        self.handler = Handler()
-        self.handler.connect('join-room', self.join_room)
-        self.handler.connect('on-connected', self.on_connected)
-        self.handler.connect('send-version', self.send_version)
-        self.handler.connect('send-time', self.send_time)
-
-    def on_connected(self, jid):
-        self.own_jid = jid
-        rooms = config.get('rooms', '')
-        if rooms == '' or type(rooms) != str:
-            return
-        else:
-            rooms = rooms.split(':')
-        for room in rooms:
-            args = room.split('/')
-            if args[0] == '':
-                return
-            roomname = args[0]
-            if len(args) == 2:
-                nick = args[1]
-            else:
-                default = os.environ.get('USER') if os.environ.get('USER') else 'poezio'
-                nick = config.get('default_nick', '')
-                if nick == '':
-                    nick = default
-            self.handler.emit('join-room', room=roomname, nick=nick)
-        if config.get('jid', '') == '': # Don't send the vcard if we're not anonymous
-            self.vcard_sender.start()   # because the user ALREADY has one on the server
-
-    def send_message(self, room, message):
-        mes = Message(to=room)
-        mes.setBody(message)
-        mes.setType('groupchat')
-        self.connection.send(mes)
-
-    def send_private_message(self, user_jid, message):
-        mes = Message(to=user_jid)
-        mes.setBody(message)
-        mes.setType('chat')
-        self.connection.send(mes)
-
-    def request_vcard(self, room_name, nickname):
-        """
-        Request the vCard of an user, over a MUC or not
-        """
-        request = Iq(typ='get', to='%s/%s'% (room_name, nickname))
-        vcard_tag = request.addChild(name='vCard', namespace='vcard-temp')
-        self.connection.send(request)
-
-    def join_room(self, room, nick, password=None):
-        """Join a new room"""
-        pres = Presence(to='%s/%s' % (room, nick))
-        pres.setFrom('%s'%self.own_jid)
-        x_tag = pres.addChild(name='x', namespace=NS_MUC)
-        if password:
-            passwd = x_tag.addChild(name='password')
-            passwd.setData(password)
-        muc_history_length = config.get('muc_history_length', -1)
-        if muc_history_length >= 0:
-            history_tag = x_tag.addChild(name='history')
-            if muc_history_length == 0:
-                history_tag.setAttr('maxchars', 0)
-            else:
-                history_tag.setAttr('maxstanzas', muc_history_length)
-        self.connection.send(pres)
-
-    def quit_room(self, room, nick, msg=None):
-        """Quit a room"""
-        if room is None and nick is None:
-            self.on_disconnect()
-            return
-
-        pres = Presence(to='%s/%s' % (room, nick), typ='unavailable')
-        if msg:
-            pres.setStatus(msg)
-        self.connection.send(pres)
-
-    def disconnect(self, rooms, msg):
-        """
-        """
-        for room in rooms:
-            if room.jid is None and room.joined:
-                pres = Presence(to='%s' % room.name,
-                                typ='unavailable')
-                pres.setStatus(msg)
-                self.connection.send(pres)
-
-    def on_disconnect(self):
-        """Called at disconnection"""
-        for room in self.rooms:
-            pres = Presence(to='%s/%s' % (room, self.rn[room]),
-                            typ='unavailable')
-            self.connection.send(pres)
-
-    def on_iq(self, iq):
-        """Receive a MUC iq notification"""
-        from_ = iq.getFrom().__str__()
-
-        if get_stripped_jid(from_) in self.rooms:
-            children = iq.getChildren()
-            for child in children:
-                if child.getName() == 'error':
-                    code = int(child.getAttr('code'))
-                    msg = None
-
-                    echildren = child.getChildren()
-                    for echild in echildren:
-                        if echild.getName() == 'text':
-                            msg = echild.getData()
-
-                    self.handler.emit('on-muc-error',
-                                      room=from_,
-                                      code=code,
-                                      msg=msg)
-
-    def on_presence(self, presence):
-        """Receive a MUC presence notification"""
-        from_ = presence.getFrom().__str__()
-        if get_stripped_jid(from_) in self.rooms:
-            self.handler.emit('on-muc-presence-changed',
-                               jid=from_.encode('utf-8'),
-                               priority=presence.getPriority(),
-                               show=presence.getShow(),
-                               status=presence.getStatus(),
-                               stanza=presence
-                              )
-
-    def on_message(self, message):
-        """Receive a MUC message notification"""
-        from_ = message.getFrom().__str__().encode('utf-8')
-
-        if get_stripped_jid(from_) in self.rooms:
-            body_ = message.getBody()
-            type_ = message.getType()
-            subj_ = message.getSubject()
-            self.handler.emit('on-muc-message-received',
-                              jid=from_, msg=body_, subject=subj_,
-                              typ=type_, stanza=message)
-
-    def eject_user(self, room, action, nick, reason):
-        """Eject an user from a room"""
-        iq = Iq(typ='set', to=room)
-        query = iq.addChild('query', namespace=NS_MUC_ADMIN)
-        item = query.addChild('item')
-
-        if action == 'kick':
-            item.setAttr('role', 'none')
-            if is_jid(nick):
-                item.setAttr('jid', nick)
-            else:
-                item.setAttr('nick', nick)
-        elif action == 'ban':
-            item.setAttr('affiliation', 'outcast')
-            item.setAttr('jid', nick)
-
-        if reason is not None:
-            rson = item.addChild('reason')
-            rson.setData(reason)
-
-        self.connection.send(iq)
-
-    def change_role(self, room, nick, role):
-        """Change the role of an user"""
-        iq = Iq(typ='set', to=room)
-        query = iq.addChild('query', namespace=NS_MUC_ADMIN)
-        item = query.addChild('item')
-        item.setAttr('nick', nick)
-        item.setAttr('role', role)
-
-        self.connection.send(iq)
-
-    def change_aff(self, room, jid, aff):
-        """Change the affiliation of an user"""
-        iq = Iq(typ='set', to=room)
-        query = iq.addChild('query', namespace=NS_MUC_ADMIN)
-        item = query.addChild('item')
-        item.setAttr('jid', jid)
-        item.setAttr('affiliation', aff)
-
-        self.connection.send(iq)
-
-    def change_subject(self, room, subject):
-        """Change the subject of a room"""
-        message = Message(typ='groupchat', to=room)
-        subj = message.addChild('subject')
-        subj.setData(subject)
+def send_groupchat_message(xmpp, jid, line):
+    """
+    Send a message to the groupchat
+    """
+    msg = xmpp.makeMessage(jid)
+    msg['type'] = 'groupchat'
+    msg['body'] = line
+    msg.send()
 
-        self.connection.send(message)
+def change_show(xmpp, jid, own_nick, show, status):
+    """
+    Change our 'Show'
+    """
+    pres = xmpp.makePresence(pto='%s/%s' % (jid, own_nick),
+                             pfrom=xmpp.fulljid)
+    if show: # if show is None, don't put a <show /> tag. It means "online"
+        pres['type'] = show
+    if status:
+        pres['status'] = status
+    debug('Change presence: %s\n' % (pres))
+    pres.send()
+
+def change_subject(xmpp, jid, subject):
+    """
+    Change the room subject
+    """
+    msg = xmpp.makeMessage(jid)
+    msg['type'] = 'groupchat'
+    msg['subject'] = subject
+    msg['from'] = xmpp.jid
+    msg.send()
 
-    def change_nick(self, room, nick):
-        """Change the nickname"""
-        pres = Presence(to='%s/%s' % (room, nick))
-        self.connection.send(pres)
+def change_nick(xmpp, jid, nick):
+    """
+    Change our own nick in a room
+    """
+    xmpp.makePresence(pto='%s/%s' % (jid, nick),
+                            pfrom=xmpp.jid).send()
 
-    def change_show(self, room, nick, show, status):
-        pres = Presence(to='%s/%s' % (room, nick))
-        if show: # if show is None, don't put a <show /> tag. It means "online"
-            pres.setShow(show)
-        if status:
-            pres.setStatus(status)
-        self.connection.send(pres)
+def join_groupchat(xmpp, jid, nick, password=None):
+    """
+    Join the groupchat
+    """
+    xmpp.plugin['xep_0045'].joinMUC(jid, nick, password)
 
-    def send_version(self, iq_obj):
-        """
-        from gajim and modified
-        """
-        iq_obj = iq_obj.buildReply('result')
-        qp = iq_obj.getTag('query')
-        if config.get('send_poezio_info', 'true') == 'true':
-            qp.setTagData('name', 'Poezio')
-            qp.setTagData('version', '0.6.3 dev')
-        else:
-            qp.setTagData('name', 'Unknown')
-            qp.setTagData('version', 'Unknown')
-        if config.get('send_os_info', 'true') == 'true':
-            qp.setTagData('os', common.get_os_info())
-        else:
-            qp.setTagData('os', 'Unknown')
-        self.connection.send(iq_obj)
-        raise xmpp.protocol.NodeProcessed
+def leave_groupchat(xmpp, jid, own_nick, msg):
+    """
+    Leave the groupchat
+    """
+    xmpp.plugin['xep_0045'].leaveMUC(jid, own_nick, msg)
 
-    def send_time(self, iq_obj):
-        """
-        from gajim
-        """
-        iq_obj = iq_obj.buildReply('result')
-        qp = iq_obj.setTag('time',
-                           namespace="urn:xmpp:time")
-        if config.get('send_time', 'true') == 'true':
-            qp.setTagData('utc', strftime('%Y-%m-%dT%H:%M:%SZ', gmtime()))
-            isdst = localtime().tm_isdst
-            zone = -(timezone, altzone)[isdst] / 60
-            tzo = (zone / 60, abs(zone % 60))
-            qp.setTagData('tzo', '%+03d:%02d' % (tzo))
-            self.connection.send(iq_obj)
-            raise xmpp.protocol.NodeProcessed
+def eject_user(xmpp, jid, nick, reason):
+    """
+    (try to) Eject an user from the room
+    """
+    iq = xmpp.makeIqSet()
+    query = ET.Element('{http://jabber.org/protocol/muc#admin}query')
+    item = ET.Element('{http://jabber.org/protocol/muc#admin}item', {'nick':nick, 'role':'none'})
+    if reason:
+        reason_el = ET.Element('{http://jabber.org/protocol/muc#admin}reason')
+        reason_el.text = reason
+        item.append(reason_el)
+    query.append(item)
+    iq.append(query)
+    iq['to'] = jid
+    return iq.send()