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

Add get_dialog_lists method. 

GitOrigin-RevId: 9d228b0681ddeef07963223f7fc93c96a1798168
This commit is contained in:
levlam 2020-05-04 01:47:26 +03:00
parent 2408b46150
commit 4fe08330d9
2 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
td/telegram/MessagesManager.cpp
View File

@ -29093,6 +29093,12 @@ MessagesManager::Dialog *MessagesManager::on_load_dialog_from_database(DialogId
return add_new_dialog(parse_dialog(dialog_id, value), true);
}
MessagesManager::DialogListView MessagesManager::get_dialog_lists(const Dialog *d) {
vector<FolderId> dialog_list_ids;
dialog_list_ids.push_back(d->folder_id);
return DialogListView(this, std::move(dialog_list_ids));
}
MessagesManager::DialogList &MessagesManager::get_dialog_list(FolderId folder_id) {
CHECK(!td_->auth_manager_->is_bot());
if (folder_id != FolderId::archive()) {

43
td/telegram/MessagesManager.h
View File

@ -1247,6 +1247,47 @@ class MessagesManager : public Actor {
int32 load_dialog_list_limit_max_ = 0;
};
class DialogListViewIterator {
MessagesManager *messages_manager_;
FolderId *dialog_list_id_;
public:
DialogListViewIterator(MessagesManager *messages_manager, FolderId *dialog_list_id)
: messages_manager_(messages_manager), dialog_list_id_(dialog_list_id) {
}
DialogList &operator*() const {
return messages_manager_->get_dialog_list(*dialog_list_id_);
}
bool operator!=(const DialogListViewIterator &other) const {
return dialog_list_id_ != other.dialog_list_id_;
}
void operator++() {
dialog_list_id_++;
}
};
class DialogListView {
MessagesManager *messages_manager_;
// TODO can be optimized to store only mask of dialog lists
vector<FolderId> dialog_list_ids_;
public:
DialogListView(MessagesManager *messages_manager, vector<FolderId> dialog_list_ids)
: messages_manager_(messages_manager), dialog_list_ids_(std::move(dialog_list_ids)) {
}
DialogListViewIterator begin() {
return DialogListViewIterator(messages_manager_, &dialog_list_ids_[0]);
}
DialogListViewIterator end() {
return DialogListViewIterator(messages_manager_, &dialog_list_ids_[0] + dialog_list_ids_.size());
}
};
class MessagesIteratorBase {
vector<const Message *> stack_;
@ -2125,6 +2166,8 @@ class MessagesManager : public Actor {
void update_dialogs_hints(const Dialog *d);
void update_dialogs_hints_rating(const Dialog *d);
DialogListView get_dialog_lists(const Dialog *d);
DialogList &get_dialog_list(FolderId folder_id);
const DialogList *get_dialog_list(FolderId folder_id) const;