plugins/mod_vcard.lua

   1
   2 require "util.datamanager"
   3 local datamanager = datamanager;
   4
   5 local st = require "util.stanza"
   6 local send = require "core.sessionmanager".send_to_session
   7 local t_concat, t_insert = table.concat, table.insert;
   8
   9 require "util.jid"
  10 local jid_split = jid.split;
  11
  12 add_iq_handler("c2s", "vcard-temp",
  13                 function (session, stanza)
  14                         if stanza.tags[1].name == "vCard" then
  15                                 local to = stanza.attr.to;
  16                                 if stanza.attr.type == "get" then
  17                                         local vCard;
  18                                         if to then
  19                                                 local node, host = jid_split(to);
  20                                                 if hosts[host] and hosts[host].type == "local" then
  21                                                         vCard = st.deserialize(datamanager.load(node, host, "vCard")); -- load vCard for user or server
  22                                                 end
  23                                         else
  24                                                 vCard = st.deserialize(datamanager.load(session.username, session.host, "vCard"));-- load user's own vCard
  25                                         end
  26                                         if vCard then
  27                                                 local iq = st.reply(stanza);
  28                                                 iq:add_child(vCard);
  29                                                 send(session, iq); -- send vCard!
  30                                         else
  31                                                 send(session, st.error_reply(stanza, "cancel", "item-not-found"));
  32                                         end
  33                                 elseif stanza.attr.type == "set" then
  34                                         if not to or to == session.username.."@"..session.host then
  35                                                 if datamanager.store(session.username, session.host, "vCard", st.preserialize(stanza.tags[1])) then
  36                                                         send(session, st.reply(stanza));
  37                                                 else
  38                                                         -- TODO unable to write file, file may be locked, etc, what's the correct error?
  39                                                         send(session, st.error_reply(stanza, "wait", "internal-server-error"));
  40                                                 end
  41                                         else
  42                                                 send(session, st.error_reply(stanza, "auth", "forbidden"));
  43                                         end
  44                                 end
  45                                 return true;
  46                         end
  47                 end);
  48
  49 add_event_hook("stream-features",
  50                                         function (session, features)
  51                                                 if session.type == "c2s" then
  52                                                         t_insert(features, "<feature var='vcard-temp'/>");
  53                                                 end
  54                                         end);