1 files changed, 112 insertions, 0 deletions
diff --git a/Sluift/LuaElementConvertors.cpp b/Sluift/LuaElementConvertors.cpp
new file mode 100644
index 0000000..71957c1
--- /dev/null
+++ b/Sluift/LuaElementConvertors.cpp
@@ -0,0 +1,112 @@
+/*
+ * Copyright (c) 2013 Remko Tronçon
+ * Licensed under the GNU General Public License.
+ * See the COPYING file for more information.
+ */
+
+#include <Sluift/LuaElementConvertors.h>
+
+#include <boost/smart_ptr/make_shared.hpp>
+
+#include <Swiften/Base/foreach.h>
+#include <Sluift/LuaElementConvertor.h>
+#include <Sluift/ElementConvertors/RawXMLElementConvertor.h>
+#include <Sluift/ElementConvertors/PubSubEventConvertor.h>
+#include <Sluift/ElementConvertors/DOMElementConvertor.h>
+#include <Sluift/ElementConvertors/DefaultElementConvertor.h>
+#include <Sluift/ElementConvertors/DiscoItemsConvertor.h>
+#include <Sluift/ElementConvertors/DiscoInfoConvertor.h>
+#include <Sluift/ElementConvertors/FormConvertor.h>
+#include <Sluift/ElementConvertors/SoftwareVersionConvertor.h>
+#include <Sluift/ElementConvertors/VCardUpdateConvertor.h>
+#include <Sluift/ElementConvertors/VCardConvertor.h>
+#include <Sluift/ElementConvertors/BodyConvertor.h>
+#include <Sluift/ElementConvertors/CommandConvertor.h>
+#include <Sluift/Lua/LuaUtils.h>
+#include <Sluift/Lua/Exception.h>
+
+using namespace Swift;
+
+LuaElementConvertors::LuaElementConvertors() {
+	registerConvertors();
+	convertors.push_back(boost::make_shared<CommandConvertor>(this));
+	convertors.push_back(boost::make_shared<PubSubEventConvertor>(this));
+	convertors.push_back(boost::make_shared<BodyConvertor>());
+	convertors.push_back(boost::make_shared<VCardConvertor>());
+	convertors.push_back(boost::make_shared<VCardUpdateConvertor>());
+	convertors.push_back(boost::make_shared<FormConvertor>());
+	convertors.push_back(boost::make_shared<SoftwareVersionConvertor>());
+	convertors.push_back(boost::make_shared<DiscoInfoConvertor>());
+	convertors.push_back(boost::make_shared<DiscoItemsConvertor>());
+	convertors.push_back(boost::make_shared<DOMElementConvertor>());
+	convertors.push_back(boost::make_shared<RawXMLElementConvertor>());
+	convertors.push_back(boost::make_shared<DefaultElementConvertor>());
+}
+
+LuaElementConvertors::~LuaElementConvertors() {
+}
+
+#include <Sluift/ElementConvertors/ElementConvertors.ipp>
+
+boost::shared_ptr<Payload> LuaElementConvertors::convertFromLua(lua_State* L, int index) {
+	if (lua_isstring(L, index)) {
+		return convertFromLuaUntyped(L, index, "xml");
+	}
+	else if (lua_istable(L, index)) {
+		lua_getfield(L, index, "_type");
+		if (lua_isstring(L, -1)) {
+			std::string type = lua_tostring(L, -1);
+			lua_pop(L, 1);
+			return convertFromLuaUntyped(L, index, type);
+		}
+		lua_pop(L, 1);
+	}
+	throw Lua::Exception("Unable to determine type");
+}
+
+boost::shared_ptr<Payload> LuaElementConvertors::convertFromLuaUntyped(lua_State* L, int index, const std::string& type) {
+	index = Lua::absoluteOffset(L, index);
+	foreach (boost::shared_ptr<LuaElementConvertor> convertor, convertors) {
+		if (boost::shared_ptr<Payload> result = convertor->convertFromLua(L, index, type)) {
+			return result;
+		}
+	}
+	return boost::shared_ptr<Payload>();
+}
+
+
+int LuaElementConvertors::convertToLua(lua_State* L, boost::shared_ptr<Payload> payload) {
+	if (boost::optional<std::string> type = doConvertToLuaUntyped(L, payload)) {
+		if (lua_istable(L, -1)) {
+			lua_pushstring(L, type->c_str());
+			lua_setfield(L, -2, "_type");
+			Lua::registerTableToString(L, -1);
+		}
+		else {
+			assert(*type == "xml");
+		}
+		return 1;
+	}
+	return 0;
+}
+
+int LuaElementConvertors::convertToLuaUntyped(lua_State* L, boost::shared_ptr<Payload> payload) {
+	if (doConvertToLuaUntyped(L, payload)) {
+		return 1;
+	}
+	return 0;
+}
+
+boost::optional<std::string> LuaElementConvertors::doConvertToLuaUntyped(
+		lua_State* L, boost::shared_ptr<Payload> payload) {
+	if (!payload) {
+		return LuaElementConvertor::NO_RESULT;
+	}
+	foreach (boost::shared_ptr<LuaElementConvertor> convertor, convertors) {
+		if (boost::optional<std::string> type = convertor->convertToLua(L, payload)) {
+			return *type;
+		}
+	}
+	return LuaElementConvertor::NO_RESULT;
+}
+