annotate sat.tac @ 6:5799493fa548

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