1 files changed, 88 insertions, 0 deletions
diff --git a/Sluift/ElementConvertors/MAMQueryConvertor.cpp b/Sluift/ElementConvertors/MAMQueryConvertor.cpp
new file mode 100644
index 0000000..52337ce
--- /dev/null
+++ b/Sluift/ElementConvertors/MAMQueryConvertor.cpp
@@ -0,0 +1,88 @@
+/*
+ * Copyright (c) 2014-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
+ */
+
+#include <Sluift/ElementConvertors/MAMQueryConvertor.h>
+
+#include <memory>
+
+#include <boost/numeric/conversion/cast.hpp>
+
+#include <lua.hpp>
+
+#include <Swiften/Elements/Form.h>
+#include <Swiften/Elements/ResultSet.h>
+
+#include <Sluift/LuaElementConvertors.h>
+
+using namespace Swift;
+
+MAMQueryConvertor::MAMQueryConvertor(LuaElementConvertors* convertors) :
+        GenericLuaElementConvertor<MAMQuery>("mam"),
+        convertors(convertors) {
+}
+
+MAMQueryConvertor::~MAMQueryConvertor() {
+}
+
+std::shared_ptr<MAMQuery> MAMQueryConvertor::doConvertFromLua(lua_State* L) {
+    std::shared_ptr<MAMQuery> result = std::make_shared<MAMQuery>();
+    lua_getfield(L, -1, "query_id");
+    if (lua_isstring(L, -1)) {
+        result->setQueryID(std::string(lua_tostring(L, -1)));
+    }
+    lua_pop(L, 1);
+    lua_getfield(L, -1, "node");
+    if (lua_isstring(L, -1)) {
+        result->setNode(std::string(lua_tostring(L, -1)));
+    }
+    lua_pop(L, 1);
+    lua_getfield(L, -1, "form");
+    if (!lua_isnil(L, -1)) {
+        std::shared_ptr<Form> form = std::dynamic_pointer_cast<Form>(convertors->convertFromLuaUntyped(L, -1, "form"));
+        if (!!form) {
+            result->setForm(form);
+        }
+    }
+    lua_pop(L, 1);
+    lua_getfield(L, -1, "result_set");
+    if (!lua_isnil(L, -1)) {
+        std::shared_ptr<ResultSet> resultSet = std::dynamic_pointer_cast<ResultSet>(convertors->convertFromLuaUntyped(L, -1, "result_set"));
+        if (!!resultSet) {
+            result->setResultSet(resultSet);
+        }
+    }
+    lua_pop(L, 1);
+    return result;
+}
+
+void MAMQueryConvertor::doConvertToLua(lua_State* L, std::shared_ptr<MAMQuery> payload) {
+    lua_createtable(L, 0, 0);
+    if (payload->getQueryID()) {
+        lua_pushstring(L, (*payload->getQueryID()).c_str());
+        lua_setfield(L, -2, "query_id");
+    }
+    if (payload->getNode()) {
+        lua_pushstring(L, (*payload->getNode()).c_str());
+        lua_setfield(L, -2, "node");
+    }
+    if (convertors->convertToLuaUntyped(L, payload->getForm()) > 0) {
+        lua_setfield(L, -2, "form");
+    }
+    if (convertors->convertToLuaUntyped(L, payload->getResultSet()) > 0) {
+        lua_setfield(L, -2, "result_set");
+    }
+}
+
+boost::optional<LuaElementConvertor::Documentation> MAMQueryConvertor::getDocumentation() const {
+    return Documentation(
+        "MAMQuery",
+        "This table has the following fields:\n\n"
+        "- `query_id`: string (Optional)\n"
+        "- `node`: string (Optional)\n"
+        "- `form`: string @{Form} (Optional)\n"
+        "- `result_set`: @{ResultSet} (Optional)\n"
+    );
+}