Making the best attempt out of a bad merge from waqas

[prosody.git] / util / stanza.lua

diff --git a/util/stanza.lua b/util/stanza.lua
index cc5b762315018632976f28b1dbdf1d09092a912b..6bc70ab91d8cb609e0392b6c69371fc564c977d0 100644 (file)
--- a/util/stanza.lua
+++ b/util/stanza.lua
@@ -6,6 +6,7 @@ local setmetatable  =  setmetatable;
 local pairs         =         pairs;
 local ipairs        =        ipairs;
 local type          =          type;
+local unpack        =        unpack;
 local s_gsub        =   string.gsub;
 module "stanza"
 
@@ -17,15 +18,6 @@ function stanza(name, attr)
        return setmetatable(stanza, stanza_mt);
 end
 
-function stanza_mt:iq(attrs)
-       return self + stanza("iq", attrs)
-end
-function stanza_mt:message(attrs)
-       return self + stanza("message", attrs)
-end
-function stanza_mt:presence(attrs)
-       return self + stanza("presence", attrs)
-end
 function stanza_mt:query(xmlns)
        return self:tag("query", { xmlns = xmlns });
 end
@@ -116,6 +108,42 @@ do
         end
 end
 
+function preserialize(stanza)
+       local s = { name = stanza.name, attr = stanza.attr };
+       for _, child in ipairs(stanza) do
+               if type(child) == "table" then
+                       t_insert(s, preserialize(child));
+               else
+                       t_insert(s, child);
+               end
+       end
+       return s;
+end
+
+function deserialize(stanza)
+       -- Set metatable
+       if stanza then
+               setmetatable(stanza, stanza_mt);
+               for _, child in ipairs(stanza) do
+                       if type(child) == "table" then
+                               deserialize(child);
+                       end
+               end
+               if not stanza.tags then
+                       -- Rebuild tags
+                       local tags = {};
+                       for _, child in ipairs(stanza) do
+                               if type(child) == "table" then
+                                       t_insert(tags, child);
+                               end
+                       end
+                       stanza.tags = tags;
+               end
+       end
+       
+       return stanza;
+end
+
 function message(attr, body)
        if not body then
                return stanza("message", attr);
@@ -133,7 +161,7 @@ function reply(orig)
 end
 
 function error_reply(orig, type, condition, message, clone)
-       local r = reply(orig);
+       local t = reply(orig);
        t.attr.type = "error";
        -- TODO use clone
        t:tag("error", {type = type})
@@ -146,4 +174,4 @@ function presence(attr)
        return stanza("presence", attr);
 end
 
-return _M;
\ No newline at end of file
+return _M;