From b1db807868e3d4430010890f8fd6de627fd6cc45 Mon Sep 17 00:00:00 2001 From: levlam Date: Tue, 31 Aug 2021 23:37:21 +0300 Subject: [PATCH] Add forwardMessages.only_preview. --- td/generate/scheme/td_api.tl | 9 ++-- td/telegram/MessagesManager.cpp | 87 +++++++++++++++++++++++---------- td/telegram/MessagesManager.h | 10 ++-- td/telegram/Td.cpp | 2 +- td/telegram/cli.cpp | 8 +-- 5 files changed, 77 insertions(+), 39 deletions(-) diff --git a/td/generate/scheme/td_api.tl b/td/generate/scheme/td_api.tl index bc3fb45c3..590c4e847 100644 --- a/td/generate/scheme/td_api.tl +++ b/td/generate/scheme/td_api.tl @@ -1391,7 +1391,7 @@ paymentForm id:int64 invoice:invoice url:string seller_bot_user_id:int32 payment //@description Contains a temporary identifier of validated order information, which is stored for one hour. Also contains the available shipping options @order_info_id Temporary identifier of the order information @shipping_options Available shipping options validatedOrderInfo order_info_id:string shipping_options:vector = ValidatedOrderInfo; -//@description Contains the result of a payment request @success True, if the payment request was successful; otherwise the verification_url will be not empty @verification_url URL for additional payment credentials verification +//@description Contains the result of a payment request @success True, if the payment request was successful; otherwise the verification_url will be non-empty @verification_url URL for additional payment credentials verification paymentResult success:Bool verification_url:string = PaymentResult; //@description Contains information about a successful payment @@ -4302,9 +4302,10 @@ sendInlineQueryResultMessage chat_id:int53 message_thread_id:int53 reply_to_mess //@from_chat_id Identifier of the chat from which to forward messages //@message_ids Identifiers of the messages to forward. Message identifiers must be in a strictly increasing order. At most 100 messages can be forwarded simultaneously //@options Options to be used to send the messages -//@send_copy True, if content of the messages needs to be copied without links to the original messages. Always true if the messages are forwarded to a secret chat -//@remove_caption True, if media caption of message copies needs to be removed. Ignored if send_copy is false -forwardMessages chat_id:int53 from_chat_id:int53 message_ids:vector options:messageSendOptions send_copy:Bool remove_caption:Bool = Messages; +//@send_copy If true, content of the messages will be copied without links to the original messages. Always true if the messages are forwarded to a secret chat +//@remove_caption If true, media caption of message copies will be removed. Ignored if send_copy is false +//@only_preview If true, messages will not be forwarded and instead fake messages will be returned +forwardMessages chat_id:int53 from_chat_id:int53 message_ids:vector options:messageSendOptions send_copy:Bool remove_caption:Bool only_preview:Bool = Messages; //@description Resends messages which failed to send. Can be called only for messages for which messageSendingStateFailed.can_retry is true and after specified in messageSendingStateFailed.retry_after time passed. //-If a message is re-sent, the corresponding failed to send message is deleted. Returns the sent messages in the same order as the message identifiers passed in message_ids. If a message can't be re-sent, null will be returned instead of the message diff --git a/td/telegram/MessagesManager.cpp b/td/telegram/MessagesManager.cpp index e2a5c882e..b05b9f25e 100644 --- a/td/telegram/MessagesManager.cpp +++ b/td/telegram/MessagesManager.cpp @@ -23215,24 +23215,25 @@ MessagesManager::Message *MessagesManager::get_message_to_send( unique_ptr forward_info, bool is_copy) { d->was_opened = true; - auto m = create_message_to_send(d, top_thread_message_id, reply_to_message_id, options, std::move(content), - suppress_reply_info, std::move(forward_info), is_copy); + auto message = create_message_to_send(d, top_thread_message_id, reply_to_message_id, options, std::move(content), + suppress_reply_info, std::move(forward_info), is_copy); MessageId message_id = options.schedule_date != 0 ? get_next_yet_unsent_scheduled_message_id(d, options.schedule_date) : get_next_yet_unsent_message_id(d); - set_message_id(m, message_id); + set_message_id(message, message_id); - m->have_previous = true; - m->have_next = true; + message->have_previous = true; + message->have_next = true; do { - m->random_id = Random::secure_int64(); - } while (m->random_id == 0 || message_random_ids_.find(m->random_id) != message_random_ids_.end()); - message_random_ids_.insert(m->random_id); + message->random_id = Random::secure_int64(); + } while (message->random_id == 0 || message_random_ids_.find(message->random_id) != message_random_ids_.end()); + message_random_ids_.insert(message->random_id); bool need_update = false; CHECK(have_input_peer(d->dialog_id, AccessRights::Read)); - auto result = add_message_to_dialog(d, std::move(m), true, &need_update, need_update_dialog_pos, "send message"); + auto result = + add_message_to_dialog(d, std::move(message), true, &need_update, need_update_dialog_pos, "send message"); LOG_CHECK(result != nullptr) << message_id << " " << debug_add_message_to_dialog_fail_reason_; if (result->message_id.is_scheduled()) { send_update_chat_has_scheduled_messages(d, false); @@ -26280,7 +26281,7 @@ Result> MessagesManager::forward_message( vector all_copy_options; all_copy_options.push_back(std::move(copy_options)); TRY_RESULT(result, forward_messages(to_dialog_id, from_dialog_id, {message_id}, std::move(options), in_game_share, - std::move(all_copy_options))); + std::move(all_copy_options), false)); CHECK(result->messages_.size() == 1); if (result->messages_[0] == nullptr) { return Status::Error(400, @@ -26454,7 +26455,7 @@ Result MessagesManager::get_forwarded_messag for (size_t i = 0; i < message_ids.size(); i++) { MessageId message_id = get_persistent_message_id(from_dialog, message_ids[i]); - const Message *forwarded_message = get_message_force(from_dialog, message_id, "forward_messages"); + const Message *forwarded_message = get_message_force(from_dialog, message_id, "get_forwarded_messages"); if (forwarded_message == nullptr) { LOG(INFO) << "Can't find " << message_id << " to forward"; continue; @@ -26562,8 +26563,8 @@ Result MessagesManager::get_forwarded_messag Result> MessagesManager::forward_messages( DialogId to_dialog_id, DialogId from_dialog_id, vector message_ids, - tl_object_ptr &&options, bool in_game_share, - vector &©_options) { + tl_object_ptr &&options, bool in_game_share, vector &©_options, + bool only_preview) { TRY_RESULT(forwarded_messages_info, get_forwarded_messages(to_dialog_id, from_dialog_id, message_ids, std::move(options), in_game_share, std::move(copy_options))); @@ -26584,41 +26585,75 @@ Result> MessagesManager::forward_messages( auto content = std::move(forwarded_message_contents[j].content); auto forward_info = create_message_forward_info(from_dialog_id, to_dialog_id, forwarded_message); - Message *m = get_message_to_send(to_dialog, MessageId(), MessageId(), message_send_options, std::move(content), - &need_update_dialog_pos, j + 1 != forwarded_message_contents.size(), - std::move(forward_info)); + + unique_ptr message; + Message *m; + if (only_preview) { + message = create_message_to_send(to_dialog, MessageId(), MessageId(), message_send_options, std::move(content), + j + 1 != forwarded_message_contents.size(), std::move(forward_info), false); + MessageId new_message_id = + message_send_options.schedule_date != 0 + ? get_next_yet_unsent_scheduled_message_id(to_dialog, message_send_options.schedule_date) + : get_next_yet_unsent_message_id(to_dialog); + set_message_id(message, new_message_id); + m = message.get(); + } else { + m = get_message_to_send(to_dialog, MessageId(), MessageId(), message_send_options, std::move(content), + &need_update_dialog_pos, j + 1 != forwarded_message_contents.size(), + std::move(forward_info)); + } fix_forwarded_message(m, to_dialog_id, forwarded_message, forwarded_message_contents[j].media_album_id); m->in_game_share = in_game_share; m->real_forward_from_dialog_id = from_dialog_id; m->real_forward_from_message_id = message_id; - send_update_new_message(to_dialog, m); + if (!only_preview) { + send_update_new_message(to_dialog, m); + forwarded_messages.push_back(m); + forwarded_message_ids.push_back(message_id); + } result[forwarded_message_contents[j].index] = get_message_object(to_dialog_id, m, false); - forwarded_messages.push_back(m); - forwarded_message_ids.push_back(message_id); } if (!forwarded_messages.empty()) { + CHECK(!only_preview); do_forward_messages(to_dialog_id, from_dialog_id, forwarded_messages, forwarded_message_ids, 0); } for (auto &copied_message : copied_messages) { - Message *m = get_message_to_send(to_dialog, copied_message.top_thread_message_id, - copied_message.reply_to_message_id, message_send_options, - std::move(copied_message.content), &need_update_dialog_pos, false, nullptr, true); + unique_ptr message; + Message *m; + if (only_preview) { + message = + create_message_to_send(to_dialog, copied_message.top_thread_message_id, copied_message.reply_to_message_id, + message_send_options, std::move(copied_message.content), false, nullptr, true); + MessageId new_message_id = + message_send_options.schedule_date != 0 + ? get_next_yet_unsent_scheduled_message_id(to_dialog, message_send_options.schedule_date) + : get_next_yet_unsent_message_id(to_dialog); + set_message_id(message, new_message_id); + m = message.get(); + } else { + m = get_message_to_send(to_dialog, copied_message.top_thread_message_id, copied_message.reply_to_message_id, + message_send_options, std::move(copied_message.content), &need_update_dialog_pos, false, + nullptr, true); + } m->disable_web_page_preview = copied_message.disable_web_page_preview; m->media_album_id = copied_message.media_album_id; m->reply_markup = std::move(copied_message.reply_markup); - save_send_message_log_event(to_dialog_id, m); - do_send_message(to_dialog_id, m); - result[copied_message.index] = get_message_object(to_dialog_id, m, false); + if (!only_preview) { + save_send_message_log_event(to_dialog_id, m); + do_send_message(to_dialog_id, m); + send_update_new_message(to_dialog, m); + } - send_update_new_message(to_dialog, m); + result[copied_message.index] = get_message_object(to_dialog_id, m, false); } if (need_update_dialog_pos) { + CHECK(!only_preview); send_update_chat_last_message(to_dialog, "forward_messages"); } diff --git a/td/telegram/MessagesManager.h b/td/telegram/MessagesManager.h index f18d128f5..da7c3988d 100644 --- a/td/telegram/MessagesManager.h +++ b/td/telegram/MessagesManager.h @@ -393,10 +393,12 @@ class MessagesManager final : public Actor { int64 query_id, const string &result_id, bool hide_via_bot) TD_WARN_UNUSED_RESULT; - Result> forward_messages( - DialogId to_dialog_id, DialogId from_dialog_id, vector message_ids, - tl_object_ptr &&options, bool in_game_share, - vector &©_options) TD_WARN_UNUSED_RESULT; + Result> forward_messages(DialogId to_dialog_id, DialogId from_dialog_id, + vector message_ids, + tl_object_ptr &&options, + bool in_game_share, + vector &©_options, + bool only_preview) TD_WARN_UNUSED_RESULT; Result> resend_messages(DialogId dialog_id, vector message_ids) TD_WARN_UNUSED_RESULT; diff --git a/td/telegram/Td.cpp b/td/telegram/Td.cpp index 4bce981f6..26945c106 100644 --- a/td/telegram/Td.cpp +++ b/td/telegram/Td.cpp @@ -5837,7 +5837,7 @@ void Td::on_request(uint64 id, td_api::forwardMessages &request) { MessageId) { return MessageCopyOptions(send_copy, remove_caption); }); auto r_messages = messages_manager_->forward_messages(DialogId(request.chat_id_), DialogId(request.from_chat_id_), std::move(input_message_ids), std::move(request.options_), - false, std::move(message_copy_options)); + false, std::move(message_copy_options), request.only_preview_); if (r_messages.is_error()) { send_closure(actor_id(this), &Td::send_error, id, r_messages.move_as_error()); } else { diff --git a/td/telegram/cli.cpp b/td/telegram/cli.cpp index ac8e505a2..020a2aefb 100644 --- a/td/telegram/cli.cpp +++ b/td/telegram/cli.cpp @@ -2698,15 +2698,15 @@ class CliClient final : public Actor { get_args(args, chat_id, message_ids); send_request( td_api::make_object(as_chat_id(chat_id), as_message_ids(message_ids), op == "dmr")); - } else if (op == "fm" || op == "cm") { + } else if (op == "fm" || op == "cm" || op == "fmp" || op == "cmp") { string chat_id; string from_chat_id; string message_ids; get_args(args, chat_id, from_chat_id, message_ids); auto chat = as_chat_id(chat_id); - send_request( - td_api::make_object(chat, as_chat_id(from_chat_id), as_message_ids(message_ids), - default_message_send_options(), op[0] == 'c', rand_bool())); + send_request(td_api::make_object( + chat, as_chat_id(from_chat_id), as_message_ids(message_ids), default_message_send_options(), op[0] == 'c', + rand_bool(), op.back() == 'p')); } else if (op == "resend") { string chat_id; string message_ids;