Swiftdiff options
| author | Roger Planas <roger.planas@isode.com> | 2017-06-28 15:09:37 (GMT) |
|---|---|---|
| committer | Tobias Markmann <tm@ayena.de> | 2017-07-04 08:56:54 (GMT) |
| commit | 93a0f41d4af17db3e3eedb61e786fec23dbd8db9 (patch) | |
| tree | 419daecd39e31d98cfb6c1b9dc47c4a584580393 /Sluift/ElementConvertors/DiscoInfoConvertor.cpp | |
| parent | 4f61e116c83be5e38a8406ac3285a07b84712ec1 (diff) | |
| download | swift-93a0f41d4af17db3e3eedb61e786fec23dbd8db9.zip swift-93a0f41d4af17db3e3eedb61e786fec23dbd8db9.tar.bz2 | |
Sluift: Add extensions support to disco#info queries
Only identities and features were reported. Now extensions,
if any, will also be reported
Test-information:
Sent disco#info to a muc room, and got the following extensions back
(along with identities and features, so no regression)
Change-Id: Ib4ab17114df4da5de51945df9e53c12bd42716d1
Diffstat (limited to 'Sluift/ElementConvertors/DiscoInfoConvertor.cpp')
| -rw-r--r-- | Sluift/ElementConvertors/DiscoInfoConvertor.cpp | 31 |
1 files changed, 28 insertions, 3 deletions
diff --git a/Sluift/ElementConvertors/DiscoInfoConvertor.cpp b/Sluift/ElementConvertors/DiscoInfoConvertor.cpp index 2aa4a77..a528211 100644 --- a/Sluift/ElementConvertors/DiscoInfoConvertor.cpp +++ b/Sluift/ElementConvertors/DiscoInfoConvertor.cpp @@ -1,119 +1,144 @@ /* * Copyright (c) 2013-2016 Isode Limited. * All rights reserved. * See the COPYING file for more information. */ #include <Sluift/ElementConvertors/DiscoInfoConvertor.h> #include <memory> #include <boost/numeric/conversion/cast.hpp> #include <lua.hpp> +#include <Swiften/Elements/Form.h> +#include <Sluift/LuaElementConvertors.h> #include <Sluift/Lua/LuaUtils.h> using namespace Swift; -DiscoInfoConvertor::DiscoInfoConvertor() : GenericLuaElementConvertor<DiscoInfo>("disco_info") { +DiscoInfoConvertor::DiscoInfoConvertor(LuaElementConvertors* convertors) : + GenericLuaElementConvertor<DiscoInfo>("disco_info"), + convertors(convertors) { } DiscoInfoConvertor::~DiscoInfoConvertor() { } std::shared_ptr<DiscoInfo> DiscoInfoConvertor::doConvertFromLua(lua_State* L) { std::shared_ptr<DiscoInfo> result = std::make_shared<DiscoInfo>(); if (boost::optional<std::string> value = Lua::getStringField(L, -1, "node")) { result->setNode(*value); } lua_getfield(L, -1, "identities"); if (lua_istable(L, -1)) { for (lua_pushnil(L); lua_next(L, -2); ) { result->addIdentity(DiscoInfo::Identity( Lua::getStringField(L, -1, "name").get_value_or(""), Lua::getStringField(L, -1, "category").get_value_or("client"), Lua::getStringField(L, -1, "type").get_value_or("pc"), Lua::getStringField(L, -1, "language").get_value_or(""))); lua_pop(L, 1); } } lua_pop(L, 1); lua_getfield(L, -1, "features"); if (lua_istable(L, -1)) { for (lua_pushnil(L); lua_next(L, -2); ) { if (lua_isstring(L, -1)) { result->addFeature(lua_tostring(L, -1)); } lua_pop(L, 1); } } lua_pop(L, 1); - // TODO: Extension + lua_getfield(L, -1, "extensions"); + if (lua_istable(L, -1)) { + for (lua_pushnil(L); lua_next(L, -2); ) { + std::shared_ptr<Form> form = std::dynamic_pointer_cast<Form>(convertors->convertFromLuaUntyped(L, -1, "form")); + if (!!form) { + result->addExtension(form); + } + lua_pop(L, 1); + } + } + lua_pop(L, 1); return result; } void DiscoInfoConvertor::doConvertToLua(lua_State* L, std::shared_ptr<DiscoInfo> payload) { lua_newtable(L); if (!payload->getNode().empty()) { lua_pushstring(L, payload->getNode().c_str()); lua_setfield(L, -2, "node"); } const std::vector<DiscoInfo::Identity>& identities = payload->getIdentities(); if (!identities.empty()) { lua_createtable(L, boost::numeric_cast<int>(identities.size()), 0); for (size_t i = 0; i < identities.size(); ++i) { lua_createtable(L, 0, 0); if (!identities[i].getName().empty()) { lua_pushstring(L, identities[i].getName().c_str()); lua_setfield(L, -2, "name"); } if (!identities[i].getCategory().empty()) { lua_pushstring(L, identities[i].getCategory().c_str()); lua_setfield(L, -2, "category"); } if (!identities[i].getType().empty()) { lua_pushstring(L, identities[i].getType().c_str()); lua_setfield(L, -2, "type"); } if (!identities[i].getLanguage().empty()) { lua_pushstring(L, identities[i].getLanguage().c_str()); lua_setfield(L, -2, "language"); } lua_rawseti(L, -2, boost::numeric_cast<int>(i+1)); } lua_setfield(L, -2, "identities"); } const std::vector<std::string>& features = payload->getFeatures(); if (!features.empty()) { lua_createtable(L, boost::numeric_cast<int>(features.size()), 0); for (size_t i = 0; i < features.size(); ++i) { lua_pushstring(L, features[i].c_str()); lua_rawseti(L, -2, boost::numeric_cast<int>(i+1)); } lua_setfield(L, -2, "features"); } - // TODO: Extension + const std::vector<Form::ref>& extensions = payload->getExtensions(); + if (!extensions.empty()) { + lua_createtable(L, boost::numeric_cast<int>(extensions.size()), 0); + for (size_t i = 0; i < extensions.size(); ++i) { + if (convertors->convertToLuaUntyped(L, extensions[i]) > 0) { + lua_rawseti(L, -2, boost::numeric_cast<int>(i+1)); + } + } + lua_setfield(L, -2, "extensions"); + } } boost::optional<LuaElementConvertor::Documentation> DiscoInfoConvertor::getDocumentation() const { return Documentation( "DiscoInfo", "Represents `disco#info` service discovery data.\n\n" "This table has the following structure:\n\n" "- `node`: string\n" "- `identities`: array(table)\n" " - `name`: string\n" " - `category`: string\n" " - `type`: string\n" " - `language`: string\n" "- `features`: array(string)\n" + "- `extensions`: array(table)\n" + "- `form`: string @{Form} (Optional)\n" ); } |