147 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			147 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| // This file is part of Desktop App Toolkit,
 | |
| // a set of libraries for developing nice desktop applications.
 | |
| //
 | |
| // For license and copyright information please follow this link:
 | |
| // https://github.com/desktop-app/legal/blob/master/LEGAL
 | |
| //
 | |
| #include "ui/toast/toast_manager.h"
 | |
| 
 | |
| #include "ui/toast/toast_widget.h"
 | |
| 
 | |
| namespace Ui {
 | |
| namespace Toast {
 | |
| namespace internal {
 | |
| namespace {
 | |
| 
 | |
| base::flat_map<not_null<QObject*>, not_null<Manager*>> ManagersMap;
 | |
| 
 | |
| } // namespace
 | |
| 
 | |
| Manager::Manager(not_null<QWidget*> parent, const CreateTag &)
 | |
| : QObject(parent)
 | |
| , _hideTimer([=] { hideByTimer(); }) {
 | |
| }
 | |
| 
 | |
| bool Manager::eventFilter(QObject *o, QEvent *e) {
 | |
| 	if (e->type() == QEvent::Resize) {
 | |
| 		for (auto i = _toastByWidget.cbegin(), e = _toastByWidget.cend(); i != e; ++i) {
 | |
| 			if (i->first->parentWidget() == o) {
 | |
| 				i->first->onParentResized();
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return QObject::eventFilter(o, e);
 | |
| }
 | |
| 
 | |
| not_null<Manager*> Manager::instance(not_null<QWidget*> parent) {
 | |
| 	auto i = ManagersMap.find(parent);
 | |
| 	if (i == end(ManagersMap)) {
 | |
| 		i = ManagersMap.emplace(
 | |
| 			parent,
 | |
| 			Ui::CreateChild<Manager>(parent.get(), CreateTag())
 | |
| 		).first;
 | |
| 	}
 | |
| 	return i->second;
 | |
| }
 | |
| 
 | |
| base::weak_ptr<Instance> Manager::addToast(
 | |
| 		std::unique_ptr<Instance> &&toast) {
 | |
| 	_toasts.push_back(std::move(toast));
 | |
| 	const auto t = _toasts.back().get();
 | |
| 	const auto widget = t->_widget.get();
 | |
| 
 | |
| 	_toastByWidget.emplace(widget, t);
 | |
| 	connect(widget, &QObject::destroyed, [=] {
 | |
| 		toastWidgetDestroyed(widget);
 | |
| 	});
 | |
| 	if (const auto parent = widget->parentWidget()) {
 | |
| 		auto found = false;
 | |
| 		for (auto i = _toastParents.begin(); i != _toastParents.cend();) {
 | |
| 			if (*i == parent) {
 | |
| 				found = true;
 | |
| 				break;
 | |
| 			} else if (!*i) {
 | |
| 				i = _toastParents.erase(i);
 | |
| 			} else {
 | |
| 				++i;
 | |
| 			}
 | |
| 		}
 | |
| 		if (!found) {
 | |
| 			_toastParents.push_back(parent);
 | |
| 			parent->installEventFilter(this);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	const auto nearestHide = _toastByHideTime.empty()
 | |
| 		? 0LL
 | |
| 		: _toastByHideTime.begin()->first;
 | |
| 	_toastByHideTime.emplace(t->_hideAt, t);
 | |
| 	if (!nearestHide || _toastByHideTime.begin()->first < nearestHide) {
 | |
| 		startNextHideTimer();
 | |
| 	}
 | |
| 	return make_weak(t);
 | |
| }
 | |
| 
 | |
| void Manager::hideByTimer() {
 | |
| 	auto now = crl::now();
 | |
| 	for (auto i = _toastByHideTime.begin(); i != _toastByHideTime.cend();) {
 | |
| 		if (i->first <= now) {
 | |
| 			const auto toast = i->second;
 | |
| 			i = _toastByHideTime.erase(i);
 | |
| 			toast->hideAnimated();
 | |
| 		} else {
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 	startNextHideTimer();
 | |
| }
 | |
| 
 | |
| void Manager::toastWidgetDestroyed(QObject *widget) {
 | |
| 	const auto i = _toastByWidget.find(static_cast<Widget*>(widget));
 | |
| 	if (i == _toastByWidget.cend()) {
 | |
| 		return;
 | |
| 	}
 | |
| 	const auto toast = i->second;
 | |
| 	_toastByWidget.erase(i);
 | |
| 	toast->_widget.release();
 | |
| 
 | |
| 	for (auto i = begin(_toastByHideTime); i != end(_toastByHideTime); ++i) {
 | |
| 		if (i->second == toast) {
 | |
| 			_toastByHideTime.erase(i);
 | |
| 			break;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	const auto j = ranges::find(
 | |
| 		_toasts,
 | |
| 		toast.get(),
 | |
| 		&std::unique_ptr<Instance>::get);
 | |
| 	if (j != end(_toasts)) {
 | |
| 		_toasts.erase(j);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| void Manager::startNextHideTimer() {
 | |
| 	if (_toastByHideTime.empty()) {
 | |
| 		return;
 | |
| 	}
 | |
| 
 | |
| 	auto ms = crl::now();
 | |
| 	if (ms >= _toastByHideTime.begin()->first) {
 | |
| 		crl::on_main(this, [=] {
 | |
| 			hideByTimer();
 | |
| 		});
 | |
| 	} else {
 | |
| 		_hideTimer.callOnce(_toastByHideTime.begin()->first - ms);
 | |
| 	}
 | |
| }
 | |
| 
 | |
| Manager::~Manager() {
 | |
| 	ManagersMap.remove(parent());
 | |
| 	_toastByWidget.clear();
 | |
| 	_toasts.clear();
 | |
| }
 | |
| 
 | |
| } // namespace internal
 | |
| } // namespace Toast
 | |
| } // namespace Ui
 | 
