From 92c03c4b9ac9355c79ab2d49dc1bd9a9bfd5d933 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Sat, 20 Jan 2018 23:24:07 +0300
Subject: [PATCH] Draft of get_message_content_caption.

GitOrigin-RevId: c01d6c4696315701681c1f50afa4011224a1fa2a
---
 td/telegram/MessagesManager.cpp | 25 +++++++++++++++++++++++--
 td/telegram/MessagesManager.h   |  2 ++
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/td/telegram/MessagesManager.cpp b/td/telegram/MessagesManager.cpp
index 7dc86cc2..57509e3e 100644
--- a/td/telegram/MessagesManager.cpp
+++ b/td/telegram/MessagesManager.cpp
@@ -14514,6 +14514,26 @@ MessageId MessagesManager::get_reply_to_message_id(Dialog *d, MessageId message_
   return message_id;
 }
 
+string MessagesManager::get_message_content_caption(const MessageContent *content) {
+  // TODO return with entities
+  switch (content->get_id()) {
+    case MessageAnimation::ID:
+      return static_cast<const MessageAnimation *>(content)->caption;
+    case MessageAudio::ID:
+      return static_cast<const MessageAudio *>(content)->caption;
+    case MessageDocument::ID:
+      return static_cast<const MessageDocument *>(content)->caption;
+    case MessagePhoto::ID:
+      return static_cast<const MessagePhoto *>(content)->caption;
+    case MessageVideo::ID:
+      return static_cast<const MessageVideo *>(content)->caption;
+    case MessageVoiceNote::ID:
+      return static_cast<const MessageVoiceNote *>(content)->caption;
+    default:
+      return string();
+  }
+}
+
 int32 MessagesManager::get_message_content_duration(const MessageContent *content) const {
   CHECK(content != nullptr);
   switch (content->get_id()) {
@@ -15674,8 +15694,6 @@ void MessagesManager::on_secret_message_media_uploaded(DialogId dialog_id, Messa
 
 void MessagesManager::on_upload_message_media_success(DialogId dialog_id, MessageId message_id,
                                                       tl_object_ptr<telegram_api::MessageMedia> &&media) {
-  auto content = get_message_content(string(), std::move(media), Auto(), dialog_id, false, UserId(), nullptr);
-
   Dialog *d = get_dialog(dialog_id);
   CHECK(d != nullptr);
 
@@ -15693,6 +15711,9 @@ void MessagesManager::on_upload_message_media_success(DialogId dialog_id, Messag
     return;  // the message should be deleted soon
   }
 
+  // TODO use get_message_content_caption()
+  auto content = get_message_content(string(), std::move(media), Auto(), dialog_id, false, UserId(), nullptr);
+
   update_message_content(dialog_id, m, m->content, std::move(content), true);
 
   auto input_media = get_input_media(m->content.get(), nullptr, nullptr, m->ttl);
diff --git a/td/telegram/MessagesManager.h b/td/telegram/MessagesManager.h
index b87953c2..64f7d5df 100644
--- a/td/telegram/MessagesManager.h
+++ b/td/telegram/MessagesManager.h
@@ -2207,6 +2207,8 @@ class MessagesManager : public Actor {
 
   tl_object_ptr<td_api::MessageContent> get_message_content_object(const MessageContent *content) const;
 
+  static string get_message_content_caption(const MessageContent *content);
+
   int32 get_message_content_duration(const MessageContent *content) const;
 
   static FileId get_message_content_file_id(const MessageContent *content);