Add class ChatAvailableReactions.

1 files changed, 99 insertions, 0 deletions

diff --git a/td/telegram/ChatReactions.cpp b/td/telegram/ChatReactions.cpp
new file mode 100644
index 000000000..8e9e4d87d
--- /dev/null
+++ b/td/telegram/ChatReactions.cpp
@@ -0,0 +1,99 @@
+//
+// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
+//
+// Distributed under the Boost Software License, Version 1.0. (See accompanying
+// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
+//
+#include "td/telegram/ChatReactions.h"
+
+#include "td/telegram/MessageReaction.h"
+
+#include "td/utils/algorithm.h"
+
+namespace td {
+
+ChatReactions::ChatReactions(telegram_api::object_ptr<telegram_api::ChatReactions> &&chat_reactions_ptr) {
+  if (chat_reactions_ptr == nullptr) {
+    return;
+  }
+  switch (chat_reactions_ptr->get_id()) {
+    case telegram_api::chatReactionsNone::ID:
+      break;
+    case telegram_api::chatReactionsAll::ID: {
+      auto chat_reactions = move_tl_object_as<telegram_api::chatReactionsAll>(chat_reactions_ptr);
+      allow_all_ = true;
+      allow_custom_ = chat_reactions->allow_custom_;
+      break;
+    }
+    case telegram_api::chatReactionsSome::ID: {
+      auto chat_reactions = move_tl_object_as<telegram_api::chatReactionsSome>(chat_reactions_ptr);
+      reactions_ =
+          transform(chat_reactions->reactions_, [](const telegram_api::object_ptr<telegram_api::Reaction> &reaction) {
+            return get_message_reaction_string(reaction);
+          });
+      break;
+    }
+    default:
+      UNREACHABLE();
+  }
+}
+
+ChatReactions::ChatReactions(td_api::object_ptr<td_api::ChatAvailableReactions> &&chat_reactions_ptr,
+                             bool allow_custom) {
+  if (chat_reactions_ptr == nullptr) {
+    return;
+  }
+  switch (chat_reactions_ptr->get_id()) {
+    case td_api::chatAvailableReactionsAll::ID:
+      allow_all_ = true;
+      allow_custom_ = allow_custom;
+      break;
+    case td_api::chatAvailableReactionsSome::ID: {
+      auto chat_reactions = move_tl_object_as<td_api::chatAvailableReactionsSome>(chat_reactions_ptr);
+      reactions_ = transform(chat_reactions->reactions_, [](const td_api::object_ptr<td_api::ReactionType> &reaction) {
+        return get_message_reaction_string(reaction);
+      });
+      break;
+    }
+    default:
+      UNREACHABLE();
+  }
+}
+
+td_api::object_ptr<td_api::ChatAvailableReactions> ChatReactions::get_chat_available_reactions_object() const {
+  if (allow_all_) {
+    return td_api::make_object<td_api::chatAvailableReactionsAll>();
+  }
+  return td_api::make_object<td_api::chatAvailableReactionsSome>(transform(reactions_, get_reaction_type_object));
+}
+
+telegram_api::object_ptr<telegram_api::ChatReactions> ChatReactions::get_input_chat_reactions() const {
+  if (allow_all_) {
+    int32 flags = 0;
+    if (allow_custom_) {
+      flags |= telegram_api::chatReactionsAll::ALLOW_CUSTOM_MASK;
+    }
+    return telegram_api::make_object<telegram_api::chatReactionsAll>(flags, false /*ignored*/);
+  }
+  if (!reactions_.empty()) {
+    return telegram_api::make_object<telegram_api::chatReactionsSome>(transform(reactions_, get_input_reaction));
+  }
+  return telegram_api::make_object<telegram_api::chatReactionsNone>();
+}
+
+bool operator==(const ChatReactions &lhs, const ChatReactions &rhs) {
+  // don't compare allow_custom_
+  return lhs.reactions_ == rhs.reactions_ && lhs.allow_all_ == rhs.allow_all_;
+}
+
+StringBuilder &operator<<(StringBuilder &string_builder, const ChatReactions &reactions) {
+  if (reactions.allow_all_) {
+    if (reactions.allow_custom_) {
+      return string_builder << "AllReactions";
+    }
+    return string_builder << "AllRegularReactions";
+  }
+  return string_builder << '[' << reactions.reactions_ << ']';
+}
+
+}  // namespace td