From ad36468b84c1575fd46824ff02c0308dd9f3e138 Mon Sep 17 00:00:00 2001
From: sha512sum <linuxnyasha@tutanota.com>
Date: Wed, 9 Oct 2024 19:19:32 +0000
Subject: [PATCH] Add iq types

---
 library/include/larra/iq.hpp | 56 ++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 library/include/larra/iq.hpp

diff --git a/library/include/larra/iq.hpp b/library/include/larra/iq.hpp
new file mode 100644
index 0000000..93b07cd
--- /dev/null
+++ b/library/include/larra/iq.hpp
@@ -0,0 +1,56 @@
+#pragma once
+#include <larra/serialization.hpp>
+#include <larra/stream_error.hpp>
+#include <larra/utils.hpp>
+#include <string>
+
+namespace larra::xmpp {
+
+namespace iq {
+
+template <auto& Name, typename PayloadType>
+struct BaseImplWithPayload {
+  std::string id;
+  PayloadType payload;
+  static const inline std::string kName = Name;
+  template <typename Self>
+  [[nodiscard]] constexpr auto Id(this Self&& self, std::string id) -> std::decay_t<Self> {
+    return utils::FieldSetHelper::With<"id", BaseImplWithPayload>(std::forward<Self>(self), std::move(id));
+  }
+  template <typename NewPayloadType, typename Self>
+  [[nodiscard]] constexpr auto Payload(this Self&& self, NewPayloadType value) {
+    return utils::FieldSetHelper::With<"payload", BaseImplWithPayload, false>(std::forward<Self>(self), std::move(value));
+  }
+  friend constexpr auto operator<<(xmlpp::Element* element, const BaseImplWithPayload& self) {
+    element->set_attribute("id", self.id);
+    element->set_attribute("type", kName);
+    using S = Serialization<PayloadType>;
+    S::Serialize(element->add_child_element(S::kDefaultName, S::kPrefix), self.payload);
+  }
+};
+static constexpr auto kGetName = "get";
+
+template <typename Payload>
+using Get = BaseImplWithPayload<kGetName, Payload>;
+
+static constexpr auto kSetName = "set";
+
+template <typename Payload>
+using Set = BaseImplWithPayload<kSetName, Payload>;
+
+static constexpr auto kResultName = "result";
+
+template <typename Payload>
+using Result = BaseImplWithPayload<kResultName, Payload>;
+
+static constexpr auto kErrorName = "error";
+
+template <typename Payload>
+using Error = BaseImplWithPayload<kErrorName, Payload>;
+
+}  // namespace iq
+
+template <typename Payload>
+using Iq = std::variant<iq::Get<Payload>, iq::Set<Payload>, iq::Result<Payload>, iq::Error<Payload>>;
+
+}  // namespace larra::xmpp