Add RichMessage::get_cover_any_file_ids.

3 files changed, 8 insertions, 5 deletions

diff --git a/td/telegram/MessageContent.cpp b/td/telegram/MessageContent.cpp
index 55f625695..c21d34fce 100644
--- a/td/telegram/MessageContent.cpp
+++ b/td/telegram/MessageContent.cpp
@@ -12455,11 +12455,7 @@ vector<FileId> get_message_content_cover_any_file_ids(const Td *td, const Messag
     }
     case MessageContentType::RichText: {
       const auto *m = static_cast<const MessageRichText *>(content);
-      vector<FileId> file_ids;
-      for (const auto *rich_text_content : m->rich_message.get_individual_message_content_refs()) {
-        file_ids.push_back(get_message_content_cover_any_file_id(rich_text_content));
-      }
-      return file_ids;
+      return m->rich_message.get_cover_any_file_ids();
     }
     default:
       break;
diff --git a/td/telegram/RichMessage.cpp b/td/telegram/RichMessage.cpp
index 14691973e..0b7b4db8f 100644
--- a/td/telegram/RichMessage.cpp
+++ b/td/telegram/RichMessage.cpp
@@ -152,6 +152,11 @@ vector<FileId> RichMessage::get_any_file_ids() const {
                    [](const MessageContent *content) { return get_message_content_any_file_id(content); });
 }
 
+vector<FileId> RichMessage::get_cover_any_file_ids() const {
+  return transform(get_individual_message_content_refs(),
+                   [](const MessageContent *content) { return get_message_content_cover_any_file_id(content); });
+}
+
 void RichMessage::append_file_ids(const Td *td, vector<FileId> &file_ids) const {
   for (const auto &block : blocks_) {
     block->append_file_ids(td, file_ids);
diff --git a/td/telegram/RichMessage.h b/td/telegram/RichMessage.h
index 77d98f25b..471b4c472 100644
--- a/td/telegram/RichMessage.h
+++ b/td/telegram/RichMessage.h
@@ -68,6 +68,8 @@ class RichMessage {
 
   vector<FileId> get_any_file_ids() const;
 
+  vector<FileId> get_cover_any_file_ids() const;
+
   void append_file_ids(const Td *td, vector<FileId> &file_ids) const;
 
   void add_dependencies(Dependencies &dependencies) const;