Comparison

tools/ejabberdsql2prosody.lua @ 1628:c3ce7cbd123d

ejabberdsql2prosody: Added an XML parser
author Waqas Hussain <waqas20@gmail.com>
date Sun, 02 Aug 2009 14:35:03 +0500
parent 1611:e20f90743863
child 1629:f6af348cf497
comparison
equal deleted inserted replaced
1623:099364ec1ab2 1628:c3ce7cbd123d
153 153
154 return readFile(filename); 154 return readFile(filename);
155 155
156 ------ 156 ------
157 end 157 end
158
159 -- XML parser
160 local parse_xml = (function()
161 local entity_map = setmetatable({
162 ["amp"] = "&";
163 ["gt"] = ">";
164 ["lt"] = "<";
165 ["apos"] = "'";
166 ["quot"] = "\"";
167 }, {__index = function(_, s)
168 if s:sub(1,1) == "#" then
169 if s:sub(2,2) == "x" then
170 return string.char(tonumber(s:sub(3), 16));
171 else
172 return string.char(tonumber(s:sub(2)));
173 end
174 end
175 end
176 });
177 local function xml_unescape(str)
178 return (str:gsub("&(.-);", entity_map));
179 end
180 local function parse_tag(s)
181 local name,sattr=(s):gmatch("([^%s]+)(.*)")();
182 local attr = {};
183 for a,b in (sattr):gmatch("([^=%s]+)=['\"]([^'\"]*)['\"]") do attr[a] = xml_unescape(b); end
184 return name, attr;
185 end
186 return function(xml)
187 local stanza = st.stanza("root");
188 local regexp = "<([^>]*)>([^<]*)";
189 for elem, text in xml:gmatch(regexp) do
190 if elem:sub(1,1) == "!" or elem:sub(1,1) == "?" then -- neglect comments and processing-instructions
191 elseif elem:sub(1,1) == "/" then -- end tag
192 elem = elem:sub(2);
193 stanza:up(); -- TODO check for start-end tag name match
194 elseif elem:sub(-1,-1) == "/" then -- empty tag
195 elem = elem:sub(1,-2);
196 local name,attr = parse_tag(elem);
197 stanza:tag(name, attr):up();
198 else -- start tag
199 local name,attr = parse_tag(elem);
200 stanza:tag(name, attr);
201 end
202 if #text ~= 0 then -- text
203 stanza:text(xml_unescape(text));
204 end
205 end
206 return stanza.tags[1];
207 end
208 end)();
209 -- end of XML parser
158 210
159 local arg, host = ...; 211 local arg, host = ...;
160 local help = "/? -? ? /h -h /help -help --help"; 212 local help = "/? -? ? /h -h /help -help --help";
161 if not(arg and host) or help:find(arg, 1, true) then 213 if not(arg and host) or help:find(arg, 1, true) then
162 print([[ejabberd SQL DB dump importer for Prosody 214 print([[ejabberd SQL DB dump importer for Prosody