plugins/mod_vcard_legacy.lua

   1 local st = require "util.stanza"
   2 local jid_split = require "util.jid".split;
   3
   4 local mod_pep = module:depends("pep");
   5
   6 local sha1 = require "util.hashes".sha1;
   7 local base64_decode = require "util.encodings".base64.decode;
   8
   9 local vcards = module:open_store("vcard");
  10
  11 module:add_feature("vcard-temp");
  12 module:hook("account-disco-info", function (event)
  13         event.reply:tag("feature", { var = "urn:xmpp:pep-vcard-conversion:0" }):up();
  14 end);
  15
  16 local function handle_error(origin, stanza, err)
  17         if err == "forbidden" then
  18                 origin.send(st.error_reply(stanza, "auth", "forbidden"));
  19         elseif err == "internal-server-error" then
  20                 origin.send(st.error_reply(stanza, "wait", "internal-server-error"));
  21         else
  22                 origin.send(st.error_reply(stanza, "modify", "undefined-condition", err));
  23         end
  24 end
  25
  26 -- Simple translations
  27 -- <foo><text>hey</text></foo> -> <FOO>hey</FOO>
  28 local simple_map = {
  29         nickname = "text";
  30         title = "text";
  31         role = "text";
  32         categories = "text";
  33         note = "text";
  34         url = "uri";
  35         bday = "date";
  36 }
  37
  38 module:hook("iq-get/bare/vcard-temp:vCard", function (event)
  39         local origin, stanza = event.origin, event.stanza;
  40         local pep_service = mod_pep.get_pep_service(jid_split(stanza.attr.to) or origin.username);
  41         local ok, id, vcard4_item = pep_service:get_last_item("urn:xmpp:vcard4", stanza.attr.from);
  42
  43         local vcard_temp = st.stanza("vCard", { xmlns = "vcard-temp" });
  44         if ok and vcard4_item then
  45                 local vcard4 = vcard4_item.tags[1];
  46
  47                 local fn = vcard4:get_child("fn");
  48                 vcard_temp:text_tag("FN", fn and fn:get_child_text("text"));
  49
  50                 local v4n = vcard4:get_child("n");
  51                 vcard_temp:tag("N")
  52                         :text_tag("FAMILY", v4n and v4n:get_child_text("surname"))
  53                         :text_tag("GIVEN", v4n and v4n:get_child_text("given"))
  54                         :text_tag("MIDDLE", v4n and v4n:get_child_text("additional"))
  55                         :text_tag("PREFIX", v4n and v4n:get_child_text("prefix"))
  56                         :text_tag("SUFFIX", v4n and v4n:get_child_text("suffix"))
  57                         :up();
  58
  59                 for tag in vcard4:childtags() do
  60                         local typ = simple_map[tag.name];
  61                         if typ then
  62                                 local text = tag:get_child_text(typ);
  63                                 if text then
  64                                         vcard_temp:text_tag(tag.name:upper(), text);
  65                                 end
  66                         elseif tag.name == "email" then
  67                                 local text = tag:get_child_text("text");
  68                                 if text then
  69                                         vcard_temp:tag("EMAIL")
  70                                                 :text_tag("USERID", text)
  71                                                 :tag("INTERNET"):up();
  72                                         if tag:find"parameters/type/text#" == "home" then
  73                                                 vcard_temp:tag("HOME"):up();
  74                                         elseif tag:find"parameters/type/text#" == "work" then
  75                                                 vcard_temp:tag("WORK"):up();
  76                                         end
  77                                         vcard_temp:up();
  78                                 end
  79                         elseif tag.name == "tel" then
  80                                 local text = tag:get_child_text("uri");
  81                                 if text then
  82                                         if text:sub(1, 4) == "tel:" then
  83                                                 text = text:sub(5)
  84                                         end
  85                                         vcard_temp:tag("TEL"):text_tag("NUMBER", text);
  86                                         if tag:find"parameters/type/text#" == "home" then
  87                                                 vcard_temp:tag("HOME"):up();
  88                                         elseif tag:find"parameters/type/text#" == "work" then
  89                                                 vcard_temp:tag("WORK"):up();
  90                                         end
  91                                         vcard_temp:up();
  92                                 end
  93                         elseif tag.name == "adr" then
  94                                 vcard_temp:tag("ADR")
  95                                         :text_tag("POBOX", tag:get_child_text("pobox"))
  96                                         :text_tag("EXTADD", tag:get_child_text("ext"))
  97                                         :text_tag("STREET", tag:get_child_text("street"))
  98                                         :text_tag("LOCALITY", tag:get_child_text("locality"))
  99                                         :text_tag("REGION", tag:get_child_text("region"))
 100                                         :text_tag("PCODE", tag:get_child_text("code"))
 101                                         :text_tag("CTRY", tag:get_child_text("country"));
 102                                 if tag:find"parameters/type/text#" == "home" then
 103                                         vcard_temp:tag("HOME"):up();
 104                                 elseif tag:find"parameters/type/text#" == "work" then
 105                                         vcard_temp:tag("WORK"):up();
 106                                 end
 107                                 vcard_temp:up();
 108                         end
 109                 end
 110         end
 111
 112         local meta_ok, avatar_meta = pep_service:get_items("urn:xmpp:avatar:metadata", stanza.attr.from);
 113         local data_ok, avatar_data = pep_service:get_items("urn:xmpp:avatar:data", stanza.attr.from);
 114
 115         if data_ok then
 116                 for _, hash in ipairs(avatar_data) do
 117                         local meta = meta_ok and avatar_meta[hash];
 118                         local data = avatar_data[hash];
 119                         local info = meta and meta.tags[1]:get_child("info");
 120                         vcard_temp:tag("PHOTO");
 121                         if info and info.attr.type then
 122                                 vcard_temp:text_tag("TYPE", info.attr.type);
 123                         end
 124                         if data then
 125                                 vcard_temp:text_tag("BINVAL", data.tags[1]:get_text());
 126                         elseif info and info.attr.url then
 127                                 vcard_temp:text_tag("EXTVAL", info.attr.url);
 128                         end
 129                         vcard_temp:up();
 130                 end
 131         end
 132
 133         if not vcard_temp.tags[1] then
 134                 vcard_temp = st.deserialize(vcards:get(jid_split(stanza.attr.to) or origin.username)) or vcard_temp;
 135         end
 136
 137         origin.send(st.reply(stanza):add_child(vcard_temp));
 138         return true;
 139 end);
 140
 141 local node_defaults = {
 142         access_model = "open";
 143         _defaults_only = true;
 144 };
 145
 146 module:hook("iq-set/self/vcard-temp:vCard", function (event)
 147         local origin, stanza = event.origin, event.stanza;
 148         local pep_service = mod_pep.get_pep_service(origin.username);
 149
 150         local vcard_temp = stanza.tags[1];
 151
 152         local vcard4 = st.stanza("item", { xmlns = "http://jabber.org/protocol/pubsub", id = "current" })
 153                 :tag("vcard", { xmlns = 'urn:ietf:params:xml:ns:vcard-4.0' });
 154
 155         if pep_service:purge("urn:xmpp:avatar:metadata", origin.full_jid) then
 156                 pep_service:purge("urn:xmpp:avatar:data", origin.full_jid);
 157         end
 158
 159         vcard4:tag("fn"):text_tag("text", vcard_temp:get_child_text("FN")):up();
 160
 161         local N = vcard_temp:get_child("N");
 162
 163         vcard4:tag("n")
 164                 :text_tag("surname", N and N:get_child_text("FAMILY"))
 165                 :text_tag("given", N and N:get_child_text("GIVEN"))
 166                 :text_tag("additional", N and N:get_child_text("MIDDLE"))
 167                 :text_tag("prefix", N and N:get_child_text("PREFIX"))
 168                 :text_tag("suffix", N and N:get_child_text("SUFFIX"))
 169         :up();
 170
 171         for tag in vcard_temp:childtags() do
 172                 local typ = simple_map[tag.name:lower()];
 173                 if typ then
 174                         local text = tag:get_text();
 175                         if text then
 176                                 vcard4:tag(tag.name:lower()):text_tag(typ, text):up();
 177                         end
 178                 elseif tag.name == "EMAIL" then
 179                         local text = tag:get_child_text("USERID");
 180                         if text then
 181                                 vcard4:tag("email")
 182                                 vcard4:text_tag("text", text)
 183                                 vcard4:tag("parameters"):tag("type");
 184                                 if tag:get_child("HOME") then
 185                                         vcard4:text_tag("text", "home");
 186                                 elseif tag:get_child("WORK") then
 187                                         vcard4:text_tag("text", "work");
 188                                 end
 189                                 vcard4:up():up():up();
 190                         end
 191                 elseif tag.name == "TEL" then
 192                         local text = tag:get_child_text("NUMBER");
 193                         if text then
 194                                 vcard4:tag("tel"):text_tag("uri", "tel:"..text);
 195                         end
 196                         vcard4:tag("parameters"):tag("type");
 197                         if tag:get_child("HOME") then
 198                                 vcard4:text_tag("text", "home");
 199                         elseif tag:get_child("WORK") then
 200                                 vcard4:text_tag("text", "work");
 201                         end
 202                         vcard4:up():up():up();
 203                 elseif tag.name == "ORG" then
 204                         local text = tag:get_child_text("ORGNAME");
 205                         if text then
 206                                 vcard4:tag("org"):text_tag("text", text):up();
 207                         end
 208                 elseif tag.name == "DESC" then
 209                         local text = tag:get_text();
 210                         if text then
 211                                 vcard4:tag("note"):text_tag("text", text):up();
 212                         end
 213                         -- <note> gets mapped into <NOTE> in the other direction
 214                 elseif tag.name == "ADR" then
 215                         vcard4:tag("adr")
 216                                 :text_tag("pobox", tag:get_child_text("POBOX"))
 217                                 :text_tag("ext", tag:get_child_text("EXTADD"))
 218                                 :text_tag("street", tag:get_child_text("STREET"))
 219                                 :text_tag("locality", tag:get_child_text("LOCALITY"))
 220                                 :text_tag("region", tag:get_child_text("REGION"))
 221                                 :text_tag("code", tag:get_child_text("PCODE"))
 222                                 :text_tag("country", tag:get_child_text("CTRY"));
 223                         vcard4:tag("parameters"):tag("type");
 224                         if tag:get_child("HOME") then
 225                                 vcard4:text_tag("text", "home");
 226                         elseif tag:get_child("WORK") then
 227                                 vcard4:text_tag("text", "work");
 228                         end
 229                         vcard4:up():up():up();
 230                 elseif tag.name == "PHOTO" then
 231                         local avatar_type = tag:get_child_text("TYPE");
 232                         local avatar_payload = tag:get_child_text("BINVAL");
 233                         -- Can EXTVAL be translated? No way to know the sha1 of the data?
 234
 235                         if avatar_payload then
 236                                 local avatar_raw = base64_decode(avatar_payload);
 237                                 local avatar_hash = sha1(avatar_raw, true);
 238
 239                                 local avatar_meta = st.stanza("item", { id = avatar_hash, xmlns = "http://jabber.org/protocol/pubsub" })
 240                                         :tag("metadata", { xmlns="urn:xmpp:avatar:metadata" })
 241                                                 :tag("info", {
 242                                                         bytes = tostring(#avatar_raw),
 243                                                         id = avatar_hash,
 244                                                         type = avatar_type,
 245                                                 });
 246
 247                                 local avatar_data = st.stanza("item", { id = avatar_hash, xmlns = "http://jabber.org/protocol/pubsub" })
 248                                         :tag("data", { xmlns="urn:xmpp:avatar:data" })
 249                                                 :text(avatar_payload);
 250
 251                                 local ok, err = pep_service:publish("urn:xmpp:avatar:data", origin.full_jid, avatar_hash, avatar_data, node_defaults)
 252                                 if ok then
 253                                         ok, err = pep_service:publish("urn:xmpp:avatar:metadata", origin.full_jid, avatar_hash, avatar_meta, node_defaults);
 254                                 end
 255                                 if not ok then
 256                                         handle_error(origin, stanza, err);
 257                                         return true;
 258                                 end
 259                         end
 260                 end
 261         end
 262
 263         local ok, err = pep_service:publish("urn:xmpp:vcard4", origin.full_jid, "current", vcard4, node_defaults);
 264         if ok then
 265                 origin.send(st.reply(stanza));
 266         else
 267                 handle_error(origin, stanza, err);
 268         end
 269
 270         return true;
 271 end);
 272
 273 local function inject_xep153(event)
 274         local origin, stanza = event.origin, event.stanza;
 275         local username = origin.username;
 276         if not username then return end
 277         if stanza.attr.type then return end
 278         local pep_service = mod_pep.get_pep_service(username);
 279
 280         stanza:remove_children("x", "vcard-temp:x:update");
 281         local x_update = st.stanza("x", { xmlns = "vcard-temp:x:update" });
 282         local ok, avatar_hash = pep_service:get_last_item("urn:xmpp:avatar:metadata", true);
 283         if ok and avatar_hash then
 284                 x_update:text_tag("photo", avatar_hash);
 285         end
 286         stanza:add_direct_child(x_update);
 287 end
 288
 289 module:hook("pre-presence/full", inject_xep153, 1);
 290 module:hook("pre-presence/bare", inject_xep153, 1);
 291 module:hook("pre-presence/host", inject_xep153, 1);