init

Telegram/SourceFiles/data/data_groups.cpp

@@ -0,0 +1,177 @@
+/*
+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.h"
+#include "history/history_item.h"
+#include "dialogs/ui/dialogs_message_view.h"
+#include "data/data_media_types.h"
+#include "data/data_session.h"
+#include "data/data_forum.h"
+#include "data/data_forum_topic.h"
+
+namespace Data {
+namespace {
+
+constexpr auto kMaxItemsInGroup = 10;
+
+} // namespace
+
+Groups::Groups(not_null<Session*> data) : _data(data) {
+}
+
+bool Groups::isGrouped(not_null<const HistoryItem*> item) const {
+	if (!item->groupId()) {
+		return false;
+	}
+	const auto media = item->media();
+	return media && media->canBeGrouped();
+}
+
+bool Groups::isGroupOfOne(not_null<const HistoryItem*> item) const {
+	if (const auto groupId = item->groupId()) {
+		const auto i = _groups.find(groupId);
+		return (i != _groups.end()) && (i->second.items.size() == 1);
+	}
+	return false;
+}
+
+void Groups::registerMessage(not_null<HistoryItem*> item) {
+	if (!isGrouped(item)) {
+		return;
+	}
+	const auto i = _groups.emplace(item->groupId(), Group()).first;
+	auto &items = i->second.items;
+	if (items.size() < kMaxItemsInGroup) {
+		items.insert(findPositionForItem(items, item), item);
+		if (items.size() > 1) {
+			refreshViews(items);
+		}
+	}
+}
+
+void Groups::unregisterMessage(not_null<const HistoryItem*> item) {
+	const auto groupId = item->groupId();
+	if (!groupId) {
+		return;
+	}
+	const auto i = _groups.find(groupId);
+	if (i != end(_groups)) {
+		auto &items = i->second.items;
+		const auto removed = ranges::remove(items, item);
+		const auto last = end(items);
+		if (removed != last) {
+			items.erase(removed, last);
+			if (!items.empty()) {
+				refreshViews(items);
+			} else {
+				_groups.erase(i);
+			}
+		}
+	}
+}
+
+void Groups::refreshMessage(
+		not_null<HistoryItem*> item,
+		bool justRefreshViews) {
+	if (!isGrouped(item)) {
+		unregisterMessage(item);
+		_data->requestItemViewRefresh(item);
+		return;
+	}
+	if (!item->isRegular() && !item->isScheduled() && !item->isUploading()) {
+		return;
+	}
+	const auto groupId = item->groupId();
+	const auto i = _groups.find(groupId);
+	if (i == end(_groups)) {
+		registerMessage(item);
+		return;
+	}
+	auto &items = i->second.items;
+
+	if (justRefreshViews) {
+		refreshViews(items);
+		return;
+	}
+
+	const auto position = findPositionForItem(items, item);
+	auto current = ranges::find(items, item);
+	if (current == end(items)) {
+		items.insert(position, item);
+	} else if (position == current + 1) {
+		return;
+	} else if (position > current + 1) {
+		for (++current; current != position; ++current) {
+			std::swap(*(current - 1), *current);
+		}
+	} else if (position < current) {
+		for (; current != position; --current) {
+			std::swap(*(current - 1), *current);
+		}
+	} else {
+		Unexpected("Position of item in Groups::refreshMessage().");
+	}
+	refreshViews(items);
+}
+
+HistoryItemsList::const_iterator Groups::findPositionForItem(
+		const HistoryItemsList &group,
+		not_null<HistoryItem*> item) {
+	const auto last = end(group);
+	const auto itemId = item->id;
+	for (auto result = begin(group); result != last; ++result) {
+		if ((*result)->id > itemId) {
+			return result;
+		}
+	}
+	return last;
+}
+
+const Group *Groups::find(not_null<const HistoryItem*> item) const {
+	const auto groupId = item->groupId();
+	if (!groupId) {
+		return nullptr;
+	}
+	const auto i = _groups.find(groupId);
+	if (i != _groups.end()) {
+		const auto &result = i->second;
+		if (result.items.size() > 1) {
+			return &result;
+		}
+	}
+	return nullptr;
+}
+
+void Groups::refreshViews(const HistoryItemsList &items) {
+	if (items.empty()) {
+		return;
+	}
+	for (const auto &item : items) {
+		_data->requestItemViewRefresh(item);
+		item->invalidateChatListEntry();
+	}
+}
+
+not_null<HistoryItem*> Groups::findItemToEdit(
+		not_null<HistoryItem*> item) const {
+	const auto group = find(item);
+	if (!group) {
+		return item;
+	}
+	const auto &list = group->items;
+	const auto it = ranges::find_if(
+		list,
+		ranges::not_fn(&HistoryItem::emptyText));
+	if (it == end(list)) {
+		return list.front();
+	}
+	return (*it);
+}
+
+} // namespace Data