libervia-backend: src/plugins/plugin_xep

annotate src/plugins/plugin_xep_0054.py @ 447:485a6d125498

Wix: fixed asynchronous call to get profile's data in profile manager

author	Goffi <goffi@goffi.org>
date	Sun, 04 Dec 2011 00:58:20 +0100
parents	d67cae85c2f9
children	cf005701624b

rev	line source
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/python
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	2 # -- coding: utf-8 --
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	3
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	4 """
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	5 SAT plugin for managing xep-0054
228 b1794cbb88e5 2011 copyright upgrade Goffi <goffi@goffi.org> parents: 223 diff changeset	6 Copyright (C) 2009, 2010, 2011 Jérôme Poisson (goffi@goffi.org)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	7
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	8 This program is free software: you can redistribute it and/or modify
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	9 it under the terms of the GNU General Public License as published by
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	10 the Free Software Foundation, either version 3 of the License, or
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	11 (at your option) any later version.
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	12
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	13 This program is distributed in the hope that it will be useful,
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	14 but WITHOUT ANY WARRANTY; without even the implied warranty of
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	16 GNU General Public License for more details.
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	17
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	18 You should have received a copy of the GNU General Public License
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	19 along with this program. If not, see <http://www.gnu.org/licenses/>.
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	20 """
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	21
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	22 from logging import debug, info, error
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	23 from twisted.words.xish import domish
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	24 from twisted.internet import protocol, defer, threads, reactor
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.words.protocols.jabber import client, jid, xmlstream
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	26 from twisted.words.protocols.jabber import error as jab_error
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	27 from twisted.words.protocols.jabber.xmlstream import IQ
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	28 import os.path
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	29
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	30 from zope.interface import implements
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	31
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	32 from wokkel import disco, iwokkel
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	33
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	34 from base64 import b64decode
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	35 from hashlib import sha1
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	36 from time import sleep
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	37 from sat.memory.persistent import PersistentBinaryDict
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	38
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	39 try:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	40 from twisted.words.protocols.xmlstream import XMPPHandler
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	41 except ImportError:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	42 from wokkel.subprotocols import XMPPHandler
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	43
365 efbfccfed623 core: local_dir moved to config file Goffi <goffi@goffi.org> parents: 291 diff changeset	44 AVATAR_PATH = "avatars"
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	45
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	46 IQ_GET = '/iq[@type="get"]'
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	47 NS_VCARD = 'vcard-temp'
44 bfa7086d26d6 plugin XEP-0054: fixed forgotten copy/paste change Goffi <goffi@goffi.org> parents: 43 diff changeset	48 VCARD_REQUEST = IQ_GET + '/vCard[@xmlns="' + NS_VCARD + '"]' #TODO: manage requests
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	49
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	50 PRESENCE = '/presence'
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	51 NS_VCARD_UPDATE = 'vcard-temp:x:update'
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	52 VCARD_UPDATE = PRESENCE + '/x[@xmlns="' + NS_VCARD_UPDATE + '"]'
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	53
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	54 PLUGIN_INFO = {
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	55 "name": "XEP 0054 Plugin",
291 7c79d4a8c9e6 plugins: fixed bad import names Goffi <goffi@goffi.org> parents: 281 diff changeset	56 "import_name": "XEP-0054",
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	57 "type": "XEP",
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	58 "protocols": ["XEP-0054", "XEP-0153"],
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	59 "dependencies": [],
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	60 "main": "XEP_0054",
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	61 "handler": "yes",
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	62 "description": _("""Implementation of vcard-temp""")
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	63 }
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	64
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	65 class XEP_0054():
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	66
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	67 def __init__(self, host):
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	68 info(_("Plugin XEP_0054 initialization"))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	69 self.host = host
365 efbfccfed623 core: local_dir moved to config file Goffi <goffi@goffi.org> parents: 291 diff changeset	70 self.avatar_path = os.path.join(self.host.memory.getConfig('', 'local_dir'), AVATAR_PATH)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	71 if not os.path.exists(self.avatar_path):
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	72 os.makedirs(self.avatar_path)
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	73 host.bridge.addMethod("getCard", ".plugin", in_sign='ss', out_sign='s', method=self.getCard)
f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	74 host.bridge.addMethod("getAvatarFile", ".plugin", in_sign='s', out_sign='s', method=self.getAvatarFile)
444 d67cae85c2f9 plugin xep 0054: fixed bad bridge signature Goffi <goffi@goffi.org> parents: 437 diff changeset	75 host.bridge.addMethod("getCardCache", ".plugin", in_sign='ss', out_sign='a{ss}', method=self.getCardCache)
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	76
72 f271fff3a713 MUC implementation: first draft Goffi <goffi@goffi.org> parents: 69 diff changeset	77 def getHandler(self, profile):
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	78 return XEP_0054_handler(self)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	79
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	80 def update_cache(self, jid, name, value, profile):
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	81 """update cache value
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	82 - save value in memory in case of change
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	83 - send updatedValue signal if the value is new or updated
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	84 @param jid: jid of the owner of the vcard
f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	85 @param name: name of the item which changed
f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	86 @param value: new value of the item
f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	87 @param profile: profile which received the update
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	88 """
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	89 client = self.host.getClient(profile)
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	90 if not jid.userhost() in client._vcard_cache:
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	91 client._vcard_cache[jid.userhost()] = {}
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	92
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	93 cache = client._vcard_cache[jid.userhost()]
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	94 old_value = cache[name] if name in cache else None
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	95 if not old_value or value != old_value:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	96 cache[name] = value
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	97 client._vcard_cache.force(jid.userhost()) #we force saving of data to storage
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	98 self.host.bridge.updatedValue('card_'+name, {'jid':jid.userhost(), name:value}, profile)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	99
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	100 def get_cache(self, jid, name, profile):
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	101 """return cached value for jid
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	102 @param jid: target contact
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	103 @param name: name of the value ('nick' or 'avatar')
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	104 @param profile: %(doc_profile)s
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	105 @return: wanted value or None"""
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	106 client = self.host.getClient(profile)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	107 try:
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	108 return client._vcard_cache[jid.userhost()][name]
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	109 except KeyError:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	110 return None
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	111
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	112
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	113 def save_photo(self, photo_xml):
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	114 """Parse a <PHOTO> elem and save the picture"""
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	115 for elem in photo_xml.elements():
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	116 if elem.name == 'TYPE':
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	117 info(_('Photo of type [%s] found') % str(elem))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	118 if elem.name == 'BINVAL':
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	119 debug(_('Decoding binary'))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	120 decoded = b64decode(str(elem))
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	121 hash = sha1(decoded).hexdigest()
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	122 filename = self.avatar_path+'/'+hash
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	123 if not os.path.exists(filename):
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	124 with open(filename,'wb') as file:
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	125 file.write(decoded)
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	126 debug(_("file saved to %s") % hash)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	127 else:
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	128 debug(_("file [%s] already in cache") % hash)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	129 return hash
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	130
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	131 @defer.deferredGenerator
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	132 def vCard2Dict(self, vcard, target, profile):
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	133 """Convert a VCard to a dict, and save binaries"""
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	134 debug (_("parsing vcard"))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	135 dictionary = {}
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	136 d = defer.Deferred()
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	137
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	138 for elem in vcard.elements():
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	139 if elem.name == 'FN':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	140 dictionary['fullname'] = unicode(elem)
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	141 elif elem.name == 'NICKNAME':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	142 dictionary['nick'] = unicode(elem)
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	143 self.update_cache(target, 'nick', dictionary['nick'], profile)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	144 elif elem.name == 'URL':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	145 dictionary['website'] = unicode(elem)
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	146 elif elem.name == 'EMAIL':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	147 dictionary['email'] = unicode(elem)
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	148 elif elem.name == 'BDAY':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	149 dictionary['birthday'] = unicode(elem)
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	150 elif elem.name == 'PHOTO':
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	151 d2 = defer.waitForDeferred(
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	152 threads.deferToThread(self.save_photo, elem))
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	153 yield d2
43 8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	154 dictionary["avatar"] = d2.getResult()
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	155 if not dictionary["avatar"]: #can happen in case of e.g. empty photo elem
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	156 del dictionary['avatar']
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	157 else:
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	158 self.update_cache(target, 'avatar', dictionary['avatar'], profile)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	159 else:
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	160 info (_('FIXME: [%s] VCard tag is not managed yet') % elem.name)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	161
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	162 yield dictionary
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	163
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	164 def vcard_ok(self, answer, profile):
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	165 """Called after the first get IQ"""
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	166 debug (_("VCard found"))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	167
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	168 if answer.firstChildElement().name == "vCard":
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	169 d = self.vCard2Dict(answer.firstChildElement(), jid.JID(answer["from"]), profile)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	170 d.addCallback(lambda data: self.host.bridge.actionResult("RESULT", answer['id'], data))
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	171 else:
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	172 error (_("FIXME: vCard not found as first child element"))
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	173 self.host.bridge.actionResult("SUPPRESS", answer['id'], {}) #FIXME: maybe an error message would be best
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	174
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	175 def vcard_err(self, failure):
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	176 """Called when something is wrong with registration"""
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	177 error (_("Can't find VCard of %s") % failure.value.stanza['from'])
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	178 self.host.bridge.actionResult("SUPPRESS", failure.value.stanza['id'], {}) #FIXME: maybe an error message would be best
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	179
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	180 def getCard(self, target, profile_key='@DEFAULT@'):
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	181 """Ask server for VCard
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	182 @param target: jid from which we want the VCard
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	183 @result: id to retrieve the profile"""
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	184 current_jid, xmlstream = self.host.getJidNStream(profile_key)
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	185 if not xmlstream:
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	186 error (_('Asking vcard for an non-existant or not connected profile'))
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	187 return ""
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	188 profile = self.host.memory.getProfileName(profile_key)
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	189 to_jid = jid.JID(target)
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	190 debug(_("Asking for %s's VCard") % to_jid.userhost())
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	191 reg_request=IQ(xmlstream,'get')
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	192 reg_request["from"]=current_jid.full()
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	193 reg_request["to"] = to_jid.userhost()
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	194 query=reg_request.addElement('vCard', NS_VCARD)
372 f964dcec1611 core: plugins refactored according to bridge + updatedValue now use profile Goffi <goffi@goffi.org> parents: 365 diff changeset	195 reg_request.send(to_jid.userhost()).addCallbacks(self.vcard_ok, self.vcard_err, [profile])
42 874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	196 return reg_request["id"]
874de3020e1c Initial VCard (XEP-0054) support + misc fixes Goffi <goffi@goffi.org> parents: diff changeset	197
43 8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	198 def getAvatarFile(self, hash):
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	199 """Give the full path of avatar from hash
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	200 @param hash: SHA1 hash
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	201 @return full_path
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	202 """
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	203 filename = self.avatar_path+'/'+hash
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	204 if not os.path.exists(filename):
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	205 error (_("Asking for an uncached avatar [%s]") % hash)
43 8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	206 return ""
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	207 return filename
8a438a6ff587 Wix: added avatar in profile Goffi <goffi@goffi.org> parents: 42 diff changeset	208
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	209 def getCardCache(self, target, profile_key):
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	210 """Request for cached values of profile
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	211 return the cached nickname and avatar if exists, else get VCard
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	212 @param target: target's jid
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	213 @param profile_key: %(doc_profile_key)s
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	214 """
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	215 profile = self.host.memory.getProfileName(profile_key)
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	216 if not profile:
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	217 error(_("Profile not found"))
c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	218 return {}
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	219 to_jid = jid.JID(target)
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	220 result = {}
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	221 nick = self.get_cache(to_jid, 'nick', profile)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	222 if nick:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	223 result['nick'] = nick
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	224 avatar = self.get_cache(to_jid, 'avatar', profile)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	225 if avatar:
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	226 result['avatar'] = avatar
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	227 return result
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	228
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	229
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	230
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	231 class XEP_0054_handler(XMPPHandler):
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	232 implements(iwokkel.IDisco)
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	233
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	234 def __init__(self, plugin_parent):
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	235 self.plugin_parent = plugin_parent
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	236 self.host = plugin_parent.host
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	237
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	238 def connectionInitialized(self):
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	239 self.parent._vcard_cache = PersistentBinaryDict(NS_VCARD, self.parent.profile)
437 02c633a0ddcf plugin XEP-0115: entity capabilities now use PersistentBinaryDict to store hash cache Goffi <goffi@goffi.org> parents: 435 diff changeset	240 self.parent._vcard_cache.load()
64 d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	241 self.xmlstream.addObserver(VCARD_UPDATE, self.update)
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	242
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	243 def getDiscoInfo(self, requestor, target, nodeIdentifier=''):
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	244 return [disco.DiscoFeature(NS_VCARD)]
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	245
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	246 def getDiscoItems(self, requestor, target, nodeIdentifier=''):
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	247 return []
d46f849664aa SàT: multi-profile, plugins updated Goffi <goffi@goffi.org> parents: 57 diff changeset	248
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	249 def update(self, presence):
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	250 """Request for VCard's nickname
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	251 return the cached nickname if exists, else get VCard
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	252 """
281 1e3e169955b2 typo Goffi <goffi@goffi.org> parents: 228 diff changeset	253 from_jid = jid.JID(presence['from'])
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	254 #FIXME: wokkel's data_form should be used here
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	255 x_elem = filter (lambda x:x.name == "x", presence.elements())[0] #We only want the "x" element
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	256 for elem in x_elem.elements():
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	257 if elem.name == 'photo':
4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	258 hash = str(elem)
435 c243f4cb2ad9 plugin XEP-0054: cache now use storage Goffi <goffi@goffi.org> parents: 372 diff changeset	259 old_avatar = self.plugin_parent.get_cache(from_jid, 'avatar', self.parent.profile)
48 4392f1fdb064 plugins improvement Goffi <goffi@goffi.org> parents: 44 diff changeset	260 if not old_avatar or old_avatar != hash:
69 86f1f7f6d332 i18n first draft Goffi <goffi@goffi.org> parents: 66 diff changeset	261 debug(_('New avatar found, requesting vcard'))
281 1e3e169955b2 typo Goffi <goffi@goffi.org> parents: 228 diff changeset	262 self.plugin_parent.getCard(from_jid.userhost(), self.parent.profile)

Mercurial > libervia-backend

annotate src/plugins/plugin_xep_0054.py @ 447:485a6d125498