Support messageRichText in update_used_hashtags.

Author: levlam

td/telegram/MessageContent.cpp

@@ -13289,6 +13289,11 @@ void on_dialog_used(TopDialogCategory category, DialogId dialog_id, int32 date)
 void update_used_hashtags(Td *td, const MessageContent *content) {
   const FormattedText *text = get_message_content_text(content);
   if (text == nullptr || text->text.empty()) {
+    if (content->get_type() == MessageContentType::RichText) {
+      for (const auto &hashtag : static_cast<const MessageRichText *>(content)->text.get_hashtags()) {
+        send_closure(td->hashtag_hints_, &HashtagHints::hashtag_used, hashtag);
+      }
+    }
     return;
   }
 

td/telegram/RichMessage.cpp

@@ -15,6 +15,7 @@
 #include "td/telegram/Td.h"
 #include "td/telegram/telegram_api.h"
 
+#include "td/utils/algorithm.h"
 #include "td/utils/common.h"
 #include "td/utils/FlatHashMap.h"
 #include "td/utils/logging.h"
@@ -116,6 +117,14 @@ bool RichMessage::has_bot_commands() const {
   return false;
 }
 
+vector<string> RichMessage::get_hashtags() const {
+  vector<string> hashtags;
+  for (const auto &block : blocks_) {
+    td::append(hashtags, block->get_hashtags());
+  }
+  return hashtags;
+}
+
 td_api::object_ptr<td_api::richMessage> RichMessage::get_rich_message_object(Td *td, bool skip_bot_commands) const {
   return td_api::make_object<td_api::richMessage>(
       get_page_blocks_object(blocks_, td, string(), string(), skip_bot_commands), is_rtl_, is_full_);

td/telegram/RichMessage.h

@@ -52,6 +52,8 @@ class RichMessage {
 
   bool has_bot_commands() const;
 
+  vector<string> get_hashtags() const;
+
   td_api::object_ptr<td_api::richMessage> get_rich_message_object(Td *td, bool skip_bot_commands) const;
 
   RichMessage clone() const;

td/telegram/WebPageBlock.cpp

@@ -3526,6 +3526,29 @@ bool WebPageBlock::has_bot_commands() const {
   return result;
 }
 
+vector<string> WebPageBlock::get_hashtags() const {
+  vector<string> result;
+  for_each_rich_text([&](const RichText *text) {
+    if (text->type == RichText::Type::Hashtag) {
+      string hashtag;
+      text->for_each_rich_text([&](const RichText *text) {
+        if (text->type == RichText::Type::Plain) {
+          hashtag += text->content;
+        }
+      });
+      if (!hashtag.empty()) {
+        if (hashtag[0] == '#') {
+          hashtag = hashtag.substr(1);
+        }
+        if (!hashtag.empty()) {
+          result.push_back(hashtag);
+        }
+      }
+    }
+  });
+  return result;
+}
+
 template <class F>
 void WebPageBlock::call_impl(Type type, const WebPageBlock *ptr, F &&f) {
   switch (type) {

td/telegram/WebPageBlock.h

@@ -108,6 +108,8 @@ class WebPageBlock {
 
   bool has_bot_commands() const;
 
+  vector<string> get_hashtags() const;
+
   virtual unique_ptr<WebPageBlock> clone() const = 0;
 
   virtual td_api::object_ptr<td_api::PageBlock> get_page_block_object(Context *context) const = 0;