50 lines
1.1 KiB
C++
50 lines
1.1 KiB
C++
|
/*
|
||
|
|
This file is part of Telegram Desktop,
|
||
|
|
the official desktop application for the Telegram messaging service.
|
||
|
|
|
||
|
|
For license and copyright information please follow this link:
|
||
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
||
|
|
*/
|
||
|
|
#include "data/data_groups.h"
|
||
|
|
|
||
|
|
#include "history/history_item.h"
|
||
|
|
|
||
|
|
namespace Data {
|
||
|
|
|
||
|
|
void Groups::registerMessage(not_null<HistoryItem*> item) {
|
||
|
|
const auto groupId = item->groupId();
|
||
|
|
if (!groupId) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const auto i = _data.emplace(groupId, Group()).first;
|
||
|
|
i->second.items.push_back(item);
|
||
|
|
}
|
||
|
|
|
||
|
|
void Groups::unregisterMessage(not_null<HistoryItem*> item) {
|
||
|
|
const auto groupId = item->groupId();
|
||
|
|
if (!groupId) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const auto i = _data.find(groupId);
|
||
|
|
if (i != _data.end()) {
|
||
|
|
auto &group = i->second;
|
||
|
|
group.items.erase(
|
||
|
|
ranges::remove(group.items, item),
|
||
|
|
group.items.end());
|
||
|
|
if (group.items.empty()) {
|
||
|
|
_data.erase(i);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const Group *Groups::find(not_null<HistoryItem*> item) const {
|
||
|
|
const auto groupId = item->groupId();
|
||
|
|
if (!groupId) {
|
||
|
|
return nullptr;
|
||
|
|
}
|
||
|
|
const auto i = _data.find(groupId);
|
||
|
|
return (i != _data.end()) ? &i->second : nullptr;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace Data
|