1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*
* Copyright (C) 2015 Canonical, Ltd.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 3.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include "SessionsModel.h"
#include <QtCore/QFile>
#include <QtCore/QSortFilterProxyModel>
QHash<int, QByteArray> SessionsModel::roleNames() const
{
return m_roleNames;
}
int SessionsModel::rowCount(const QModelIndex& parent) const
{
return m_model->rowCount(parent);
}
QList<QUrl> SessionsModel::iconSearchDirectories() const
{
return m_iconSearchDirectories;
}
void SessionsModel::setIconSearchDirectories(const QList<QUrl> searchDirectories)
{
// QML gives us a url with file:// prepended which breaks QFile::exists()
// so convert the url to a local file
QList<QUrl> localList = {};
Q_FOREACH(const QUrl& searchDirectory, searchDirectories)
{
localList.append(searchDirectory.toLocalFile());
}
m_iconSearchDirectories = localList;
Q_EMIT iconSearchDirectoriesChanged();
}
QUrl SessionsModel::iconUrl(const QString sessionName) const
{
Q_FOREACH(const QUrl& searchDirectory, m_iconSearchDirectories)
{
// This is an established icon naming convention
QString customIconUrl = searchDirectory.toString(QUrl::StripTrailingSlash) +
"/custom_" + sessionName + "_badge.png";
QString iconUrl = searchDirectory.toString(QUrl::StripTrailingSlash) +
"/" + sessionName + "_badge.png";
QFile customIconFile(customIconUrl);
QFile iconFile(iconUrl);
if (customIconFile.exists()) {
return QUrl(customIconUrl);
} else if (iconFile.exists()) {
return QUrl(iconUrl);
} else{
// Search the legacy way
QString path = searchDirectory.toString(QUrl::StripTrailingSlash) + "/";
if (sessionName == "ubuntu" || sessionName == "ubuntu-2d") {
path += "ubuntu_badge.png";
} else if(
sessionName == "gnome-classic" ||
sessionName == "gnome-flashback-compiz" ||
sessionName == "gnome-flashback-metacity" ||
sessionName == "gnome-shell" ||
sessionName == "gnome-wayland" ||
sessionName == "gnome"
){
path += "gnome_badge.png";
} else if (sessionName == "plasma") {
path += "kde_badge.png";
} else if (sessionName == "xterm") {
path += "recovery_console_badge.png";
} else if (sessionName == "remote-login") {
path += "remote_login_help.png";
}
if (QFile(path).exists()) {
return path;
}
}
}
// FIXME make this smarter
return QUrl("./graphics/session_icons/unknown_badge.png");
}
QVariant SessionsModel::data(const QModelIndex& index, int role) const
{
switch (role) {
case SessionsModel::IconRole:
return iconUrl(m_model->data(index, Qt::DisplayRole).toString());
default:
return m_model->data(index, role);
}
}
SessionsModel::SessionsModel(QObject* parent)
: UnitySortFilterProxyModelQML(parent)
{
// Add a custom IconRole that isn't in either of the lightdm implementations
m_model = new QLightDM::SessionsModel(this);
m_roleNames = m_model->roleNames();
m_roleNames[IconRole] = "icon_url";
setModel(m_model);
setSourceModel(m_model);
setSortCaseSensitivity(Qt::CaseInsensitive);
setSortLocaleAware(true);
setSortRole(Qt::DisplayRole);
sort(0);
}
|