summaryrefslogtreecommitdiffstats
blob: cadfbc4536aa857ee6bf21475c05fcb7077f6986 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
 * 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/Lua/LuaUtils.h>
#include <Sluift/Lua/Exception.h>

using namespace Swift;

LuaElementConvertors::LuaElementConvertors() {
	registerConvertors();
	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;
}