annotate plugins/plugin_xep_0096.py @ 0:c4bc297b82f0

sat: - first public release, initial commit
author goffi@necton2
date Sat, 29 Aug 2009 13:34:59 +0200
parents
children a06a151fc31f
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 plugin for managing xep-0096
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 from logging import debug, info, error
goffi@necton2
parents:
diff changeset
23 from twisted.words.xish import domish
goffi@necton2
parents:
diff changeset
24 from twisted.internet import protocol
goffi@necton2
parents:
diff changeset
25 from twisted.words.protocols.jabber import client, jid, xmlstream, error
goffi@necton2
parents:
diff changeset
26 import os.path
goffi@necton2
parents:
diff changeset
27 from twisted.internet import reactor #FIXME best way ???
goffi@necton2
parents:
diff changeset
28
goffi@necton2
parents:
diff changeset
29
goffi@necton2
parents:
diff changeset
30 PLUGIN_INFO = {
goffi@necton2
parents:
diff changeset
31 "name": "XEP 0096 Plugin",
goffi@necton2
parents:
diff changeset
32 "import_name": "XEP_0096",
goffi@necton2
parents:
diff changeset
33 "type": "XEP",
goffi@necton2
parents:
diff changeset
34 "dependencies": ["XEP_0065"],
goffi@necton2
parents:
diff changeset
35 "main": "XEP_0096",
goffi@necton2
parents:
diff changeset
36 "description": """Implementation of SI File Transfert"""
goffi@necton2
parents:
diff changeset
37 }
goffi@necton2
parents:
diff changeset
38
goffi@necton2
parents:
diff changeset
39 class XEP_0096:
goffi@necton2
parents:
diff changeset
40 def __init__(self, host):
goffi@necton2
parents:
diff changeset
41 info("Plugin XEP_0096 initialization")
goffi@necton2
parents:
diff changeset
42 self.host = host
goffi@necton2
parents:
diff changeset
43 self._waiting_for_approval = {}
goffi@necton2
parents:
diff changeset
44 host.add_IQ_cb("http://jabber.org/protocol/si", self.xep_96)
goffi@necton2
parents:
diff changeset
45 host.bridge.register("sendFile", self.sendFile)
goffi@necton2
parents:
diff changeset
46
goffi@necton2
parents:
diff changeset
47
goffi@necton2
parents:
diff changeset
48 def xep_96(self, IQ):
goffi@necton2
parents:
diff changeset
49 info ("XEP-0096 management")
goffi@necton2
parents:
diff changeset
50 SI_elem = IQ.firstChildElement()
goffi@necton2
parents:
diff changeset
51 debug(SI_elem.toXml())
goffi@necton2
parents:
diff changeset
52 filename = ""
goffi@necton2
parents:
diff changeset
53 file_size = ""
goffi@necton2
parents:
diff changeset
54 for element in SI_elem.elements():
goffi@necton2
parents:
diff changeset
55 if element.name == "file":
goffi@necton2
parents:
diff changeset
56 info ("File proposed: name=[%s] size=%s", element['name'], element['size'])
goffi@necton2
parents:
diff changeset
57 filename = element["name"]
goffi@necton2
parents:
diff changeset
58 file_size = element["size"]
goffi@necton2
parents:
diff changeset
59 elif element.name == "feature":
goffi@necton2
parents:
diff changeset
60 from_jid = IQ["from"]
goffi@necton2
parents:
diff changeset
61 self._waiting_for_approval[IQ["id"]] = (element, from_jid, file_size)
goffi@necton2
parents:
diff changeset
62 data={ "filename":filename, "from":from_jid, "size":file_size }
goffi@necton2
parents:
diff changeset
63 self.host.askConfirmation(IQ["id"], "FILE_TRANSFERT", data, self.confirmationCB)
goffi@necton2
parents:
diff changeset
64
goffi@necton2
parents:
diff changeset
65 def confirmationCB(self, id, accepted, data):
goffi@necton2
parents:
diff changeset
66 """Called on confirmation answer"""
goffi@necton2
parents:
diff changeset
67 if accepted:
goffi@necton2
parents:
diff changeset
68 data['size'] = self._waiting_for_approval[id][2]
goffi@necton2
parents:
diff changeset
69 self.host.plugins["XEP_0065"].setData(data, id)
goffi@necton2
parents:
diff changeset
70 self.approved(id)
goffi@necton2
parents:
diff changeset
71 else:
goffi@necton2
parents:
diff changeset
72 debug ("Transfert [%s] refused", id)
goffi@necton2
parents:
diff changeset
73 del(self._waiting_for_approval[id])
goffi@necton2
parents:
diff changeset
74
goffi@necton2
parents:
diff changeset
75 def approved(self, id):
goffi@necton2
parents:
diff changeset
76 """must be called when a file transfert has be accepted by client"""
goffi@necton2
parents:
diff changeset
77 debug ("Transfert [%s] accepted", id)
goffi@necton2
parents:
diff changeset
78
goffi@necton2
parents:
diff changeset
79 if ( not self._waiting_for_approval.has_key(id) ):
goffi@necton2
parents:
diff changeset
80 error ("Approved unknow id !")
goffi@necton2
parents:
diff changeset
81 #TODO: manage this (maybe approved by several frontends)
goffi@necton2
parents:
diff changeset
82 else:
goffi@necton2
parents:
diff changeset
83 element, from_id, size = self._waiting_for_approval[id]
goffi@necton2
parents:
diff changeset
84 del(self._waiting_for_approval[id])
goffi@necton2
parents:
diff changeset
85 self.negociate(element, id, from_id)
goffi@necton2
parents:
diff changeset
86
goffi@necton2
parents:
diff changeset
87 def negociate(self, feat_elem, id, to_jid):
goffi@necton2
parents:
diff changeset
88 #TODO: put this in a plugin
goffi@necton2
parents:
diff changeset
89 #FIXME: over ultra mega ugly, need to be generic
goffi@necton2
parents:
diff changeset
90 info ("Feature negociation")
goffi@necton2
parents:
diff changeset
91 data = feat_elem.firstChildElement()
goffi@necton2
parents:
diff changeset
92 field = data.firstChildElement()
goffi@necton2
parents:
diff changeset
93 #FIXME: several options ! Q&D code for test only
goffi@necton2
parents:
diff changeset
94 option = field.firstChildElement()
goffi@necton2
parents:
diff changeset
95 value = option.firstChildElement()
goffi@necton2
parents:
diff changeset
96 if unicode(value) == "http://jabber.org/protocol/bytestreams":
goffi@necton2
parents:
diff changeset
97 #ugly, as usual, need to be entirely rewritten (just for test !)
goffi@necton2
parents:
diff changeset
98 result = domish.Element(('', 'iq'))
goffi@necton2
parents:
diff changeset
99 result['type'] = 'result'
goffi@necton2
parents:
diff changeset
100 result['id'] = id
goffi@necton2
parents:
diff changeset
101 result['to'] = to_jid
goffi@necton2
parents:
diff changeset
102 si = result.addElement('si', 'http://jabber.org/protocol/si')
goffi@necton2
parents:
diff changeset
103 file = si.addElement('file', 'http://jabber.org/protocol/si/profile/file-transfer')
goffi@necton2
parents:
diff changeset
104 feature = si.addElement('feature', 'http://jabber.org/protocol/feature-neg')
goffi@necton2
parents:
diff changeset
105 x = feature.addElement('x', 'jabber:x:data')
goffi@necton2
parents:
diff changeset
106 x['type'] = 'submit'
goffi@necton2
parents:
diff changeset
107 field = x.addElement('field')
goffi@necton2
parents:
diff changeset
108 field['var'] = 'stream-method'
goffi@necton2
parents:
diff changeset
109 value = field.addElement('value')
goffi@necton2
parents:
diff changeset
110 value.addContent('http://jabber.org/protocol/bytestreams')
goffi@necton2
parents:
diff changeset
111 self.host.xmlstream.send(result)
goffi@necton2
parents:
diff changeset
112
goffi@necton2
parents:
diff changeset
113 def fileCB(self, answer):
goffi@necton2
parents:
diff changeset
114 if answer['type']=="result": #FIXME FIXME FIXME ugly ugly ugly ! and temp FIXME FIXME FIXME
goffi@necton2
parents:
diff changeset
115 info("SENDING UGLY ANSWER")
goffi@necton2
parents:
diff changeset
116 offer=client.IQ(self.host.xmlstream,'set')
goffi@necton2
parents:
diff changeset
117 offer["from"]=self.host.me.full()
goffi@necton2
parents:
diff changeset
118 offer["to"]=answer['from']
goffi@necton2
parents:
diff changeset
119 query=offer.addElement('query', 'http://jabber.org/protocol/bytestreams')
goffi@necton2
parents:
diff changeset
120 query['mode']='tcp'
goffi@necton2
parents:
diff changeset
121 streamhost=query.addElement('streamhost')
goffi@necton2
parents:
diff changeset
122 streamhost['host']=self.host.memory.getParamV("IP", "File Transfert")
goffi@necton2
parents:
diff changeset
123 streamhost['port']=self.host.memory.getParamV("Port", "File Transfert")
goffi@necton2
parents:
diff changeset
124 streamhost['jid']=self.host.me.full()
goffi@necton2
parents:
diff changeset
125 offer.send()
goffi@necton2
parents:
diff changeset
126
goffi@necton2
parents:
diff changeset
127
goffi@necton2
parents:
diff changeset
128
goffi@necton2
parents:
diff changeset
129
goffi@necton2
parents:
diff changeset
130 def sendFile(self, to, filepath):
goffi@necton2
parents:
diff changeset
131 """send a file using XEP-0096
goffi@necton2
parents:
diff changeset
132 Return an unique id to identify the transfert
goffi@necton2
parents:
diff changeset
133 """
goffi@necton2
parents:
diff changeset
134 debug ("sendfile (%s) to %s", filepath, to )
goffi@necton2
parents:
diff changeset
135 print type(filepath), type(to)
goffi@necton2
parents:
diff changeset
136
goffi@necton2
parents:
diff changeset
137 statinfo = os.stat(filepath)
goffi@necton2
parents:
diff changeset
138
goffi@necton2
parents:
diff changeset
139 offer=client.IQ(self.host.xmlstream,'set')
goffi@necton2
parents:
diff changeset
140 debug ("Transfert ID: %s", offer["id"])
goffi@necton2
parents:
diff changeset
141
goffi@necton2
parents:
diff changeset
142 self.host.plugins["XEP_0065"].sendFile(offer["id"], filepath, str(statinfo.st_size))
goffi@necton2
parents:
diff changeset
143
goffi@necton2
parents:
diff changeset
144 offer["from"]=self.host.me.full()
goffi@necton2
parents:
diff changeset
145 offer["to"]=jid.JID(to).full()
goffi@necton2
parents:
diff changeset
146 si=offer.addElement('si','http://jabber.org/protocol/si')
goffi@necton2
parents:
diff changeset
147 si["mime-type"]='text/plain'
goffi@necton2
parents:
diff changeset
148 si["profile"]='http://jabber.org/protocol/si/profile/file-transfer'
goffi@necton2
parents:
diff changeset
149 file = si.addElement('file', 'http://jabber.org/protocol/si')
goffi@necton2
parents:
diff changeset
150 file['name']=os.path.basename(filepath)
goffi@necton2
parents:
diff changeset
151 file['size']=str(statinfo.st_size)
goffi@necton2
parents:
diff changeset
152
goffi@necton2
parents:
diff changeset
153 ###
goffi@necton2
parents:
diff changeset
154 # FIXME: Ugly temporary hard coded implementation of XEP-0020 & XEP-0004,
goffi@necton2
parents:
diff changeset
155 # Need to be recoded elsewhere in a more generic way
goffi@necton2
parents:
diff changeset
156 ###
goffi@necton2
parents:
diff changeset
157
goffi@necton2
parents:
diff changeset
158 feature=si.addElement('feature', "http://jabber.org/protocol/feature-neg")
goffi@necton2
parents:
diff changeset
159 x=feature.addElement('x', "jabber:x:data")
goffi@necton2
parents:
diff changeset
160 x['type']='form'
goffi@necton2
parents:
diff changeset
161 field=x.addElement('field')
goffi@necton2
parents:
diff changeset
162 field['type']='list-single'
goffi@necton2
parents:
diff changeset
163 field['var']='stream-method'
goffi@necton2
parents:
diff changeset
164 option = field.addElement('option')
goffi@necton2
parents:
diff changeset
165 value = option.addElement('value', content='http://jabber.org/protocol/bytestreams')
goffi@necton2
parents:
diff changeset
166
goffi@necton2
parents:
diff changeset
167 offer.addCallback(self.fileCB)
goffi@necton2
parents:
diff changeset
168 offer.send()
goffi@necton2
parents:
diff changeset
169 return offer["id"] #XXX: using IQ id as file transfert id seems OK as IQ id are required
goffi@necton2
parents:
diff changeset
170