150 lines
		
	
	
	
		
			3.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			150 lines
		
	
	
	
		
			3.7 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 "platform/linux/linux_wayland_integration.h"
 | 
						|
 | 
						|
#include "base/platform/linux/base_linux_wayland_utilities.h"
 | 
						|
#include "base/qt_signal_producer.h"
 | 
						|
#include "base/flat_map.h"
 | 
						|
 | 
						|
#include <QtGui/QGuiApplication>
 | 
						|
#include <QtGui/QWindow>
 | 
						|
#include <qpa/qplatformnativeinterface.h>
 | 
						|
#include <qpa/qplatformwindow_p.h>
 | 
						|
 | 
						|
#include <qwayland-wayland.h>
 | 
						|
#include <qwayland-plasma-shell.h>
 | 
						|
 | 
						|
using QWlApp = QNativeInterface::QWaylandApplication;
 | 
						|
using namespace QNativeInterface::Private;
 | 
						|
using namespace base::Platform::Wayland;
 | 
						|
 | 
						|
namespace Platform {
 | 
						|
namespace internal {
 | 
						|
namespace {
 | 
						|
 | 
						|
class PlasmaShell : public Global<QtWayland::org_kde_plasma_shell> {
 | 
						|
public:
 | 
						|
	using Global::Global;
 | 
						|
 | 
						|
	using Surface = AutoDestroyer<QtWayland::org_kde_plasma_surface>;
 | 
						|
	base::flat_map<wl_surface*, Surface> surfaces;
 | 
						|
};
 | 
						|
 | 
						|
} // namespace
 | 
						|
 | 
						|
struct WaylandIntegration::Private
 | 
						|
		: public AutoDestroyer<QtWayland::wl_registry> {
 | 
						|
	Private(not_null<QWlApp*> native)
 | 
						|
	: AutoDestroyer(wl_display_get_registry(native->display()))
 | 
						|
	, display(native->display()) {
 | 
						|
		wl_display_roundtrip(display);
 | 
						|
	}
 | 
						|
 | 
						|
	QtWayland::org_kde_plasma_surface plasmaSurface(QWindow *window);
 | 
						|
 | 
						|
	const not_null<wl_display*> display;
 | 
						|
	std::optional<PlasmaShell> plasmaShell;
 | 
						|
 | 
						|
protected:
 | 
						|
	void registry_global(
 | 
						|
			uint32_t name,
 | 
						|
			const QString &interface,
 | 
						|
			uint32_t version) override {
 | 
						|
		if (interface == qstr("org_kde_plasma_shell")) {
 | 
						|
			plasmaShell.emplace(object(), name, version);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	void registry_global_remove(uint32_t name) override {
 | 
						|
		if (plasmaShell && name == plasmaShell->id()) {
 | 
						|
			plasmaShell = std::nullopt;
 | 
						|
		}
 | 
						|
	}
 | 
						|
};
 | 
						|
 | 
						|
QtWayland::org_kde_plasma_surface WaylandIntegration::Private::plasmaSurface(
 | 
						|
		QWindow *window) {
 | 
						|
	if (!plasmaShell) {
 | 
						|
		return {};
 | 
						|
	}
 | 
						|
 | 
						|
	const auto native = window->nativeInterface<QWaylandWindow>();
 | 
						|
	if (!native) {
 | 
						|
		return {};
 | 
						|
	}
 | 
						|
 | 
						|
	const auto surface = native->surface();
 | 
						|
	if (!surface) {
 | 
						|
		return {};
 | 
						|
	}
 | 
						|
 | 
						|
	const auto it = plasmaShell->surfaces.find(surface);
 | 
						|
	if (it != plasmaShell->surfaces.cend()) {
 | 
						|
		return it->second;
 | 
						|
	}
 | 
						|
 | 
						|
	const auto plasmaSurface = plasmaShell->get_surface(surface);
 | 
						|
	if (!plasmaSurface) {
 | 
						|
		return {};
 | 
						|
	}
 | 
						|
 | 
						|
	const auto result = plasmaShell->surfaces.emplace(surface, plasmaSurface);
 | 
						|
 | 
						|
	base::qt_signal_producer(
 | 
						|
		native,
 | 
						|
		&QWaylandWindow::surfaceDestroyed
 | 
						|
	) | rpl::start_with_next([=] {
 | 
						|
		auto it = plasmaShell->surfaces.find(surface);
 | 
						|
		if (it != plasmaShell->surfaces.cend()) {
 | 
						|
			plasmaShell->surfaces.erase(it);
 | 
						|
		}
 | 
						|
	}, result.first->second.lifetime());
 | 
						|
 | 
						|
	return result.first->second;
 | 
						|
}
 | 
						|
 | 
						|
WaylandIntegration::WaylandIntegration()
 | 
						|
: _private(std::make_unique<Private>(qApp->nativeInterface<QWlApp>())) {
 | 
						|
}
 | 
						|
 | 
						|
WaylandIntegration::~WaylandIntegration() = default;
 | 
						|
 | 
						|
WaylandIntegration *WaylandIntegration::Instance() {
 | 
						|
	const auto native = qApp->nativeInterface<QWlApp>();
 | 
						|
	if (!native) return nullptr;
 | 
						|
	static std::optional<WaylandIntegration> instance;
 | 
						|
	if (instance && native->display() != instance->_private->display) {
 | 
						|
		instance.reset();
 | 
						|
	}
 | 
						|
	if (!instance) {
 | 
						|
		instance.emplace();
 | 
						|
		base::qt_signal_producer(
 | 
						|
			QGuiApplication::platformNativeInterface(),
 | 
						|
			&QObject::destroyed
 | 
						|
		) | rpl::start_with_next([] {
 | 
						|
			instance = std::nullopt;
 | 
						|
		}, instance->_private->lifetime());
 | 
						|
	}
 | 
						|
	return &*instance;
 | 
						|
}
 | 
						|
 | 
						|
bool WaylandIntegration::skipTaskbarSupported() {
 | 
						|
	return _private->plasmaShell.has_value();
 | 
						|
}
 | 
						|
 | 
						|
void WaylandIntegration::skipTaskbar(QWindow *window, bool skip) {
 | 
						|
	auto plasmaSurface = _private->plasmaSurface(window);
 | 
						|
	if (!plasmaSurface.isInitialized()) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	plasmaSurface.set_skip_taskbar(skip);
 | 
						|
}
 | 
						|
 | 
						|
} // namespace internal
 | 
						|
} // namespace Platform
 |