plugins/mod_vcard.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10
  11 local hosts = _G.hosts;
  12 local datamanager = require "util.datamanager"
  13
  14 local st = require "util.stanza"
  15 local t_concat, t_insert = table.concat, table.insert;
  16
  17 local jid = require "util.jid"
  18 local jid_split = jid.split;
  19
  20 local xmlns_vcard = "vcard-temp";
  21
  22 module:add_feature(xmlns_vcard);
  23
  24 function handle_vcard(event)
  25         local session, stanza = event.origin, event.stanza;
  26         if stanza.tags[1].name == "vCard" then
  27                 local to = stanza.attr.to;
  28                 if stanza.attr.type == "get" then
  29                         local vCard;
  30                         if to then
  31                                 local node, host = jid_split(to);
  32                                 if hosts[host] and hosts[host].type == "local" then
  33                                         vCard = st.deserialize(datamanager.load(node, host, "vcard")); -- load vCard for user or server
  34                                 end
  35                         else
  36                                 vCard = st.deserialize(datamanager.load(session.username, session.host, "vcard"));-- load user's own vCard
  37                         end
  38                         if vCard then
  39                                 session.send(st.reply(stanza):add_child(vCard)); -- send vCard!
  40                         else
  41                                 session.send(st.error_reply(stanza, "cancel", "item-not-found"));
  42                         end
  43                 elseif stanza.attr.type == "set" then
  44                         if not to then
  45                                 if datamanager.store(session.username, session.host, "vcard", st.preserialize(stanza.tags[1])) then
  46                                         session.send(st.reply(stanza));
  47                                 else
  48                                         -- TODO unable to write file, file may be locked, etc, what's the correct error?
  49                                         session.send(st.error_reply(stanza, "wait", "internal-server-error"));
  50                                 end
  51                         else
  52                                 session.send(st.error_reply(stanza, "auth", "forbidden"));
  53                         end
  54                 end
  55                 return true;
  56         end
  57 end
  58
  59 --module:add_iq_handler({"c2s", "s2sin", "component"}, xmlns_vcard, handle_vcard);
  60 module:hook("iq/bare/vcard-temp:vCard", handle_vcard);
  61 module:hook("iq/host/vcard-temp:vCard", handle_vcard);
  62
  63 -- COMPAT: https://support.process-one.net/browse/EJAB-1045
  64 if module:get_option("vcard_compatibility") then
  65         module:hook("iq/full", function (data)
  66                 local stanza = data.stanza;
  67                 if stanza.attr.type == "get" and stanza.tags[1]
  68                         and stanza.tags[1].attr.xmlns == xmlns_vcard then
  69                                 return handle_vcard(data);
  70                 end
  71         end, 1);
  72 end
  73
  74 local feature_vcard_attr = { var=xmlns_vcard };
  75 module:add_event_hook("stream-features",
  76                                         function (session, features)
  77                                                 if session.type == "c2s" then
  78                                                         features:tag("feature", feature_vcard_attr):up();
  79                                                 end
  80                                         end);