plugins/LightDM/UsersModel.cpp · unity8

git.iko.soy/unity8

on 217d68c help sign in

unity8/plugins/LightDM/UsersModel.cpp

168 lines 5.5 kB history raw

1/*

3 *

4 * This program is free software; you can redistribute it and/or modify

5 * it under the terms of the GNU General Public License as published by

6 * the Free Software Foundation; version 3.

7 *

8 * This program is distributed in the hope that it will be useful,

9 * but WITHOUT ANY WARRANTY; without even the implied warranty of

10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

11 * GNU General Public License for more details.

12 *

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/>.

15 *

16 * Author: Michael Terry <michael.terry@canonical.com>

17 */

19#include "Greeter.h"

20#include "UsersModel.h"

21#include <QLightDM/UsersModel>

22#include <QtCore/QSortFilterProxyModel>

24#include <libintl.h>

26// 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

28// modify any empty Real Names into just normal Names.

29// (We can't modify the data directly in UsersModel below because it won't sort

30// using the modified data.)

31class MangleModel : public QSortFilterProxyModel

32{

33 Q_OBJECT

35public:

36 explicit MangleModel(QObject* parent=0);

38 QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;

39};

41MangleModel::MangleModel(QObject* parent)

42 : QSortFilterProxyModel(parent)

43{

44 setSourceModel(new QLightDM::UsersModel(this));

45}

47QVariant MangleModel::data(const QModelIndex &index, int role) const

48{

49 QVariant variantData = QSortFilterProxyModel::data(index, role);

51 // If user's real name is empty, switch to unix name

52 if (role == QLightDM::UsersModel::RealNameRole && variantData.toString().isEmpty()) {

53 variantData = QSortFilterProxyModel::data(index, QLightDM::UsersModel::NameRole);

54 } 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>";

56 variantData = stringData;

57 }

59 return variantData;

60}

62// **** Now we continue with actual UsersModel class ****

64UsersModel::UsersModel(QObject* parent)

65 : UnitySortFilterProxyModelQML(parent)

66 , m_showManual(false)

67{

68 if (!Greeter::instance()->hideUsersHint()) {

69 setModel(new MangleModel(this));

70 }

71 setSortCaseSensitivity(Qt::CaseInsensitive);

72 setSortLocaleAware(true);

73 setSortRole(QLightDM::UsersModel::RealNameRole);

74 sort(0);

76 connect(this, &UnitySortFilterProxyModelQML::countChanged,

77 this, &UsersModel::updateShowManual);

78 updateShowManual();

79}

81void UsersModel::updateShowManual()

82{

83 // Show manual login if we are asked to OR if no other entry exists

84 bool showManual = Greeter::instance()->showManualLoginHint() ||

85 (QSortFilterProxyModel::rowCount() == 0 &&

86 !Greeter::instance()->hasGuestAccount());

88 if (m_showManual != showManual) {

89 int row = QSortFilterProxyModel::rowCount();

90 if (showManual)

91 beginInsertRows(QModelIndex(), row, row);

92 else

93 beginRemoveRows(QModelIndex(), row, row);

95 m_showManual = showManual;

97 if (showManual)

98 endInsertRows();

99 else

100 endRemoveRows();

101 }

102}

103

104int UsersModel::manualRow() const

105{

106 if (!m_showManual)

107 return -1;

108

109 return QSortFilterProxyModel::rowCount();

110}

111

112int UsersModel::guestRow() const

113{

114 if (!Greeter::instance()->hasGuestAccount())

115 return -1;

116

117 int row = QSortFilterProxyModel::rowCount();

118 if (m_showManual)

119 row++;

120

121 return row;

122}

123

124int UsersModel::rowCount(const QModelIndex &parent) const

125{

126 auto count = UnitySortFilterProxyModelQML::rowCount(parent);

127

128 if (m_showManual && !parent.isValid())

129 count++;

130 if (Greeter::instance()->hasGuestAccount() && !parent.isValid())

131 count++;

132

133 return count;

134}

135

136QModelIndex UsersModel::index(int row, int column, const QModelIndex &parent) const

137{

138 if ((row == manualRow() || row == guestRow()) && !parent.isValid()) {

139 return createIndex(row, column);

140 } else {

141 return UnitySortFilterProxyModelQML::index(row, column, parent);

142 }

143}

144

145QVariant UsersModel::data(const QModelIndex &index, int role) const

146{

147 if (index.row() == manualRow() && index.column() == 0) {

148 switch (role) {

149 case QLightDM::UsersModel::NameRole: return QStringLiteral("*other");

150 case QLightDM::UsersModel::RealNameRole: return gettext("Login");

151 case QLightDM::UsersModel::LoggedInRole: return false;

152 case QLightDM::UsersModel::SessionRole: return Greeter::instance()->defaultSessionHint();

153 default: return QVariant();

154 }

155 } else if (index.row() == guestRow() && index.column() == 0) {

156 switch (role) {

157 case QLightDM::UsersModel::NameRole: return QStringLiteral("*guest");

158 case QLightDM::UsersModel::RealNameRole: return gettext("Guest Session");

159 case QLightDM::UsersModel::LoggedInRole: return false;

160 case QLightDM::UsersModel::SessionRole: return Greeter::instance()->defaultSessionHint();

161 default: return QVariant();

162 }

163 }

164

165 return QSortFilterProxyModel::data(index, role);

166}

167

168#include "UsersModel.moc"