|
980
|
1 #!/usr/bin/python |
|
|
2 # -*- coding: utf-8 -*- |
|
|
3 |
|
|
4 # SAT plugin for managing xep-0049 |
|
|
5 # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014 Jérôme Poisson (goffi@goffi.org) |
|
|
6 |
|
|
7 # This program is free software: you can redistribute it and/or modify |
|
|
8 # it under the terms of the GNU Affero General Public License as published by |
|
|
9 # the Free Software Foundation, either version 3 of the License, or |
|
|
10 # (at your option) any later version. |
|
|
11 |
|
|
12 # This program is distributed in the hope that it will be useful, |
|
|
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
15 # GNU Affero General Public License for more details. |
|
|
16 |
|
|
17 # You should have received a copy of the GNU Affero General Public License |
|
|
18 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
19 |
|
|
20 from sat.core.i18n import _ |
|
|
21 from logging import debug, info, error, warning |
|
|
22 from twisted.internet import defer |
|
|
23 from wokkel import compat |
|
|
24 from twisted.words.xish import domish |
|
|
25 |
|
|
26 |
|
|
27 |
|
|
28 PLUGIN_INFO = { |
|
|
29 "name": "XEP-0049 Plugin", |
|
|
30 "import_name": "XEP-0049", |
|
|
31 "type": "XEP", |
|
|
32 "protocols": ["XEP-0049"], |
|
|
33 "dependencies": [], |
|
|
34 "main": "XEP_0049", |
|
|
35 "handler": "no", |
|
|
36 "description": _("""Implementation of private XML storage""") |
|
|
37 } |
|
|
38 |
|
|
39 |
|
|
40 class XEP_0049(object): |
|
|
41 NS_PRIVATE = 'jabber:iq:private' |
|
|
42 |
|
|
43 def __init__(self, host): |
|
|
44 info(_("Plugin XEP-0049 initialization")) |
|
|
45 self.host = host |
|
|
46 |
|
|
47 @defer.inlineCallbacks |
|
|
48 def privateXMLStore(self, element, profile_key): |
|
|
49 """Store private data |
|
|
50 @param element: domish.Element to store (must have a namespace) |
|
|
51 @param profile_key: %(doc_profile_key)s |
|
|
52 |
|
|
53 """ |
|
|
54 assert isinstance(element, domish.Element) |
|
|
55 client = self.host.getClient(profile_key) |
|
|
56 yield self.host.checkFeature(XEP_0049.NS_PRIVATE, profile_key=client.profile) |
|
|
57 iq_elt = compat.IQ(client.xmlstream) |
|
|
58 query_elt = iq_elt.addElement('query', XEP_0049.NS_PRIVATE) |
|
|
59 query_elt.addChild(element) |
|
|
60 yield iq_elt.send() |
|
|
61 |
|
|
62 @defer.inlineCallbacks |
|
|
63 def privateXMLGet(self, node_name, namespace, profile_key): |
|
|
64 """Store private data |
|
|
65 @param node_name: name of the node to get |
|
|
66 @param namespace: namespace of the node to get |
|
|
67 @param profile_key: %(doc_profile_key)s |
|
|
68 @return (domish.Element): a deferred which fire the stored data |
|
|
69 |
|
|
70 """ |
|
|
71 client = self.host.getClient(profile_key) |
|
|
72 yield self.host.checkFeature(XEP_0049.NS_PRIVATE, profile_key=client.profile) |
|
|
73 iq_elt = compat.IQ(client.xmlstream, 'get') |
|
|
74 query_elt = iq_elt.addElement('query', XEP_0049.NS_PRIVATE) |
|
|
75 query_elt.addElement(node_name, namespace) |
|
|
76 answer_iq_elt = yield iq_elt.send() |
|
|
77 answer_query_elt = answer_iq_elt.elements(XEP_0049.NS_PRIVATE, 'query').next() |
|
|
78 defer.returnValue(answer_query_elt.firstChildElement()) |
|
|
79 |
