Mercurial > prosody-modules
changeset 1419:877dc87539ff
mod_profile: Replacement for mod_vcard with vcard4 support and integration with mod_pep_plus
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Wed, 28 May 2014 22:02:27 +0200 (2014-05-28) |
parents | a0375f84d65a |
children | 808950ab007b |
files | mod_profile/mod_profile.lua |
diffstat | 1 files changed, 151 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_profile/mod_profile.lua Wed May 28 22:02:27 2014 +0200 @@ -0,0 +1,151 @@ +-- mod_profile + +local st = require"util.stanza"; +local jid_split, jid_bare = import("util.jid", "split", "bare"); +local is_admin = require"core.usermanager".is_admin; +local vcard = require"util.vcard"; +local base64 = require"util.encodings".base64; +local sha1 = require"util.hashes".sha1; + +local pep_plus; +if module:get_host_type() == "local" and module:get_option_boolean("vcard_to_pep") then + pep_plus = module:depends"pep_plus"; +end + +local storage = module:open_store(); +local legacy_storage = module:open_store("vcard"); + +local function get_item(vcard, name) + local item; + for i=1, #vcard do + item=vcard[i]; + if item.name == name then + return item, i; + end + end +end + +local magic_mime = { + ["\137PNG\r\n\026\n"] = "image/png"; + ["\255\216"] = "image/jpeg"; + ["GIF87a"] = "image/gif"; + ["GIF89a"] = "image/gif"; + ["<?xml"] = "image/svg+xml"; +} +local function identify(data) + for magic, mime in pairs(magic_mime) do + if data:sub(1, #magic) == magic then + return mime; + end + end + return "application/octet-stream"; +end + +local function update_pep(username, data) + local pep = pep_plus.get_pep_service(username.."@"..module.host); + if vcard.to_vcard4 then + pep:publish("urn:xmpp:vcard4", true, "", st.stanza("item"):add_child(vcard.to_vcard4(data))); + end + + local nickname = get_item(data, "NICKNAME"); + if nickname and nickname[1] then + pep:publish("http://jabber.org/protocol/nick", true, "", st.stanza("item") + :tag("nick", { xmlns="http://jabber.org/protocol/nick" }):text(nickname[1])); + end + + local photo = get_item(data, "PHOTO"); + if photo and photo[1] then + local photo_raw = base64.decode(photo[1]); + local photo_hash = sha1(photo_raw, true); + + pep:publish("urn:xmpp:avatar:metadata", true, "", st.stanza("item") + :tag("metadata", { + xmlns="urn:xmpp:avatar:metadata", + bytes = tostring(#photo_raw), + id = photo_hash, + type = identify(photo_raw), + })); + pep:publish("urn:xmpp:avatar:data", true, photo_hash, st.stanza("item") + :tag("data", { xmlns="urn:xmpp:avatar:data" }):text(photo[1])); + end +end + +-- The "temporary" vCard XEP-0054 part +module:add_feature("vcard-temp"); + +local function handle_get(event) + local origin, stanza = event.origin, event.stanza; + local username = origin.username; + local to = stanza.attr.to; + if to then username = jid_split(to); end + local data = storage:get(username); + if not data then + data = legacy_storage:get(username); + data = data and st.deserialize(data); + end + if not data then + return origin.send(st.error_reply(stanza, "cancel", "item-not-found")); + end + return origin.send(st.reply(stanza):add_child(vcard.to_xep54(data))); +end + +local function handle_set(event) + local origin, stanza = event.origin, event.stanza; + local data = vcard.from_xep54(stanza.tags[1]); + local username = origin.username; + local to = stanza.attr.to; + if to then + if not is_admin(jid_bare(stanza.attr.from), module.host) then + return origin.send(st.error_reply(stanza, "auth", "forbidden")); + end + username = jid_split(to); + end + local ok, err = storage:set(username, data); + if not ok then + return origin.send(st.error_reply(stanza, "cancel", "internal-server-error", err)); + end + + if pep_plus and username then + update_pep(username, data); + end + + return origin.send(st.reply(stanza)); +end + +module:hook("iq-get/bare/vcard-temp:vCard", handle_get); +module:hook("iq-get/host/vcard-temp:vCard", handle_get); + +module:hook("iq-set/bare/vcard-temp:vCard", handle_set); +module:hook("iq-set/host/vcard-temp:vCard", handle_set); + +-- The vCard4 part +if vcard.to_vcard4 then + module:add_feature("urn:ietf:params:xml:ns:vcard-4.0"); + + module:hook("iq-get/bare/urn:ietf:params:xml:ns:vcard-4.0:vcard", function(event) + local origin, stanza = event.origin, event.stanza; + local username = jid_split(stanza.attr.to) or origin.username; + local data = storage:get(username); + if not data then + return origin.send(st.error_reply(stanza, "cancel", "item-not-found")); + end + return origin.send(st.reply(stanza):add_child(vcard.to_vcard4(data))); + end); + + if vcard.from_vcard4 then + module:hook("iq-set/self/urn:ietf:params:xml:ns:vcard-4.0:vcard", function(event) + local origin, stanza = event.origin, event.stanza; + local ok, err = storage:set(origin.username, vcard.from_vcard4(stanza.tags[1])); + if not ok then + return origin.send(st.error_reply(stanza, "cancel", "internal-server-error", err)); + end + return origin.send(st.reply(stanza)); + end); + else + module:hook("iq-set/self/urn:ietf:params:xml:ns:vcard-4.0:vcard", function(event) + local origin, stanza = event.origin, event.stanza; + return origin.send(st.error_reply(stanza, "cancel", "feature-not-implemented")); + end); + end +end +