Fixed errors and improve roster tests

.vscode/launch.json

@@ -20,7 +20,7 @@
             "program": "${workspaceFolder}/build/larra_xmpp_tests",
             "args": [
                 // --gtest_filter=POSTIVE_PATTERNS[-NEGATIVE_PATTERNS]
-                "--gtest_filter=Roster*"
+                // "--gtest_filter=Roster*"
             ],
             "cwd": "${workspaceFolder}",
             "preLaunchTask": "GCC: Build"

library/include/larra/bind.hpp

@@ -47,7 +47,4 @@ using SetBind = Set<Bind>;
 using ResultBind = Result<Bind>;
 using IqBind = Iq<Bind>;
 
-inline auto MakeSetBind() {
-  return SetBind{.id = "1", .payload = Bind{}};
-}
 }  // namespace larra::xmpp::iq

library/include/larra/client/client.hpp

@@ -80,21 +80,35 @@ struct Client {
 
   auto CreateResourceBind() -> boost::asio::awaitable<void> {
     SPDLOG_INFO("Send IQ: Set::Bind");
-    co_await this->Send(::iq::MakeSetBind());
+    co_await this->Send(::iq::SetBind{.id = "1", .payload = {}});
 
-    auto bind_result = co_await connection.template Read<::iq::ResultBind>();
-    jid.resource = std::move(bind_result.payload.jid->resource);
+    auto set_bind_response = co_await connection.template Read<Iq<::iq::Bind>>();
+    std::visit(utempl::Overloaded(
+                   [](auto error) {
+                     throw "Error response on IQ: Set::Bind: ''";  // TODO(unknown): Add exact error parsing
+                   },
+                   [&](::iq::ResultBind r) {
+                     jid.resource = std::move(r.payload.jid->resource);
+                     SPDLOG_INFO("Allocated resource: {}", jid.resource);
+                   }),
+               set_bind_response);
     co_return;
   }
 
   auto UpdateListOfContacts() -> boost::asio::awaitable<void> {
     SPDLOG_INFO("Send IQ: Get::Roster");
-    co_await this->Send(::iq::MakeGetRoster(jid));
+    co_await this->Send(::iq::GetRoster{.id = "1", .from = std::format("{}@{}", "invalid_user", jid.server), .payload = {}});
 
-    const auto roster_result = co_await connection.template Read<::iq::ResultRoster>();
-    roster = std::move(roster_result.payload);
-
-    SPDLOG_INFO("New roster: {}", ToString(roster));
+    const auto get_roster_response = co_await connection.template Read<Iq<::iq::Roster>>();
+    std::visit(utempl::Overloaded(
+                   [](auto error) {
+                     throw "Error response on IQ: Get::Roster: ''";  // TODO(unknown): Add exact error parsing
+                   },
+                   [&](::iq::ResultRoster r) {
+                     roster = std::move(r.payload);
+                     SPDLOG_INFO("New roster: {}", ToString(roster));
+                   }),
+               get_roster_response);
     co_return;
   }
 
@@ -143,14 +157,15 @@ struct Challenge {
       throw std::runtime_error(std::format("Invalid name {} for challenge", node->get_name()));
     }
     std::string decoded = DecodeBase64(node->get_first_child_text()->get_content());
-    auto params = std::views::split(decoded, ',') | std::views::transform([](auto param) {
-                    return std::string_view{param};
-                  }) |
-                  std::views::transform([](std::string_view param) -> std::pair<std::string_view, std::string_view> {
-                    auto v = param.find("=");
-                    return {param.substr(0, v), param.substr(v + 1)};
-                  }) |
-                  std::ranges::to<std::unordered_map<std::string_view, std::string_view>>();
+    auto params = std::views::split(decoded, ',')                                                                        //
+                  | std::views::transform([](auto param) {                                                               //
+                      return std::string_view{param};                                                                    //
+                    })                                                                                                   //
+                  | std::views::transform([](std::string_view param) -> std::pair<std::string_view, std::string_view> {  //
+                      auto v = param.find("=");                                                                          //
+                      return {param.substr(0, v), param.substr(v + 1)};                                                  //
+                    })                                                                                                   //
+                  | std::ranges::to<std::unordered_map<std::string_view, std::string_view>>();
     return {.body = std::move(decoded),
             .serverNonce = params.at("r"),
             .salt = DecodeBase64(params.at("s")),

library/include/larra/iq.hpp

@@ -37,10 +37,10 @@ struct BaseImplWithPayload {
   friend constexpr auto operator<<(xmlpp::Element* element, const BaseImplWithPayload& self) {
     element->set_attribute("id", self.id);
 
-    if (self.to) {
+    if(self.to) {
       element->set_attribute("to", *self.to);
     }
-    if (self.from) {
+    if(self.from) {
       element->set_attribute("from", *self.from);
     }
     element->set_attribute("type", kName);
@@ -48,6 +48,19 @@ struct BaseImplWithPayload {
     S::Serialize(element->add_child_element(S::kDefaultName, S::kPrefix), self.payload);
   }
 
+  [[nodiscard]] static constexpr auto TryParse(xmlpp::Element* element) -> std::optional<BaseImplWithPayload> {
+    return [&] -> std::optional<BaseImplWithPayload> {
+      auto node = element->get_attribute("type");
+      if(!node) {
+        return std::nullopt;
+      }
+      if(node->get_value() != Name) {
+        return std::nullopt;
+      }
+      return Parse(element);
+    }();
+  }
+
   [[nodiscard]] static constexpr auto Parse(xmlpp::Element* element) -> BaseImplWithPayload {
     auto node = element->get_attribute("type");
     if(!node) {
@@ -63,7 +76,6 @@ struct BaseImplWithPayload {
     auto from = element->get_attribute("from");
     auto to = element->get_attribute("to");
 
-
     using S = Serialization<PayloadType>;
     auto payload = element->get_first_child(S::kDefaultName);
     if(!payload) {
@@ -73,13 +85,13 @@ struct BaseImplWithPayload {
     if(!payload2) {
       throw std::runtime_error("Invalid payload for parse Iq");
     }
-    return {
-      .id = idNode->get_value(),
-      .from = (from ? std::optional{from->get_value()} : std::nullopt),
-      .to = (to ? std::optional{to->get_value()} : std::nullopt),
-      .payload = S::Parse(payload2)};
+    return {.id = idNode->get_value(),
+            .from = (from ? std::optional{from->get_value()} : std::nullopt),
+            .to = (to ? std::optional{to->get_value()} : std::nullopt),
+            .payload = S::Parse(payload2)};
   }
 };
+
 static constexpr auto kGetName = "get";
 
 template <typename Payload>

library/include/larra/jid.hpp

@@ -54,9 +54,10 @@ struct FullJid {
 struct BareJid {
   std::string username;
   std::string server;
-  constexpr operator FullJid() const {
-    return FullJid{.username = username, .server = server, .resource = ""};
+  constexpr operator FullJid(this auto&& self) {
+    return {.username = std::forward_like<decltype(self)>(self.username), .server = std::forward_like<decltype(self)>(self.server)};
   }
+
   [[nodiscard]] static auto Parse(std::string_view jid) -> BareJid;
   friend auto ToString(const BareJid& jid) -> std::string;
 

library/include/larra/roster.hpp

@@ -38,9 +38,9 @@ struct Roster {
   }
   friend constexpr auto operator<<(xmlpp::Element* element, const Roster& roster) {
     element->set_attribute("xmlns", Roster::kDefaultNamespace);
-    std::ranges::for_each(roster.items, [element](const auto& item) {
+    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");
@@ -69,8 +69,4 @@ using GetRoster = Get<Roster>;
 using ResultRoster = Result<Roster>;
 using IqRoster = Iq<Roster>;
 
-inline auto MakeGetRoster(const FullJid& jid) {
-  return GetRoster{.id = "1", .from = ToString(jid), .payload = Roster{}};
-}
-
 }  // namespace larra::xmpp::iq

library/include/larra/stream_error.hpp

@@ -73,6 +73,25 @@ struct ErrorImpl : BaseError {
   }
 };
 
+// Helper class to prevent parsing response stream into an expected return type if its name is a 'stream:error'
+struct UnknownXmppError : BaseError {
+  static constexpr auto kDefaultName = "stream:error";
+  static constexpr std::string_view kErrorMessage = "Unknown XMPP stream error";
+
+  static constexpr auto TryParse(xmlpp::Element* element) {
+    return std::optional{UnknownXmppError{}};
+  }
+  static constexpr auto Parse(xmlpp::Element* element) {
+    return TryParse(element).value();
+  }
+  friend constexpr auto operator<<(xmlpp::Element* element, const UnknownXmppError& obj) -> void {
+    throw std::format("'{}' must never be written into the real stream!", kErrorMessage);
+  }
+  [[nodiscard]] constexpr auto what() const noexcept -> const char* override {
+    return kErrorMessage.data();
+  }
+};
+
 struct BadFormat : ErrorImpl<BadFormat> {};
 struct BadNamespacePrefix : ErrorImpl<BadNamespacePrefix> {};
 struct Conflict : ErrorImpl<Conflict> {};
@@ -81,7 +100,7 @@ struct HostGone : ErrorImpl<HostGone> {};
 struct HostUnknown : ErrorImpl<HostUnknown> {};
 struct ImproperAdressing : ErrorImpl<ImproperAdressing> {};
 struct InternalServerError : ErrorImpl<InternalServerError> {};
-struct InvalidForm : ErrorImpl<InvalidForm> {};
+struct InvalidFrom : ErrorImpl<InvalidFrom> {};
 struct InvalidNamespace : ErrorImpl<InvalidNamespace> {};
 struct InvalidXml : ErrorImpl<InvalidXml> {};
 struct NotAuthorized : ErrorImpl<NotAuthorized> {};
@@ -109,7 +128,7 @@ using StreamError = std::variant<error::stream::BadFormat,
                                  error::stream::HostUnknown,
                                  error::stream::ImproperAdressing,
                                  error::stream::InternalServerError,
-                                 error::stream::InvalidForm,
+                                 error::stream::InvalidFrom,
                                  error::stream::InvalidNamespace,
                                  error::stream::InvalidXml,
                                  error::stream::NotAuthorized,
@@ -123,6 +142,10 @@ using StreamError = std::variant<error::stream::BadFormat,
                                  error::stream::UnsupportedEncoding,
                                  error::stream::UnsupportedFeature,
                                  error::stream::UnsupportedStanzaType,
-                                 error::stream::UnsupportedVersion>;
+                                 error::stream::UnsupportedVersion,
+                                 error::stream::UnknownXmppError>;
+
+static_assert(!std::is_same_v<typename std::variant_alternative_t<std::variant_size_v<StreamError> - 1, StreamError>, StreamError>,
+              "'UnknownXmppError' must be at the end of 'StreamError' variant");
 
 }  // namespace larra::xmpp

library/include/larra/utils.hpp

@@ -297,7 +297,7 @@ struct RangeToWrapper : T {
 
 template <typename T>
 concept LengthCalculatable = requires(const T& obj) {
-  { obj.length() } -> std::convertible_to<std::size_t>;  // Checks if obj has a length() method returning a type convertible to std::size_t
+  { obj.length() } -> std::convertible_to<std::size_t>;
 } || std::convertible_to<T, std::string>;
 
 template <typename T>

tests/roster.cpp

@@ -1,17 +1,13 @@
 #include <gtest/gtest.h>
 
+#include <larra/jid.hpp>
 #include <larra/roster.hpp>
 
-#include "larra/jid.hpp"
-
 namespace larra::xmpp {
 
 TEST(Roster, SerializeAndParse) {
   FullJid jid{.username = "test", .server = "server", .resource = "res"};  // NOLINT
-  auto roster = iq::MakeGetRoster(jid);
-  roster.payload.items.emplace_back("u1", "s1");
-  roster.payload.items.emplace_back("u2", "s2");
-  roster.payload.items.emplace_back("u3", "s3");
+  auto roster = iq::GetRoster{.id = "1", .from = ToString(jid), .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
 
   xmlpp::Document doc;
   auto node = doc.create_root_node("iq");
@@ -22,18 +18,19 @@ TEST(Roster, SerializeAndParse) {
   ASSERT_EQ(roster.payload.items.size(), parse_res.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)) {
     EXPECT_EQ(expect_el, parsed_el) << "Mismatched on idx: " << idx;
-    // std::cerr << "        " << "idx: " << idx << "; expect_el: " << expect_el << "; parsed_el: " << parsed_el << '\n';
   }
 }
 
+static constexpr std::string_view kRosterPrintExpectedData = "Roster: [\n\tu1@s1\n\tu2@s2\n\tu3@s3\n\t]";
 TEST(Roster, Print) {
   FullJid jid{.username = "test", .server = "server", .resource = "res"};  // NOLINT
-  auto roster = iq::MakeGetRoster(jid);
-  roster.payload.items.emplace_back("u1", "s1");
-  roster.payload.items.emplace_back("u2", "s2");
-  roster.payload.items.emplace_back("u3", "s3");
+  auto roster = iq::GetRoster{.id = "1", .from = ToString(jid), .payload = iq::Roster{.items = {{"u1", "s1"}, {"u2", "s2"}, {"u3", "s3"}}}};
 
-  EXPECT_NO_THROW({ std::cerr << "[          ] Roster payload: " << ToString(roster.payload) << '\n'; });
+  EXPECT_NO_THROW({
+    auto roster_str = ToString(roster.payload);
+    EXPECT_EQ(kRosterPrintExpectedData.length(), roster_str.capacity());
+    EXPECT_EQ(kRosterPrintExpectedData, roster_str);
+  });
 }
 
 }  // namespace larra::xmpp

tests/serialization.cpp

@@ -5,6 +5,9 @@
 #include <larra/serialization/auto.hpp>
 #include <larra/serialization/error.hpp>
 #include <larra/stream_error.hpp>
+#include <variant>
+
+#include "utempl/utils.hpp"
 
 using namespace std::literals;