Move check/join by chat invite link to DialogInviteLinkManager.
This commit is contained in:
parent
ebf2f95681
commit
409778b760
@ -16,6 +16,7 @@
|
||||
#include "td/telegram/ConfigManager.h"
|
||||
#include "td/telegram/Dependencies.h"
|
||||
#include "td/telegram/DialogInviteLink.h"
|
||||
#include "td/telegram/DialogInviteLinkManager.h"
|
||||
#include "td/telegram/DialogLocation.h"
|
||||
#include "td/telegram/DialogManager.h"
|
||||
#include "td/telegram/DialogOnlineMemberManager.h"
|
||||
@ -2912,81 +2913,6 @@ class DeleteRevokedExportedChatInvitesQuery final : public Td::ResultHandler {
|
||||
}
|
||||
};
|
||||
|
||||
class CheckChatInviteQuery final : public Td::ResultHandler {
|
||||
Promise<Unit> promise_;
|
||||
string invite_link_;
|
||||
|
||||
public:
|
||||
explicit CheckChatInviteQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
||||
}
|
||||
|
||||
void send(const string &invite_link) {
|
||||
invite_link_ = invite_link;
|
||||
send_query(G()->net_query_creator().create(
|
||||
telegram_api::messages_checkChatInvite(LinkManager::get_dialog_invite_link_hash(invite_link_))));
|
||||
}
|
||||
|
||||
void on_result(BufferSlice packet) final {
|
||||
auto result_ptr = fetch_result<telegram_api::messages_checkChatInvite>(packet);
|
||||
if (result_ptr.is_error()) {
|
||||
return on_error(result_ptr.move_as_error());
|
||||
}
|
||||
|
||||
auto ptr = result_ptr.move_as_ok();
|
||||
LOG(INFO) << "Receive result for CheckChatInviteQuery: " << to_string(ptr);
|
||||
|
||||
td_->contacts_manager_->on_get_dialog_invite_link_info(invite_link_, std::move(ptr), std::move(promise_));
|
||||
}
|
||||
|
||||
void on_error(Status status) final {
|
||||
promise_.set_error(std::move(status));
|
||||
}
|
||||
};
|
||||
|
||||
class ImportChatInviteQuery final : public Td::ResultHandler {
|
||||
Promise<DialogId> promise_;
|
||||
|
||||
string invite_link_;
|
||||
|
||||
public:
|
||||
explicit ImportChatInviteQuery(Promise<DialogId> &&promise) : promise_(std::move(promise)) {
|
||||
}
|
||||
|
||||
void send(const string &invite_link) {
|
||||
invite_link_ = invite_link;
|
||||
send_query(G()->net_query_creator().create(
|
||||
telegram_api::messages_importChatInvite(LinkManager::get_dialog_invite_link_hash(invite_link_))));
|
||||
}
|
||||
|
||||
void on_result(BufferSlice packet) final {
|
||||
auto result_ptr = fetch_result<telegram_api::messages_importChatInvite>(packet);
|
||||
if (result_ptr.is_error()) {
|
||||
return on_error(result_ptr.move_as_error());
|
||||
}
|
||||
|
||||
auto ptr = result_ptr.move_as_ok();
|
||||
LOG(INFO) << "Receive result for ImportChatInviteQuery: " << to_string(ptr);
|
||||
|
||||
auto dialog_ids = UpdatesManager::get_chat_dialog_ids(ptr.get());
|
||||
if (dialog_ids.size() != 1u) {
|
||||
LOG(ERROR) << "Receive wrong result for ImportChatInviteQuery: " << to_string(ptr);
|
||||
return on_error(Status::Error(500, "Internal Server Error: failed to join chat via invite link"));
|
||||
}
|
||||
auto dialog_id = dialog_ids[0];
|
||||
|
||||
td_->contacts_manager_->invalidate_invite_link_info(invite_link_);
|
||||
td_->updates_manager_->on_get_updates(
|
||||
std::move(ptr), PromiseCreator::lambda([promise = std::move(promise_), dialog_id](Unit) mutable {
|
||||
promise.set_value(std::move(dialog_id));
|
||||
}));
|
||||
}
|
||||
|
||||
void on_error(Status status) final {
|
||||
td_->contacts_manager_->invalidate_invite_link_info(invite_link_);
|
||||
promise_.set_error(std::move(status));
|
||||
}
|
||||
};
|
||||
|
||||
class DeleteChatUserQuery final : public Td::ResultHandler {
|
||||
Promise<Unit> promise_;
|
||||
|
||||
@ -4134,11 +4060,11 @@ ContactsManager::~ContactsManager() {
|
||||
invalidated_channels_full_, channel_full_file_source_ids_, secret_chats_, unknown_secret_chats_,
|
||||
secret_chats_with_user_);
|
||||
Scheduler::instance()->destroy_on_scheduler(
|
||||
G()->get_gc_scheduler_id(), invite_link_infos_, dialog_access_by_invite_link_, loaded_from_database_users_,
|
||||
unavailable_user_fulls_, loaded_from_database_chats_, unavailable_chat_fulls_, loaded_from_database_channels_,
|
||||
unavailable_channel_fulls_, loaded_from_database_secret_chats_, dialog_administrators_,
|
||||
user_online_member_dialogs_, cached_channel_participants_, resolved_phone_numbers_, channel_participants_,
|
||||
all_imported_contacts_, linked_channel_ids_, restricted_user_ids_, restricted_channel_ids_);
|
||||
G()->get_gc_scheduler_id(), dialog_access_by_invite_link_, loaded_from_database_users_, unavailable_user_fulls_,
|
||||
loaded_from_database_chats_, unavailable_chat_fulls_, loaded_from_database_channels_, unavailable_channel_fulls_,
|
||||
loaded_from_database_secret_chats_, dialog_administrators_, user_online_member_dialogs_,
|
||||
cached_channel_participants_, resolved_phone_numbers_, channel_participants_, all_imported_contacts_,
|
||||
linked_channel_ids_, restricted_user_ids_, restricted_channel_ids_);
|
||||
}
|
||||
|
||||
void ContactsManager::start_up() {
|
||||
@ -4345,7 +4271,7 @@ void ContactsManager::on_invite_link_info_expire_timeout(DialogId dialog_id) {
|
||||
if (access_it == dialog_access_by_invite_link_.end()) {
|
||||
return;
|
||||
}
|
||||
auto expires_in = access_it->second.accessible_before - G()->unix_time() - 1;
|
||||
auto expires_in = access_it->second.accessible_before_date - G()->unix_time() - 1;
|
||||
if (expires_in >= 3) {
|
||||
invite_link_info_expire_timeout_.set_timeout_in(dialog_id.get(), expires_in);
|
||||
return;
|
||||
@ -9581,33 +9507,6 @@ void ContactsManager::delete_all_revoked_dialog_invite_links(DialogId dialog_id,
|
||||
->send(dialog_id, std::move(input_user));
|
||||
}
|
||||
|
||||
void ContactsManager::check_dialog_invite_link(const string &invite_link, bool force, Promise<Unit> &&promise) {
|
||||
auto it = invite_link_infos_.find(invite_link);
|
||||
if (it != invite_link_infos_.end()) {
|
||||
auto dialog_id = it->second->dialog_id;
|
||||
if (!force && dialog_id.get_type() == DialogType::Chat && !get_chat_is_active(dialog_id.get_chat_id())) {
|
||||
invite_link_infos_.erase(it);
|
||||
} else {
|
||||
return promise.set_value(Unit());
|
||||
}
|
||||
}
|
||||
|
||||
if (!DialogInviteLink::is_valid_invite_link(invite_link)) {
|
||||
return promise.set_error(Status::Error(400, "Wrong invite link"));
|
||||
}
|
||||
|
||||
CHECK(!invite_link.empty());
|
||||
td_->create_handler<CheckChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
}
|
||||
|
||||
void ContactsManager::import_dialog_invite_link(const string &invite_link, Promise<DialogId> &&promise) {
|
||||
if (!DialogInviteLink::is_valid_invite_link(invite_link)) {
|
||||
return promise.set_error(Status::Error(400, "Wrong invite link"));
|
||||
}
|
||||
|
||||
td_->create_handler<ImportChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
}
|
||||
|
||||
void ContactsManager::delete_chat_participant(ChatId chat_id, UserId user_id, bool revoke_messages,
|
||||
Promise<Unit> &&promise) {
|
||||
const Chat *c = get_chat(chat_id);
|
||||
@ -16165,119 +16064,24 @@ void ContactsManager::on_update_channel_full_slow_mode_next_send_date(ChannelFul
|
||||
}
|
||||
}
|
||||
|
||||
void ContactsManager::on_get_dialog_invite_link_info(const string &invite_link,
|
||||
tl_object_ptr<telegram_api::ChatInvite> &&chat_invite_ptr,
|
||||
Promise<Unit> &&promise) {
|
||||
CHECK(chat_invite_ptr != nullptr);
|
||||
switch (chat_invite_ptr->get_id()) {
|
||||
case telegram_api::chatInviteAlready::ID:
|
||||
case telegram_api::chatInvitePeek::ID: {
|
||||
telegram_api::object_ptr<telegram_api::Chat> chat = nullptr;
|
||||
int32 accessible_before = 0;
|
||||
if (chat_invite_ptr->get_id() == telegram_api::chatInviteAlready::ID) {
|
||||
auto chat_invite_already = move_tl_object_as<telegram_api::chatInviteAlready>(chat_invite_ptr);
|
||||
chat = std::move(chat_invite_already->chat_);
|
||||
} else {
|
||||
auto chat_invite_peek = move_tl_object_as<telegram_api::chatInvitePeek>(chat_invite_ptr);
|
||||
chat = std::move(chat_invite_peek->chat_);
|
||||
accessible_before = chat_invite_peek->expires_;
|
||||
}
|
||||
auto chat_id = get_chat_id(chat);
|
||||
if (chat_id != ChatId() && !chat_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << chat_id;
|
||||
chat_id = ChatId();
|
||||
}
|
||||
auto channel_id = get_channel_id(chat);
|
||||
if (channel_id != ChannelId() && !channel_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << channel_id;
|
||||
channel_id = ChannelId();
|
||||
}
|
||||
if (accessible_before != 0 && (!channel_id.is_valid() || accessible_before < 0)) {
|
||||
LOG(ERROR) << "Receive expires = " << accessible_before << " for invite link " << invite_link << " to "
|
||||
<< to_string(chat);
|
||||
accessible_before = 0;
|
||||
}
|
||||
on_get_chat(std::move(chat), "chatInviteAlready");
|
||||
|
||||
CHECK(chat_id == ChatId() || channel_id == ChannelId());
|
||||
|
||||
// the access is already expired, reget the info
|
||||
if (accessible_before != 0 && accessible_before <= G()->unix_time() + 1) {
|
||||
td_->create_handler<CheckChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
return;
|
||||
}
|
||||
|
||||
DialogId dialog_id = chat_id.is_valid() ? DialogId(chat_id) : DialogId(channel_id);
|
||||
auto &invite_link_info = invite_link_infos_[invite_link];
|
||||
if (invite_link_info == nullptr) {
|
||||
invite_link_info = make_unique<InviteLinkInfo>();
|
||||
}
|
||||
invite_link_info->dialog_id = dialog_id;
|
||||
if (accessible_before != 0 && dialog_id.is_valid()) {
|
||||
void ContactsManager::add_dialog_access_by_invite_link(DialogId dialog_id, const string &invite_link,
|
||||
int32 accessible_before_date) {
|
||||
auto &access = dialog_access_by_invite_link_[dialog_id];
|
||||
access.invite_links.insert(invite_link);
|
||||
if (access.accessible_before < accessible_before) {
|
||||
access.accessible_before = accessible_before;
|
||||
if (access.accessible_before_date < accessible_before_date) {
|
||||
access.accessible_before_date = accessible_before_date;
|
||||
|
||||
auto expires_in = accessible_before - G()->unix_time() - 1;
|
||||
auto expires_in = accessible_before_date - G()->unix_time() - 1;
|
||||
invite_link_info_expire_timeout_.set_timeout_in(dialog_id.get(), expires_in);
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
case telegram_api::chatInvite::ID: {
|
||||
auto chat_invite = move_tl_object_as<telegram_api::chatInvite>(chat_invite_ptr);
|
||||
vector<UserId> participant_user_ids;
|
||||
for (auto &user : chat_invite->participants_) {
|
||||
auto user_id = get_user_id(user);
|
||||
if (!user_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << user_id;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
on_get_user(std::move(user), "chatInvite");
|
||||
participant_user_ids.push_back(user_id);
|
||||
int32 ContactsManager::get_dialog_accessible_by_invite_link_before_date(DialogId dialog_id) const {
|
||||
auto it = dialog_access_by_invite_link_.find(dialog_id);
|
||||
if (it != dialog_access_by_invite_link_.end()) {
|
||||
return td::max(1, it->second.accessible_before_date - G()->unix_time() - 1);
|
||||
}
|
||||
|
||||
auto &invite_link_info = invite_link_infos_[invite_link];
|
||||
if (invite_link_info == nullptr) {
|
||||
invite_link_info = make_unique<InviteLinkInfo>();
|
||||
}
|
||||
invite_link_info->dialog_id = DialogId();
|
||||
invite_link_info->title = chat_invite->title_;
|
||||
invite_link_info->photo = get_photo(td_, std::move(chat_invite->photo_), DialogId());
|
||||
invite_link_info->accent_color_id = AccentColorId(chat_invite->color_);
|
||||
invite_link_info->description = std::move(chat_invite->about_);
|
||||
invite_link_info->participant_count = chat_invite->participants_count_;
|
||||
invite_link_info->participant_user_ids = std::move(participant_user_ids);
|
||||
invite_link_info->creates_join_request = std::move(chat_invite->request_needed_);
|
||||
invite_link_info->is_chat = !chat_invite->channel_;
|
||||
invite_link_info->is_channel = chat_invite->channel_;
|
||||
|
||||
bool is_broadcast = chat_invite->broadcast_;
|
||||
bool is_public = chat_invite->public_;
|
||||
bool is_megagroup = chat_invite->megagroup_;
|
||||
|
||||
if (!invite_link_info->is_channel) {
|
||||
if (is_broadcast || is_public || is_megagroup) {
|
||||
LOG(ERROR) << "Receive wrong chat invite: " << to_string(chat_invite);
|
||||
is_public = is_megagroup = false;
|
||||
}
|
||||
} else {
|
||||
LOG_IF(ERROR, is_broadcast == is_megagroup) << "Receive wrong chat invite: " << to_string(chat_invite);
|
||||
}
|
||||
|
||||
invite_link_info->is_public = is_public;
|
||||
invite_link_info->is_megagroup = is_megagroup;
|
||||
invite_link_info->is_verified = chat_invite->verified_;
|
||||
invite_link_info->is_scam = chat_invite->scam_;
|
||||
invite_link_info->is_fake = chat_invite->fake_;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
UNREACHABLE();
|
||||
}
|
||||
promise.set_value(Unit());
|
||||
return 0;
|
||||
}
|
||||
|
||||
void ContactsManager::remove_dialog_access_by_invite_link(DialogId dialog_id) {
|
||||
@ -16287,7 +16091,7 @@ void ContactsManager::remove_dialog_access_by_invite_link(DialogId dialog_id) {
|
||||
}
|
||||
|
||||
for (auto &invite_link : access_it->second.invite_links) {
|
||||
invalidate_invite_link_info(invite_link);
|
||||
td_->dialog_invite_link_manager_->invalidate_invite_link_info(invite_link);
|
||||
}
|
||||
dialog_access_by_invite_link_.erase(access_it);
|
||||
|
||||
@ -16298,7 +16102,7 @@ bool ContactsManager::update_permanent_invite_link(DialogInviteLink &invite_link
|
||||
if (new_invite_link != invite_link) {
|
||||
if (invite_link.is_valid() && invite_link.get_invite_link() != new_invite_link.get_invite_link()) {
|
||||
// old link was invalidated
|
||||
invite_link_infos_.erase(invite_link.get_invite_link());
|
||||
td_->dialog_invite_link_manager_->invalidate_invite_link_info(invite_link.get_invite_link());
|
||||
}
|
||||
|
||||
invite_link = std::move(new_invite_link);
|
||||
@ -16307,11 +16111,6 @@ bool ContactsManager::update_permanent_invite_link(DialogInviteLink &invite_link
|
||||
return false;
|
||||
}
|
||||
|
||||
void ContactsManager::invalidate_invite_link_info(const string &invite_link) {
|
||||
LOG(INFO) << "Invalidate info about invite link " << invite_link;
|
||||
invite_link_infos_.erase(invite_link);
|
||||
}
|
||||
|
||||
bool ContactsManager::need_poll_user_active_stories(const User *u, UserId user_id) const {
|
||||
return u != nullptr && user_id != get_my_id() && !is_user_contact(u, user_id, false) && !is_user_bot(u) &&
|
||||
!is_user_support(u) && !is_user_deleted(u) && u->was_online != 0;
|
||||
@ -18548,6 +18347,22 @@ bool ContactsManager::get_channel_is_verified(ChannelId channel_id) const {
|
||||
return c->is_verified;
|
||||
}
|
||||
|
||||
bool ContactsManager::get_channel_is_scam(ChannelId channel_id) const {
|
||||
auto c = get_channel(channel_id);
|
||||
if (c == nullptr) {
|
||||
return false;
|
||||
}
|
||||
return c->is_scam;
|
||||
}
|
||||
|
||||
bool ContactsManager::get_channel_is_fake(ChannelId channel_id) const {
|
||||
auto c = get_channel(channel_id);
|
||||
if (c == nullptr) {
|
||||
return false;
|
||||
}
|
||||
return c->is_fake;
|
||||
}
|
||||
|
||||
bool ContactsManager::get_channel_sign_messages(ChannelId channel_id) const {
|
||||
auto c = get_channel(channel_id);
|
||||
if (c == nullptr) {
|
||||
@ -20575,117 +20390,6 @@ tl_object_ptr<td_api::secretChat> ContactsManager::get_secret_chat_object_const(
|
||||
secret_chat->is_outbound, secret_chat->key_hash, secret_chat->layer);
|
||||
}
|
||||
|
||||
tl_object_ptr<td_api::chatInviteLinkInfo> ContactsManager::get_chat_invite_link_info_object(const string &invite_link) {
|
||||
auto it = invite_link_infos_.find(invite_link);
|
||||
if (it == invite_link_infos_.end()) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
auto invite_link_info = it->second.get();
|
||||
CHECK(invite_link_info != nullptr);
|
||||
|
||||
DialogId dialog_id = invite_link_info->dialog_id;
|
||||
bool is_chat = false;
|
||||
bool is_megagroup = false;
|
||||
string title;
|
||||
const DialogPhoto *photo = nullptr;
|
||||
DialogPhoto invite_link_photo;
|
||||
int32 accent_color_id_object;
|
||||
string description;
|
||||
int32 participant_count = 0;
|
||||
vector<int64> member_user_ids;
|
||||
bool creates_join_request = false;
|
||||
bool is_public = false;
|
||||
bool is_member = false;
|
||||
bool is_verified = false;
|
||||
bool is_scam = false;
|
||||
bool is_fake = false;
|
||||
|
||||
if (dialog_id.is_valid()) {
|
||||
switch (dialog_id.get_type()) {
|
||||
case DialogType::Chat: {
|
||||
auto chat_id = dialog_id.get_chat_id();
|
||||
const Chat *c = get_chat_force(chat_id, "get_chat_invite_link_info_object");
|
||||
is_chat = true;
|
||||
|
||||
if (c != nullptr) {
|
||||
title = c->title;
|
||||
photo = &c->photo;
|
||||
participant_count = c->participant_count;
|
||||
is_member = c->status.is_member();
|
||||
} else {
|
||||
LOG(ERROR) << "Have no information about " << chat_id;
|
||||
}
|
||||
accent_color_id_object = get_chat_accent_color_id_object(chat_id);
|
||||
break;
|
||||
}
|
||||
case DialogType::Channel: {
|
||||
auto channel_id = dialog_id.get_channel_id();
|
||||
const Channel *c = get_channel_force(channel_id, "get_chat_invite_link_info_object");
|
||||
|
||||
if (c != nullptr) {
|
||||
title = c->title;
|
||||
photo = &c->photo;
|
||||
is_public = is_channel_public(c);
|
||||
is_megagroup = c->is_megagroup;
|
||||
participant_count = c->participant_count;
|
||||
is_member = c->status.is_member();
|
||||
is_verified = c->is_verified;
|
||||
is_scam = c->is_scam;
|
||||
is_fake = c->is_fake;
|
||||
} else {
|
||||
LOG(ERROR) << "Have no information about " << channel_id;
|
||||
}
|
||||
accent_color_id_object = get_channel_accent_color_id_object(channel_id);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
UNREACHABLE();
|
||||
}
|
||||
description = get_dialog_about(dialog_id);
|
||||
} else {
|
||||
is_chat = invite_link_info->is_chat;
|
||||
is_megagroup = invite_link_info->is_megagroup;
|
||||
title = invite_link_info->title;
|
||||
invite_link_photo = as_fake_dialog_photo(invite_link_info->photo, dialog_id, false);
|
||||
photo = &invite_link_photo;
|
||||
accent_color_id_object = td_->theme_manager_->get_accent_color_id_object(invite_link_info->accent_color_id);
|
||||
description = invite_link_info->description;
|
||||
participant_count = invite_link_info->participant_count;
|
||||
member_user_ids = get_user_ids_object(invite_link_info->participant_user_ids, "get_chat_invite_link_info_object");
|
||||
creates_join_request = invite_link_info->creates_join_request;
|
||||
is_public = invite_link_info->is_public;
|
||||
is_verified = invite_link_info->is_verified;
|
||||
is_scam = invite_link_info->is_scam;
|
||||
is_fake = invite_link_info->is_fake;
|
||||
}
|
||||
|
||||
td_api::object_ptr<td_api::InviteLinkChatType> chat_type;
|
||||
if (is_chat) {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeBasicGroup>();
|
||||
} else if (is_megagroup) {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeSupergroup>();
|
||||
} else {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeChannel>();
|
||||
}
|
||||
|
||||
if (dialog_id.is_valid()) {
|
||||
td_->dialog_manager_->force_create_dialog(dialog_id, "get_chat_invite_link_info_object");
|
||||
}
|
||||
int32 accessible_for = 0;
|
||||
if (dialog_id.is_valid() && !is_member) {
|
||||
auto access_it = dialog_access_by_invite_link_.find(dialog_id);
|
||||
if (access_it != dialog_access_by_invite_link_.end()) {
|
||||
accessible_for = td::max(1, access_it->second.accessible_before - G()->unix_time() - 1);
|
||||
}
|
||||
}
|
||||
|
||||
return td_api::make_object<td_api::chatInviteLinkInfo>(
|
||||
td_->dialog_manager_->get_chat_id_object(dialog_id, "chatInviteLinkInfo"), accessible_for, std::move(chat_type),
|
||||
title, get_chat_photo_info_object(td_->file_manager_.get(), photo), accent_color_id_object, description,
|
||||
participant_count, std::move(member_user_ids), creates_join_request, is_public, is_verified, is_scam, is_fake);
|
||||
}
|
||||
|
||||
void ContactsManager::get_support_user(Promise<td_api::object_ptr<td_api::user>> &&promise) {
|
||||
if (support_user_id_.is_valid()) {
|
||||
return promise.set_value(get_user_object(support_user_id_));
|
||||
|
@ -301,12 +301,6 @@ class ContactsManager final : public Actor {
|
||||
|
||||
void on_get_permanent_dialog_invite_link(DialogId dialog_id, const DialogInviteLink &invite_link);
|
||||
|
||||
void on_get_dialog_invite_link_info(const string &invite_link,
|
||||
tl_object_ptr<telegram_api::ChatInvite> &&chat_invite_ptr,
|
||||
Promise<Unit> &&promise);
|
||||
|
||||
void invalidate_invite_link_info(const string &invite_link);
|
||||
|
||||
void on_get_created_public_channels(PublicDialogType type, vector<tl_object_ptr<telegram_api::Chat>> &&chats);
|
||||
|
||||
void on_get_dialogs_for_discussion(vector<tl_object_ptr<telegram_api::Chat>> &&chats);
|
||||
@ -315,6 +309,10 @@ class ContactsManager final : public Actor {
|
||||
|
||||
void remove_inactive_channel(ChannelId channel_id);
|
||||
|
||||
void add_dialog_access_by_invite_link(DialogId dialog_id, const string &invite_link, int32 accessible_before_date);
|
||||
|
||||
int32 get_dialog_accessible_by_invite_link_before_date(DialogId dialog_id) const;
|
||||
|
||||
void register_message_users(MessageFullId message_full_id, vector<UserId> user_ids);
|
||||
|
||||
void register_message_channels(MessageFullId message_full_id, vector<ChannelId> channel_ids);
|
||||
@ -583,10 +581,6 @@ class ContactsManager final : public Actor {
|
||||
|
||||
void delete_all_revoked_dialog_invite_links(DialogId dialog_id, UserId creator_user_id, Promise<Unit> &&promise);
|
||||
|
||||
void check_dialog_invite_link(const string &invite_link, bool force, Promise<Unit> &&promise);
|
||||
|
||||
void import_dialog_invite_link(const string &invite_link, Promise<DialogId> &&promise);
|
||||
|
||||
ChannelId migrate_chat_to_megagroup(ChatId chat_id, Promise<Unit> &promise);
|
||||
|
||||
void get_channel_recommendations(DialogId dialog_id, bool return_local,
|
||||
@ -699,6 +693,8 @@ class ContactsManager final : public Actor {
|
||||
DialogParticipantStatus get_channel_status(ChannelId channel_id) const;
|
||||
DialogParticipantStatus get_channel_permissions(ChannelId channel_id) const;
|
||||
bool get_channel_is_verified(ChannelId channel_id) const;
|
||||
bool get_channel_is_scam(ChannelId channel_id) const;
|
||||
bool get_channel_is_fake(ChannelId channel_id) const;
|
||||
int32 get_channel_participant_count(ChannelId channel_id) const;
|
||||
bool get_channel_sign_messages(ChannelId channel_id) const;
|
||||
bool get_channel_has_linked_channel(ChannelId channel_id) const;
|
||||
@ -769,8 +765,6 @@ class ContactsManager final : public Actor {
|
||||
tl_object_ptr<td_api::chatMember> get_chat_member_object(const DialogParticipant &dialog_participant,
|
||||
const char *source) const;
|
||||
|
||||
tl_object_ptr<td_api::chatInviteLinkInfo> get_chat_invite_link_info_object(const string &invite_link);
|
||||
|
||||
void get_support_user(Promise<td_api::object_ptr<td_api::user>> &&promise);
|
||||
|
||||
void on_view_dialog_active_stories(vector<DialogId> dialog_ids);
|
||||
@ -1186,27 +1180,6 @@ class ContactsManager final : public Actor {
|
||||
void parse(ParserT &parser);
|
||||
};
|
||||
|
||||
struct InviteLinkInfo {
|
||||
// known dialog
|
||||
DialogId dialog_id;
|
||||
|
||||
// unknown dialog
|
||||
string title;
|
||||
Photo photo;
|
||||
AccentColorId accent_color_id;
|
||||
int32 participant_count = 0;
|
||||
vector<UserId> participant_user_ids;
|
||||
string description;
|
||||
bool creates_join_request = false;
|
||||
bool is_chat = false;
|
||||
bool is_channel = false;
|
||||
bool is_public = false;
|
||||
bool is_megagroup = false;
|
||||
bool is_verified = false;
|
||||
bool is_scam = false;
|
||||
bool is_fake = false;
|
||||
};
|
||||
|
||||
struct PendingGetPhotoRequest {
|
||||
int32 offset = 0;
|
||||
int32 limit = 0;
|
||||
@ -2068,9 +2041,8 @@ class ContactsManager final : public Actor {
|
||||
|
||||
struct DialogAccessByInviteLink {
|
||||
FlatHashSet<string> invite_links;
|
||||
int32 accessible_before = 0;
|
||||
int32 accessible_before_date = 0;
|
||||
};
|
||||
FlatHashMap<string, unique_ptr<InviteLinkInfo>> invite_link_infos_;
|
||||
FlatHashMap<DialogId, DialogAccessByInviteLink, DialogIdHash> dialog_access_by_invite_link_;
|
||||
|
||||
FlatHashMap<ChannelId, RecommendedDialogs, ChannelIdHash> channel_recommended_dialogs_;
|
||||
|
@ -6,8 +6,96 @@
|
||||
//
|
||||
#include "td/telegram/DialogInviteLinkManager.h"
|
||||
|
||||
#include "td/telegram/ChannelId.h"
|
||||
#include "td/telegram/ChatId.h"
|
||||
#include "td/telegram/ContactsManager.h"
|
||||
#include "td/telegram/DialogManager.h"
|
||||
#include "td/telegram/Global.h"
|
||||
#include "td/telegram/LinkManager.h"
|
||||
#include "td/telegram/Td.h"
|
||||
#include "td/telegram/ThemeManager.h"
|
||||
#include "td/telegram/UpdatesManager.h"
|
||||
|
||||
#include "td/utils/buffer.h"
|
||||
#include "td/utils/logging.h"
|
||||
|
||||
namespace td {
|
||||
|
||||
class CheckChatInviteQuery final : public Td::ResultHandler {
|
||||
Promise<Unit> promise_;
|
||||
string invite_link_;
|
||||
|
||||
public:
|
||||
explicit CheckChatInviteQuery(Promise<Unit> &&promise) : promise_(std::move(promise)) {
|
||||
}
|
||||
|
||||
void send(const string &invite_link) {
|
||||
invite_link_ = invite_link;
|
||||
send_query(G()->net_query_creator().create(
|
||||
telegram_api::messages_checkChatInvite(LinkManager::get_dialog_invite_link_hash(invite_link_))));
|
||||
}
|
||||
|
||||
void on_result(BufferSlice packet) final {
|
||||
auto result_ptr = fetch_result<telegram_api::messages_checkChatInvite>(packet);
|
||||
if (result_ptr.is_error()) {
|
||||
return on_error(result_ptr.move_as_error());
|
||||
}
|
||||
|
||||
auto ptr = result_ptr.move_as_ok();
|
||||
LOG(INFO) << "Receive result for CheckChatInviteQuery: " << to_string(ptr);
|
||||
|
||||
td_->dialog_invite_link_manager_->on_get_dialog_invite_link_info(invite_link_, std::move(ptr), std::move(promise_));
|
||||
}
|
||||
|
||||
void on_error(Status status) final {
|
||||
promise_.set_error(std::move(status));
|
||||
}
|
||||
};
|
||||
|
||||
class ImportChatInviteQuery final : public Td::ResultHandler {
|
||||
Promise<DialogId> promise_;
|
||||
|
||||
string invite_link_;
|
||||
|
||||
public:
|
||||
explicit ImportChatInviteQuery(Promise<DialogId> &&promise) : promise_(std::move(promise)) {
|
||||
}
|
||||
|
||||
void send(const string &invite_link) {
|
||||
invite_link_ = invite_link;
|
||||
send_query(G()->net_query_creator().create(
|
||||
telegram_api::messages_importChatInvite(LinkManager::get_dialog_invite_link_hash(invite_link_))));
|
||||
}
|
||||
|
||||
void on_result(BufferSlice packet) final {
|
||||
auto result_ptr = fetch_result<telegram_api::messages_importChatInvite>(packet);
|
||||
if (result_ptr.is_error()) {
|
||||
return on_error(result_ptr.move_as_error());
|
||||
}
|
||||
|
||||
auto ptr = result_ptr.move_as_ok();
|
||||
LOG(INFO) << "Receive result for ImportChatInviteQuery: " << to_string(ptr);
|
||||
|
||||
auto dialog_ids = UpdatesManager::get_chat_dialog_ids(ptr.get());
|
||||
if (dialog_ids.size() != 1u) {
|
||||
LOG(ERROR) << "Receive wrong result for ImportChatInviteQuery: " << to_string(ptr);
|
||||
return on_error(Status::Error(500, "Internal Server Error: failed to join chat via invite link"));
|
||||
}
|
||||
auto dialog_id = dialog_ids[0];
|
||||
|
||||
td_->dialog_invite_link_manager_->invalidate_invite_link_info(invite_link_);
|
||||
td_->updates_manager_->on_get_updates(
|
||||
std::move(ptr), PromiseCreator::lambda([promise = std::move(promise_), dialog_id](Unit) mutable {
|
||||
promise.set_value(std::move(dialog_id));
|
||||
}));
|
||||
}
|
||||
|
||||
void on_error(Status status) final {
|
||||
td_->dialog_invite_link_manager_->invalidate_invite_link_info(invite_link_);
|
||||
promise_.set_error(std::move(status));
|
||||
}
|
||||
};
|
||||
|
||||
DialogInviteLinkManager::DialogInviteLinkManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
|
||||
}
|
||||
|
||||
@ -15,4 +103,248 @@ void DialogInviteLinkManager::tear_down() {
|
||||
parent_.reset();
|
||||
}
|
||||
|
||||
DialogInviteLinkManager::~DialogInviteLinkManager() {
|
||||
Scheduler::instance()->destroy_on_scheduler(G()->get_gc_scheduler_id(), invite_link_infos_);
|
||||
}
|
||||
|
||||
void DialogInviteLinkManager::check_dialog_invite_link(const string &invite_link, bool force, Promise<Unit> &&promise) {
|
||||
auto it = invite_link_infos_.find(invite_link);
|
||||
if (it != invite_link_infos_.end()) {
|
||||
auto dialog_id = it->second->dialog_id;
|
||||
if (!force && dialog_id.get_type() == DialogType::Chat &&
|
||||
!td_->contacts_manager_->get_chat_is_active(dialog_id.get_chat_id())) {
|
||||
invite_link_infos_.erase(it);
|
||||
} else {
|
||||
return promise.set_value(Unit());
|
||||
}
|
||||
}
|
||||
|
||||
if (!DialogInviteLink::is_valid_invite_link(invite_link)) {
|
||||
return promise.set_error(Status::Error(400, "Wrong invite link"));
|
||||
}
|
||||
|
||||
CHECK(!invite_link.empty());
|
||||
td_->create_handler<CheckChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
}
|
||||
|
||||
void DialogInviteLinkManager::import_dialog_invite_link(const string &invite_link, Promise<DialogId> &&promise) {
|
||||
if (!DialogInviteLink::is_valid_invite_link(invite_link)) {
|
||||
return promise.set_error(Status::Error(400, "Wrong invite link"));
|
||||
}
|
||||
|
||||
td_->create_handler<ImportChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
}
|
||||
|
||||
void DialogInviteLinkManager::on_get_dialog_invite_link_info(
|
||||
const string &invite_link, telegram_api::object_ptr<telegram_api::ChatInvite> &&chat_invite_ptr,
|
||||
Promise<Unit> &&promise) {
|
||||
CHECK(chat_invite_ptr != nullptr);
|
||||
switch (chat_invite_ptr->get_id()) {
|
||||
case telegram_api::chatInviteAlready::ID:
|
||||
case telegram_api::chatInvitePeek::ID: {
|
||||
telegram_api::object_ptr<telegram_api::Chat> chat = nullptr;
|
||||
int32 accessible_before_date = 0;
|
||||
if (chat_invite_ptr->get_id() == telegram_api::chatInviteAlready::ID) {
|
||||
auto chat_invite_already = telegram_api::move_object_as<telegram_api::chatInviteAlready>(chat_invite_ptr);
|
||||
chat = std::move(chat_invite_already->chat_);
|
||||
} else {
|
||||
auto chat_invite_peek = telegram_api::move_object_as<telegram_api::chatInvitePeek>(chat_invite_ptr);
|
||||
chat = std::move(chat_invite_peek->chat_);
|
||||
accessible_before_date = chat_invite_peek->expires_;
|
||||
}
|
||||
auto chat_id = ContactsManager::get_chat_id(chat);
|
||||
if (chat_id != ChatId() && !chat_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << chat_id;
|
||||
chat_id = ChatId();
|
||||
}
|
||||
auto channel_id = ContactsManager::get_channel_id(chat);
|
||||
if (channel_id != ChannelId() && !channel_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << channel_id;
|
||||
channel_id = ChannelId();
|
||||
}
|
||||
if (accessible_before_date != 0 && (!channel_id.is_valid() || accessible_before_date < 0)) {
|
||||
LOG(ERROR) << "Receive expires = " << accessible_before_date << " for invite link " << invite_link << " to "
|
||||
<< to_string(chat);
|
||||
accessible_before_date = 0;
|
||||
}
|
||||
td_->contacts_manager_->on_get_chat(std::move(chat), "chatInviteAlready");
|
||||
|
||||
CHECK(chat_id == ChatId() || channel_id == ChannelId());
|
||||
|
||||
// the access is already expired, reget the info
|
||||
if (accessible_before_date != 0 && accessible_before_date <= G()->unix_time() + 1) {
|
||||
td_->create_handler<CheckChatInviteQuery>(std::move(promise))->send(invite_link);
|
||||
return;
|
||||
}
|
||||
|
||||
DialogId dialog_id = chat_id.is_valid() ? DialogId(chat_id) : DialogId(channel_id);
|
||||
auto &invite_link_info = invite_link_infos_[invite_link];
|
||||
if (invite_link_info == nullptr) {
|
||||
invite_link_info = make_unique<InviteLinkInfo>();
|
||||
}
|
||||
invite_link_info->dialog_id = dialog_id;
|
||||
if (accessible_before_date != 0 && dialog_id.is_valid()) {
|
||||
td_->contacts_manager_->add_dialog_access_by_invite_link(dialog_id, invite_link, accessible_before_date);
|
||||
}
|
||||
break;
|
||||
}
|
||||
case telegram_api::chatInvite::ID: {
|
||||
auto chat_invite = telegram_api::move_object_as<telegram_api::chatInvite>(chat_invite_ptr);
|
||||
vector<UserId> participant_user_ids;
|
||||
for (auto &user : chat_invite->participants_) {
|
||||
auto user_id = ContactsManager::get_user_id(user);
|
||||
if (!user_id.is_valid()) {
|
||||
LOG(ERROR) << "Receive invalid " << user_id;
|
||||
continue;
|
||||
}
|
||||
|
||||
td_->contacts_manager_->on_get_user(std::move(user), "chatInvite");
|
||||
participant_user_ids.push_back(user_id);
|
||||
}
|
||||
|
||||
auto &invite_link_info = invite_link_infos_[invite_link];
|
||||
if (invite_link_info == nullptr) {
|
||||
invite_link_info = make_unique<InviteLinkInfo>();
|
||||
}
|
||||
invite_link_info->dialog_id = DialogId();
|
||||
invite_link_info->title = chat_invite->title_;
|
||||
invite_link_info->photo = get_photo(td_, std::move(chat_invite->photo_), DialogId());
|
||||
invite_link_info->accent_color_id = AccentColorId(chat_invite->color_);
|
||||
invite_link_info->description = std::move(chat_invite->about_);
|
||||
invite_link_info->participant_count = chat_invite->participants_count_;
|
||||
invite_link_info->participant_user_ids = std::move(participant_user_ids);
|
||||
invite_link_info->creates_join_request = std::move(chat_invite->request_needed_);
|
||||
invite_link_info->is_chat = !chat_invite->channel_;
|
||||
invite_link_info->is_channel = chat_invite->channel_;
|
||||
|
||||
bool is_broadcast = chat_invite->broadcast_;
|
||||
bool is_public = chat_invite->public_;
|
||||
bool is_megagroup = chat_invite->megagroup_;
|
||||
|
||||
if (!invite_link_info->is_channel) {
|
||||
if (is_broadcast || is_public || is_megagroup) {
|
||||
LOG(ERROR) << "Receive wrong chat invite: " << to_string(chat_invite);
|
||||
is_public = is_megagroup = false;
|
||||
}
|
||||
} else {
|
||||
LOG_IF(ERROR, is_broadcast == is_megagroup) << "Receive wrong chat invite: " << to_string(chat_invite);
|
||||
}
|
||||
|
||||
invite_link_info->is_public = is_public;
|
||||
invite_link_info->is_megagroup = is_megagroup;
|
||||
invite_link_info->is_verified = chat_invite->verified_;
|
||||
invite_link_info->is_scam = chat_invite->scam_;
|
||||
invite_link_info->is_fake = chat_invite->fake_;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
UNREACHABLE();
|
||||
}
|
||||
promise.set_value(Unit());
|
||||
}
|
||||
|
||||
void DialogInviteLinkManager::invalidate_invite_link_info(const string &invite_link) {
|
||||
LOG(INFO) << "Invalidate info about invite link " << invite_link;
|
||||
invite_link_infos_.erase(invite_link);
|
||||
}
|
||||
|
||||
td_api::object_ptr<td_api::chatInviteLinkInfo> DialogInviteLinkManager::get_chat_invite_link_info_object(
|
||||
const string &invite_link) {
|
||||
auto it = invite_link_infos_.find(invite_link);
|
||||
if (it == invite_link_infos_.end()) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
auto invite_link_info = it->second.get();
|
||||
CHECK(invite_link_info != nullptr);
|
||||
|
||||
DialogId dialog_id = invite_link_info->dialog_id;
|
||||
bool is_chat = false;
|
||||
bool is_megagroup = false;
|
||||
string title;
|
||||
const DialogPhoto *photo = nullptr;
|
||||
DialogPhoto invite_link_photo;
|
||||
int32 accent_color_id_object;
|
||||
string description;
|
||||
int32 participant_count = 0;
|
||||
vector<int64> member_user_ids;
|
||||
bool creates_join_request = false;
|
||||
bool is_public = false;
|
||||
bool is_member = false;
|
||||
bool is_verified = false;
|
||||
bool is_scam = false;
|
||||
bool is_fake = false;
|
||||
|
||||
if (dialog_id.is_valid()) {
|
||||
switch (dialog_id.get_type()) {
|
||||
case DialogType::Chat: {
|
||||
auto chat_id = dialog_id.get_chat_id();
|
||||
is_chat = true;
|
||||
|
||||
title = td_->contacts_manager_->get_chat_title(chat_id);
|
||||
photo = td_->contacts_manager_->get_chat_dialog_photo(chat_id);
|
||||
participant_count = td_->contacts_manager_->get_chat_participant_count(chat_id);
|
||||
is_member = td_->contacts_manager_->get_chat_status(chat_id).is_member();
|
||||
accent_color_id_object = td_->contacts_manager_->get_chat_accent_color_id_object(chat_id);
|
||||
break;
|
||||
}
|
||||
case DialogType::Channel: {
|
||||
auto channel_id = dialog_id.get_channel_id();
|
||||
title = td_->contacts_manager_->get_channel_title(channel_id);
|
||||
photo = td_->contacts_manager_->get_channel_dialog_photo(channel_id);
|
||||
is_public = td_->contacts_manager_->is_channel_public(channel_id);
|
||||
is_megagroup = td_->contacts_manager_->is_megagroup_channel(channel_id);
|
||||
participant_count = td_->contacts_manager_->get_channel_participant_count(channel_id);
|
||||
is_member = td_->contacts_manager_->get_channel_status(channel_id).is_member();
|
||||
is_verified = td_->contacts_manager_->get_channel_is_verified(channel_id);
|
||||
is_scam = td_->contacts_manager_->get_channel_is_scam(channel_id);
|
||||
is_fake = td_->contacts_manager_->get_channel_is_fake(channel_id);
|
||||
accent_color_id_object = td_->contacts_manager_->get_channel_accent_color_id_object(channel_id);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
UNREACHABLE();
|
||||
}
|
||||
description = td_->contacts_manager_->get_dialog_about(dialog_id);
|
||||
} else {
|
||||
is_chat = invite_link_info->is_chat;
|
||||
is_megagroup = invite_link_info->is_megagroup;
|
||||
title = invite_link_info->title;
|
||||
invite_link_photo = as_fake_dialog_photo(invite_link_info->photo, dialog_id, false);
|
||||
photo = &invite_link_photo;
|
||||
accent_color_id_object = td_->theme_manager_->get_accent_color_id_object(invite_link_info->accent_color_id);
|
||||
description = invite_link_info->description;
|
||||
participant_count = invite_link_info->participant_count;
|
||||
member_user_ids = td_->contacts_manager_->get_user_ids_object(invite_link_info->participant_user_ids,
|
||||
"get_chat_invite_link_info_object");
|
||||
creates_join_request = invite_link_info->creates_join_request;
|
||||
is_public = invite_link_info->is_public;
|
||||
is_verified = invite_link_info->is_verified;
|
||||
is_scam = invite_link_info->is_scam;
|
||||
is_fake = invite_link_info->is_fake;
|
||||
}
|
||||
|
||||
td_api::object_ptr<td_api::InviteLinkChatType> chat_type;
|
||||
if (is_chat) {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeBasicGroup>();
|
||||
} else if (is_megagroup) {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeSupergroup>();
|
||||
} else {
|
||||
chat_type = td_api::make_object<td_api::inviteLinkChatTypeChannel>();
|
||||
}
|
||||
|
||||
if (dialog_id.is_valid()) {
|
||||
td_->dialog_manager_->force_create_dialog(dialog_id, "get_chat_invite_link_info_object");
|
||||
}
|
||||
int32 accessible_for = 0;
|
||||
if (dialog_id.is_valid() && !is_member) {
|
||||
accessible_for = td_->contacts_manager_->get_dialog_accessible_by_invite_link_before_date(dialog_id);
|
||||
}
|
||||
|
||||
return td_api::make_object<td_api::chatInviteLinkInfo>(
|
||||
td_->dialog_manager_->get_chat_id_object(dialog_id, "chatInviteLinkInfo"), accessible_for, std::move(chat_type),
|
||||
title, get_chat_photo_info_object(td_->file_manager_.get(), photo), accent_color_id_object, description,
|
||||
participant_count, std::move(member_user_ids), creates_join_request, is_public, is_verified, is_scam, is_fake);
|
||||
}
|
||||
|
||||
} // namespace td
|
||||
|
@ -6,9 +6,19 @@
|
||||
//
|
||||
#pragma once
|
||||
|
||||
#include "td/telegram/AccentColorId.h"
|
||||
#include "td/telegram/DialogId.h"
|
||||
#include "td/telegram/DialogInviteLink.h"
|
||||
#include "td/telegram/Photo.h"
|
||||
#include "td/telegram/td_api.h"
|
||||
#include "td/telegram/telegram_api.h"
|
||||
#include "td/telegram/UserId.h"
|
||||
|
||||
#include "td/actor/actor.h"
|
||||
|
||||
#include "td/utils/common.h"
|
||||
#include "td/utils/FlatHashMap.h"
|
||||
#include "td/utils/Promise.h"
|
||||
|
||||
namespace td {
|
||||
|
||||
@ -17,10 +27,49 @@ class Td;
|
||||
class DialogInviteLinkManager final : public Actor {
|
||||
public:
|
||||
DialogInviteLinkManager(Td *td, ActorShared<> parent);
|
||||
DialogInviteLinkManager(const DialogInviteLinkManager &) = delete;
|
||||
DialogInviteLinkManager &operator=(const DialogInviteLinkManager &) = delete;
|
||||
DialogInviteLinkManager(DialogInviteLinkManager &&) = delete;
|
||||
DialogInviteLinkManager &operator=(DialogInviteLinkManager &&) = delete;
|
||||
~DialogInviteLinkManager() final;
|
||||
|
||||
void check_dialog_invite_link(const string &invite_link, bool force, Promise<Unit> &&promise);
|
||||
|
||||
void import_dialog_invite_link(const string &invite_link, Promise<DialogId> &&promise);
|
||||
|
||||
void on_get_dialog_invite_link_info(const string &invite_link,
|
||||
telegram_api::object_ptr<telegram_api::ChatInvite> &&chat_invite_ptr,
|
||||
Promise<Unit> &&promise);
|
||||
|
||||
void invalidate_invite_link_info(const string &invite_link);
|
||||
|
||||
td_api::object_ptr<td_api::chatInviteLinkInfo> get_chat_invite_link_info_object(const string &invite_link);
|
||||
|
||||
private:
|
||||
void tear_down() final;
|
||||
|
||||
struct InviteLinkInfo {
|
||||
// known dialog
|
||||
DialogId dialog_id;
|
||||
|
||||
// unknown dialog
|
||||
string title;
|
||||
Photo photo;
|
||||
AccentColorId accent_color_id;
|
||||
int32 participant_count = 0;
|
||||
vector<UserId> participant_user_ids;
|
||||
string description;
|
||||
bool creates_join_request = false;
|
||||
bool is_chat = false;
|
||||
bool is_channel = false;
|
||||
bool is_public = false;
|
||||
bool is_megagroup = false;
|
||||
bool is_verified = false;
|
||||
bool is_scam = false;
|
||||
bool is_fake = false;
|
||||
};
|
||||
FlatHashMap<string, unique_ptr<InviteLinkInfo>> invite_link_infos_;
|
||||
|
||||
Td *td_;
|
||||
ActorShared<> parent_;
|
||||
};
|
||||
|
@ -8,6 +8,7 @@
|
||||
|
||||
#include "td/telegram/ChannelId.h"
|
||||
#include "td/telegram/ContactsManager.h"
|
||||
#include "td/telegram/DialogInviteLinkManager.h"
|
||||
#include "td/telegram/DialogManager.h"
|
||||
#include "td/telegram/Global.h"
|
||||
#include "td/telegram/LinkManager.h"
|
||||
@ -273,7 +274,7 @@ td_api::object_ptr<td_api::messageSponsor> SponsoredMessageManager::get_message_
|
||||
break;
|
||||
}
|
||||
auto invite_link = LinkManager::get_dialog_invite_link(sponsored_message.invite_hash, false);
|
||||
auto chat_invite_link_info = td_->contacts_manager_->get_chat_invite_link_info_object(invite_link);
|
||||
auto chat_invite_link_info = td_->dialog_invite_link_manager_->get_chat_invite_link_info_object(invite_link);
|
||||
if (chat_invite_link_info == nullptr) {
|
||||
LOG(ERROR) << "Failed to get invite link info for " << invite_link;
|
||||
return nullptr;
|
||||
@ -404,9 +405,9 @@ void SponsoredMessageManager::on_get_dialog_sponsored_messages(
|
||||
continue;
|
||||
}
|
||||
auto chat_invite = to_string(sponsored_message->chat_invite_);
|
||||
td_->contacts_manager_->on_get_dialog_invite_link_info(
|
||||
td_->dialog_invite_link_manager_->on_get_dialog_invite_link_info(
|
||||
invite_link, std::move(sponsored_message->chat_invite_), Promise<Unit>());
|
||||
auto chat_invite_link_info = td_->contacts_manager_->get_chat_invite_link_info_object(invite_link);
|
||||
auto chat_invite_link_info = td_->dialog_invite_link_manager_->get_chat_invite_link_info_object(invite_link);
|
||||
if (chat_invite_link_info == nullptr) {
|
||||
LOG(ERROR) << "Failed to get invite link info from " << chat_invite << " for "
|
||||
<< to_string(sponsored_message);
|
||||
|
@ -292,9 +292,9 @@ class GetRecentMeUrlsQuery final : public Td::ResultHandler {
|
||||
case telegram_api::recentMeUrlChatInvite::ID: {
|
||||
auto url = move_tl_object_as<telegram_api::recentMeUrlChatInvite>(url_ptr);
|
||||
result->url_ = std::move(url->url_);
|
||||
td_->contacts_manager_->on_get_dialog_invite_link_info(result->url_, std::move(url->chat_invite_),
|
||||
td_->dialog_invite_link_manager_->on_get_dialog_invite_link_info(result->url_, std::move(url->chat_invite_),
|
||||
Promise<Unit>());
|
||||
auto info_object = td_->contacts_manager_->get_chat_invite_link_info_object(result->url_);
|
||||
auto info_object = td_->dialog_invite_link_manager_->get_chat_invite_link_info_object(result->url_);
|
||||
if (info_object == nullptr) {
|
||||
result = nullptr;
|
||||
break;
|
||||
@ -1779,11 +1779,11 @@ class CheckChatInviteLinkRequest final : public RequestActor<> {
|
||||
string invite_link_;
|
||||
|
||||
void do_run(Promise<Unit> &&promise) final {
|
||||
td_->contacts_manager_->check_dialog_invite_link(invite_link_, get_tries() < 2, std::move(promise));
|
||||
td_->dialog_invite_link_manager_->check_dialog_invite_link(invite_link_, get_tries() < 2, std::move(promise));
|
||||
}
|
||||
|
||||
void do_send_result() final {
|
||||
auto result = td_->contacts_manager_->get_chat_invite_link_info_object(invite_link_);
|
||||
auto result = td_->dialog_invite_link_manager_->get_chat_invite_link_info_object(invite_link_);
|
||||
CHECK(result != nullptr);
|
||||
send_result(std::move(result));
|
||||
}
|
||||
@ -1804,7 +1804,7 @@ class JoinChatByInviteLinkRequest final : public RequestActor<DialogId> {
|
||||
promise.set_value(std::move(dialog_id_));
|
||||
return;
|
||||
}
|
||||
td_->contacts_manager_->import_dialog_invite_link(invite_link_, std::move(promise));
|
||||
td_->dialog_invite_link_manager_->import_dialog_invite_link(invite_link_, std::move(promise));
|
||||
}
|
||||
|
||||
void do_set_result(DialogId &&result) final {
|
||||
|
Loading…
Reference in New Issue
Block a user