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

Add StoryManager::on_get_stories.

Browse Source
This commit is contained in:
levlam 2023-05-19 23:25:52 +03:00
parent 8c9bf81359
commit 0b8c71419f
2 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
td/telegram/StoryManager.cpp
View File

@ -14,6 +14,8 @@
#include "td/telegram/StoryContentType.h" #include "td/telegram/StoryContentType.h"
#include "td/telegram/Td.h" #include "td/telegram/Td.h"
#include "td/utils/logging.h"
namespace td { namespace td {
StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { StoryManager::StoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
@ -85,6 +87,7 @@ StoryId StoryManager::on_get_story(DialogId owner_dialog_id,
CHECK(story_item != nullptr); CHECK(story_item != nullptr);
StoryId story_id(story_item->id_); StoryId story_id(story_item->id_);
if (!story_id.is_valid() || is_local_story_id(story_id)) { if (!story_id.is_valid() || is_local_story_id(story_id)) {
LOG(ERROR) << "Receive " << to_string(story_item);
return StoryId(); return StoryId();
} }
@ -142,4 +145,35 @@ StoryId StoryManager::on_get_story(DialogId owner_dialog_id,
return story_id; return story_id;
} }
std::pair<int32, vector<StoryId>> StoryManager::on_get_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");
vector<StoryId> story_ids;
for (auto &story : stories->stories_) {
switch (story->get_id()) {
case telegram_api::storyItemDeleted::ID:
LOG(ERROR) << "Receive storyItemDeleted";
break;
case telegram_api::storyItemSkipped::ID:
LOG(ERROR) << "Receive storyItemSkipped";
break;
case telegram_api::storyItem::ID: {
auto story_id = on_get_story(owner_dialog_id, telegram_api::move_object_as<telegram_api::storyItem>(story));
if (story_id.is_valid()) {
story_ids.push_back(story_id);
}
break;
}
default:
UNREACHABLE();
}
}
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)};
}
} // namespace td } // namespace td

5
td/telegram/StoryManager.h
View File

@ -21,6 +21,8 @@
#include "td/utils/common.h" #include "td/utils/common.h"
#include "td/utils/WaitFreeHashMap.h" #include "td/utils/WaitFreeHashMap.h"
#include <utility>
namespace td { namespace td {
class StoryContent; class StoryContent;
@ -37,6 +39,9 @@ class StoryManager final : public Actor {
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,
telegram_api::object_ptr<telegram_api::stories_stories> &&stories);
td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id) const; td_api::object_ptr<td_api::story> get_story_object(StoryFullId story_full_id) const;
private: private: