Differential D1722 Diff 4228 libtaskmanager/startuptasksmodel.cpp

Changeset View

Standalone View

libtaskmanager/startuptasksmodel.cpp

This file was added.

1		/********************************************************************
2		Copyright 2016 Eike Hein <hein@kde.org>
3
4		This library is free software; you can redistribute it and/or
5		modify it under the terms of the GNU Lesser General Public
6		License as published by the Free Software Foundation; either
7		version 2.1 of the License, or (at your option) version 3, or any
8		later version accepted by the membership of KDE e.V. (or its
9		successor approved by the membership of KDE e.V.), which shall
10		act as a proxy defined in Section 6 of version 3 of the license.
11
12		This library is distributed in the hope that it will be useful,
13		but WITHOUT ANY WARRANTY; without even the implied warranty of
14		MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15		Lesser General Public License for more details.
16
17		You should have received a copy of the GNU Lesser General Public
18		License along with this library. If not, see <http://www.gnu.org/licenses/>.
19		*********************************************************************/
20
21		#include "startuptasksmodel.h"
22
23		#include <KConfig>
24		#include <KConfigGroup>
25		#include <KDirWatch>
26		#include <KService>
27		#include <KServiceTypeTrader>
28		#include <KStartupInfo>
29
30		#include <QIcon>
31		#include <QTimer>
32		#include <QUrl>
33
34		namespace TaskManager
35		{
36
37		class StartupTasksModel::Private
38		{
39		public:
40		Private(StartupTasksModel *q);
41		KDirWatch* configWatcher = nullptr;
42		KStartupInfo *startupInfo = nullptr;
43		QVector<KStartupInfoId> startups;
44		QHash<QByteArray, KStartupInfoData> startupData;
45		QHash<QByteArray, QUrl> launcherUrls;
46
47		void init();
48		void loadConfig();
49		QUrl launcherUrl(const KStartupInfoData &data);
50
51		private:
52		StartupTasksModel *q;
53		};
54
55		StartupTasksModel::Private::Private(StartupTasksModel *q)
56		: q(q)
57		{
58		}
59
60		void StartupTasksModel::Private::init()
61		{
62		configWatcher = new KDirWatch(q);
63		configWatcher->addFile(QStandardPaths::writableLocation(QStandardPaths::GenericConfigLocation)
64		+ QLatin1String("/klaunchrc"));
65
	broulikUnsubmitted Done QLatin1String broulik: QLatin1String
	heinAuthorUnsubmitted Not Done Done. hein: Done.
66		QObject::connect(configWatcher, &KDirWatch::dirty, [this] { loadConfig(); });
67		QObject::connect(configWatcher, &KDirWatch::created, [this] { loadConfig(); });
68		QObject::connect(configWatcher, &KDirWatch::deleted, [this] { loadConfig(); });
69
70		loadConfig();
71		}
72
73		void StartupTasksModel::Private::loadConfig()
74		{
75		const KConfig _c("klaunchrc");
76		KConfigGroup c(&_c, "FeedbackStyle");
77
78		if (!c.readEntry("TaskbarButton", true)) {
79		delete startupInfo;
80		startupInfo = nullptr;
81
82		q->beginResetModel();
83		startups.clear();
84		startupData.clear();
85		q->endResetModel();
86
87		return;
88		}
89
90		if (!startupInfo) {
91		startupInfo = new KStartupInfo(KStartupInfo::CleanOnCantDetect, q);
92
93		QObject::connect(startupInfo, &KStartupInfo::gotNewStartup, q,
94		[this](const KStartupInfoId &id, const KStartupInfoData &data) {
95		if (startups.contains(id)) {
96		return;
97		}
98
99		const QString appId = data.applicationId();
100		const QString bin = data.bin();
101
102		foreach(const KStartupInfoData &known, startupData) {
103		// Reject if we already have a startup notification for this app.
104		if (known.applicationId() == appId && known.bin() == bin) {
105		return;
106		}
107		}
108
109		const int count = startups.count();
110		q->beginInsertRows(QModelIndex(), count, count);
111		startups.append(id);
112		startupData.insert(id.id(), data);
113		launcherUrls.insert(id.id(), launcherUrl(data));
114		q->endInsertRows();
115		}
116		);
117
118		QObject::connect(startupInfo, &KStartupInfo::gotRemoveStartup, q,
119		[this](const KStartupInfoId &id) {
120		// The order in which startups are cancelled and corresponding
121		// windows appear is not reliable. Add some grace time to make
122		// an overlap more likely, giving a proxy some time to arbitrate
123		// between the two.
124		QTimer::singleShot(500,
125		[this, id]() {
126		const int row = startups.indexOf(id);
127
128		if (row != -1) {
129		q->beginRemoveRows(QModelIndex(), row, row);
130		startups.removeAt(row);
131		startupData.remove(id.id());
132		launcherUrls.remove(id.id());
133		q->endRemoveRows();
134		}
135		}
136		);
137		}
138		);
139
140		QObject::connect(startupInfo, &KStartupInfo::gotStartupChange, q,
141		[this](const KStartupInfoId &id, const KStartupInfoData &data) {
142		const int row = startups.indexOf(id);
143		if (row != -1) {
144		startupData.insert(id.id(), data);
145		launcherUrls.insert(id.id(), launcherUrl(data));
146		QModelIndex idx = q->index(row);
147		emit q->dataChanged(idx, idx);
	davidedmundsonUnsubmitted Done q->index(row) davidedmundson: q->index(row)
	heinAuthorUnsubmitted Not Done Done. hein: Done.
148		}
149		}
150		);
151		}
152
153		c = KConfigGroup(&_c, "TaskbarButtonSettings");
154		startupInfo->setTimeout(c.readEntry("Timeout", 5));
155		}
156
157		QUrl StartupTasksModel::Private::launcherUrl(const KStartupInfoData &data)
158		{
159		QUrl launcherUrl;
160		KService::List services;
161
162		QString appId = data.applicationId();
163
164		// Try to match via desktop filename ...
165		if (!appId.isEmpty() && appId.endsWith(QLatin1String(".desktop"))) {
166		if (appId.startsWith(QLatin1String("/"))) {
167		launcherUrl = QUrl::fromLocalFile(appId);
168		return launcherUrl;
169		} else {
170		if (appId.endsWith(QLatin1String(".desktop"))) {
171		appId = appId.mid(appId.length() - 8);
172		}
173
174		services = KServiceTypeTrader::self()->query(QStringLiteral("Application"),
175		QStringLiteral("exist Exec and ('%1' =~ DesktopEntryName)").arg(appId));
176		}
177		}
178
179		const QString wmClass = data.WMClass();
180
181		// Try StartupWMClass.
182		if (services.empty() && !wmClass.isEmpty()) {
183		services = KServiceTypeTrader::self()->query(QStringLiteral("Application"),
	broulikUnsubmitted Done I prefer isEmpty(), also consistent with isNull() broulik: I prefer isEmpty(), also consistent with isNull()
	heinAuthorUnsubmitted Not Done Done. hein: Done.
184		QStringLiteral("exist Exec and ('%1' =~ StartupWMClass)").arg(wmClass));
185		}
186
187		const QString name = data.findName();
188
189		// Try via name ...
190		if (services.empty() && !name.isEmpty()) {
191		services = KServiceTypeTrader::self()->query(QStringLiteral("Application"),
192		QStringLiteral("exist Exec and ('%1' =~ Name)").arg(name));
193		}
194
195		if (!services.empty()) {
196		QString path = services.at(0)->entryPath();
197
	broulikUnsubmitted Done .at(0)? operator[] is weird broulik: .at(0)? operator[] is weird
	heinAuthorUnsubmitted Not Done Done. hein: Done.
198		if (path.isEmpty()) {
199		path = services.at(0)->exec();
200		}
201
202		if (!path.isEmpty()) {
203		launcherUrl = QUrl::fromLocalFile(path);
204		}
205		}
206
207		return launcherUrl;
208		}
209
210		StartupTasksModel::StartupTasksModel(QObject *parent)
211		: AbstractTasksModel(parent)
212		, d(new Private(this))
213		{
214		d->init();
215		}
216
217		StartupTasksModel::~StartupTasksModel()
218		{
219		}
220
221		QVariant StartupTasksModel::data(const QModelIndex &index, int role) const
222		{
223		if (!index.isValid() \|\| index.row() >= d->startups.count()) {
224		return QVariant();
225		}
226
227		const QByteArray &id = d->startups.at(index.row()).id();
228
229		if (!d->startupData.contains(id)) {
230		return QVariant();
231		}
232
233		const KStartupInfoData &data = d->startupData.value(id);
234
235		if (role == Qt::DisplayRole) {
236		return data.findName();
237		} else if (role == Qt::DecorationRole) {
238		return QIcon::fromTheme(data.findIcon(), QIcon::fromTheme(QLatin1String("unknown")));
239		} else if (role == AppId) {
	broulikUnsubmitted Done QStringLiteral/QLatin1String broulik: QStringLiteral/QLatin1String
	heinAuthorUnsubmitted Not Done Done. hein: Done.
240		QString idFromPath = QUrl::fromLocalFile(data.applicationId()).fileName();
241
242		if (idFromPath.endsWith(QLatin1String(".desktop"))) {
243		idFromPath = idFromPath.left(idFromPath.length() - 8);
	broulikUnsubmitted Done QLatin1String broulik: QLatin1String
	heinAuthorUnsubmitted Not Done Done. hein: Done.
244		}
245
246		return idFromPath;
247		} else if (role == AppName) {
248		return data.findName();
249		} else if (role == LauncherUrl) {
250		return d->launcherUrls.value(id);
251		} else if (role == IsStartup) {
252		return true;
253		} else if (role == VirtualDesktop) {
254		return data.desktop();
255		} else if (role == IsOnAllVirtualDesktops) {
256		return (data.desktop() == 0);
257		} /* else if (role == Screen) {
258		// You might be tempted to do this, but KStartupInfoData::screen()
259		// is actually the X11 screen.
260		return (data.screen() == 0);
261		} */
262
263		return QVariant();
264		}
265
266		int StartupTasksModel::rowCount(const QModelIndex &parent) const
267		{
268		return parent.isValid() ? 0 : d->startups.count();
269		}
270
271		void StartupTasksModel::requestNewInstance(const QModelIndex &index)
272		{
273		Q_UNUSED(index)
274
275		// FIXME Implement.
276		}
277
278		}