tools/openfire2prosody: Initial commit.

diff --git a/tools/openfire2prosody.lua b/tools/openfire2prosody.lua
new file mode 100644 (file)
index 0000000..49d0dc5
--- /dev/null
+++ b/tools/openfire2prosody.lua
@@ -0,0 +1,153 @@
+#!/usr/bin/env lua
+-- Prosody IM
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+package.path = package.path..";../?.lua";
+package.cpath = package.cpath..";../?.so"; -- needed for util.pposix used in datamanager
+
+-- ugly workaround for getting datamanager to work outside of prosody :(
+prosody = { };
+prosody.platform = "unknown";
+if os.getenv("WINDIR") then
+       prosody.platform = "windows";
+elseif package.config:sub(1,1) == "/" then
+       prosody.platform = "posix";
+end
+
+local lxp = require "lxp";
+local st = require "util.stanza";
+
+local parse_xml = (function()
+       local ns_prefixes = {
+               ["http://www.w3.org/XML/1998/namespace"] = "xml";
+       };
+       local ns_separator = "\1";
+       local ns_pattern = "^([^"..ns_separator.."]*)"..ns_separator.."?(.*)$";
+       return function(xml)
+               local handler = {};
+               local stanza = st.stanza("root");
+               function handler:StartElement(tagname, attr)
+                       local curr_ns,name = tagname:match(ns_pattern);
+                       if name == "" then
+                               curr_ns, name = "", curr_ns;
+                       end
+                       if curr_ns ~= "" then
+                               attr.xmlns = curr_ns;
+                       end
+                       for i=1,#attr do
+                               local k = attr[i];
+                               attr[i] = nil;
+                               local ns, nm = k:match(ns_pattern);
+                               if nm ~= "" then
+                                       ns = ns_prefixes[ns]; 
+                                       if ns then 
+                                               attr[ns..":"..nm] = attr[k];
+                                               attr[k] = nil;
+                                       end
+                               end
+                       end
+                       stanza:tag(name, attr);
+               end
+               function handler:CharacterData(data)
+                       stanza:text(data);
+               end
+               function handler:EndElement(tagname)
+                       stanza:up();
+               end
+               local parser = lxp.new(handler, "\1");
+               local ok, err, line, col = parser:parse(xml);
+               if ok then ok, err, line, col = parser:parse(); end
+               --parser:close();
+               if ok then
+                       return stanza.tags[1];
+               else
+                       return ok, err.." (line "..line..", col "..col..")";
+               end
+       end;
+end)();
+
+-----------------------------------------------------------------------
+
+package.loaded["util.logger"] = {init = function() return function() end; end}
+local dm = require "util.datamanager"
+dm.set_data_path("data");
+
+local arg = ...;
+local help = "/? -? ? /h -h /help -help --help";
+if not arg or help:find(arg, 1, true) then
+       print([[Openfire importer for Prosody
+
+  Usage: openfire2prosody.lua filename.xml hostname
+
+]]);
+       os.exit(1);
+end
+
+local host = select(2, ...) or "localhost";
+
+local file = assert(io.open(arg));
+local data = assert(file:read("*a"));
+file:close();
+
+local xml = assert(parse_xml(data));
+
+assert(xml.name == "Openfire", "The input file is not an Openfire XML export");
+
+local substatus_mapping = { ["0"] = "none", ["1"] = "to", ["2"] = "from", ["3"] = "both" };
+
+for _,tag in ipairs(xml.tags) do
+       if tag.name == "User" then
+               local username, password, roster;
+
+               for _,tag in ipairs(tag.tags) do
+                       if tag.name == "Username" then
+                               username = tag:get_text();
+                       elseif tag.name == "Password" then
+                               password = tag:get_text();
+                       elseif tag.name == "Roster" then
+                               roster = {};
+                               local pending = {};
+                               for _,tag in ipairs(tag.tags) do
+                                       if tag.name == "Item" then
+                                               local jid = assert(tag.attr.jid, "Roster item has no JID");
+                                               if tag.attr.substatus ~= "-1" then
+                                                       local item = {};
+                                                       item.name = tag.attr.name;
+                                                       item.subscription = assert(substatus_mapping[tag.attr.substatus], "invalid substatus");
+                                                       item.ask = tag.attr.askstatus == "0" and "subscribe" or nil;
+
+                                                       local groups = {};
+                                                       for _,tag in ipairs(tag) do
+                                                               if tag.name == "Group" then
+                                                                       groups[tag:get_text()] = true;
+                                                               end
+                                                       end
+                                                       item.groups = groups;
+                                                       roster[jid] = item;
+                                               end
+                                               if tag.attr.recvstatus == "1" then pending[jid] = true; end
+                                       end
+                               end
+
+                               if next(pending) then
+                                       roster[false] = { pending = pending };
+                               end
+                       end
+               end
+
+               assert(username and password, "No username or password");
+
+               local ret, err = dm.store(username, host, "accounts", {password = password});
+               print("["..(err or "success").."] stored account: "..username.."@"..host.." = "..password);
+
+               if roster then
+                       local ret, err = dm.store(username, host, "roster", roster);
+                       print("["..(err or "success").."] stored roster: "..username.."@"..host.." = "..password);
+               end
+       end
+end
+