tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow clicking sponsored message after it was viewed.

Browse Source
This commit is contained in:
levlam 2023-05-04 17:33:35 +03:00
parent 1a5f81163d
commit e8e1236e31
2 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
td/telegram/SponsoredMessageManager.cpp
View File

@ -148,10 +148,16 @@ struct SponsoredMessageManager::SponsoredMessage {
} }
}; };
struct SponsoredMessageManager::SponsoredMessageInfo {
string random_id_;
bool is_viewed_ = false;
bool is_clicked_ = false;
};
struct SponsoredMessageManager::DialogSponsoredMessages { struct SponsoredMessageManager::DialogSponsoredMessages {
vector<Promise<td_api::object_ptr<td_api::sponsoredMessages>>> promises; vector<Promise<td_api::object_ptr<td_api::sponsoredMessages>>> promises;
vector<SponsoredMessage> messages; vector<SponsoredMessage> messages;
FlatHashMap<int64, string> message_random_ids; FlatHashMap<int64, SponsoredMessageInfo> message_infos;
int32 messages_between = 0; int32 messages_between = 0;
bool is_premium = false; bool is_premium = false;
}; };
@ -302,7 +308,7 @@ void SponsoredMessageManager::on_get_dialog_sponsored_messages(
auto promises = std::move(messages->promises); auto promises = std::move(messages->promises);
reset_to_empty(messages->promises); reset_to_empty(messages->promises);
CHECK(messages->messages.empty()); CHECK(messages->messages.empty());
CHECK(messages->message_random_ids.empty()); CHECK(messages->message_infos.empty());
if (result.is_error()) { if (result.is_error()) {
dialog_sponsored_messages_.erase(dialog_id); dialog_sponsored_messages_.erase(dialog_id);
@ -378,8 +384,9 @@ void SponsoredMessageManager::on_get_dialog_sponsored_messages(
auto local_id = current_sponsored_message_id_.get(); auto local_id = current_sponsored_message_id_.get();
CHECK(!current_sponsored_message_id_.is_valid()); CHECK(!current_sponsored_message_id_.is_valid());
CHECK(!current_sponsored_message_id_.is_scheduled()); CHECK(!current_sponsored_message_id_.is_scheduled());
auto is_inserted = SponsoredMessageInfo message_info;
messages->message_random_ids.emplace(local_id, sponsored_message->random_id_.as_slice().str()).second; message_info.random_id_ = sponsored_message->random_id_.as_slice().str();
auto is_inserted = messages->message_infos.emplace(local_id, std::move(message_info)).second;
CHECK(is_inserted); CHECK(is_inserted);
messages->messages.emplace_back( messages->messages.emplace_back(
local_id, sponsored_message->recommended_, sponsored_message->show_peer_photo_, sponsor_dialog_id, local_id, sponsored_message->recommended_, sponsored_message->show_peer_photo_, sponsor_dialog_id,
@ -407,14 +414,13 @@ void SponsoredMessageManager::view_sponsored_message(DialogId dialog_id, Message
if (it == dialog_sponsored_messages_.end()) { if (it == dialog_sponsored_messages_.end()) {
return; return;
} }
auto random_id_it = it->second->message_random_ids.find(sponsored_message_id.get()); auto random_id_it = it->second->message_infos.find(sponsored_message_id.get());
if (random_id_it == it->second->message_random_ids.end()) { if (random_id_it == it->second->message_infos.end() || random_id_it->second.is_viewed_) {
return; return;
} }
auto random_id = std::move(random_id_it->second); random_id_it->second.is_viewed_ = true;
it->second->message_random_ids.erase(random_id_it); td_->create_handler<ViewSponsoredMessageQuery>()->send(dialog_id.get_channel_id(), random_id_it->second.random_id_);
td_->create_handler<ViewSponsoredMessageQuery>()->send(dialog_id.get_channel_id(), random_id);
} }
void SponsoredMessageManager::click_sponsored_message(DialogId dialog_id, MessageId sponsored_message_id, void SponsoredMessageManager::click_sponsored_message(DialogId dialog_id, MessageId sponsored_message_id,
@ -426,14 +432,14 @@ void SponsoredMessageManager::click_sponsored_message(DialogId dialog_id, Messag
if (it == dialog_sponsored_messages_.end()) { if (it == dialog_sponsored_messages_.end()) {
return promise.set_value(Unit()); return promise.set_value(Unit());
} }
auto random_id_it = it->second->message_random_ids.find(sponsored_message_id.get()); auto random_id_it = it->second->message_infos.find(sponsored_message_id.get());
if (random_id_it == it->second->message_random_ids.end()) { if (random_id_it == it->second->message_infos.end() || random_id_it->second.is_clicked_) {
return promise.set_value(Unit()); return promise.set_value(Unit());
} }
auto random_id = std::move(random_id_it->second); random_id_it->second.is_clicked_ = true;
it->second->message_random_ids.erase(random_id_it); td_->create_handler<ClickSponsoredMessageQuery>(std::move(promise))
td_->create_handler<ClickSponsoredMessageQuery>(std::move(promise))->send(dialog_id.get_channel_id(), random_id); ->send(dialog_id.get_channel_id(), random_id_it->second.random_id_);
} }
} // namespace td } // namespace td

1
td/telegram/SponsoredMessageManager.h
View File

@ -41,6 +41,7 @@ class SponsoredMessageManager final : public Actor {
private: private:
struct SponsoredMessage; struct SponsoredMessage;
struct SponsoredMessageInfo;
struct DialogSponsoredMessages; struct DialogSponsoredMessages;
void tear_down() final; void tear_down() final;