diff --git a/libtaskmanager/launchertasksmodel.cpp b/libtaskmanager/launchertasksmodel.cpp index b3cba5ed4..a0b3745b7 100644 --- a/libtaskmanager/launchertasksmodel.cpp +++ b/libtaskmanager/launchertasksmodel.cpp @@ -1,601 +1,602 @@ /******************************************************************** Copyright 2016 Eike Hein This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) version 3, or any later version accepted by the membership of KDE e.V. (or its successor approved by the membership of KDE e.V.), which shall act as a proxy defined in Section 6 of version 3 of the license. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library. If not, see . *********************************************************************/ #include "launchertasksmodel.h" #include "tasktools.h" #include #include #include #include #include #include #include #include #include #include #include #if HAVE_X11 #include #endif #include "launchertasksmodel_p.h" namespace TaskManager { typedef QSet ActivitiesSet; template inline bool isOnAllActivities(const ActivitiesCollection &activities) { return activities.isEmpty() || activities.contains(NULL_UUID); } class LauncherTasksModel::Private { public: Private(LauncherTasksModel *q); KActivities::Consumer activitiesConsumer; QList launchersOrder; QHash activitiesForLauncher; inline void setActivitiesForLauncher(const QUrl &url, const ActivitiesSet &activities) { if (activities.size() == activitiesConsumer.activities().size()) { activitiesForLauncher[url] = { NULL_UUID }; } else { activitiesForLauncher[url] = activities; } } QHash appDataCache; QTimer sycocaChangeTimer; void init(); AppData appData(const QUrl &url); bool requestAddLauncherToActivities(const QUrl &_url, const QStringList &activities); bool requestRemoveLauncherFromActivities(const QUrl &_url, const QStringList &activities); private: LauncherTasksModel *q; }; LauncherTasksModel::Private::Private(LauncherTasksModel *q) : q(q) { } void LauncherTasksModel::Private::init() { sycocaChangeTimer.setSingleShot(true); sycocaChangeTimer.setInterval(100); QObject::connect(&sycocaChangeTimer, &QTimer::timeout, q, [this]() { if (!launchersOrder.count()) { return; } appDataCache.clear(); // Emit changes of all roles satisfied from app data cache. q->dataChanged(q->index(0, 0), q->index(launchersOrder.count() - 1, 0), QVector{Qt::DisplayRole, Qt::DecorationRole, AbstractTasksModel::AppId, AbstractTasksModel::AppName, AbstractTasksModel::GenericName, AbstractTasksModel::LauncherUrl}); } ); void (KSycoca::*myDatabaseChangeSignal)(const QStringList &) = &KSycoca::databaseChanged; QObject::connect(KSycoca::self(), myDatabaseChangeSignal, q, [this](const QStringList &changedResources) { if (changedResources.contains(QLatin1String("services")) || changedResources.contains(QLatin1String("apps")) || changedResources.contains(QLatin1String("xdgdata-apps"))) { sycocaChangeTimer.start(); } } ); } AppData LauncherTasksModel::Private::appData(const QUrl &url) { const auto &it = appDataCache.constFind(url); if (it != appDataCache.constEnd()) { return *it; } const AppData &data = appDataFromUrl(url, QIcon::fromTheme(QLatin1String("unknown"))); appDataCache.insert(url, data); return data; } bool LauncherTasksModel::Private::requestAddLauncherToActivities(const QUrl &_url, const QStringList &_activities) { // isValid() for the passed-in URL might return true if it was // constructed in TolerantMode, but we want to reject invalid URLs. QUrl url(_url.toString(), QUrl::StrictMode); const auto activities = ActivitiesSet::fromList(_activities); if (url.isEmpty() || !url.isValid()) { return false; } // Merge duplicates int row = -1; foreach(const QUrl &launcher, launchersOrder) { ++row; if (launcherUrlsMatch(url, launcher, IgnoreQueryItems)) { ActivitiesSet newActivities; if (!activitiesForLauncher.contains(url)) { // If we don't have the activities assigned to this url // for some reason newActivities = activities; } else { if (isOnAllActivities(activities)) { // If the new list is empty, or has a null uuid, this // launcher should be on all activities newActivities = ActivitiesSet { NULL_UUID }; } else if (isOnAllActivities(activitiesForLauncher[url])) { // If we have been on all activities before, and we have // been asked to be on a specific one, lets make an // exception - we will set the activities to exactly // what we have been asked newActivities = activities; } else { newActivities += activities; newActivities += activitiesForLauncher[url]; } } if (newActivities != activitiesForLauncher[url]) { setActivitiesForLauncher(url, newActivities); emit q->dataChanged( q->index(row, 0), q->index(row, 0)); emit q->launcherListChanged(); return true; } return false; } } // This is a new one const auto count = launchersOrder.count(); q->beginInsertRows(QModelIndex(), count, count); setActivitiesForLauncher(url, activities); launchersOrder.append(url); q->endInsertRows(); emit q->launcherListChanged(); return true; } bool LauncherTasksModel::Private::requestRemoveLauncherFromActivities(const QUrl &url, const QStringList &activities) { for (int row = 0; row < launchersOrder.count(); ++row) { const QUrl &launcher = launchersOrder.at(row); if (launcherUrlsMatch(url, launcher, IgnoreQueryItems) || launcherUrlsMatch(url, appData(launcher).url, IgnoreQueryItems)) { const auto currentActivities = activitiesForLauncher[url]; ActivitiesSet newActivities; bool remove = false; bool update = false; if (isOnAllActivities(currentActivities)) { // We are currently on all activities. // Should we go away, or just remove from the current one? if (isOnAllActivities(activities)) { remove = true; } else { for (const auto& activity: activitiesConsumer.activities()) { if (!activities.contains(activity)) { newActivities << activity; } else { update = true; } } } } else if (isOnAllActivities(activities)) { remove = true; } else { // We weren't on all activities, just remove those that // we were on for (const auto& activity: currentActivities) { if (!activities.contains(activity)) { newActivities << activity; } } if (newActivities.isEmpty()) { remove = true; } else { update = true; } } if (remove) { q->beginRemoveRows(QModelIndex(), row, row); launchersOrder.removeAt(row); activitiesForLauncher.remove(url); appDataCache.remove(launcher); q->endRemoveRows(); } else if (update) { setActivitiesForLauncher(url, newActivities); emit q->dataChanged( q->index(row, 0), q->index(row, 0)); } if (remove || update) { emit q->launcherListChanged(); return true; } } } return false; } LauncherTasksModel::LauncherTasksModel(QObject *parent) : AbstractTasksModel(parent) , d(new Private(this)) { d->init(); } LauncherTasksModel::~LauncherTasksModel() { } QVariant LauncherTasksModel::data(const QModelIndex &index, int role) const { if (!index.isValid() || index.row() >= d->launchersOrder.count()) { return QVariant(); } const QUrl &url = d->launchersOrder.at(index.row()); const AppData &data = d->appData(url); if (role == Qt::DisplayRole) { return data.name; } else if (role == Qt::DecorationRole) { return data.icon; } else if (role == AppId) { return data.id; } else if (role == AppName) { return data.name; } else if (role == GenericName) { return data.genericName; } else if (role == LauncherUrl) { // Take resolved URL from cache. return data.url; } else if (role == LauncherUrlWithoutIcon) { // Take resolved URL from cache. QUrl url = data.url; if (url.hasQuery()) { QUrlQuery query(url); query.removeQueryItem(QLatin1String("iconData")); url.setQuery(query); } return url; } else if (role == IsLauncher) { return true; } else if (role == IsOnAllVirtualDesktops) { return true; } else if (role == Activities) { return QStringList(d->activitiesForLauncher[url].toList()); } return QVariant(); } int LauncherTasksModel::rowCount(const QModelIndex &parent) const { return parent.isValid() ? 0 : d->launchersOrder.count(); } QStringList LauncherTasksModel::launcherList() const { // Serializing the launchers QStringList result; for (const auto &launcher: d->launchersOrder) { const auto &activities = d->activitiesForLauncher[launcher]; QString serializedLauncher; if (isOnAllActivities(activities)) { serializedLauncher = launcher.toString(); } else { serializedLauncher = "[" + d->activitiesForLauncher[launcher].toList().join(",") + "]\n" + launcher.toString(); } result << serializedLauncher; } return result; } void LauncherTasksModel::setLauncherList(const QStringList &serializedLaunchers) { // Clearing everything QList newLaunchersOrder; QHash newActivitiesForLauncher; // Loading the activity to launchers map QHash> launchersForActivitiesCandidates; for (const auto& serializedLauncher: serializedLaunchers) { QStringList _activities; QUrl url; std::tie(url, _activities) = deserializeLauncher(serializedLauncher); auto activities = ActivitiesSet::fromList(_activities); // Is url is not valid, ignore it if (!url.isValid()) continue; // If we have a null uuid, it means we are on all activities // and we should contain only the null uuid if (isOnAllActivities(activities)) { activities = { NULL_UUID }; } else { // Filter out invalid activities const auto allActivities = d->activitiesConsumer.activities(); ActivitiesSet validActivities; for (const auto& activity: activities) { if (allActivities.contains(activity)) { validActivities << activity; } } if (validActivities.isEmpty()) { // If all activities that had this launcher are // removed, we are killing the launcher as well continue; } activities = validActivities; } // Is the url a duplicate? const auto location = std::find_if(newLaunchersOrder.begin(), newLaunchersOrder.end(), [&url] (const QUrl &item) { return launcherUrlsMatch(url, item, IgnoreQueryItems); }); if (location != newLaunchersOrder.end()) { // It is a duplicate url = *location; } else { // It is not a duplicate, we need to add it // to the list of registered launchers newLaunchersOrder << url; } if (!newActivitiesForLauncher.contains(url)) { // This is the first time we got this url newActivitiesForLauncher[url] = activities; } else if (newActivitiesForLauncher[url].contains(NULL_UUID)) { // Do nothing, we are already on all activities } else if (activities.contains(NULL_UUID)) { newActivitiesForLauncher[url] = { NULL_UUID }; } else { // We are not on all activities, append the new ones newActivitiesForLauncher[url] += activities; } } - if (newActivitiesForLauncher != d->activitiesForLauncher) { + if (newLaunchersOrder != d->launchersOrder + || newActivitiesForLauncher != d->activitiesForLauncher) { // Common case optimization: If the list changed but its size // did not (e.g. due to reordering by a user of this model), // just clear the caches and announce new data instead of // resetting. if (newLaunchersOrder.count() == d->launchersOrder.count()) { d->appDataCache.clear(); std::swap(newLaunchersOrder, d->launchersOrder); std::swap(newActivitiesForLauncher, d->activitiesForLauncher); emit dataChanged( index(0, 0), index(d->launchersOrder.count() - 1, 0)); } else { beginResetModel(); std::swap(newLaunchersOrder, d->launchersOrder); std::swap(newActivitiesForLauncher, d->activitiesForLauncher); d->appDataCache.clear(); endResetModel(); } emit launcherListChanged(); } } bool LauncherTasksModel::requestAddLauncher(const QUrl &url) { return d->requestAddLauncherToActivities(url, { NULL_UUID }); } bool LauncherTasksModel::requestRemoveLauncher(const QUrl &url) { return d->requestRemoveLauncherFromActivities(url, { NULL_UUID }); } bool LauncherTasksModel::requestAddLauncherToActivity(const QUrl &url, const QString &activity) { return d->requestAddLauncherToActivities(url, { activity }); } bool LauncherTasksModel::requestRemoveLauncherFromActivity(const QUrl &url, const QString &activity) { return d->requestRemoveLauncherFromActivities(url, { activity }); } QStringList LauncherTasksModel::launcherActivities(const QUrl &_url) const { const auto position = launcherPosition(_url); if (position == -1) { // If we do not have this launcher, return an empty list return {}; } else { const auto url = d->launchersOrder.at(position); // If the launcher is on all activities, return a null uuid return d->activitiesForLauncher.contains(url) ? d->activitiesForLauncher[url].toList() : QStringList { NULL_UUID }; } } int LauncherTasksModel::launcherPosition(const QUrl &url) const { for (int i = 0; i < d->launchersOrder.count(); ++i) { if (launcherUrlsMatch(url, d->launchersOrder.at(i), IgnoreQueryItems)) { return i; } } return -1; } void LauncherTasksModel::requestActivate(const QModelIndex &index) { requestNewInstance(index); } void LauncherTasksModel::requestNewInstance(const QModelIndex &index) { if (!index.isValid() || index.model() != this || index.row() < 0 || index.row() >= d->launchersOrder.count()) { return; } const QUrl &url = d->launchersOrder.at(index.row()); quint32 timeStamp = 0; #if HAVE_X11 if (KWindowSystem::isPlatformX11()) { timeStamp = QX11Info::appUserTime(); } #endif if (url.scheme() == QLatin1String("preferred")) { KService::Ptr service = KService::serviceByStorageId(defaultApplication(url)); if (!service) { return; } new KRun(QUrl::fromLocalFile(service->entryPath()), 0, false, KStartupInfo::createNewStartupIdForTimestamp(timeStamp)); } else { new KRun(url, 0, false, KStartupInfo::createNewStartupIdForTimestamp(timeStamp)); } } void LauncherTasksModel::requestOpenUrls(const QModelIndex &index, const QList &urls) { if (!index.isValid() || index.model() != this || index.row() < 0 || index.row() >= d->launchersOrder.count() || urls.isEmpty()) { return; } const QUrl &url = d->launchersOrder.at(index.row()); quint32 timeStamp = 0; #if HAVE_X11 if (KWindowSystem::isPlatformX11()) { timeStamp = QX11Info::appUserTime(); } #endif KService::Ptr service; if (url.scheme() == QLatin1String("preferred")) { service = KService::serviceByStorageId(defaultApplication(url)); } else { service = KService::serviceByDesktopPath(url.toLocalFile()); } if (!service) { return; } KRun::runApplication(*service, urls, nullptr, 0, {}, KStartupInfo::createNewStartupIdForTimestamp(timeStamp)); } }