@@ -12,23 +12,22 @@
13 * You should have received a copy of the GNU General Public License
13 * You should have received a copy of the GNU General Public License
14 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16− * Author: Michael Terry <michael.terry@canonical.com>
19#include "Greeter.h"
17#include "Greeter.h"
20#include "UsersModel.h"
18#include "UsersModel.h"
19+#include <QIdentityProxyModel>
21#include <QLightDM/UsersModel>
20#include <QLightDM/UsersModel>
22−#include <QtCore/QSortFilterProxyModel>
24#include <libintl.h>
22#include <libintl.h>
26// First, we define an internal class that wraps LightDM's UsersModel. This
24// First, we define an internal class that wraps LightDM's UsersModel. This
27// class will modify some of the data coming from LightDM. For example, we
25// class will modify some of the data coming from LightDM. For example, we
28−// modify any empty Real Names into just normal Names.
26+// modify any empty Real Names into just normal Names. We also add optional
27+// rows, depending on configuration.
29// (We can't modify the data directly in UsersModel below because it won't sort
28// (We can't modify the data directly in UsersModel below because it won't sort
30// using the modified data.)
29// using the modified data.)
31−class MangleModel : public QSortFilterProxyModel
30+class MangleModel : public QIdentityProxyModel
@@ -36,21 +35,41 @@ public:
36 explicit MangleModel(QObject* parent=0);
35 explicit MangleModel(QObject* parent=0);
38 QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
37 QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
38+ int rowCount(const QModelIndex &parent = QModelIndex()) const override;
39+ QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override;
42+ int guestRow() const;
41MangleModel::MangleModel(QObject* parent)
45MangleModel::MangleModel(QObject* parent)
42− : QSortFilterProxyModel(parent)
46+ : QIdentityProxyModel(parent)
44 setSourceModel(new QLightDM::UsersModel(this));
48 setSourceModel(new QLightDM::UsersModel(this));
47QVariant MangleModel::data(const QModelIndex &index, int role) const
51QVariant MangleModel::data(const QModelIndex &index, int role) const
49− QVariant variantData = QSortFilterProxyModel::data(index, role);
53+ QVariant variantData;
55+ if (index.row() == guestRow() && index.column() == 0) {
57+ case QLightDM::UsersModel::NameRole:
58+ variantData = QStringLiteral("*guest"); break;
59+ case QLightDM::UsersModel::RealNameRole:
60+ variantData = gettext("Guest Session"); break;
61+ case QLightDM::UsersModel::LoggedInRole:
62+ variantData = false; break;
63+ case QLightDM::UsersModel::SessionRole:
64+ variantData = Greeter::instance()->defaultSessionHint(); break;
67+ variantData = QIdentityProxyModel::data(index, role);
51 // If user's real name is empty, switch to unix name
70 // If user's real name is empty, switch to unix name
52 if (role == QLightDM::UsersModel::RealNameRole && variantData.toString().isEmpty()) {
71 if (role == QLightDM::UsersModel::RealNameRole && variantData.toString().isEmpty()) {
53− variantData = QSortFilterProxyModel::data(index, QLightDM::UsersModel::NameRole);
72+ variantData = QIdentityProxyModel::data(index, QLightDM::UsersModel::NameRole);
54 } else if (role == QLightDM::UsersModel::BackgroundPathRole && variantData.toString().startsWith('#')) {
73 } else if (role == QLightDM::UsersModel::BackgroundPathRole && variantData.toString().startsWith('#')) {
55 const QString stringData = "data:image/svg+xml,<svg><rect width='100%' height='100%' fill='" + variantData.toString() + "'/></svg>";
74 const QString stringData = "data:image/svg+xml,<svg><rect width='100%' height='100%' fill='" + variantData.toString() + "'/></svg>";
56 variantData = stringData;
75 variantData = stringData;
@@ -59,27 +78,15 @@ QVariant MangleModel::data(const QModelIndex &index, int role) const
59 return variantData;
78 return variantData;
62−// **** Now we continue with actual UsersModel class ****
64−UsersModel::UsersModel(QObject* parent)
65− : UnitySortFilterProxyModelQML(parent)
67− setModel(new MangleModel(this));
68− setSortCaseSensitivity(Qt::CaseInsensitive);
69− setSortLocaleAware(true);
70− setSortRole(QLightDM::UsersModel::RealNameRole);
74−int UsersModel::guestRow() const
81+int MangleModel::guestRow() const
76 return Greeter::instance()->hasGuestAccount() ?
83 return Greeter::instance()->hasGuestAccount() ?
77− UnitySortFilterProxyModelQML::rowCount() : -1;
84+ QIdentityProxyModel::rowCount() : -1;
80−int UsersModel::rowCount(const QModelIndex &parent) const
87+int MangleModel::rowCount(const QModelIndex &parent) const
82− auto count = UnitySortFilterProxyModelQML::rowCount(parent);
89+ auto count = QIdentityProxyModel::rowCount(parent);
84 if (Greeter::instance()->hasGuestAccount() && !parent.isValid())
91 if (Greeter::instance()->hasGuestAccount() && !parent.isValid())
@@ -87,28 +94,38 @@ int UsersModel::rowCount(const QModelIndex &parent) const
87 return count;
94 return count;
90−QModelIndex UsersModel::index(int row, int column, const QModelIndex &parent) const
97+QModelIndex MangleModel::index(int row, int column, const QModelIndex &parent) const
92 if (row == guestRow() && !parent.isValid()) {
99 if (row == guestRow() && !parent.isValid()) {
93 return createIndex(row, column);
100 return createIndex(row, column);
95− return UnitySortFilterProxyModelQML::index(row, column, parent);
102+ return QIdentityProxyModel::index(row, column, parent);
99−QVariant UsersModel::data(const QModelIndex &index, int role) const
106+// **** Now we continue with actual UsersModel class ****
108+UsersModel::UsersModel(QObject* parent)
109+ : UnitySortFilterProxyModelQML(parent)
101− if (index.row() == guestRow() && index.column() == 0) {
111+ setModel(new MangleModel(this));
102− switch (role) {
112+ setSortCaseSensitivity(Qt::CaseInsensitive);
103− case QLightDM::UsersModel::NameRole: return QStringLiteral("*guest");
113+ setSortLocaleAware(true);
104− case QLightDM::UsersModel::RealNameRole: return gettext("Guest Session");
114+ setSortRole(QLightDM::UsersModel::RealNameRole);
105− case QLightDM::UsersModel::LoggedInRole: return false;
115+ sort(0);
106− case QLightDM::UsersModel::SessionRole: return Greeter::instance()->defaultSessionHint();
116+}
107− default: return QVariant();
117+
108− }
118+bool UsersModel::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const
120+ auto leftName = source_left.data(QLightDM::UsersModel::NameRole);
121+ auto rightName = source_right.data(QLightDM::UsersModel::NameRole);
123+ if (leftName == QStringLiteral("*guest"))
125+ if (rightName == QStringLiteral("*guest"))
111− return QSortFilterProxyModel::data(index, role);
128+ return UnitySortFilterProxyModelQML::lessThan(source_left, source_right);
114#include "UsersModel.moc"
131#include "UsersModel.moc"