Added auto serialization for Bind and Roster
All checks were successful
PR Check / on-push-commit-check (push) Successful in 11m42s
All checks were successful
PR Check / on-push-commit-check (push) Successful in 11m42s
This commit is contained in:
parent
30a5e69d14
commit
a4e973ca32
9 changed files with 100 additions and 80 deletions
|
@ -14,37 +14,11 @@ struct Bind {
|
||||||
|
|
||||||
std::optional<FullJid> jid;
|
std::optional<FullJid> jid;
|
||||||
|
|
||||||
friend constexpr auto operator<<(xmlpp::Element* element, const Bind& bind) {
|
friend auto operator<<(xmlpp::Element* element, const Bind& bind) -> void;
|
||||||
element->set_attribute("xmlns", Bind::kDefaultNamespace);
|
[[nodiscard]] static auto Parse(xmlpp::Element* element) -> Bind;
|
||||||
|
|
||||||
if(bind.jid) {
|
|
||||||
auto* jid_el = element->add_child_element("jid");
|
|
||||||
jid_el->add_child_text(ToString(*bind.jid));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
[[nodiscard]] static constexpr auto Parse(xmlpp::Element* element) -> Bind {
|
|
||||||
const auto* jid_node = element->get_first_child("jid");
|
|
||||||
if(!jid_node) {
|
|
||||||
SPDLOG_DEBUG("No Jid Node at Iq::Bind");
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
auto* jid_el = dynamic_cast<const xmlpp::Element*>(jid_node);
|
|
||||||
if(!jid_el) {
|
|
||||||
throw std::runtime_error("dynamic_cast to const xmlpp::Element* failed");
|
|
||||||
}
|
|
||||||
|
|
||||||
const auto* text = jid_el->get_first_child_text();
|
|
||||||
if(!jid_el) {
|
|
||||||
throw std::runtime_error("No text at Iq::Bind jid child");
|
|
||||||
}
|
|
||||||
|
|
||||||
return {.jid = (jid_node ? std::optional{FullJid::Parse(text->get_content())} : std::nullopt)};
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
using SetBind = Set<Bind>;
|
using SetBind = Set<Bind>;
|
||||||
using ResultBind = Result<Bind>;
|
using ResultBind = Result<Bind>;
|
||||||
using IqBind = Iq<Bind>;
|
|
||||||
|
|
||||||
} // namespace larra::xmpp::iq
|
} // namespace larra::xmpp::iq
|
||||||
|
|
|
@ -97,7 +97,7 @@ struct Client {
|
||||||
|
|
||||||
auto UpdateListOfContacts() -> boost::asio::awaitable<void> {
|
auto UpdateListOfContacts() -> boost::asio::awaitable<void> {
|
||||||
SPDLOG_INFO("Send IQ: Get::Roster");
|
SPDLOG_INFO("Send IQ: Get::Roster");
|
||||||
co_await this->Send(::iq::GetRoster{.id = "1", .from = std::format("{}@{}", "invalid_user", jid.server), .payload = {}});
|
co_await this->Send(::iq::GetRoster{.id = "1", .from = jid, .payload = {}});
|
||||||
|
|
||||||
const auto get_roster_response = co_await connection.template Read<Iq<::iq::Roster>>();
|
const auto get_roster_response = co_await connection.template Read<Iq<::iq::Roster>>();
|
||||||
std::visit(utempl::Overloaded(
|
std::visit(utempl::Overloaded(
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
#pragma once
|
#pragma once
|
||||||
|
#include <larra/jid.hpp>
|
||||||
#include <larra/serialization.hpp>
|
#include <larra/serialization.hpp>
|
||||||
#include <larra/stream_error.hpp>
|
#include <larra/stream_error.hpp>
|
||||||
#include <larra/utils.hpp>
|
#include <larra/utils.hpp>
|
||||||
|
@ -12,8 +13,8 @@ namespace iq {
|
||||||
template <auto& Name, typename PayloadType>
|
template <auto& Name, typename PayloadType>
|
||||||
struct BaseImplWithPayload {
|
struct BaseImplWithPayload {
|
||||||
std::string id;
|
std::string id;
|
||||||
std::optional<std::string> from{};
|
std::optional<Jid> from{};
|
||||||
std::optional<std::string> to{};
|
std::optional<Jid> to{};
|
||||||
PayloadType payload;
|
PayloadType payload;
|
||||||
static const inline std::string kName = Name;
|
static const inline std::string kName = Name;
|
||||||
static constexpr auto kDefaultName = "iq";
|
static constexpr auto kDefaultName = "iq";
|
||||||
|
@ -23,11 +24,11 @@ struct BaseImplWithPayload {
|
||||||
return utils::FieldSetHelper::With<"id", BaseImplWithPayload>(std::forward<Self>(self), std::move(id));
|
return utils::FieldSetHelper::With<"id", BaseImplWithPayload>(std::forward<Self>(self), std::move(id));
|
||||||
}
|
}
|
||||||
template <typename Self>
|
template <typename Self>
|
||||||
[[nodiscard]] constexpr auto To(this Self&& self, std::string to) -> std::decay_t<Self> {
|
[[nodiscard]] constexpr auto To(this Self&& self, Jid to) -> std::decay_t<Self> {
|
||||||
return utils::FieldSetHelper::With<"to", BaseImplWithPayload>(std::forward<Self>(self), std::move(to));
|
return utils::FieldSetHelper::With<"to", BaseImplWithPayload>(std::forward<Self>(self), std::move(to));
|
||||||
}
|
}
|
||||||
template <typename Self>
|
template <typename Self>
|
||||||
[[nodiscard]] constexpr auto From(this Self&& self, std::string from) -> std::decay_t<Self> {
|
[[nodiscard]] constexpr auto From(this Self&& self, Jid from) -> std::decay_t<Self> {
|
||||||
return utils::FieldSetHelper::With<"from", BaseImplWithPayload>(std::forward<Self>(self), std::move(from));
|
return utils::FieldSetHelper::With<"from", BaseImplWithPayload>(std::forward<Self>(self), std::move(from));
|
||||||
}
|
}
|
||||||
template <typename NewPayloadType, typename Self>
|
template <typename NewPayloadType, typename Self>
|
||||||
|
@ -38,10 +39,10 @@ struct BaseImplWithPayload {
|
||||||
element->set_attribute("id", self.id);
|
element->set_attribute("id", self.id);
|
||||||
|
|
||||||
if(self.to) {
|
if(self.to) {
|
||||||
element->set_attribute("to", *self.to);
|
element->set_attribute("to", ToString(*self.to));
|
||||||
}
|
}
|
||||||
if(self.from) {
|
if(self.from) {
|
||||||
element->set_attribute("from", *self.from);
|
element->set_attribute("from", ToString(*self.from));
|
||||||
}
|
}
|
||||||
element->set_attribute("type", kName);
|
element->set_attribute("type", kName);
|
||||||
using S = Serialization<PayloadType>;
|
using S = Serialization<PayloadType>;
|
||||||
|
@ -86,8 +87,8 @@ struct BaseImplWithPayload {
|
||||||
throw std::runtime_error("Invalid payload for parse Iq");
|
throw std::runtime_error("Invalid payload for parse Iq");
|
||||||
}
|
}
|
||||||
return {.id = idNode->get_value(),
|
return {.id = idNode->get_value(),
|
||||||
.from = (from ? std::optional{from->get_value()} : std::nullopt),
|
.from = (from ? std::optional{Jid::Parse(from->get_value())} : std::nullopt),
|
||||||
.to = (to ? std::optional{to->get_value()} : std::nullopt),
|
.to = (to ? std::optional{Jid::Parse(to->get_value())} : std::nullopt),
|
||||||
.payload = S::Parse(payload2)};
|
.payload = S::Parse(payload2)};
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
|
@ -11,18 +11,28 @@
|
||||||
|
|
||||||
namespace larra::xmpp::iq {
|
namespace larra::xmpp::iq {
|
||||||
|
|
||||||
|
struct RosterItem {
|
||||||
|
BareJid jid;
|
||||||
|
friend constexpr auto ToString(const RosterItem& item) {
|
||||||
|
return ToString(item.jid);
|
||||||
|
}
|
||||||
|
constexpr auto operator==(const RosterItem&) const -> bool = default;
|
||||||
|
friend auto operator<<(xmlpp::Element* element, const RosterItem& item) -> void;
|
||||||
|
[[nodiscard]] static auto Parse(xmlpp::Element* element) -> RosterItem;
|
||||||
|
};
|
||||||
|
|
||||||
struct Roster {
|
struct Roster {
|
||||||
static constexpr auto kDefaultName = "query";
|
static constexpr auto kDefaultName = "query";
|
||||||
static constexpr auto kDefaultNamespace = "jabber:iq:roster";
|
static constexpr auto kDefaultNamespace = "jabber:iq:roster";
|
||||||
|
|
||||||
std::vector<BareJid> items;
|
std::vector<RosterItem> items;
|
||||||
|
|
||||||
friend auto ToString(const Roster& roster) -> std::string {
|
friend auto ToString(const Roster& roster) -> std::string {
|
||||||
static constexpr std::string_view prefix = "Roster: [\n\t";
|
static constexpr std::string_view prefix = "Roster: [\n\t";
|
||||||
static constexpr std::string_view suffix = "]";
|
static constexpr std::string_view suffix = "]";
|
||||||
// \n\r\t
|
// \n\r\t
|
||||||
std::size_t total_length = std::ranges::fold_left(roster.items | std::views::transform([](const auto& el) {
|
std::size_t total_length = std::ranges::fold_left(roster.items | std::views::transform([](const auto& el) {
|
||||||
return larra::xmpp::utils::AccumulateFieldLength(el) + 3;
|
return larra::xmpp::utils::AccumulateFieldLength(el.jid) + 3;
|
||||||
}),
|
}),
|
||||||
prefix.length() + suffix.length(),
|
prefix.length() + suffix.length(),
|
||||||
std::plus<>{});
|
std::plus<>{});
|
||||||
|
@ -36,37 +46,11 @@ struct Roster {
|
||||||
}
|
}
|
||||||
return s += suffix;
|
return s += suffix;
|
||||||
}
|
}
|
||||||
friend constexpr auto operator<<(xmlpp::Element* element, const Roster& roster) {
|
friend auto operator<<(xmlpp::Element* element, const Roster& roster) -> void;
|
||||||
element->set_attribute("xmlns", Roster::kDefaultNamespace);
|
[[nodiscard]] static auto Parse(xmlpp::Element* element) -> Roster;
|
||||||
for(const auto& item : roster.items) {
|
|
||||||
element->add_child_element("item")->set_attribute("jid", ToString(item));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
[[nodiscard]] static constexpr auto Parse(xmlpp::Element* element) -> Roster {
|
|
||||||
const auto& item_nodes = element->get_children("item");
|
|
||||||
if(item_nodes.empty()) {
|
|
||||||
SPDLOG_DEBUG("No items at Iq::Roster");
|
|
||||||
}
|
|
||||||
|
|
||||||
return {.items = item_nodes | std::views::transform([](const xmlpp::Node* node) {
|
|
||||||
auto item_element = dynamic_cast<const xmlpp::Element*>(node);
|
|
||||||
if(!item_element) {
|
|
||||||
throw std::runtime_error("Can't convert xmlpp::Node to xmlpp::Element");
|
|
||||||
}
|
|
||||||
|
|
||||||
auto jid_ptr = item_element->get_attribute("jid");
|
|
||||||
if(!jid_ptr) {
|
|
||||||
throw std::runtime_error("Not found attribute 'jid' for parse Roster item");
|
|
||||||
}
|
|
||||||
|
|
||||||
return BareJid::Parse(jid_ptr->get_value());
|
|
||||||
}) |
|
|
||||||
std::ranges::to<std::vector<BareJid>>()};
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
using GetRoster = Get<Roster>;
|
using GetRoster = Get<Roster>;
|
||||||
using ResultRoster = Result<Roster>;
|
using ResultRoster = Result<Roster>;
|
||||||
using IqRoster = Iq<Roster>;
|
|
||||||
|
|
||||||
} // namespace larra::xmpp::iq
|
} // namespace larra::xmpp::iq
|
||||||
|
|
|
@ -115,7 +115,7 @@ struct Config : V {
|
||||||
requires(!HasParse<T>)
|
requires(!HasParse<T>)
|
||||||
: V(AttributeConfig{}) {
|
: V(AttributeConfig{}) {
|
||||||
}
|
}
|
||||||
constexpr auto Base() const -> const V& {
|
[[nodiscard]] constexpr auto Base() const -> const V& {
|
||||||
return static_cast<const V&>(*this);
|
return static_cast<const V&>(*this);
|
||||||
}
|
}
|
||||||
using type = T;
|
using type = T;
|
||||||
|
|
|
@ -302,11 +302,11 @@ concept LengthCalculatable = requires(const T& obj) {
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
auto AccumulateFieldLength(const T& obj) -> std::size_t {
|
auto AccumulateFieldLength(const T& obj) -> std::size_t {
|
||||||
std::size_t total_length = 0;
|
std::size_t totalLength = 0;
|
||||||
boost::pfr::for_each_field(obj, [&](const LengthCalculatable auto& field) {
|
boost::pfr::for_each_field(obj, [&](const LengthCalculatable auto& field) {
|
||||||
total_length += field.length(); // Accumulate length of each field
|
totalLength += field.length(); // Accumulate length of each field
|
||||||
});
|
});
|
||||||
return total_length;
|
return totalLength;
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace larra::xmpp::utils
|
} // namespace larra::xmpp::utils
|
||||||
|
|
31
library/src/bind.cpp
Normal file
31
library/src/bind.cpp
Normal file
|
@ -0,0 +1,31 @@
|
||||||
|
#include <larra/bind.hpp>
|
||||||
|
|
||||||
|
namespace larra::xmpp::iq {
|
||||||
|
auto operator<<(xmlpp::Element* element, const Bind& bind) -> void {
|
||||||
|
element->set_attribute("xmlns", Bind::kDefaultNamespace);
|
||||||
|
|
||||||
|
if(bind.jid) {
|
||||||
|
auto* jid_el = element->add_child_element("jid");
|
||||||
|
jid_el->add_child_text(ToString(*bind.jid));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
[[nodiscard]] auto Bind::Parse(xmlpp::Element* element) -> Bind {
|
||||||
|
const auto* jid_node = element->get_first_child("jid");
|
||||||
|
if(!jid_node) {
|
||||||
|
SPDLOG_DEBUG("No Jid Node at Iq::Bind");
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
auto* jid_el = dynamic_cast<const xmlpp::Element*>(jid_node);
|
||||||
|
if(!jid_el) {
|
||||||
|
throw std::runtime_error("dynamic_cast to const xmlpp::Element* failed");
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto* text = jid_el->get_first_child_text();
|
||||||
|
if(!jid_el) {
|
||||||
|
throw std::runtime_error("No text at Iq::Bind jid child");
|
||||||
|
}
|
||||||
|
|
||||||
|
return {.jid = (jid_node ? std::optional{FullJid::Parse(text->get_content())} : std::nullopt)};
|
||||||
|
}
|
||||||
|
} // namespace larra::xmpp::iq
|
30
library/src/roster.cpp
Normal file
30
library/src/roster.cpp
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
#include <larra/roster.hpp>
|
||||||
|
#include <larra/serialization/auto.hpp>
|
||||||
|
|
||||||
|
namespace larra::xmpp::serialization {
|
||||||
|
namespace iq = larra::xmpp::iq;
|
||||||
|
|
||||||
|
template <>
|
||||||
|
constexpr auto kSerializationConfig<iq::RosterItem> = SerializationConfig<iq::RosterItem>{};
|
||||||
|
template <>
|
||||||
|
constexpr auto kSerializationConfig<iq::Roster> = SerializationConfig<iq::Roster>{}.With<"items">({Config<std::vector<iq::RosterItem>>{}});
|
||||||
|
} // namespace larra::xmpp::serialization
|
||||||
|
|
||||||
|
namespace larra::xmpp::iq {
|
||||||
|
namespace S = larra::xmpp::serialization;
|
||||||
|
|
||||||
|
auto operator<<(xmlpp::Element* element, const RosterItem& self) -> void {
|
||||||
|
S::Serialize(element, self);
|
||||||
|
}
|
||||||
|
auto RosterItem::Parse(xmlpp::Element* element) -> RosterItem {
|
||||||
|
return S::Parse<RosterItem>(element);
|
||||||
|
}
|
||||||
|
|
||||||
|
auto operator<<(xmlpp::Element* element, const Roster& self) -> void {
|
||||||
|
element->set_attribute("xmlns", Roster::kDefaultNamespace);
|
||||||
|
S::Serialize(element, self);
|
||||||
|
}
|
||||||
|
auto Roster::Parse(xmlpp::Element* element) -> Roster {
|
||||||
|
return S::Parse<Roster>(element);
|
||||||
|
}
|
||||||
|
} // namespace larra::xmpp::iq
|
|
@ -7,29 +7,29 @@ namespace larra::xmpp {
|
||||||
|
|
||||||
TEST(Roster, SerializeAndParse) {
|
TEST(Roster, SerializeAndParse) {
|
||||||
FullJid jid{.username = "test", .server = "server", .resource = "res"}; // NOLINT
|
FullJid jid{.username = "test", .server = "server", .resource = "res"}; // NOLINT
|
||||||
auto roster = iq::GetRoster{.id = "1", .from = ToString(jid), .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
|
auto roster = iq::GetRoster{.id = "1", .from = jid, .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
|
||||||
|
|
||||||
xmlpp::Document doc;
|
xmlpp::Document doc;
|
||||||
auto node = doc.create_root_node("iq");
|
auto node = doc.create_root_node("iq");
|
||||||
node << roster;
|
node << roster;
|
||||||
|
|
||||||
auto parse_res = decltype(roster)::Parse(node);
|
auto parseRes = decltype(roster)::Parse(node);
|
||||||
|
|
||||||
ASSERT_EQ(roster.payload.items.size(), parse_res.payload.items.size());
|
ASSERT_EQ(roster.payload.items.size(), parseRes.payload.items.size());
|
||||||
for(const auto& [idx, expect_el, parsed_el] : std::views::zip(std::views::iota(0), roster.payload.items, parse_res.payload.items)) {
|
for(const auto& [idx, expectEl, parsedEl] : std::views::zip(std::views::iota(0), roster.payload.items, parseRes.payload.items)) {
|
||||||
EXPECT_EQ(expect_el, parsed_el) << "Mismatched on idx: " << idx;
|
EXPECT_EQ(expectEl, parsedEl) << "Mismatched on idx: " << idx;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static constexpr std::string_view kRosterPrintExpectedData = "Roster: [\n\tu1@s1\n\tu2@s2\n\tu3@s3\n\t]";
|
static constexpr std::string_view kRosterPrintExpectedData = "Roster: [\n\tu1@s1\n\tu2@s2\n\tu3@s3\n\t]";
|
||||||
TEST(Roster, Print) {
|
TEST(Roster, Print) {
|
||||||
FullJid jid{.username = "test", .server = "server", .resource = "res"}; // NOLINT
|
FullJid jid{.username = "test", .server = "server", .resource = "res"}; // NOLINT
|
||||||
auto roster = iq::GetRoster{.id = "1", .from = ToString(jid), .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
|
auto roster = iq::GetRoster{.id = "1", .from = jid, .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
|
||||||
|
|
||||||
EXPECT_NO_THROW({
|
EXPECT_NO_THROW({
|
||||||
auto roster_str = ToString(roster.payload);
|
auto rosterStr = ToString(roster.payload);
|
||||||
EXPECT_EQ(kRosterPrintExpectedData.length(), roster_str.capacity());
|
EXPECT_EQ(kRosterPrintExpectedData.length(), rosterStr.capacity());
|
||||||
EXPECT_EQ(kRosterPrintExpectedData, roster_str);
|
EXPECT_EQ(kRosterPrintExpectedData, rosterStr);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue