tools/xep227toprosody.lua

   1 #!/usr/bin/env lua
   2 -- Prosody IM
   3 -- Copyright (C) 2008-2009 Matthew Wild
   4 -- Copyright (C) 2008-2009 Waqas Hussain
   5 -- Copyright (C) 2010      Stefan Gehn
   6 --
   7 -- This project is MIT/X11 licensed. Please see the
   8 -- COPYING file in the source package for more information.
   9 --
  10
  11 -- FIXME: XEP-0227 supports XInclude but luaexpat does not
  12 --
  13 -- XEP-227 elements and their current level of support:
  14 -- Hosts : supported
  15 -- Users : supported
  16 -- Rosters : supported, needs testing
  17 -- Offline Messages : supported, needs testing
  18 -- Private XML Storage : supported, needs testing
  19 -- vCards : supported, needs testing
  20 -- Privacy Lists: UNSUPPORTED
  21 --   http://xmpp.org/extensions/xep-0227.html#privacy-lists
  22 --   mod_privacy uses dm.load(username, host, "privacy"); and stores stanzas 1:1
  23 -- Incoming Subscription Requests : supported
  24
  25 package.path = package.path..";../?.lua";
  26 package.cpath = package.cpath..";../?.so"; -- needed for util.pposix used in datamanager
  27
  28 local my_name = arg[0];
  29 if my_name:match("[/\\]") then
  30         package.path = package.path..";"..my_name:gsub("[^/\\]+$", "../?.lua");
  31         package.cpath = package.cpath..";"..my_name:gsub("[^/\\]+$", "../?.so");
  32 end
  33
  34 -- ugly workaround for getting datamanager to work outside of prosody :(
  35 prosody = { };
  36 prosody.platform = "unknown";
  37 if os.getenv("WINDIR") then
  38         prosody.platform = "windows";
  39 elseif package.config:sub(1,1) == "/" then
  40         prosody.platform = "posix";
  41 end
  42
  43 local lxp = require "lxp";
  44 local st = require "util.stanza";
  45 local xmppstream = require "util.xmppstream";
  46 local new_xmpp_handlers = xmppstream.new_sax_handlers;
  47 local dm = require "util.datamanager"
  48 dm.set_data_path("data");
  49
  50 local ns_separator = xmppstream.ns_separator;
  51 local ns_pattern = xmppstream.ns_pattern;
  52
  53 local xmlns_xep227 = "http://www.xmpp.org/extensions/xep-0227.html#ns";
  54
  55 -----------------------------------------------------------------------
  56
  57 function store_vcard(username, host, stanza)
  58         -- create or update vCard for username@host
  59         local ret, err = dm.store(username, host, "vcard", st.preserialize(stanza));
  60         print("["..(err or "success").."] stored vCard: "..username.."@"..host);
  61 end
  62
  63 function store_password(username, host, password)
  64         -- create or update account for username@host
  65         local ret, err = dm.store(username, host, "accounts", {password = password});
  66         print("["..(err or "success").."] stored account: "..username.."@"..host.." = "..password);
  67 end
  68
  69 function store_roster(username, host, roster_items)
  70         -- fetch current roster-table for username@host if he already has one
  71         local roster = dm.load(username, host, "roster") or {};
  72         -- merge imported roster-items with loaded roster
  73         for item_tag in roster_items:childtags("item") do
  74                 -- jid for this roster-item
  75                 local item_jid = item_tag.attr.jid
  76                 -- validate item stanzas
  77                 if (item_jid ~= "") then
  78                         -- prepare roster item
  79                         -- TODO: is the subscription attribute optional?
  80                         local item = {subscription = item_tag.attr.subscription, groups = {}};
  81                         -- optional: give roster item a real name
  82                         if item_tag.attr.name then
  83                                 item.name = item_tag.attr.name;
  84                         end
  85                         -- optional: iterate over group stanzas inside item stanza
  86                         for group_tag in item_tag:childtags("group") do
  87                                 local group_name = group_tag:get_text();
  88                                 if (group_name ~= "") then
  89                                         item.groups[group_name] = true;
  90                                 else
  91                                         print("[error] invalid group stanza: "..group_tag:pretty_print());
  92                                 end
  93                         end
  94                         -- store item in roster
  95                         roster[item_jid] = item;
  96                         print("[success] roster entry: " ..username.."@"..host.." - "..item_jid);
  97                 else
  98                         print("[error] invalid roster stanza: " ..item_tag:pretty_print());
  99                 end
 100
 101         end
 102         -- store merged roster-table
 103         local ret, err = dm.store(username, host, "roster", roster);
 104         print("["..(err or "success").."] stored roster: " ..username.."@"..host);
 105 end
 106
 107 function store_private(username, host, private_items)
 108         local private = dm.load(username, host, "private") or {};
 109         for _, ch in ipairs(private_items.tags) do
 110                 --print("private :"..ch:pretty_print());
 111                 private[ch.name..":"..ch.attr.xmlns] = st.preserialize(ch);
 112                 print("[success] private item: " ..username.."@"..host.." - "..ch.name);
 113         end
 114         local ret, err = dm.store(username, host, "private", private);
 115         print("["..(err or "success").."] stored private: " ..username.."@"..host);
 116 end
 117
 118 function store_offline_messages(username, host, offline_messages)
 119         -- TODO: maybe use list_load(), append and list_store() instead
 120         --       of constantly reopening the file with list_append()?
 121         for ch in offline_messages:childtags("message", "jabber:client") do
 122                 --print("message :"..ch:pretty_print());
 123                 local ret, err = dm.list_append(username, host, "offline", st.preserialize(ch));
 124                 print("["..(err or "success").."] stored offline message: " ..username.."@"..host.." - "..ch.attr.from);
 125         end
 126 end
 127
 128
 129 function store_subscription_request(username, host, presence_stanza)
 130         local from_bare = presence_stanza.attr.from;
 131
 132         -- fetch current roster-table for username@host if he already has one
 133         local roster = dm.load(username, host, "roster") or {};
 134
 135         local item = roster[from_bare];
 136         if item and (item.subscription == "from" or item.subscription == "both") then
 137                 return; -- already subscribed, do nothing
 138         end
 139
 140         -- add to table of pending subscriptions
 141         if not roster.pending then roster.pending = {}; end
 142         roster.pending[from_bare] = true;
 143
 144         -- store updated roster-table
 145         local ret, err = dm.store(username, host, "roster", roster);
 146         print("["..(err or "success").."] stored subscription request: " ..username.."@"..host.." - "..from_bare);
 147 end
 148
 149 -----------------------------------------------------------------------
 150
 151 local curr_host = "";
 152 local user_name = "";
 153
 154
 155 local cb = {
 156         stream_tag = "user",
 157         stream_ns = xmlns_xep227,
 158 };
 159 function cb.streamopened(session, attr)
 160         session.notopen = false;
 161         user_name = attr.name;
 162         store_password(user_name, curr_host, attr.password);
 163 end
 164 function cb.streamclosed(session)
 165         session.notopen = true;
 166         user_name = "";
 167 end
 168 function cb.handlestanza(session, stanza)
 169         --print("Parsed stanza "..stanza.name.." xmlns: "..(stanza.attr.xmlns or ""));
 170         if (stanza.name == "vCard") and (stanza.attr.xmlns == "vcard-temp") then
 171                 store_vcard(user_name, curr_host, stanza);
 172         elseif (stanza.name == "query") then
 173                 if (stanza.attr.xmlns == "jabber:iq:roster") then
 174                         store_roster(user_name, curr_host, stanza);
 175                 elseif (stanza.attr.xmlns == "jabber:iq:private") then
 176                         store_private(user_name, curr_host, stanza);
 177                 end
 178         elseif (stanza.name == "offline-messages") then
 179                 store_offline_messages(user_name, curr_host, stanza);
 180         elseif (stanza.name == "presence") and (stanza.attr.xmlns == "jabber:client") then
 181                 store_subscription_request(user_name, curr_host, stanza);
 182         else
 183                 print("UNHANDLED stanza "..stanza.name.." xmlns: "..(stanza.attr.xmlns or ""));
 184         end
 185 end
 186
 187 local user_handlers = new_xmpp_handlers({ notopen = true }, cb);
 188
 189 -----------------------------------------------------------------------
 190
 191 local lxp_handlers = {
 192         --count = 0
 193 };
 194
 195 -- TODO: error handling for invalid opening elements if curr_host is empty
 196 function lxp_handlers.StartElement(parser, elementname, attributes)
 197         local curr_ns, name = elementname:match(ns_pattern);
 198         if name == "" then
 199                 curr_ns, name = "", curr_ns;
 200         end
 201         --io.write("+ ", string.rep(" ", count), name, "  (", curr_ns, ")", "\n")
 202         --count = count + 1;
 203         if curr_host ~= "" then
 204                 -- forward to xmlhandlers
 205                 user_handlers.StartElement(parser, elementname, attributes);
 206         elseif (curr_ns == xmlns_xep227) and (name == "host") then
 207                 curr_host = attributes["jid"]; -- start of host element
 208                 print("Begin parsing host "..curr_host);
 209         elseif (curr_ns ~= xmlns_xep227) or (name ~= "server-data") then
 210                 io.stderr:write("Unhandled XML element: ", name, "\n");
 211                 os.exit(1);
 212         end
 213 end
 214
 215 -- TODO: error handling for invalid closing elements if host is empty
 216 function lxp_handlers.EndElement(parser, elementname)
 217         local curr_ns, name = elementname:match(ns_pattern);
 218         if name == "" then
 219                 curr_ns, name = "", curr_ns;
 220         end
 221         --count = count - 1;
 222         --io.write("- ", string.rep(" ", count), name, "  (", curr_ns, ")", "\n")
 223         if curr_host ~= "" then
 224                 if (curr_ns == xmlns_xep227) and (name == "host") then
 225                         print("End parsing host "..curr_host);
 226                         curr_host = "" -- end of host element
 227                 else
 228                         -- forward to xmlhandlers
 229                         user_handlers.EndElement(parser, elementname);
 230                 end
 231         elseif (curr_ns ~= xmlns_xep227) or (name ~= "server-data") then
 232                 io.stderr:write("Unhandled XML element: ", name, "\n");
 233                 os.exit(1);
 234         end
 235 end
 236
 237 function lxp_handlers.CharacterData(parser, string)
 238         if curr_host ~= "" then
 239                 -- forward to xmlhandlers
 240                 user_handlers.CharacterData(parser, string);
 241         end
 242 end
 243
 244 -----------------------------------------------------------------------
 245
 246 local arg = ...;
 247 local help = "/? -? ? /h -h /help -help --help";
 248 if not arg or help:find(arg, 1, true) then
 249         print([[XEP-227 importer for Prosody
 250
 251   Usage: xep227toprosody.lua filename.xml
 252
 253 ]]);
 254         os.exit(1);
 255 end
 256
 257 local file = io.open(arg);
 258 if not file then
 259         io.stderr:write("Could not open file: ", arg, "\n");
 260         os.exit(0);
 261 end
 262
 263 local parser = lxp.new(lxp_handlers, ns_separator);
 264 for l in file:lines() do
 265         parser:parse(l);
 266 end
 267 parser:parse();
 268 parser:close();
 269 file:close();