1 files changed, 75 insertions, 0 deletions
diff --git a/Sluift/ElementConvertors/MAMQueryConvertor.cpp b/Sluift/ElementConvertors/MAMQueryConvertor.cpp
new file mode 100644
index 0000000..7d7224e
--- /dev/null
+++ b/Sluift/ElementConvertors/MAMQueryConvertor.cpp
@@ -0,0 +1,75 @@
+/*
+ * Copyright (c) 2014 Kevin Smith and Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <boost/numeric/conversion/cast.hpp>
+#include <boost/smart_ptr/make_shared.hpp>
+#include <lua.hpp>
+#include <Sluift/ElementConvertors/MAMQueryConvertor.h>
+#include <Sluift/LuaElementConvertors.h>
+#include <Swiften/Elements/Form.h>
+#include <Swiften/Elements/ResultSet.h>
+
+#pragma clang diagnostic ignored "-Wunused-private-field"
+
+using namespace Swift;
+
+MAMQueryConvertor::MAMQueryConvertor(LuaElementConvertors* convertors) : 
+		GenericLuaElementConvertor<MAMQuery>("mam_query"),
+		convertors(convertors) {
+}
+
+MAMQueryConvertor::~MAMQueryConvertor() {
+}
+
+boost::shared_ptr<MAMQuery> MAMQueryConvertor::doConvertFromLua(lua_State* L) {
+	boost::shared_ptr<MAMQuery> result = boost::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, "form");
+	if (!lua_isnil(L, -1)) {
+		boost::shared_ptr<Form> form = boost::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)) {
+		boost::shared_ptr<ResultSet> resultSet = boost::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, boost::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 (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"
+		"- `form`: string @{Form} (Optional)\n"
+		"- `result_set`: @{ResultSet} (Optional)\n"
+	);
+}