annotate sat @ 1:a06a151fc31f

Disconnect first draft
author Goffi <goffi@goffi.org>
date Sun, 18 Oct 2009 23:20:45 +0200
parents c4bc297b82f0
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
goffi@necton2
parents:
diff changeset
1 #!/usr/bin/python
goffi@necton2
parents:
diff changeset
2 # -*- coding: utf-8 -*-
goffi@necton2
parents:
diff changeset
3
goffi@necton2
parents:
diff changeset
4 """
goffi@necton2
parents:
diff changeset
5 SAT: a jabber client
goffi@necton2
parents:
diff changeset
6 Copyright (C) 2009 Jérôme Poisson (goffi@goffi.org)
goffi@necton2
parents:
diff changeset
7
goffi@necton2
parents:
diff changeset
8 This program is free software: you can redistribute it and/or modify
goffi@necton2
parents:
diff changeset
9 it under the terms of the GNU General Public License as published by
goffi@necton2
parents:
diff changeset
10 the Free Software Foundation, either version 3 of the License, or
goffi@necton2
parents:
diff changeset
11 (at your option) any later version.
goffi@necton2
parents:
diff changeset
12
goffi@necton2
parents:
diff changeset
13 This program is distributed in the hope that it will be useful,
goffi@necton2
parents:
diff changeset
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
goffi@necton2
parents:
diff changeset
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
goffi@necton2
parents:
diff changeset
16 GNU General Public License for more details.
goffi@necton2
parents:
diff changeset
17
goffi@necton2
parents:
diff changeset
18 You should have received a copy of the GNU General Public License
goffi@necton2
parents:
diff changeset
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
goffi@necton2
parents:
diff changeset
20 """
goffi@necton2
parents:
diff changeset
21
goffi@necton2
parents:
diff changeset
22
goffi@necton2
parents:
diff changeset
23 from twisted.internet import glib2reactor, protocol
goffi@necton2
parents:
diff changeset
24 glib2reactor.install()
goffi@necton2
parents:
diff changeset
25
goffi@necton2
parents:
diff changeset
26 from twisted.words.protocols.jabber import client, jid, xmlstream, error
goffi@necton2
parents:
diff changeset
27 from twisted.words.xish import domish
goffi@necton2
parents:
diff changeset
28
goffi@necton2
parents:
diff changeset
29 from twisted.internet import reactor
goffi@necton2
parents:
diff changeset
30 import pdb
goffi@necton2
parents:
diff changeset
31
goffi@necton2
parents:
diff changeset
32 from sat_bridge.DBus import DBusBridge
goffi@necton2
parents:
diff changeset
33 import logging
goffi@necton2
parents:
diff changeset
34 from logging import debug, info, error
goffi@necton2
parents:
diff changeset
35
goffi@necton2
parents:
diff changeset
36 import signal, sys
goffi@necton2
parents:
diff changeset
37 import os.path
goffi@necton2
parents:
diff changeset
38
goffi@necton2
parents:
diff changeset
39 from tools.memory import Memory
goffi@necton2
parents:
diff changeset
40 from glob import glob
goffi@necton2
parents:
diff changeset
41
goffi@necton2
parents:
diff changeset
42
goffi@necton2
parents:
diff changeset
43 ### logging configuration FIXME: put this elsewhere ###
goffi@necton2
parents:
diff changeset
44 logging.basicConfig(level=logging.DEBUG,
goffi@necton2
parents:
diff changeset
45 format='%(message)s')
goffi@necton2
parents:
diff changeset
46 ###
goffi@necton2
parents:
diff changeset
47
goffi@necton2
parents:
diff changeset
48
goffi@necton2
parents:
diff changeset
49
goffi@necton2
parents:
diff changeset
50
goffi@necton2
parents:
diff changeset
51 class SAT:
goffi@necton2
parents:
diff changeset
52
goffi@necton2
parents:
diff changeset
53 def __init__(self):
goffi@necton2
parents:
diff changeset
54 self.reactor=reactor
goffi@necton2
parents:
diff changeset
55 self.memory=Memory()
goffi@necton2
parents:
diff changeset
56 self.server_features=[] #XXX: temp dic, need to be transfered into self.memory in the future
goffi@necton2
parents:
diff changeset
57 self.connected=False #FIXME: use twisted var instead
goffi@necton2
parents:
diff changeset
58
goffi@necton2
parents:
diff changeset
59 self._iq_cb_map = {} #callback called when ns is found on IQ
goffi@necton2
parents:
diff changeset
60 self._waiting_conf = {} #callback called when a confirmation is received
goffi@necton2
parents:
diff changeset
61 self._progress_cb_map = {} #callback called when a progress is requested (key = progress id)
goffi@necton2
parents:
diff changeset
62 self.plugins = {}
goffi@necton2
parents:
diff changeset
63
goffi@necton2
parents:
diff changeset
64 self.bridge=DBusBridge()
goffi@necton2
parents:
diff changeset
65 self.bridge.register("connect", self.connect)
1
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
66 self.bridge.register("disconnect", self.disconnect)
0
goffi@necton2
parents:
diff changeset
67 self.bridge.register("getContacts", self.memory.getContacts)
goffi@necton2
parents:
diff changeset
68 self.bridge.register("getPresenceStatus", self.memory.getPresenceStatus)
goffi@necton2
parents:
diff changeset
69 self.bridge.register("sendMessage", self.sendMessage)
goffi@necton2
parents:
diff changeset
70 self.bridge.register("setParam", self.setParam)
goffi@necton2
parents:
diff changeset
71 self.bridge.register("getParam", self.memory.getParam)
goffi@necton2
parents:
diff changeset
72 self.bridge.register("getParams", self.memory.getParams)
goffi@necton2
parents:
diff changeset
73 self.bridge.register("getParamsCategories", self.memory.getParamsCategories)
goffi@necton2
parents:
diff changeset
74 self.bridge.register("getHistory", self.memory.getHistory)
goffi@necton2
parents:
diff changeset
75 self.bridge.register("setPresence", self.setPresence)
goffi@necton2
parents:
diff changeset
76 self.bridge.register("addContact", self.addContact)
goffi@necton2
parents:
diff changeset
77 self.bridge.register("delContact", self.delContact)
goffi@necton2
parents:
diff changeset
78 self.bridge.register("isConnected", self.isConnected)
goffi@necton2
parents:
diff changeset
79 self.bridge.register("confirmationAnswer", self.confirmationAnswer)
goffi@necton2
parents:
diff changeset
80 self.bridge.register("getProgress", self.getProgress)
goffi@necton2
parents:
diff changeset
81
goffi@necton2
parents:
diff changeset
82 self._import_plugins()
goffi@necton2
parents:
diff changeset
83 self.connect()
goffi@necton2
parents:
diff changeset
84
goffi@necton2
parents:
diff changeset
85
goffi@necton2
parents:
diff changeset
86 def _import_plugins(self):
goffi@necton2
parents:
diff changeset
87 """Import all plugins found in plugins directory"""
goffi@necton2
parents:
diff changeset
88 #TODO: manage dependencies
goffi@necton2
parents:
diff changeset
89 plug_lst = [os.path.splitext(plugin)[0] for plugin in map(os.path.basename,glob ("plugins/plugin*.py"))]
goffi@necton2
parents:
diff changeset
90
goffi@necton2
parents:
diff changeset
91 for plug in plug_lst:
goffi@necton2
parents:
diff changeset
92 plug_path = 'plugins.'+plug
goffi@necton2
parents:
diff changeset
93 __import__(plug_path)
goffi@necton2
parents:
diff changeset
94 mod = sys.modules[plug_path]
goffi@necton2
parents:
diff changeset
95 plug_info = mod.PLUGIN_INFO
goffi@necton2
parents:
diff changeset
96 info ("importing plugin: %s", plug_info['name'])
goffi@necton2
parents:
diff changeset
97 self.plugins[plug_info['import_name']] = getattr(mod, plug_info['main'])(self)
goffi@necton2
parents:
diff changeset
98
goffi@necton2
parents:
diff changeset
99 def connect(self):
1
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
100 print "Connect !"
0
goffi@necton2
parents:
diff changeset
101 if (self.connected):
goffi@necton2
parents:
diff changeset
102 info("already connected !")
goffi@necton2
parents:
diff changeset
103 return
goffi@necton2
parents:
diff changeset
104 info("Connecting...")
goffi@necton2
parents:
diff changeset
105 self.me = jid.JID(self.memory.getParamV("JabberID", "Connection"))
goffi@necton2
parents:
diff changeset
106 self.factory = client.XMPPClientFactory(self.me, self.memory.getParamV("Password", "Connection"))
goffi@necton2
parents:
diff changeset
107 self.factory.addBootstrap(xmlstream.STREAM_AUTHD_EVENT,self.authd)
goffi@necton2
parents:
diff changeset
108 self.factory.addBootstrap(xmlstream.INIT_FAILED_EVENT,self.failed)
goffi@necton2
parents:
diff changeset
109 reactor.connectTCP(self.memory.getParamV("Server", "Connection"), 5222, self.factory)
goffi@necton2
parents:
diff changeset
110 self.connectionStatus="online" #TODO: check if connection is OK
goffi@necton2
parents:
diff changeset
111 self.connected=True #TODO: use startedConnecting and clientConnectionLost of XMPPClientFactory
goffi@necton2
parents:
diff changeset
112
1
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
113 def disconnect(self):
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
114 if (not self.connected):
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
115 info("not connected !")
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
116 return
a06a151fc31f Disconnect first draft
Goffi <goffi@goffi.org>
parents: 0
diff changeset
117 info("Disconnecting...")
0
goffi@necton2
parents:
diff changeset
118
goffi@necton2
parents:
diff changeset
119 def run(self):
goffi@necton2
parents:
diff changeset
120 debug("running app")
goffi@necton2
parents:
diff changeset
121 reactor.run()
goffi@necton2
parents:
diff changeset
122
goffi@necton2
parents:
diff changeset
123 def stop(self):
goffi@necton2
parents:
diff changeset
124 debug("stopping app")
goffi@necton2
parents:
diff changeset
125 reactor.stop()
goffi@necton2
parents:
diff changeset
126
goffi@necton2
parents:
diff changeset
127 def authd(self,xmlstream):
goffi@necton2
parents:
diff changeset
128 self.xmlstream=xmlstream
goffi@necton2
parents:
diff changeset
129 roster=client.IQ(xmlstream,'get')
goffi@necton2
parents:
diff changeset
130 roster.addElement(('jabber:iq:roster', 'query'))
goffi@necton2
parents:
diff changeset
131 roster.addCallback(self.rosterCb)
goffi@necton2
parents:
diff changeset
132 roster.send()
goffi@necton2
parents:
diff changeset
133 debug("server = %s",self.memory.getParamV("Server", "Connection"))
goffi@necton2
parents:
diff changeset
134
goffi@necton2
parents:
diff changeset
135 ###FIXME: tmp disco ###
goffi@necton2
parents:
diff changeset
136 self.memory.registerFeature("http://jabber.org/protocol/disco#info")
goffi@necton2
parents:
diff changeset
137 self.disco(self.memory.getParamV("Server", "Connection"), self.serverDisco)
goffi@necton2
parents:
diff changeset
138
goffi@necton2
parents:
diff changeset
139
goffi@necton2
parents:
diff changeset
140 #we now send our presence status
goffi@necton2
parents:
diff changeset
141 self.setPresence(status="Online")
goffi@necton2
parents:
diff changeset
142
goffi@necton2
parents:
diff changeset
143 # add a callback for the messages
goffi@necton2
parents:
diff changeset
144 xmlstream.addObserver('/message', self.gotMessage)
goffi@necton2
parents:
diff changeset
145 xmlstream.addObserver('/presence', self.presenceCb)
goffi@necton2
parents:
diff changeset
146 xmlstream.addObserver("/iq[@type='set' or @type='get']", self.iqCb)
goffi@necton2
parents:
diff changeset
147 #reactor.callLater(2,self.sendFile,"goffi2@jabber.goffi.int/Psi", "/tmp/fakefile")
goffi@necton2
parents:
diff changeset
148
goffi@necton2
parents:
diff changeset
149 def sendMessage(self,to,msg,type='chat'):
goffi@necton2
parents:
diff changeset
150 #FIXME: check validity of recipient
goffi@necton2
parents:
diff changeset
151 debug("Sending jabber message to %s...", to)
goffi@necton2
parents:
diff changeset
152 message = domish.Element(('jabber:client','message'))
goffi@necton2
parents:
diff changeset
153 message["to"] = jid.JID(to).full()
goffi@necton2
parents:
diff changeset
154 message["from"] = self.me.full()
goffi@necton2
parents:
diff changeset
155 message["type"] = type
goffi@necton2
parents:
diff changeset
156 message.addElement("body", "jabber:client", msg)
goffi@necton2
parents:
diff changeset
157 self.xmlstream.send(message)
goffi@necton2
parents:
diff changeset
158 self.memory.addToHistory(self.me, self.me, jid.JID(to), message["type"], unicode(msg))
goffi@necton2
parents:
diff changeset
159 self.bridge.newMessage(message['from'], unicode(msg), to=message['to']) #We send back the message, so all clients are aware of it
goffi@necton2
parents:
diff changeset
160
goffi@necton2
parents:
diff changeset
161 def setParam(self, name, value, namespace):
goffi@necton2
parents:
diff changeset
162 """set wanted paramater and notice observers"""
goffi@necton2
parents:
diff changeset
163 info ("setting param: %s=%s in namespace %s", name, value, namespace)
goffi@necton2
parents:
diff changeset
164 self.memory.setParam(name, value, namespace)
goffi@necton2
parents:
diff changeset
165 self.bridge.paramUpdate(name, value, namespace)
goffi@necton2
parents:
diff changeset
166
goffi@necton2
parents:
diff changeset
167 def setRoster(self, to):
goffi@necton2
parents:
diff changeset
168 """Add a contact to roster list"""
goffi@necton2
parents:
diff changeset
169 to_jid=jid.JID(to)
goffi@necton2
parents:
diff changeset
170 roster=client.IQ(self.xmlstream,'set')
goffi@necton2
parents:
diff changeset
171 query=roster.addElement(('jabber:iq:roster', 'query'))
goffi@necton2
parents:
diff changeset
172 item=query.addElement("item")
goffi@necton2
parents:
diff changeset
173 item.attributes["jid"]=to_jid.userhost()
goffi@necton2
parents:
diff changeset
174 roster.send()
goffi@necton2
parents:
diff changeset
175 #TODO: check IQ result
goffi@necton2
parents:
diff changeset
176
goffi@necton2
parents:
diff changeset
177 def delRoster(self, to):
goffi@necton2
parents:
diff changeset
178 """Remove a contact from roster list"""
goffi@necton2
parents:
diff changeset
179 to_jid=jid.JID(to)
goffi@necton2
parents:
diff changeset
180 roster=client.IQ(self.xmlstream,'set')
goffi@necton2
parents:
diff changeset
181 query=roster.addElement(('jabber:iq:roster', 'query'))
goffi@necton2
parents:
diff changeset
182 item=query.addElement("item")
goffi@necton2
parents:
diff changeset
183 item.attributes["jid"]=to_jid.userhost()
goffi@necton2
parents:
diff changeset
184 item.attributes["subscription"]="remove"
goffi@necton2
parents:
diff changeset
185 roster.send()
goffi@necton2
parents:
diff changeset
186 #TODO: check IQ result
goffi@necton2
parents:
diff changeset
187
goffi@necton2
parents:
diff changeset
188
goffi@necton2
parents:
diff changeset
189 def failed(self,xmlstream):
goffi@necton2
parents:
diff changeset
190 debug("failed: %s", xmlstream.getErrorMessage())
goffi@necton2
parents:
diff changeset
191 debug("failed: %s", dir(xmlstream))
goffi@necton2
parents:
diff changeset
192
goffi@necton2
parents:
diff changeset
193 def isConnected(self):
goffi@necton2
parents:
diff changeset
194 return self.connected
goffi@necton2
parents:
diff changeset
195
goffi@necton2
parents:
diff changeset
196 ## jabber methods ##
goffi@necton2
parents:
diff changeset
197
goffi@necton2
parents:
diff changeset
198 def disco (self, item, callback, node=None):
goffi@necton2
parents:
diff changeset
199 """XEP-0030 Service discovery Feature."""
goffi@necton2
parents:
diff changeset
200 disco=client.IQ(self.xmlstream,'get')
goffi@necton2
parents:
diff changeset
201 disco["from"]=self.me.full()
goffi@necton2
parents:
diff changeset
202 disco["to"]=item
goffi@necton2
parents:
diff changeset
203 disco.addElement(('http://jabber.org/protocol/disco#info', 'query'))
goffi@necton2
parents:
diff changeset
204 disco.addCallback(callback)
goffi@necton2
parents:
diff changeset
205 disco.send()
goffi@necton2
parents:
diff changeset
206
goffi@necton2
parents:
diff changeset
207
goffi@necton2
parents:
diff changeset
208 def setPresence(self, to="", type="", show="", status="", priority=0):
goffi@necton2
parents:
diff changeset
209 """Send our presence information"""
goffi@necton2
parents:
diff changeset
210 presence = domish.Element(('jabber:client', 'presence'))
goffi@necton2
parents:
diff changeset
211 if not type in ["", "unavailable", "subscribed", "subscribe",
goffi@necton2
parents:
diff changeset
212 "unsubscribe", "unsubscribed", "prob", "error"]:
goffi@necton2
parents:
diff changeset
213 error("Type error !")
goffi@necton2
parents:
diff changeset
214 #TODO: throw an error
goffi@necton2
parents:
diff changeset
215 return
goffi@necton2
parents:
diff changeset
216
goffi@necton2
parents:
diff changeset
217 if to:
goffi@necton2
parents:
diff changeset
218 presence.attributes["to"]=to
goffi@necton2
parents:
diff changeset
219 if type:
goffi@necton2
parents:
diff changeset
220 presence.attributes["type"]=type
goffi@necton2
parents:
diff changeset
221
goffi@necton2
parents:
diff changeset
222 for element in ["show", "status", "priority"]:
goffi@necton2
parents:
diff changeset
223 if locals()[element]:
goffi@necton2
parents:
diff changeset
224 presence.addElement(element).addContent(unicode(locals()[element]))
goffi@necton2
parents:
diff changeset
225
goffi@necton2
parents:
diff changeset
226 self.xmlstream.send(presence)
goffi@necton2
parents:
diff changeset
227
goffi@necton2
parents:
diff changeset
228 def addContact(self, to):
goffi@necton2
parents:
diff changeset
229 """Add a contact in roster list"""
goffi@necton2
parents:
diff changeset
230 to_jid=jid.JID(to)
goffi@necton2
parents:
diff changeset
231 self.setRoster(to_jid.userhost())
goffi@necton2
parents:
diff changeset
232 self.setPresence(to_jid.userhost(), "subscribe")
goffi@necton2
parents:
diff changeset
233
goffi@necton2
parents:
diff changeset
234 def delContact(self, to):
goffi@necton2
parents:
diff changeset
235 """Remove contact from roster list"""
goffi@necton2
parents:
diff changeset
236 to_jid=jid.JID(to)
goffi@necton2
parents:
diff changeset
237 self.delRoster(to_jid.userhost())
goffi@necton2
parents:
diff changeset
238 self.bridge.contactDeleted(to)
goffi@necton2
parents:
diff changeset
239
goffi@necton2
parents:
diff changeset
240 def gotMessage(self,message):
goffi@necton2
parents:
diff changeset
241 debug (u"got_message from: %s", message["from"])
goffi@necton2
parents:
diff changeset
242 for e in message.elements():
goffi@necton2
parents:
diff changeset
243 if e.name == "body":
goffi@necton2
parents:
diff changeset
244 self.bridge.newMessage(message["from"], e.children[0])
goffi@necton2
parents:
diff changeset
245 self.memory.addToHistory(self.me, jid.JID(message["from"]), self.me, "chat", e.children[0])
goffi@necton2
parents:
diff changeset
246 break
goffi@necton2
parents:
diff changeset
247
goffi@necton2
parents:
diff changeset
248 ## callbacks ##
goffi@necton2
parents:
diff changeset
249
goffi@necton2
parents:
diff changeset
250 def add_IQ_cb(self, ns, cb):
goffi@necton2
parents:
diff changeset
251 """Add an IQ callback on namespace ns"""
goffi@necton2
parents:
diff changeset
252 debug ("Registered callback for namespace %s", ns)
goffi@necton2
parents:
diff changeset
253 self._iq_cb_map[ns]=cb
goffi@necton2
parents:
diff changeset
254
goffi@necton2
parents:
diff changeset
255 def iqCb(self, stanza):
goffi@necton2
parents:
diff changeset
256 info ("iqCb")
goffi@necton2
parents:
diff changeset
257 debug ("="*20)
goffi@necton2
parents:
diff changeset
258 debug ("DEBUG:\n")
goffi@necton2
parents:
diff changeset
259 debug (stanza.toXml().encode('utf-8'))
goffi@necton2
parents:
diff changeset
260 debug ("="*20)
goffi@necton2
parents:
diff changeset
261 #FIXME: temporary ugly code
goffi@necton2
parents:
diff changeset
262 uri = stanza.firstChildElement().uri
goffi@necton2
parents:
diff changeset
263 if self._iq_cb_map.has_key(uri):
goffi@necton2
parents:
diff changeset
264 self._iq_cb_map[uri](stanza)
goffi@necton2
parents:
diff changeset
265 #TODO: manage errors stanza
goffi@necton2
parents:
diff changeset
266
goffi@necton2
parents:
diff changeset
267 def presenceCb(self, elem):
goffi@necton2
parents:
diff changeset
268 info ("presence update for [%s]", elem.getAttribute("from"))
goffi@necton2
parents:
diff changeset
269 debug("\n\nXML=\n%s\n\n", elem.toXml())
goffi@necton2
parents:
diff changeset
270 presence={}
goffi@necton2
parents:
diff changeset
271 presence["jid"]=elem.getAttribute("from")
goffi@necton2
parents:
diff changeset
272 presence["type"]=elem.getAttribute("type") or ""
goffi@necton2
parents:
diff changeset
273 presence["show"]=""
goffi@necton2
parents:
diff changeset
274 presence["status"]=""
goffi@necton2
parents:
diff changeset
275 presence["priority"]=0
goffi@necton2
parents:
diff changeset
276
goffi@necton2
parents:
diff changeset
277 for item in elem.elements():
goffi@necton2
parents:
diff changeset
278 if presence.has_key(item.name):
goffi@necton2
parents:
diff changeset
279 presence[item.name]=item.children[0]
goffi@necton2
parents:
diff changeset
280
goffi@necton2
parents:
diff changeset
281 ### we check if the status is not about subscription ###
goffi@necton2
parents:
diff changeset
282 #TODO: check that from jid is one we wants to subscribe (ie: check a recent subscription asking)
goffi@necton2
parents:
diff changeset
283 if jid.JID(presence["jid"]).userhost()!=self.me.userhost():
goffi@necton2
parents:
diff changeset
284 if presence["type"]=="subscribed":
goffi@necton2
parents:
diff changeset
285 debug ("subscription answer")
goffi@necton2
parents:
diff changeset
286 elif presence["type"]=="unsubscribed":
goffi@necton2
parents:
diff changeset
287 debug ("unsubscription answer")
goffi@necton2
parents:
diff changeset
288 elif presence["type"]=="subscribe":
goffi@necton2
parents:
diff changeset
289 #FIXME: auto answer for subscribe request, must be checked !
goffi@necton2
parents:
diff changeset
290 debug ("subscription request")
goffi@necton2
parents:
diff changeset
291 self.setPresence(to=presence["jid"], type="subscribed")
goffi@necton2
parents:
diff changeset
292 else:
goffi@necton2
parents:
diff changeset
293 #We keep presence information only if it is not for subscription
goffi@necton2
parents:
diff changeset
294 self.memory.addPresenceStatus(presence["jid"], presence["type"], presence["show"],
goffi@necton2
parents:
diff changeset
295 presence["status"], int(presence["priority"]))
goffi@necton2
parents:
diff changeset
296
goffi@necton2
parents:
diff changeset
297 #now it's time to notify frontends
goffi@necton2
parents:
diff changeset
298 self.bridge.presenceUpdate(presence["jid"], presence["type"], presence["show"],
goffi@necton2
parents:
diff changeset
299 presence["status"], int(presence["priority"]))
goffi@necton2
parents:
diff changeset
300
goffi@necton2
parents:
diff changeset
301 def rosterCb(self,roster):
goffi@necton2
parents:
diff changeset
302 for contact in roster.firstChildElement().elements():
goffi@necton2
parents:
diff changeset
303 info ("new contact in roster list: %s", contact['jid'])
goffi@necton2
parents:
diff changeset
304 #and now the groups
goffi@necton2
parents:
diff changeset
305 groups=[]
goffi@necton2
parents:
diff changeset
306 for group in contact.elements():
goffi@necton2
parents:
diff changeset
307 if group.name!="group":
goffi@necton2
parents:
diff changeset
308 error("Unexpected element !")
goffi@necton2
parents:
diff changeset
309 break
goffi@necton2
parents:
diff changeset
310 groups.append(str(group))
goffi@necton2
parents:
diff changeset
311 self.memory.addContact(contact['jid'], contact.attributes, groups)
goffi@necton2
parents:
diff changeset
312 self.bridge.newContact(contact['jid'], contact.attributes, groups)
goffi@necton2
parents:
diff changeset
313
goffi@necton2
parents:
diff changeset
314 def serverDisco(self, disco):
goffi@necton2
parents:
diff changeset
315 """xep-0030 Discovery Protocol."""
goffi@necton2
parents:
diff changeset
316 for element in disco.firstChildElement().elements():
goffi@necton2
parents:
diff changeset
317 if element.name=="feature":
goffi@necton2
parents:
diff changeset
318 debug ("Feature dectetee: %s",element["var"])
goffi@necton2
parents:
diff changeset
319 self.server_features.append(element["var"])
goffi@necton2
parents:
diff changeset
320 elif element.name=="identity":
goffi@necton2
parents:
diff changeset
321 debug ("categorie= %s",element["category"])
goffi@necton2
parents:
diff changeset
322 debug ("features= %s",self.server_features)
goffi@necton2
parents:
diff changeset
323
goffi@necton2
parents:
diff changeset
324 ## Generic HMI ##
goffi@necton2
parents:
diff changeset
325
goffi@necton2
parents:
diff changeset
326 def askConfirmation(self, id, type, data, cb):
goffi@necton2
parents:
diff changeset
327 """Add a confirmation callback"""
goffi@necton2
parents:
diff changeset
328 if self._waiting_conf.has_key(id):
goffi@necton2
parents:
diff changeset
329 error ("Attempt to register two callbacks for the same confirmation")
goffi@necton2
parents:
diff changeset
330 else:
goffi@necton2
parents:
diff changeset
331 self._waiting_conf[id] = cb
goffi@necton2
parents:
diff changeset
332 self.bridge.askConfirmation(type, id, data)
goffi@necton2
parents:
diff changeset
333
goffi@necton2
parents:
diff changeset
334
goffi@necton2
parents:
diff changeset
335 def confirmationAnswer(self, id, accepted, data):
goffi@necton2
parents:
diff changeset
336 """Called by frontends to answer confirmation requests"""
goffi@necton2
parents:
diff changeset
337 debug ("Received confirmation answer for id [%s]: %s", id, "accepted" if accepted else "refused")
goffi@necton2
parents:
diff changeset
338 if not self._waiting_conf.has_key(id):
goffi@necton2
parents:
diff changeset
339 error ("Received an unknown confirmation")
goffi@necton2
parents:
diff changeset
340 else:
goffi@necton2
parents:
diff changeset
341 cb = self._waiting_conf[id]
goffi@necton2
parents:
diff changeset
342 del self._waiting_conf[id]
goffi@necton2
parents:
diff changeset
343 cb(id, accepted, data)
goffi@necton2
parents:
diff changeset
344
goffi@necton2
parents:
diff changeset
345 def registerProgressCB(self, id, CB):
goffi@necton2
parents:
diff changeset
346 """Register a callback called when progress is requested for id"""
goffi@necton2
parents:
diff changeset
347 self._progress_cb_map[id] = CB
goffi@necton2
parents:
diff changeset
348
goffi@necton2
parents:
diff changeset
349 def removeProgressCB(self, id):
goffi@necton2
parents:
diff changeset
350 """Remove a progress callback"""
goffi@necton2
parents:
diff changeset
351 if not self._progress_cb_map.has_key(id):
goffi@necton2
parents:
diff changeset
352 error ("Trying to remove an unknow progress callback")
goffi@necton2
parents:
diff changeset
353 else:
goffi@necton2
parents:
diff changeset
354 del self._progress_cb_map[id]
goffi@necton2
parents:
diff changeset
355
goffi@necton2
parents:
diff changeset
356 def getProgress(self, id):
goffi@necton2
parents:
diff changeset
357 """Return a dict with progress information
goffi@necton2
parents:
diff changeset
358 data['position'] : current possition
goffi@necton2
parents:
diff changeset
359 data['size'] : end_position
goffi@necton2
parents:
diff changeset
360 """
goffi@necton2
parents:
diff changeset
361 data = {}
goffi@necton2
parents:
diff changeset
362 try:
goffi@necton2
parents:
diff changeset
363 self._progress_cb_map[id](data)
goffi@necton2
parents:
diff changeset
364 except KeyError:
goffi@necton2
parents:
diff changeset
365 pass
goffi@necton2
parents:
diff changeset
366 #debug("Requested progress for unknown id")
goffi@necton2
parents:
diff changeset
367 return data
goffi@necton2
parents:
diff changeset
368
goffi@necton2
parents:
diff changeset
369
goffi@necton2
parents:
diff changeset
370 app=SAT() #TODO: tmp, use twisted way instead
goffi@necton2
parents:
diff changeset
371 app.run()
goffi@necton2
parents:
diff changeset
372
goffi@necton2
parents:
diff changeset
373 app.memory.save() #FIXME: not the best place
goffi@necton2
parents:
diff changeset
374 debug("Good Bye")