130 lines
		
	
	
	
		
			3.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			130 lines
		
	
	
	
		
			3.4 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_desktop_environment.h"
 | 
						|
 | 
						|
#include "base/platform/base_platform_info.h"
 | 
						|
 | 
						|
namespace Platform {
 | 
						|
namespace DesktopEnvironment {
 | 
						|
namespace {
 | 
						|
 | 
						|
QString GetEnv(const char *name) {
 | 
						|
	const auto value = qEnvironmentVariable(name);
 | 
						|
	LOG(("Getting DE, %1: '%2'").arg(name, value));
 | 
						|
	return value;
 | 
						|
}
 | 
						|
 | 
						|
QString GetWM() {
 | 
						|
	const auto result = Platform::GetWindowManager();
 | 
						|
	LOG(("Getting DE via WM: '%1'").arg(result));
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
std::vector<Type> Compute() {
 | 
						|
	auto result = std::vector<Type>();
 | 
						|
 | 
						|
	const auto xdgCurrentDesktop = GetEnv(
 | 
						|
		"XDG_CURRENT_DESKTOP").toLower().split(':', Qt::SkipEmptyParts);
 | 
						|
 | 
						|
	const auto xdgSessionDesktop = GetEnv("XDG_SESSION_DESKTOP").toLower();
 | 
						|
 | 
						|
	const auto desktopSession = [] {
 | 
						|
		const auto result = GetEnv("DESKTOP_SESSION").toLower();
 | 
						|
		const auto slash = result.lastIndexOf('/');
 | 
						|
		// DESKTOP_SESSION can contain a path
 | 
						|
		if (slash != -1) {
 | 
						|
			return result.mid(slash + 1);
 | 
						|
		}
 | 
						|
		return result;
 | 
						|
	}();
 | 
						|
 | 
						|
	const auto windowManager = GetWM().toLower();
 | 
						|
 | 
						|
	const auto desktopToType = [&](const QString &desktop) {
 | 
						|
		if (desktop == qstr("unity")) {
 | 
						|
			// gnome-fallback sessions set XDG_CURRENT_DESKTOP to Unity
 | 
						|
			// DESKTOP_SESSION can be gnome-fallback or gnome-fallback-compiz
 | 
						|
			if (desktopSession.contains(qstr("gnome-fallback"))) {
 | 
						|
				result.push_back(Type::Gnome);
 | 
						|
			}
 | 
						|
			result.push_back(Type::Unity);
 | 
						|
		} else if (desktop == qstr("gnome")) {
 | 
						|
			result.push_back(Type::Gnome);
 | 
						|
		} else if (desktop == qstr("x-cinnamon") || desktop == qstr("cinnamon")) {
 | 
						|
			result.push_back(Type::Cinnamon);
 | 
						|
		} else if (desktop == qstr("kde")) {
 | 
						|
			result.push_back(Type::KDE);
 | 
						|
		} else if (desktop == qstr("mate")) {
 | 
						|
			result.push_back(Type::MATE);
 | 
						|
		}
 | 
						|
	};
 | 
						|
 | 
						|
	for (const auto ¤t : xdgCurrentDesktop) {
 | 
						|
		desktopToType(current);
 | 
						|
	}
 | 
						|
 | 
						|
	if (!xdgSessionDesktop.isEmpty()) {
 | 
						|
		desktopToType(xdgSessionDesktop);
 | 
						|
	}
 | 
						|
 | 
						|
	if (!desktopSession.isEmpty()) {
 | 
						|
		desktopToType(desktopSession);
 | 
						|
	}
 | 
						|
 | 
						|
	// Fall back on some older environment variables.
 | 
						|
	// Useful particularly in the DESKTOP_SESSION=default case.
 | 
						|
	if (!GetEnv("GNOME_DESKTOP_SESSION_ID").isEmpty()) {
 | 
						|
		result.push_back(Type::Gnome);
 | 
						|
	}
 | 
						|
	if (!GetEnv("KDE_FULL_SESSION").isEmpty()) {
 | 
						|
		result.push_back(Type::KDE);
 | 
						|
	}
 | 
						|
 | 
						|
	// Some DEs could be detected via X11
 | 
						|
	if (!windowManager.isEmpty()) {
 | 
						|
		if (windowManager == qstr("gnome shell")) {
 | 
						|
			result.push_back(Type::Gnome);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	result = result | ranges::views::unique | ranges::to_vector;
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
std::vector<Type> ComputeAndLog() {
 | 
						|
	const auto result = Compute();
 | 
						|
	if (result.empty()) {
 | 
						|
		LOG(("DE: Other"));
 | 
						|
		return {};
 | 
						|
	}
 | 
						|
	const auto names = ranges::accumulate(
 | 
						|
		result | ranges::views::transform([](auto type) {
 | 
						|
			switch (type) {
 | 
						|
			case Type::Gnome: return u"Gnome, "_q;
 | 
						|
			case Type::Cinnamon: return u"Cinnamon, "_q;
 | 
						|
			case Type::KDE: return u"KDE, "_q;
 | 
						|
			case Type::Unity: return u"Unity, "_q;
 | 
						|
			case Type::MATE: return u"MATE, "_q;
 | 
						|
			}
 | 
						|
			Unexpected("Type in Platform::DesktopEnvironment::ComputeAndLog");
 | 
						|
		}),
 | 
						|
		QString()).chopped(2);
 | 
						|
	LOG(("DE: %1").arg(names));
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
} // namespace
 | 
						|
 | 
						|
// Thanks Chromium.
 | 
						|
std::vector<Type> Get() {
 | 
						|
	static const auto result = ComputeAndLog();
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
} // namespace DesktopEnvironment
 | 
						|
} // namespace Platform
 |