From 9d7610a23c2a1ab6dba52f12ba819d8934119387 Mon Sep 17 00:00:00 2001
From: sha512sum <linuxnyasha@tutanota.com>
Date: Thu, 24 Oct 2024 09:50:00 +0000
Subject: [PATCH] Add send AnyMessage event

---
 include/yail/core.hpp | 22 ++++++++++++++++------
 main.cpp              |  5 +++++
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/include/yail/core.hpp b/include/yail/core.hpp
index 1ae9f77..d2b31b6 100644
--- a/include/yail/core.hpp
+++ b/include/yail/core.hpp
@@ -34,6 +34,10 @@ struct Ping {
   }
 };
 
+struct AnyMessage {
+  std::string_view data;
+};
+
 }  // namespace event
 
 template <typename F>
@@ -124,12 +128,15 @@ struct IrcClient {
     co_await boost::asio::async_write(this->socket, boost::asio::buffer(request), boost::asio::transfer_all(), boost::asio::use_awaitable);
     this->nick = std::move(nick);
   };
+
+  static constexpr auto kHandledEvents = []<typename... Ids, typename... FFs>(utempl::TypeList<std::tuple<Function<Ids, FFs>...>>) {
+    return utempl::kTypeList<Ids...>;
+  }(utempl::kType<Fs>);
+
   template <typename Event>
-  auto SendEvent(Event&& event) -> boost::asio::awaitable<void> {
-    return [&]<typename... Ids, typename... FFs>(utempl::TypeList<std::tuple<Function<Ids, FFs>...>>) {
-      constexpr auto i = utempl::Find<std::decay_t<Event>>(utempl::kTypeList<Ids...>);
-      return std::get<i>(this->callbacks).f(*this, std::forward<Event>(event));
-    }(utempl::kType<Fs>);
+  auto SendEvent(Event&& event)
+      -> decltype(std::get<utempl::Find<std::decay_t<Event>>(kHandledEvents)>(this->callbacks).f(*this, std::forward<Event>(event))) {
+    return std::get<utempl::Find<std::decay_t<Event>>(kHandledEvents)>(this->callbacks).f(*this, std::forward<Event>(event));
   };
   auto Pong(std::string_view buff) -> boost::asio::awaitable<void> {
     const auto pong = std::format("PONG {}", buff);
@@ -148,13 +155,16 @@ struct IrcClient {
           [&]<typename... Ids, typename... FFs>(Function<Ids, FFs>&... fs) -> boost::asio::awaitable<void> {
             auto self = this;
             auto& str = messageStr;
+            if constexpr(requires { self->SendEvent(event::AnyMessage{messageStr}); }) {
+              co_await self->SendEvent(event::AnyMessage{messageStr});
+            };
             // clang-format off
             (co_await [](auto self, auto& messageStr, auto& fs) -> boost::asio::awaitable<void> { // NOLINT
               if constexpr(requires {Ids::TryParse(*self, messageStr);}) {
                 if(auto value = Ids::TryParse(*self, messageStr)) {
                   co_await fs.f(*self, std::move(*value));
                 };
-                };
+              };
               }(self, str, fs), ...);
             // clang-format on
           },
diff --git a/main.cpp b/main.cpp
index 271ee5c..f60da1d 100644
--- a/main.cpp
+++ b/main.cpp
@@ -16,6 +16,11 @@ auto Test() -> boost::asio::awaitable<void> {
                                                                      .server = "irc.ilita.i2p"}
             .AddState<std::unordered_set<yail::User, yail::User::Hash>, "ruUsers">()
             // NOLINTNEXTLINE
+            .With([](auto& self, yail::event::AnyMessage message) -> boost::asio::awaitable<void> {
+              std::println("Readed message: {}", message.data);
+              co_return;
+            })
+            // NOLINTNEXTLINE
             .With([](auto& self, UserEvent event) -> boost::asio::awaitable<void> {
               std::println("User event!");
               co_return;