153 lines
		
	
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			153 lines
		
	
	
	
		
			3.5 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_send_action.h"
 | 
						|
 | 
						|
#include "data/data_user.h"
 | 
						|
#include "history/history.h"
 | 
						|
#include "history/view/history_view_send_action.h"
 | 
						|
 | 
						|
namespace Data {
 | 
						|
 | 
						|
SendActionManager::SendActionManager()
 | 
						|
: _animation([=](crl::time now) { return callback(now); }) {
 | 
						|
}
 | 
						|
 | 
						|
HistoryView::SendActionPainter *SendActionManager::lookupPainter(
 | 
						|
		not_null<History*> history,
 | 
						|
		MsgId rootId) {
 | 
						|
	if (!rootId) {
 | 
						|
		return history->sendActionPainter();
 | 
						|
	}
 | 
						|
	const auto i = _painters.find(history);
 | 
						|
	if (i == end(_painters)) {
 | 
						|
		return nullptr;
 | 
						|
	}
 | 
						|
	const auto j = i->second.find(rootId);
 | 
						|
	if (j == end(i->second)) {
 | 
						|
		return nullptr;
 | 
						|
	}
 | 
						|
	const auto result = j->second.lock();
 | 
						|
	if (!result) {
 | 
						|
		i->second.erase(j);
 | 
						|
		if (i->second.empty()) {
 | 
						|
			_painters.erase(i);
 | 
						|
		}
 | 
						|
		return nullptr;
 | 
						|
	}
 | 
						|
	crl::on_main([copy = result] {
 | 
						|
	});
 | 
						|
	return result.get();
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::registerFor(
 | 
						|
		not_null<History*> history,
 | 
						|
		MsgId rootId,
 | 
						|
		not_null<UserData*> user,
 | 
						|
		const MTPSendMessageAction &action,
 | 
						|
		TimeId when) {
 | 
						|
	if (history->peer->isSelf()) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	const auto sendAction = lookupPainter(history, rootId);
 | 
						|
	if (!sendAction) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	if (sendAction->updateNeedsAnimating(user, action)) {
 | 
						|
		user->madeAction(when);
 | 
						|
 | 
						|
		if (!_sendActions.contains(std::pair{ history, rootId })) {
 | 
						|
			_sendActions.emplace(std::pair{ history, rootId }, crl::now());
 | 
						|
			_animation.start();
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
auto SendActionManager::repliesPainter(
 | 
						|
	not_null<History*> history,
 | 
						|
	MsgId rootId)
 | 
						|
-> std::shared_ptr<SendActionPainter> {
 | 
						|
	auto &weak = _painters[history][rootId];
 | 
						|
	if (auto strong = weak.lock()) {
 | 
						|
		return strong;
 | 
						|
	}
 | 
						|
	auto result = std::make_shared<SendActionPainter>(history);
 | 
						|
	weak = result;
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::repliesPainterRemoved(
 | 
						|
		not_null<History*> history,
 | 
						|
		MsgId rootId) {
 | 
						|
	const auto i = _painters.find(history);
 | 
						|
	if (i == end(_painters)) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	const auto j = i->second.find(rootId);
 | 
						|
	if (j == end(i->second) || j->second.lock()) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	i->second.erase(j);
 | 
						|
	if (i->second.empty()) {
 | 
						|
		_painters.erase(i);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::repliesPaintersClear(
 | 
						|
		not_null<History*> history,
 | 
						|
		not_null<UserData*> user) {
 | 
						|
	auto &map = _painters[history];
 | 
						|
	for (auto i = map.begin(); i != map.end();) {
 | 
						|
		if (auto strong = i->second.lock()) {
 | 
						|
			strong->clear(user);
 | 
						|
			++i;
 | 
						|
		} else {
 | 
						|
			i = map.erase(i);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (map.empty()) {
 | 
						|
		_painters.erase(history);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
bool SendActionManager::callback(crl::time now) {
 | 
						|
	for (auto i = begin(_sendActions); i != end(_sendActions);) {
 | 
						|
		const auto sendAction = lookupPainter(
 | 
						|
			i->first.first,
 | 
						|
			i->first.second);
 | 
						|
		if (sendAction && sendAction->updateNeedsAnimating(now)) {
 | 
						|
			++i;
 | 
						|
		} else {
 | 
						|
			i = _sendActions.erase(i);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return !_sendActions.empty();
 | 
						|
}
 | 
						|
 | 
						|
auto SendActionManager::animationUpdated() const
 | 
						|
-> rpl::producer<SendActionManager::AnimationUpdate> {
 | 
						|
	return _animationUpdate.events();
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::updateAnimation(AnimationUpdate &&update) {
 | 
						|
	_animationUpdate.fire(std::move(update));
 | 
						|
}
 | 
						|
 | 
						|
auto SendActionManager::speakingAnimationUpdated() const
 | 
						|
-> rpl::producer<not_null<History*>> {
 | 
						|
	return _speakingAnimationUpdate.events();
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::updateSpeakingAnimation(not_null<History*> history) {
 | 
						|
	_speakingAnimationUpdate.fire_copy(history);
 | 
						|
}
 | 
						|
 | 
						|
void SendActionManager::clear() {
 | 
						|
	_sendActions.clear();
 | 
						|
}
 | 
						|
 | 
						|
} // namespace Data
 |