1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
import sys
sys.path.append("../3rdParty")
import sleekxmpp.componentxmpp
from xml.etree import cElementTree as ET
class RegistrableComponent :
def __init__(self, jid, password, server, port, backend) :
self.xmpp = sleekxmpp.componentxmpp.ComponentXMPP(jid, password, server, port)
self.xmpp.add_event_handler("session_start", self.handleXMPPConnected)
self.xmpp.add_event_handler("changed_subscription",
self.handleXMPPPresenceSubscription)
self.xmpp.add_event_handler("got_presence_probe",
self.handleXMPPPresenceProbe)
for event in ["message", "got_online", "got_offline", "changed_status"] :
self.xmpp.add_event_handler(event, self.handleIncomingXMPPEvent)
self.backend = backend
self.backend.addMessageHandler(self.handleMessageAddedToBackend)
## BEGIN NEW
self.xmpp.registerPlugin("xep_0030")
self.xmpp.plugin["xep_0030"].add_feature("jabber:iq:register")
self.xmpp.add_handler("<iq type='get' xmlns='jabber:component:accept'>" +
"<query xmlns='jabber:iq:register'/></iq>", self.handleRegistrationFormRequest)
self.xmpp.add_handler("<iq type='set' xmlns='jabber:component:accept'>" +
"<query xmlns='jabber:iq:register'/></iq>", self.handleRegistrationRequest)
## END NEW
## BEGIN NEW
def handleRegistrationFormRequest(self, request) :
payload = ET.Element("{jabber:iq:register}query")
payload.append(ET.Element("username"))
payload.append(ET.Element("password"))
self.sendRegistrationResponse(request, "result", payload)
def handleRegistrationRequest(self, request) :
jid = request.attrib["from"]
user = request.find("{jabber:iq:register}query/{jabber:iq:register}username").text
password = request.find("{jabber:iq:register}query/{jabber:iq:register}password").text
if self.backend.registerXMPPUser(user, password, jid) :
self.sendRegistrationResponse(request, "result")
userJID = self.backend.getJIDForUser(user)
self.xmpp.sendPresenceSubscription(pto=userJID, ptype="subscribe")
else :
error = self.xmpp.makeStanzaError("forbidden", "auth")
self.sendRegistrationResponse(request, "error", error)
def sendRegistrationResponse(self, request, type, payload = None) :
iq = self.xmpp.makeIq(request.get("id"))
iq.attrib["type"] = type
iq.attrib["from"] = self.xmpp.fulljid
iq.attrib["to"] = request.get("from")
if payload :
iq.append(payload)
self.xmpp.send(iq)
## END NEW
## ...
def handleXMPPConnected(self, event) :
for user in self.backend.getAllUsers() :
if self.backend.getUserHasJID(user) :
self.xmpp.sendPresence(pto = self.backend.getJIDForUser(user))
def handleIncomingXMPPEvent(self, event) :
message = event["message"]
user = self.backend.getUserFromJID(event["jid"])
self.backend.addMessageFromUser(message, user)
def handleXMPPPresenceProbe(self, event) :
self.xmpp.sendPresence(pto = event["from"])
def handleXMPPPresenceSubscription(self, subscription) :
if subscription["type"] == "subscribe" :
userJID = subscription["from"]
self.xmpp.sendPresenceSubscription(pto=userJID, ptype="subscribed")
self.xmpp.sendPresence(pto = userJID)
self.xmpp.sendPresenceSubscription(pto=userJID, ptype="subscribe")
def handleMessageAddedToBackend(self, message) :
body = message.user + ": " + message.text
for subscriberJID in self.backend.getSubscriberJIDs(message.user) :
self.xmpp.sendMessage(subscriberJID, body)
def start(self) :
self.xmpp.connect()
self.xmpp.process()
|
