mirror of
https://github.com/correl/SleekXMPP.git
synced 2025-03-18 17:00:13 -09:00
* fixed many stanza bugs
* added stanza unhandled (unhandled iqs now reply with feature-not-implemented) * added stanza exceptions (stanzas may now reply with exceptions when their handler raises an exception)
This commit is contained in:
parent
6897a0b57c
commit
07018c0afa
13 changed files with 262 additions and 125 deletions
|
@ -30,6 +30,8 @@ from . stanza.message import Message
|
||||||
from . stanza.iq import Iq
|
from . stanza.iq import Iq
|
||||||
from . stanza.presence import Presence
|
from . stanza.presence import Presence
|
||||||
from . stanza.roster import Roster
|
from . stanza.roster import Roster
|
||||||
|
from . stanza.nick import Nick
|
||||||
|
from . stanza.htmlim import HTMLIM
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
|
@ -38,7 +40,6 @@ def stanzaPlugin(stanza, plugin):
|
||||||
stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
|
stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
|
||||||
stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin
|
stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin
|
||||||
|
|
||||||
stanzaPlugin(Iq, Roster)
|
|
||||||
|
|
||||||
class basexmpp(object):
|
class basexmpp(object):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
@ -61,9 +62,16 @@ class basexmpp(object):
|
||||||
self.registerStanza(Message)
|
self.registerStanza(Message)
|
||||||
self.registerStanza(Iq)
|
self.registerStanza(Iq)
|
||||||
self.registerStanza(Presence)
|
self.registerStanza(Presence)
|
||||||
|
self.stanzaPlugin(Iq, Roster)
|
||||||
|
self.stanzaPlugin(Message, Nick)
|
||||||
|
self.stanzaPlugin(Message, HTMLIM)
|
||||||
|
|
||||||
|
def stanzaPlugin(self, stanza, plugin):
|
||||||
|
stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
|
||||||
|
stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin
|
||||||
|
|
||||||
def Message(self, *args, **kwargs):
|
def Message(self, *args, **kwargs):
|
||||||
return Presence(self, *args, **kwargs)
|
return Message(self, *args, **kwargs)
|
||||||
|
|
||||||
def Iq(self, *args, **kwargs):
|
def Iq(self, *args, **kwargs):
|
||||||
return Iq(self, *args, **kwargs)
|
return Iq(self, *args, **kwargs)
|
||||||
|
@ -191,14 +199,13 @@ class basexmpp(object):
|
||||||
message = self.Message(sto=mto, stype=mtype, sfrom=mfrom)
|
message = self.Message(sto=mto, stype=mtype, sfrom=mfrom)
|
||||||
message['body'] = mbody
|
message['body'] = mbody
|
||||||
message['subject'] = msubject
|
message['subject'] = msubject
|
||||||
message['nick'] = mnick
|
if mnick is not None: message['nick'] = mnick
|
||||||
message['html'] = mhtml
|
if mhtml is not None: message['html'] = mhtml
|
||||||
return message
|
return message
|
||||||
|
|
||||||
def makePresence(self, pshow=None, pstatus=None, ppriority=None, pto=None, ptype=None, pfrom=None):
|
def makePresence(self, pshow=None, pstatus=None, ppriority=None, pto=None, ptype=None, pfrom=None):
|
||||||
presence = self.Presence(stype=ptype, sfrom=pfrom, sto=pto)
|
presence = self.Presence(stype=ptype, sfrom=pfrom, sto=pto)
|
||||||
if pshow is not None:
|
if pshow is not None: presence['type'] = pshow
|
||||||
presence['type'] = pshow
|
|
||||||
if pfrom is None: #maybe this should be done in stanzabase
|
if pfrom is None: #maybe this should be done in stanzabase
|
||||||
presence['from'] = self.fulljid
|
presence['from'] = self.fulljid
|
||||||
presence['priority'] = ppriority
|
presence['priority'] = ppriority
|
||||||
|
@ -237,7 +244,10 @@ class basexmpp(object):
|
||||||
def _handlePresence(self, presence):
|
def _handlePresence(self, presence):
|
||||||
"""Update roster items based on presence"""
|
"""Update roster items based on presence"""
|
||||||
self.event("presence_%s" % presence['type'], presence)
|
self.event("presence_%s" % presence['type'], presence)
|
||||||
if not presence['type'] in ('available', 'unavailable'):
|
if presence['type'] in ('subscribe', 'subscribed', 'unsubscribe', 'unsubscribed'):
|
||||||
|
self.event('changed_subscription', presence)
|
||||||
|
return
|
||||||
|
elif not presence['type'] in ('available', 'unavailable'):
|
||||||
return
|
return
|
||||||
jid = presence['from'].bare
|
jid = presence['from'].bare
|
||||||
resource = presence['from'].resource
|
resource = presence['from'].resource
|
||||||
|
@ -260,7 +270,7 @@ class basexmpp(object):
|
||||||
if wasoffline and show in ('available', 'away', 'xa', 'na', 'ffc'):
|
if wasoffline and show in ('available', 'away', 'xa', 'na', 'ffc'):
|
||||||
self.event("got_online", presence)
|
self.event("got_online", presence)
|
||||||
elif not wasoffline and show == 'unavailable':
|
elif not wasoffline and show == 'unavailable':
|
||||||
self.event("got_offline", eventdata)
|
self.event("got_offline", presence)
|
||||||
if len(self.roster[jid]['presence']) > 1:
|
if len(self.roster[jid]['presence']) > 1:
|
||||||
del self.roster[jid]['presence'][resource]
|
del self.roster[jid]['presence'][resource]
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -64,14 +64,7 @@ class ComponentXMPP(basexmpp, XMLStream):
|
||||||
self.server_port = port
|
self.server_port = port
|
||||||
self.set_jid(jid)
|
self.set_jid(jid)
|
||||||
self.secret = secret
|
self.secret = secret
|
||||||
self.registerHandler(Callback('PresenceProbe', MatchXMLMask("<presence xmlns='%s' type='probe'/>" % self.default_ns), self._handlePresenceProbe))
|
|
||||||
self.registerHandler(Callback('Handshake', MatchXPath('{jabber:component:accept}handshake'), self._handleHandshake))
|
self.registerHandler(Callback('Handshake', MatchXPath('{jabber:component:accept}handshake'), self._handleHandshake))
|
||||||
self.registerHandler(Callback('PresenceSubscription', MatchMany(\
|
|
||||||
(MatchXMLMask("<presence xmlns='%s' type='subscribe'/>" % self.default_ns), \
|
|
||||||
MatchXMLMask("<presence xmlns='%s' type='subscribed'/>" % self.default_ns), \
|
|
||||||
MatchXMLMask("<presence xmlns='%s' type='unsubscribe'/>" % self.default_ns), \
|
|
||||||
MatchXMLMask("<presence xmlns='%s' type='unsubscribed'/>" % self.default_ns) \
|
|
||||||
)), self._handlePresenceSubscription))
|
|
||||||
|
|
||||||
def incoming_filter(self, xmlobj):
|
def incoming_filter(self, xmlobj):
|
||||||
if xmlobj.tag.startswith('{jabber:client}'):
|
if xmlobj.tag.startswith('{jabber:client}'):
|
||||||
|
@ -80,22 +73,6 @@ class ComponentXMPP(basexmpp, XMLStream):
|
||||||
self.incoming_filter(sub)
|
self.incoming_filter(sub)
|
||||||
return xmlobj
|
return xmlobj
|
||||||
|
|
||||||
|
|
||||||
def _handlePresenceProbe(self, stanza):
|
|
||||||
xml = stanza.xml
|
|
||||||
self.event("got_presence_probe", ({
|
|
||||||
'from': xml.attrib['from'],
|
|
||||||
'to': xml.attrib['to']
|
|
||||||
}))
|
|
||||||
|
|
||||||
def _handlePresenceSubscription(self, presence):
|
|
||||||
xml = presence.xml
|
|
||||||
self.event("changed_subscription", {
|
|
||||||
'type' : xml.attrib['type'],
|
|
||||||
'from': xml.attrib['from'],
|
|
||||||
'to': xml.attrib['to']
|
|
||||||
})
|
|
||||||
|
|
||||||
def start_stream_handler(self, xml):
|
def start_stream_handler(self, xml):
|
||||||
sid = xml.get('id', '')
|
sid = xml.get('id', '')
|
||||||
handshake = ET.Element('{jabber:component:accept}handshake')
|
handshake = ET.Element('{jabber:component:accept}handshake')
|
||||||
|
|
|
@ -21,6 +21,98 @@ from __future__ import with_statement
|
||||||
from . import base
|
from . import base
|
||||||
import logging
|
import logging
|
||||||
from xml.etree import cElementTree as ET
|
from xml.etree import cElementTree as ET
|
||||||
|
from .. xmlstream.stanzabase import ElementBase, JID
|
||||||
|
from .. stanza.presence import Presence
|
||||||
|
from .. xmlstream.handler.callback import Callback
|
||||||
|
from .. xmlstream.matcher.xpath import MatchXPath
|
||||||
|
from .. xmlstream.matcher.xmlmask import MatchXMLMask
|
||||||
|
|
||||||
|
class MUCPresence(ElementBase):
|
||||||
|
name = 'x'
|
||||||
|
namespace = 'http://jabber.org/protocol/muc#user'
|
||||||
|
plugin_attrib = 'muc'
|
||||||
|
interfaces = set(('affiliation', 'role', 'jid', 'nick', 'room'))
|
||||||
|
affiliations = set(('', ))
|
||||||
|
roles = set(('', ))
|
||||||
|
|
||||||
|
def getXMLItem(self):
|
||||||
|
item = self.xml.find('{http://jabber.org/protocol/muc#user}item')
|
||||||
|
if item is None:
|
||||||
|
item = ET.Element('{http://jabber.org/protocol/muc#user}item')
|
||||||
|
self.xml.append(item)
|
||||||
|
return item
|
||||||
|
|
||||||
|
def getAffiliation(self):
|
||||||
|
#TODO if no affilation, set it to the default and return default
|
||||||
|
item = self.getXMLItem()
|
||||||
|
return item.get('affiliation', '')
|
||||||
|
|
||||||
|
def setAffiliation(self, value):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
#TODO check for valid affiliation
|
||||||
|
item.attrib['affiliation'] = value
|
||||||
|
return self
|
||||||
|
|
||||||
|
def delAffiliation(self):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
#TODO set default affiliation
|
||||||
|
if 'affiliation' in item.attrib: del item.attrib['affiliation']
|
||||||
|
return self
|
||||||
|
|
||||||
|
def getJid(self):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
return JID(item.get('jid', ''))
|
||||||
|
|
||||||
|
def setJid(self, value):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
if not isinstance(value, str):
|
||||||
|
value = str(value)
|
||||||
|
item.attrib['jid'] = value
|
||||||
|
return self
|
||||||
|
|
||||||
|
def delJid(self):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
if 'jid' in item.attrib: del item.attrib['jid']
|
||||||
|
return self
|
||||||
|
|
||||||
|
def getRole(self):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
#TODO get default role, set default role if none
|
||||||
|
return item.get('role', '')
|
||||||
|
|
||||||
|
def setRole(self, value):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
#TODO check for valid role
|
||||||
|
item.attrib['role'] = value
|
||||||
|
return self
|
||||||
|
|
||||||
|
def delRole(self):
|
||||||
|
item = self.getXMLItem()
|
||||||
|
#TODO set default role
|
||||||
|
if 'role' in item.attrib: del item.attrib['role']
|
||||||
|
return self
|
||||||
|
|
||||||
|
def getNick(self):
|
||||||
|
return self.parent['from'].resource
|
||||||
|
|
||||||
|
def getRoom(self):
|
||||||
|
return self.parent['from'].bare
|
||||||
|
|
||||||
|
def setNick(self, value):
|
||||||
|
logging.warning("Cannot set nick through mucpresence plugin.")
|
||||||
|
return self
|
||||||
|
|
||||||
|
def setRoom(self, value):
|
||||||
|
logging.warning("Cannot set room through mucpresence plugin.")
|
||||||
|
return self
|
||||||
|
|
||||||
|
def delNick(self):
|
||||||
|
logging.warning("Cannot delete nick through mucpresence plugin.")
|
||||||
|
return self
|
||||||
|
|
||||||
|
def delRoom(self):
|
||||||
|
logging.warning("Cannot delete room through mucpresence plugin.")
|
||||||
|
return self
|
||||||
|
|
||||||
class xep_0045(base.base_plugin):
|
class xep_0045(base.base_plugin):
|
||||||
"""
|
"""
|
||||||
|
@ -32,74 +124,40 @@ class xep_0045(base.base_plugin):
|
||||||
self.ourNicks = {}
|
self.ourNicks = {}
|
||||||
self.xep = '0045'
|
self.xep = '0045'
|
||||||
self.description = 'Multi User Chat'
|
self.description = 'Multi User Chat'
|
||||||
self.xmpp.add_handler("<message xmlns='%s' type='groupchat'><body/></message>" % self.xmpp.default_ns, self.handle_groupchat_message)
|
# load MUC support in presence stanzas
|
||||||
self.xmpp.add_handler("<presence xmlns='%s' />" % self.xmpp.default_ns, self.handle_groupchat_presence)
|
self.xmpp.stanzaPlugin(Presence, MUCPresence)
|
||||||
|
self.xmpp.registerHandler(Callback('MUCPresence', MatchXMLMask("<presence xmlns='%s' />" % self.xmpp.default_ns), self.handle_groupchat_presence))
|
||||||
|
self.xmpp.registerHandler(Callback('MUCMessage', MatchXMLMask("<message xmlns='%s' type='groupchat'><body/></message>" % self.xmpp.default_ns), self.handle_groupchat_message))
|
||||||
|
|
||||||
def handle_groupchat_presence(self, xml):
|
def handle_groupchat_presence(self, pr):
|
||||||
""" Handle a presence in a muc.
|
""" Handle a presence in a muc.
|
||||||
"""
|
"""
|
||||||
source = xml.attrib['from']
|
if pr['muc']['room'] not in self.rooms.keys():
|
||||||
room = self.xmpp.getjidbare(source)
|
|
||||||
if room not in self.rooms.keys():
|
|
||||||
return
|
return
|
||||||
nick = self.xmpp.getjidresource(source)
|
entry = pr['muc'].getValues()
|
||||||
entry = {
|
if pr['type'] == 'unavailable':
|
||||||
'nick': nick,
|
self.rooms[entry['room']][entry['nick']] = None
|
||||||
'room': room,
|
|
||||||
}
|
|
||||||
if 'type' in xml.attrib.keys():
|
|
||||||
entry['type'] = xml.attrib['type']
|
|
||||||
else:
|
else:
|
||||||
entry['type'] = 'available'
|
self.rooms[entry['room']][entry['nick']] = entry
|
||||||
for tag in ['status','show','priority']:
|
|
||||||
if xml.find(('{%s}' % self.xmpp.default_ns) + tag) != None:
|
|
||||||
entry[tag] = xml.find(('{%s}' % self.xmpp.default_ns) + tag).text
|
|
||||||
else:
|
|
||||||
entry[tag] = None
|
|
||||||
|
|
||||||
for tag in ['affiliation','role','jid']:
|
|
||||||
item = xml.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item')
|
|
||||||
if item != None:
|
|
||||||
if tag in item.attrib:
|
|
||||||
entry[tag] = item.attrib[tag]
|
|
||||||
else:
|
|
||||||
entry[tag] = None
|
|
||||||
else:
|
|
||||||
entry[tag] = None
|
|
||||||
|
|
||||||
if entry['status'] == 'unavailable':
|
|
||||||
self.rooms[room][nick] = None
|
|
||||||
else:
|
|
||||||
self.rooms[room][nick] = entry
|
|
||||||
logging.debug("MUC presence from %s/%s : %s" % (entry['room'],entry['nick'], entry))
|
logging.debug("MUC presence from %s/%s : %s" % (entry['room'],entry['nick'], entry))
|
||||||
self.xmpp.event("groupchat_presence", entry)
|
self.xmpp.event("groupchat_presence", pr)
|
||||||
|
|
||||||
def handle_groupchat_message(self, xml):
|
def handle_groupchat_message(self, msg):
|
||||||
""" Handle a message event in a muc.
|
""" Handle a message event in a muc.
|
||||||
"""
|
"""
|
||||||
mfrom = xml.attrib['from']
|
self.xmpp.event('groupchat_message', msg)
|
||||||
message = xml.find('{%s}body' % self.xmpp.default_ns).text
|
|
||||||
subject = xml.find('{%s}subject' % self.xmpp.default_ns)
|
|
||||||
if subject:
|
|
||||||
subject = subject.text
|
|
||||||
else:
|
|
||||||
subject = ''
|
|
||||||
resource = self.xmpp.getjidresource(mfrom)
|
|
||||||
mfrom = self.xmpp.getjidbare(mfrom)
|
|
||||||
mtype = xml.attrib.get('type', 'normal')
|
|
||||||
self.xmpp.event("groupchat_message", {'room': mfrom, 'name': resource, 'type': mtype, 'subject': subject, 'message': message})
|
|
||||||
|
|
||||||
def getRoomForm(self, room, ifrom=None):
|
def getRoomForm(self, room, ifrom=None):
|
||||||
iq = self.xmpp.makeIqGet()
|
iq = self.xmpp.makeIqGet()
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
if ifrom is not None:
|
if ifrom is not None:
|
||||||
iq.attrib['from'] = ifrom
|
iq['from'] = ifrom
|
||||||
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
||||||
iq.append(query)
|
iq.append(query)
|
||||||
result = self.xmpp.send(iq, self.xmpp.makeIq(id=iq.get('id')))
|
result = iq.send()
|
||||||
if result.get('type', 'error') == 'error':
|
if result['type'] == 'error':
|
||||||
return False
|
return False
|
||||||
xform = result.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
|
xform = result.xml.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
|
||||||
if xform is None: return False
|
if xform is None: return False
|
||||||
form = self.xmpp.plugin['xep_0004'].buildForm(xform)
|
form = self.xmpp.plugin['xep_0004'].buildForm(xform)
|
||||||
return form
|
return form
|
||||||
|
@ -110,15 +168,16 @@ class xep_0045(base.base_plugin):
|
||||||
#form = self.xmpp.plugin['xep_0004'].makeForm(ftype='submit')
|
#form = self.xmpp.plugin['xep_0004'].makeForm(ftype='submit')
|
||||||
#form.addField('FORM_TYPE', value='http://jabber.org/protocol/muc#roomconfig')
|
#form.addField('FORM_TYPE', value='http://jabber.org/protocol/muc#roomconfig')
|
||||||
iq = self.xmpp.makeIqSet()
|
iq = self.xmpp.makeIqSet()
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
if ifrom is not None:
|
if ifrom is not None:
|
||||||
iq.attrib['from'] = ifrom
|
iq['from'] = ifrom
|
||||||
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
||||||
form = form.getXML('submit')
|
form = form.getXML('submit')
|
||||||
query.append(form)
|
query.append(form)
|
||||||
iq.append(query)
|
iq.append(query)
|
||||||
result = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
|
#result = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
|
||||||
if result.get('type', 'error') == 'error':
|
result = iq.send()
|
||||||
|
if result['type'] == 'error':
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
@ -147,8 +206,8 @@ class xep_0045(base.base_plugin):
|
||||||
def destroy(self, room, reason='', altroom = '', ifrom=None):
|
def destroy(self, room, reason='', altroom = '', ifrom=None):
|
||||||
iq = self.xmpp.makeIqSet()
|
iq = self.xmpp.makeIqSet()
|
||||||
if ifrom is not None:
|
if ifrom is not None:
|
||||||
iq.attrib['from'] = ifrom
|
iq['from'] = ifrom
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
||||||
destroy = ET.Element('destroy')
|
destroy = ET.Element('destroy')
|
||||||
if altroom:
|
if altroom:
|
||||||
|
@ -158,8 +217,9 @@ class xep_0045(base.base_plugin):
|
||||||
destroy.append(xreason)
|
destroy.append(xreason)
|
||||||
query.append(destroy)
|
query.append(destroy)
|
||||||
iq.append(query)
|
iq.append(query)
|
||||||
r = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
|
#r = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
|
||||||
if r is None or r.get('type', 'error') == 'error':
|
r = iq.send()
|
||||||
|
if r is False or r['type'] == 'error':
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
@ -171,17 +231,18 @@ class xep_0045(base.base_plugin):
|
||||||
item = ET.Element('item', {'affiliation':affiliation, 'jid':jid})
|
item = ET.Element('item', {'affiliation':affiliation, 'jid':jid})
|
||||||
query.append(item)
|
query.append(item)
|
||||||
iq = self.xmpp.makeIqSet(query)
|
iq = self.xmpp.makeIqSet(query)
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
result = self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id'))
|
result = iq.send()
|
||||||
if result is None or result.get('type') != 'result':
|
if result is False or result['type'] != 'result':
|
||||||
raise ValueError
|
raise ValueError
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def invite(self, room, jid, reason=''):
|
def invite(self, room, jid, reason=''):
|
||||||
""" Invite a jid to a room."""
|
""" Invite a jid to a room."""
|
||||||
msg = self.xmpp.makeMessage(room, mtype='none')
|
msg = self.xmpp.makeMessage(room)
|
||||||
|
msg['from'] = self.xmpp.jid
|
||||||
x = ET.Element('{http://jabber.org/protocol/muc#user}x')
|
x = ET.Element('{http://jabber.org/protocol/muc#user}x')
|
||||||
invite = ET.Element('invite', {'to': jid})
|
invite = ET.Element('{http://jabber.org/protocol/muc#user}invite', {'to': jid})
|
||||||
if reason:
|
if reason:
|
||||||
rxml = ET.Element('reason')
|
rxml = ET.Element('reason')
|
||||||
rxml.text = reason
|
rxml.text = reason
|
||||||
|
@ -198,11 +259,11 @@ class xep_0045(base.base_plugin):
|
||||||
|
|
||||||
def getRoomConfig(self, room):
|
def getRoomConfig(self, room):
|
||||||
iq = self.xmpp.makeIqGet('http://jabber.org/protocol/muc#owner')
|
iq = self.xmpp.makeIqGet('http://jabber.org/protocol/muc#owner')
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
result = self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id'))
|
result = iq.send()
|
||||||
if result is None or result.get('type') != 'result':
|
if result is None or result['type'] != 'result':
|
||||||
raise ValueError
|
raise ValueError
|
||||||
form = result.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
|
form = result.xml.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
|
||||||
if form is None:
|
if form is None:
|
||||||
raise ValueError
|
raise ValueError
|
||||||
return self.xmpp.plugin['xep_0004'].buildForm(form)
|
return self.xmpp.plugin['xep_0004'].buildForm(form)
|
||||||
|
@ -212,15 +273,15 @@ class xep_0045(base.base_plugin):
|
||||||
x = ET.Element('{jabber:x:data}x', type='cancel')
|
x = ET.Element('{jabber:x:data}x', type='cancel')
|
||||||
query.append(x)
|
query.append(x)
|
||||||
iq = self.xmpp.makeIqSet(query)
|
iq = self.xmpp.makeIqSet(query)
|
||||||
self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id'))
|
iq.send()
|
||||||
|
|
||||||
def setRoomConfig(self, room, config):
|
def setRoomConfig(self, room, config):
|
||||||
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
|
||||||
x = config.getXML('submit')
|
x = config.getXML('submit')
|
||||||
query.append(x)
|
query.append(x)
|
||||||
iq = self.xmpp.makeIqSet(query)
|
iq = self.xmpp.makeIqSet(query)
|
||||||
iq.attrib['to'] = room
|
iq['to'] = room
|
||||||
self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id'))
|
iq.send()
|
||||||
|
|
||||||
def getJoinedRooms(self):
|
def getJoinedRooms(self):
|
||||||
return self.rooms.keys()
|
return self.rooms.keys()
|
||||||
|
|
|
@ -5,6 +5,8 @@ class HTMLIM(ElementBase):
|
||||||
name = 'html'
|
name = 'html'
|
||||||
plugin_attrib = 'html'
|
plugin_attrib = 'html'
|
||||||
interfaces = set(('html'))
|
interfaces = set(('html'))
|
||||||
|
plugin_attrib_map = set()
|
||||||
|
plugin_xml_map = set()
|
||||||
|
|
||||||
def setHtml(self, html):
|
def setHtml(self, html):
|
||||||
if issinstance(html, str):
|
if issinstance(html, str):
|
||||||
|
|
|
@ -5,7 +5,7 @@ from .. xmlstream.handler.waiter import Waiter
|
||||||
from .. xmlstream.matcher.id import MatcherId
|
from .. xmlstream.matcher.id import MatcherId
|
||||||
|
|
||||||
class Iq(StanzaBase):
|
class Iq(StanzaBase):
|
||||||
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject'))
|
interfaces = set(('type', 'to', 'from', 'id','query'))
|
||||||
types = set(('get', 'result', 'set', 'error'))
|
types = set(('get', 'result', 'set', 'error'))
|
||||||
name = 'iq'
|
name = 'iq'
|
||||||
namespace = 'jabber:client'
|
namespace = 'jabber:client'
|
||||||
|
@ -13,7 +13,19 @@ class Iq(StanzaBase):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
StanzaBase.__init__(self, *args, **kwargs)
|
StanzaBase.__init__(self, *args, **kwargs)
|
||||||
if self['id'] == '':
|
if self['id'] == '':
|
||||||
self['id'] = self.stream.getId()
|
self['id'] = self.stream.getNewId()
|
||||||
|
|
||||||
|
def exception(self, text):
|
||||||
|
self.reply()
|
||||||
|
self['error']['condition'] = 'undefined-condition'
|
||||||
|
self['error']['text'] = text
|
||||||
|
self.send()
|
||||||
|
|
||||||
|
def unhandled(self):
|
||||||
|
self.reply()
|
||||||
|
self['error']['condition'] = 'feature-not-implemented'
|
||||||
|
self['error']['text'] = 'No handlers registered for this request.'
|
||||||
|
self.send()
|
||||||
|
|
||||||
def result(self):
|
def result(self):
|
||||||
self['type'] = 'result'
|
self['type'] = 'result'
|
||||||
|
@ -36,6 +48,29 @@ class Iq(StanzaBase):
|
||||||
self.clear()
|
self.clear()
|
||||||
StanzaBase.setPayload(self, value)
|
StanzaBase.setPayload(self, value)
|
||||||
|
|
||||||
|
def setQuery(self, value):
|
||||||
|
query = self.xml.find("{%s}query" % value)
|
||||||
|
if query is None:
|
||||||
|
self.clear()
|
||||||
|
query = ET.Element("{%s}query" % value)
|
||||||
|
self.xml.append(query)
|
||||||
|
return self
|
||||||
|
|
||||||
|
def getQuery(self):
|
||||||
|
for child in self.getchildren():
|
||||||
|
if child.tag.endswith('query'):
|
||||||
|
ns =child.tag.split('}')[0]
|
||||||
|
if '{' in ns:
|
||||||
|
ns = ns[1:]
|
||||||
|
return ns
|
||||||
|
return ''
|
||||||
|
|
||||||
|
def delQuery(self):
|
||||||
|
for child in self.getchildren():
|
||||||
|
if child.tag.endswith('query'):
|
||||||
|
self.xml.remove(child)
|
||||||
|
return self
|
||||||
|
|
||||||
def unhandled(self):
|
def unhandled(self):
|
||||||
pass
|
pass
|
||||||
# returned unhandled error
|
# returned unhandled error
|
||||||
|
|
|
@ -4,7 +4,7 @@ from . error import Error
|
||||||
|
|
||||||
class Message(StanzaBase):
|
class Message(StanzaBase):
|
||||||
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject'))
|
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject'))
|
||||||
types = set((None, 'normal', 'chat', 'headline', 'error'))
|
types = set((None, 'normal', 'chat', 'headline', 'error', 'groupchat'))
|
||||||
sub_interfaces = set(('body', 'subject'))
|
sub_interfaces = set(('body', 'subject'))
|
||||||
name = 'message'
|
name = 'message'
|
||||||
namespace = 'jabber:client'
|
namespace = 'jabber:client'
|
||||||
|
@ -22,9 +22,16 @@ class Message(StanzaBase):
|
||||||
|
|
||||||
def reply(self, body=None):
|
def reply(self, body=None):
|
||||||
StanzaBase.reply(self)
|
StanzaBase.reply(self)
|
||||||
|
del self['id']
|
||||||
if body is not None:
|
if body is not None:
|
||||||
self['body'] = body
|
self['body'] = body
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
def exception(self, text):
|
||||||
|
self.reply()
|
||||||
|
self['error']['condition'] = 'undefined-condition'
|
||||||
|
self['error']['text'] = text
|
||||||
|
self.send()
|
||||||
|
|
||||||
Message.plugin_attrib_map['error'] = Error
|
Message.plugin_attrib_map['error'] = Error
|
||||||
Message.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|
Message.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|
||||||
|
|
|
@ -1,10 +1,12 @@
|
||||||
from .. xmlstream.stanzabase import ElementBase, ET
|
from .. xmlstream.stanzabase import ElementBase, ET
|
||||||
|
|
||||||
class HTMLIM(ElementBase):
|
class Nick(ElementBase):
|
||||||
namespace = 'http://jabber.org/nick/nick'
|
namespace = 'http://jabber.org/nick/nick'
|
||||||
name = 'nick'
|
name = 'nick'
|
||||||
plugin_attrib = 'nick'
|
plugin_attrib = 'nick'
|
||||||
interfaces = set(('nick'))
|
interfaces = set(('nick'))
|
||||||
|
plugin_attrib_map = set()
|
||||||
|
plugin_xml_map = set()
|
||||||
|
|
||||||
def setNick(self, nick):
|
def setNick(self, nick):
|
||||||
self.xml.text = nick
|
self.xml.text = nick
|
||||||
|
|
|
@ -19,6 +19,8 @@ class Presence(StanzaBase):
|
||||||
if value in self.types:
|
if value in self.types:
|
||||||
if show is not None:
|
if show is not None:
|
||||||
self.xml.remove(show)
|
self.xml.remove(show)
|
||||||
|
if value == 'available':
|
||||||
|
value = ''
|
||||||
self._setAttr('type', value)
|
self._setAttr('type', value)
|
||||||
elif value in self.showtypes:
|
elif value in self.showtypes:
|
||||||
if show is None:
|
if show is None:
|
||||||
|
@ -44,8 +46,18 @@ class Presence(StanzaBase):
|
||||||
out = 'available'
|
out = 'available'
|
||||||
return out
|
return out
|
||||||
|
|
||||||
def delType(self):
|
def reply(self):
|
||||||
self.setType('available')
|
if self['type'] == 'unsubscribe':
|
||||||
|
self['type'] = 'unsubscribed'
|
||||||
|
elif self['type'] == 'subscribe':
|
||||||
|
self['type'] = 'subscribed'
|
||||||
|
return StanzaBase.reply(self)
|
||||||
|
|
||||||
|
def exception(self, text):
|
||||||
|
self.reply()
|
||||||
|
self['error']['condition'] = 'undefined-condition'
|
||||||
|
self['error']['text'] = text
|
||||||
|
self.send()
|
||||||
|
|
||||||
Presence.plugin_attrib_map['error'] = Error
|
Presence.plugin_attrib_map['error'] = Error
|
||||||
Presence.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|
Presence.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|
||||||
|
|
|
@ -20,7 +20,7 @@ class Waiter(base.BaseHandler):
|
||||||
return self._payload.get(True, timeout)
|
return self._payload.get(True, timeout)
|
||||||
except queue.Empty:
|
except queue.Empty:
|
||||||
logging.warning("Timed out waiting for %s" % self.name)
|
logging.warning("Timed out waiting for %s" % self.name)
|
||||||
return StanzaBase(stype='error')
|
return False
|
||||||
|
|
||||||
def checkDelete(self):
|
def checkDelete(self):
|
||||||
return True
|
return True
|
||||||
|
|
6
sleekxmpp/xmlstream/matcher/id.py
Normal file
6
sleekxmpp/xmlstream/matcher/id.py
Normal file
|
@ -0,0 +1,6 @@
|
||||||
|
from . import base
|
||||||
|
|
||||||
|
class MatcherId(base.MatcherBase):
|
||||||
|
|
||||||
|
def match(self, xml):
|
||||||
|
return xml.get('id') == self._criteria
|
|
@ -1,4 +1,5 @@
|
||||||
from xml.etree import cElementTree as ET
|
from xml.etree import cElementTree as ET
|
||||||
|
import logging
|
||||||
|
|
||||||
class JID(object):
|
class JID(object):
|
||||||
def __init__(self, jid):
|
def __init__(self, jid):
|
||||||
|
@ -45,7 +46,12 @@ class ElementBase(object):
|
||||||
if self.xml is None:
|
if self.xml is None:
|
||||||
self.xml = xml
|
self.xml = xml
|
||||||
if self.xml is None:
|
if self.xml is None:
|
||||||
self.xml = ET.Element("{%(namespace)s}%(name)s" % {'name': self.name, 'namespace': self.namespace})
|
for ename in self.name.split('/'):
|
||||||
|
new = ET.Element("{%(namespace)s}%(name)s" % {'name': self.name, 'namespace': self.namespace})
|
||||||
|
if self.xml is None:
|
||||||
|
self.xml = new
|
||||||
|
else:
|
||||||
|
self.xml.append(new)
|
||||||
if self.parent is not None:
|
if self.parent is not None:
|
||||||
self.parent.xml.append(self.xml)
|
self.parent.xml.append(self.xml)
|
||||||
return True #had to generate XML
|
return True #had to generate XML
|
||||||
|
@ -70,7 +76,7 @@ class ElementBase(object):
|
||||||
else:
|
else:
|
||||||
return self._getAttr(attrib)
|
return self._getAttr(attrib)
|
||||||
elif attrib in self.plugin_attrib_map:
|
elif attrib in self.plugin_attrib_map:
|
||||||
self.initPlugin(attrib)
|
if attrib not in self.plugins: self.initPlugin(attrib)
|
||||||
return self.plugins[attrib]
|
return self.plugins[attrib]
|
||||||
else:
|
else:
|
||||||
return ''
|
return ''
|
||||||
|
@ -87,9 +93,10 @@ class ElementBase(object):
|
||||||
self._setAttr(attrib, value)
|
self._setAttr(attrib, value)
|
||||||
else:
|
else:
|
||||||
self.__delitem__(attrib)
|
self.__delitem__(attrib)
|
||||||
elif attrib in self.plugin_map:
|
elif attrib in self.plugin_attrib_map:
|
||||||
|
if attrib not in self.plugins: self.initPlugin(attrib)
|
||||||
self.initPlugin(attrib)
|
self.initPlugin(attrib)
|
||||||
self.plugins[attrib].setValues(value)
|
self.plugins[attrib][attrib] = value
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def __delitem__(self, attrib):
|
def __delitem__(self, attrib):
|
||||||
|
@ -101,7 +108,7 @@ class ElementBase(object):
|
||||||
return self._delSub(attrib)
|
return self._delSub(attrib)
|
||||||
else:
|
else:
|
||||||
self._delAttr(attrib)
|
self._delAttr(attrib)
|
||||||
elif attrib in self.plugin_map:
|
elif attrib in self.plugin_attrib_map:
|
||||||
if attrib in self.plugins:
|
if attrib in self.plugins:
|
||||||
del self.plugins[attrib]
|
del self.plugins[attrib]
|
||||||
return self
|
return self
|
||||||
|
@ -114,6 +121,9 @@ class ElementBase(object):
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def _setAttr(self, name, value):
|
def _setAttr(self, name, value):
|
||||||
|
if value is None or value == '':
|
||||||
|
self.__delitem__(name)
|
||||||
|
else:
|
||||||
self.xml.attrib[name] = value
|
self.xml.attrib[name] = value
|
||||||
|
|
||||||
def _delAttr(self, name):
|
def _delAttr(self, name):
|
||||||
|
@ -131,11 +141,13 @@ class ElementBase(object):
|
||||||
return stanza.text
|
return stanza.text
|
||||||
|
|
||||||
def _setSubText(self, name, attrib={}, text=None):
|
def _setSubText(self, name, attrib={}, text=None):
|
||||||
|
if text is None or text == '':
|
||||||
|
return self.__delitem__(name)
|
||||||
stanza = self.xml.find("{%s}%s" % (self.namespace, name))
|
stanza = self.xml.find("{%s}%s" % (self.namespace, name))
|
||||||
if stanza is None:
|
if stanza is None:
|
||||||
self.xml.append(ET.Element("{%s}%s" % (self.namespace, name), attrib))
|
#self.xml.append(ET.Element("{%s}%s" % (self.namespace, name), attrib))
|
||||||
stanza = self.xml.find("{%s}%s" % (self.namespace, name))
|
stanza = ET.Element("{%s}%s" % (self.namespace, name))
|
||||||
if text is not None:
|
self.xml.append(stanza)
|
||||||
stanza.text = text
|
stanza.text = text
|
||||||
return stanza
|
return stanza
|
||||||
|
|
||||||
|
@ -228,6 +240,9 @@ class StanzaBase(ElementBase):
|
||||||
def unhandled(self):
|
def unhandled(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def exception(self, text):
|
||||||
|
logging.error(text)
|
||||||
|
|
||||||
def send(self):
|
def send(self):
|
||||||
self.stream.sendRaw(str(self))
|
self.stream.sendRaw(str(self))
|
||||||
|
|
||||||
|
@ -285,7 +300,7 @@ class StanzaBase(ElementBase):
|
||||||
text[cc] = '>'
|
text[cc] = '>'
|
||||||
elif c == "'":
|
elif c == "'":
|
||||||
text[cc] = '''
|
text[cc] = '''
|
||||||
elif self.escape_quotes:
|
else:
|
||||||
text[cc] = '"'
|
text[cc] = '"'
|
||||||
cc += 1
|
cc += 1
|
||||||
return ''.join(text)
|
return ''.join(text)
|
||||||
|
|
|
@ -262,6 +262,8 @@ class XMLStream(object):
|
||||||
handler.prerun(stanza)
|
handler.prerun(stanza)
|
||||||
self.eventqueue.put(('stanza', handler, stanza))
|
self.eventqueue.put(('stanza', handler, stanza))
|
||||||
if handler.checkDelete(): self.__handlers.pop(self.__handlers.index(handler))
|
if handler.checkDelete(): self.__handlers.pop(self.__handlers.index(handler))
|
||||||
|
else:
|
||||||
|
stanza.unhandled()
|
||||||
|
|
||||||
#loop through handlers and test match
|
#loop through handlers and test match
|
||||||
#spawn threads as necessary, call handlers, sending Stanza
|
#spawn threads as necessary, call handlers, sending Stanza
|
||||||
|
@ -274,10 +276,18 @@ class XMLStream(object):
|
||||||
except queue.Empty:
|
except queue.Empty:
|
||||||
event = None
|
event = None
|
||||||
if event is not None:
|
if event is not None:
|
||||||
etype, handler, stanza = event
|
etype, handler, *args = event
|
||||||
if etype == 'stanza':
|
if etype == 'stanza':
|
||||||
handler.run(stanza)
|
try:
|
||||||
if etype == 'quit':
|
handler.run(args[0])
|
||||||
|
except:
|
||||||
|
args[0].exception(traceback.format_exc())
|
||||||
|
elif etype == 'sched':
|
||||||
|
try:
|
||||||
|
handler.run(*args)
|
||||||
|
except:
|
||||||
|
logging.error(traceback.format_exc())
|
||||||
|
elif etype == 'quit':
|
||||||
logging.debug("Quitting eventRunner thread")
|
logging.debug("Quitting eventRunner thread")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue