Add td_api::getUserExpiringStories.

This commit is contained in:
levlam 2023-05-21 12:39:53 +03:00
parent e763cb2803
commit a049076797
6 changed files with 104 additions and 7 deletions

View File

@ -7156,6 +7156,9 @@ readChatList chat_list:ChatList = Ok;
//-For optimal performance, the number of returned stories is chosen by TDLib and can be smaller than the specified limit //-For optimal performance, the number of returned stories is chosen by TDLib and can be smaller than the specified limit
getUserPinnedStories user_id:int53 from_story_id:int32 limit:int32 = Stories; getUserPinnedStories user_id:int53 from_story_id:int32 limit:int32 = Stories;
//@description Returns the list of expiring stories of a given user. The stories are returned in a reverse chronological order (i.e., in order of decreasing story_id) @user_id User identifier
getUserExpiringStories user_id:int53 = Stories;
//@description Returns information about a bot that can be added to attachment menu @bot_user_id Bot's user identifier //@description Returns information about a bot that can be added to attachment menu @bot_user_id Bot's user identifier
getAttachmentMenuBot bot_user_id:int53 = AttachmentMenuBot; getAttachmentMenuBot bot_user_id:int53 = AttachmentMenuBot;

View File

@ -92,6 +92,38 @@ class GetPinnedStoriesQuery final : public Td::ResultHandler {
} }
}; };
class GetUserStoriesQuery final : public Td::ResultHandler {
Promise<telegram_api::object_ptr<telegram_api::stories_userStories>> promise_;
public:
explicit GetUserStoriesQuery(Promise<telegram_api::object_ptr<telegram_api::stories_userStories>> &&promise)
: promise_(std::move(promise)) {
}
void send(UserId user_id) {
auto r_input_user = td_->contacts_manager_->get_input_user(user_id);
if (r_input_user.is_error()) {
return on_error(r_input_user.move_as_error());
}
send_query(G()->net_query_creator().create(telegram_api::stories_getUserStories(r_input_user.move_as_ok())));
}
void on_result(BufferSlice packet) final {
auto result_ptr = fetch_result<telegram_api::stories_getUserStories>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
}
auto result = result_ptr.move_as_ok();
LOG(DEBUG) << "Receive result for GetUserStoriesQuery: " << to_string(result);
promise_.set_value(std::move(result));
}
void on_error(Status status) final {
promise_.set_error(std::move(status));
}
};
StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
} }
@ -160,6 +192,41 @@ void StoryManager::on_get_dialog_pinned_stories(DialogId owner_dialog_id,
}))); })));
} }
void StoryManager::get_dialog_expiring_stories(DialogId owner_dialog_id,
Promise<td_api::object_ptr<td_api::stories>> &&promise) {
if (!td_->messages_manager_->have_dialog_force(owner_dialog_id, "get_dialog_pinned_stories")) {
return promise.set_error(Status::Error(400, "Chat not found"));
}
if (!td_->messages_manager_->have_input_peer(owner_dialog_id, AccessRights::Read)) {
return promise.set_error(Status::Error(400, "Can't access the chat"));
}
if (owner_dialog_id.get_type() != DialogType::User) {
return promise.set_value(td_api::make_object<td_api::stories>());
}
auto query_promise =
PromiseCreator::lambda([actor_id = actor_id(this), owner_dialog_id, promise = std::move(promise)](
Result<telegram_api::object_ptr<telegram_api::stories_userStories>> &&result) mutable {
if (result.is_error()) {
return promise.set_error(result.move_as_error());
}
send_closure(actor_id, &StoryManager::on_get_dialog_expiring_stories, owner_dialog_id, result.move_as_ok(),
std::move(promise));
});
td_->create_handler<GetUserStoriesQuery>(std::move(query_promise))->send(owner_dialog_id.get_user_id());
}
void StoryManager::on_get_dialog_expiring_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::stories_userStories> &&stories,
Promise<td_api::object_ptr<td_api::stories>> &&promise) {
TRY_STATUS_PROMISE(promise, G()->close_status());
td_->contacts_manager_->on_get_users(std::move(stories->users_), "on_get_dialog_expiring_stories");
auto story_ids = on_get_stories(owner_dialog_id, std::move(stories->stories_->stories_));
promise.set_value(get_stories_object(-1, transform(story_ids, [owner_dialog_id](StoryId story_id) {
return StoryFullId(owner_dialog_id, story_id);
})));
}
td_api::object_ptr<td_api::story> StoryManager::get_story_object(StoryFullId story_full_id) const { td_api::object_ptr<td_api::story> StoryManager::get_story_object(StoryFullId story_full_id) const {
return get_story_object(story_full_id, get_story(story_full_id)); return get_story_object(story_full_id, get_story(story_full_id));
} }
@ -301,8 +368,19 @@ StoryId StoryManager::on_get_story(DialogId owner_dialog_id,
std::pair<int32, vector<StoryId>> StoryManager::on_get_stories( std::pair<int32, vector<StoryId>> StoryManager::on_get_stories(
DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::stories_stories> &&stories) { DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::stories_stories> &&stories) {
td_->contacts_manager_->on_get_users(std::move(stories->users_), "on_get_stories"); td_->contacts_manager_->on_get_users(std::move(stories->users_), "on_get_stories");
auto story_ids = on_get_stories(owner_dialog_id, std::move(stories->stories_));
auto total_count = stories->count_;
if (total_count < static_cast<int32>(story_ids.size())) {
LOG(ERROR) << "Expected at most " << total_count << " stories, but receive " << story_ids.size();
total_count = static_cast<int32>(story_ids.size());
}
return {total_count, std::move(story_ids)};
}
vector<StoryId> StoryManager::on_get_stories(DialogId owner_dialog_id,
vector<telegram_api::object_ptr<telegram_api::StoryItem>> &&stories) {
vector<StoryId> story_ids; vector<StoryId> story_ids;
for (auto &story : stories->stories_) { for (auto &story : stories) {
switch (story->get_id()) { switch (story->get_id()) {
case telegram_api::storyItemDeleted::ID: case telegram_api::storyItemDeleted::ID:
LOG(ERROR) << "Receive storyItemDeleted"; LOG(ERROR) << "Receive storyItemDeleted";
@ -321,12 +399,7 @@ std::pair<int32, vector<StoryId>> StoryManager::on_get_stories(
UNREACHABLE(); UNREACHABLE();
} }
} }
auto total_count = stories->count_; return story_ids;
if (total_count < static_cast<int32>(story_ids.size())) {
LOG(ERROR) << "Expected at most " << total_count << " stories, but receive " << story_ids.size();
total_count = static_cast<int32>(story_ids.size());
}
return {total_count, std::move(story_ids)};
} }
FileSourceId StoryManager::get_story_file_source_id(StoryFullId story_full_id) { FileSourceId StoryManager::get_story_file_source_id(StoryFullId story_full_id) {

View File

@ -42,6 +42,8 @@ class StoryManager final : public Actor {
void get_dialog_pinned_stories(DialogId owner_dialog_id, StoryId from_story_id, int32 limit, void get_dialog_pinned_stories(DialogId owner_dialog_id, StoryId from_story_id, int32 limit,
Promise<td_api::object_ptr<td_api::stories>> &&promise); Promise<td_api::object_ptr<td_api::stories>> &&promise);
void get_dialog_expiring_stories(DialogId owner_dialog_id, Promise<td_api::object_ptr<td_api::stories>> &&promise);
StoryId on_get_story(DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::storyItem> &&story_item); StoryId on_get_story(DialogId owner_dialog_id, telegram_api::object_ptr<telegram_api::storyItem> &&story_item);
std::pair<int32, vector<StoryId>> on_get_stories(DialogId owner_dialog_id, std::pair<int32, vector<StoryId>> on_get_stories(DialogId owner_dialog_id,
@ -79,10 +81,17 @@ class StoryManager final : public Actor {
td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id, const Story *story) const; td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id, const Story *story) const;
vector<StoryId> on_get_stories(DialogId owner_dialog_id,
vector<telegram_api::object_ptr<telegram_api::StoryItem>> &&stories);
void on_get_dialog_pinned_stories(DialogId owner_dialog_id, void on_get_dialog_pinned_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::stories_stories> &&stories, telegram_api::object_ptr<telegram_api::stories_stories> &&stories,
Promise<td_api::object_ptr<td_api::stories>> &&promise); Promise<td_api::object_ptr<td_api::stories>> &&promise);
void on_get_dialog_expiring_stories(DialogId owner_dialog_id,
telegram_api::object_ptr<telegram_api::stories_userStories> &&stories,
Promise<td_api::object_ptr<td_api::stories>> &&promise);
vector<FileId> get_story_file_ids(const Story *story) const; vector<FileId> get_story_file_ids(const Story *story) const;
void delete_story_files(const Story *story) const; void delete_story_files(const Story *story) const;

View File

@ -6381,6 +6381,12 @@ void Td::on_request(uint64 id, const td_api::getUserPinnedStories &request) {
request.limit_, std::move(promise)); request.limit_, std::move(promise));
} }
void Td::on_request(uint64 id, const td_api::getUserExpiringStories &request) {
CHECK_IS_USER();
CREATE_REQUEST_PROMISE();
story_manager_->get_dialog_expiring_stories(DialogId(UserId(request.user_id_)), std::move(promise));
}
void Td::on_request(uint64 id, const td_api::getAttachmentMenuBot &request) { void Td::on_request(uint64 id, const td_api::getAttachmentMenuBot &request) {
CHECK_IS_USER(); CHECK_IS_USER();
CREATE_REQUEST_PROMISE(); CREATE_REQUEST_PROMISE();

View File

@ -988,6 +988,8 @@ class Td final : public Actor {
void on_request(uint64 id, const td_api::getUserPinnedStories &request); void on_request(uint64 id, const td_api::getUserPinnedStories &request);
void on_request(uint64 id, const td_api::getUserExpiringStories &request);
void on_request(uint64 id, const td_api::getAttachmentMenuBot &request); void on_request(uint64 id, const td_api::getAttachmentMenuBot &request);
void on_request(uint64 id, const td_api::toggleBotIsAddedToAttachmentMenu &request); void on_request(uint64 id, const td_api::toggleBotIsAddedToAttachmentMenu &request);

View File

@ -3917,6 +3917,10 @@ class CliClient final : public Actor {
string limit; string limit;
get_args(args, user_id, from_story_id, limit); get_args(args, user_id, from_story_id, limit);
send_request(td_api::make_object<td_api::getUserPinnedStories>(user_id, from_story_id, as_limit(limit))); send_request(td_api::make_object<td_api::getUserPinnedStories>(user_id, from_story_id, as_limit(limit)));
} else if (op == "gues") {
UserId user_id;
get_args(args, user_id);
send_request(td_api::make_object<td_api::getUserExpiringStories>(user_id));
} else if (op == "gamb") { } else if (op == "gamb") {
UserId user_id; UserId user_id;
get_args(args, user_id); get_args(args, user_id);