--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/qmlfrontend/players_model.cpp Fri May 24 23:39:51 2019 +0200
@@ -0,0 +1,364 @@
+#include <QDebug>
+#include <QFile>
+#include <QModelIndex>
+#include <QModelIndexList>
+#include <QPainter>
+#include <QTextStream>
+
+#include "players_model.h"
+
+PlayersListModel::PlayersListModel(QObject *parent)
+ : QAbstractListModel(parent) {
+ m_fontInRoom = QFont();
+ m_fontInRoom.setItalic(true);
+}
+
+int PlayersListModel::rowCount(const QModelIndex &parent) const {
+ if (parent.isValid())
+ return 0;
+ else
+ return m_data.size();
+}
+
+QVariant PlayersListModel::data(const QModelIndex &index, int role) const {
+ if (!index.isValid() || index.row() < 0 || index.row() >= rowCount() ||
+ index.column() != 0)
+ return QVariant(QVariant::Invalid);
+
+ return m_data.at(index.row()).value(role);
+}
+
+bool PlayersListModel::setData(const QModelIndex &index, const QVariant &value,
+ int role) {
+ if (!index.isValid() || index.row() < 0 || index.row() >= rowCount() ||
+ index.column() != 0)
+ return false;
+
+ m_data[index.row()].insert(role, value);
+
+ emit dataChanged(index, index);
+
+ return true;
+}
+
+bool PlayersListModel::insertRows(int row, int count,
+ const QModelIndex &parent) {
+ if (parent.isValid() || row > rowCount() || row < 0 || count < 1)
+ return false;
+
+ beginInsertRows(parent, row, row + count - 1);
+
+ for (int i = 0; i < count; ++i) m_data.insert(row, DataEntry());
+
+ endInsertRows();
+
+ return true;
+}
+
+bool PlayersListModel::removeRows(int row, int count,
+ const QModelIndex &parent) {
+ if (parent.isValid() || row + count > rowCount() || row < 0 || count < 1)
+ return false;
+
+ beginRemoveRows(parent, row, row + count - 1);
+
+ for (int i = 0; i < count; ++i) m_data.removeAt(row);
+
+ endRemoveRows();
+
+ return true;
+}
+
+QModelIndex PlayersListModel::nicknameIndex(const QString &nickname) {
+ QModelIndexList mil =
+ match(index(0), Qt::DisplayRole, nickname, 1, Qt::MatchExactly);
+
+ if (mil.size() > 0)
+ return mil[0];
+ else
+ return QModelIndex();
+}
+
+void PlayersListModel::addPlayer(const QString &nickname, bool notify) {
+ insertRow(rowCount());
+
+ QModelIndex mi = index(rowCount() - 1);
+ setData(mi, nickname);
+
+ checkFriendIgnore(mi);
+
+ emit nickAddedLobby(nickname, notify);
+}
+
+void PlayersListModel::removePlayer(const QString &nickname,
+ const QString &msg) {
+ if (msg.isEmpty())
+ emit nickRemovedLobby(nickname, QString());
+ else
+ emit nickRemovedLobby(nickname, msg);
+
+ QModelIndex mi = nicknameIndex(nickname);
+
+ if (mi.isValid()) removeRow(mi.row());
+}
+
+void PlayersListModel::playerJoinedRoom(const QString &nickname, bool notify) {
+ QModelIndex mi = nicknameIndex(nickname);
+
+ if (mi.isValid()) {
+ setData(mi, true, RoomFilterRole);
+ updateIcon(mi);
+ updateSortData(mi);
+ }
+
+ emit nickAdded(nickname, notify);
+}
+
+void PlayersListModel::playerLeftRoom(const QString &nickname) {
+ emit nickRemoved(nickname);
+
+ QModelIndex mi = nicknameIndex(nickname);
+
+ if (mi.isValid()) {
+ setData(mi, false, RoomFilterRole);
+ setData(mi, false, RoomAdmin);
+ setData(mi, false, Ready);
+ setData(mi, false, InGame);
+ updateIcon(mi);
+ }
+}
+
+void PlayersListModel::setFlag(const QString &nickname, StateFlag flagType,
+ bool isSet) {
+ if (flagType == Friend) {
+ if (isSet)
+ m_friendsSet.insert(nickname.toLower());
+ else
+ m_friendsSet.remove(nickname.toLower());
+
+ // FIXME: set proper file name
+ // saveSet(m_friendsSet, "friends");
+ } else if (flagType == Ignore) {
+ if (isSet)
+ m_ignoredSet.insert(nickname.toLower());
+ else
+ m_ignoredSet.remove(nickname.toLower());
+
+ // FIXME: set proper file name
+ // saveSet(m_ignoredSet, "ignore");
+ }
+
+ QModelIndex mi = nicknameIndex(nickname);
+
+ if (mi.isValid()) {
+ setData(mi, isSet, flagType);
+
+ if (flagType == Friend || flagType == ServerAdmin || flagType == Ignore ||
+ flagType == RoomAdmin)
+ updateSortData(mi);
+
+ updateIcon(mi);
+ }
+}
+
+bool PlayersListModel::isFlagSet(const QString &nickname, StateFlag flagType) {
+ QModelIndex mi = nicknameIndex(nickname);
+
+ if (mi.isValid())
+ return mi.data(flagType).toBool();
+ else if (flagType == Friend)
+ return isFriend(nickname);
+ else if (flagType == Ignore)
+ return isIgnored(nickname);
+ else
+ return false;
+}
+
+void PlayersListModel::resetRoomFlags() {
+ for (int i = rowCount() - 1; i >= 0; --i) {
+ QModelIndex mi = index(i);
+
+ if (mi.data(RoomFilterRole).toBool()) {
+ setData(mi, false, RoomFilterRole);
+ setData(mi, false, RoomAdmin);
+ setData(mi, false, Ready);
+ setData(mi, false, InGame);
+
+ updateSortData(mi);
+ updateIcon(mi);
+ }
+ }
+}
+
+void PlayersListModel::updateIcon(const QModelIndex &index) {
+ quint32 iconNum = 0;
+
+ QList<bool> flags;
+ flags << index.data(Ready).toBool() << index.data(ServerAdmin).toBool()
+ << index.data(RoomAdmin).toBool() << index.data(Registered).toBool()
+ << index.data(Friend).toBool() << index.data(Ignore).toBool()
+ << index.data(InGame).toBool() << index.data(RoomFilterRole).toBool()
+ << index.data(InRoom).toBool() << index.data(Contributor).toBool();
+
+ for (int i = flags.size() - 1; i >= 0; --i)
+ if (flags[i]) iconNum |= 1 << i;
+
+ if (m_icons().contains(iconNum)) {
+ setData(index, m_icons().value(iconNum), Qt::DecorationRole);
+ } else {
+ QPixmap result(24, 16);
+ result.fill(Qt::transparent);
+
+ QPainter painter(&result);
+
+ if (index.data(RoomFilterRole).toBool()) {
+ if (index.data(InGame).toBool()) {
+ painter.drawPixmap(0, 0, 16, 16, QPixmap(":/res/chat/ingame.png"));
+ } else {
+ if (index.data(Ready).toBool())
+ painter.drawPixmap(0, 0, 16, 16, QPixmap(":/res/chat/lamp.png"));
+ else
+ painter.drawPixmap(0, 0, 16, 16, QPixmap(":/res/chat/lamp_off.png"));
+ }
+ } else { // we're in lobby
+ if (!index.data(InRoom).toBool())
+ painter.drawPixmap(0, 0, 16, 16, QPixmap(":/res/Flake.png"));
+ }
+
+ QString mainIconName(":/res/chat/");
+
+ if (index.data(ServerAdmin).toBool())
+ mainIconName += "serveradmin";
+ else {
+ if (index.data(RoomAdmin).toBool())
+ mainIconName += "roomadmin";
+ else
+ mainIconName += "hedgehog";
+
+ if (index.data(Contributor).toBool()) mainIconName += "contributor";
+ }
+
+ if (!index.data(Registered).toBool()) mainIconName += "_gray";
+
+ painter.drawPixmap(8, 0, 16, 16, QPixmap(mainIconName + ".png"));
+
+ if (index.data(Ignore).toBool())
+ painter.drawPixmap(8, 0, 16, 16, QPixmap(":/res/chat/ignore.png"));
+ else if (index.data(Friend).toBool())
+ painter.drawPixmap(8, 0, 16, 16, QPixmap(":/res/chat/friend.png"));
+
+ painter.end();
+
+ QIcon icon(result);
+
+ setData(index, icon, Qt::DecorationRole);
+ m_icons().insert(iconNum, icon);
+ }
+
+ if (index.data(Ignore).toBool())
+ setData(index, QColor(Qt::gray), Qt::ForegroundRole);
+ else if (index.data(Friend).toBool())
+ setData(index, QColor(Qt::green), Qt::ForegroundRole);
+ else
+ setData(index, QBrush(QColor(0xff, 0xcc, 0x00)), Qt::ForegroundRole);
+}
+
+QHash<quint32, QIcon> &PlayersListModel::m_icons() {
+ static QHash<quint32, QIcon> iconsCache;
+
+ return iconsCache;
+}
+
+void PlayersListModel::updateSortData(const QModelIndex &index) {
+ QString result =
+ QString("%1%2%3%4%5%6")
+ // room admins go first, then server admins, then friends
+ .arg(1 - index.data(RoomAdmin).toInt())
+ .arg(1 - index.data(ServerAdmin).toInt())
+ .arg(1 - index.data(Friend).toInt())
+ // ignored at bottom
+ .arg(index.data(Ignore).toInt())
+ // keep nicknames starting from non-letter character at bottom within
+ // group assume there are no empty nicks in list
+ .arg(index.data(Qt::DisplayRole).toString().at(0).isLetter() ? 0 : 1)
+ // sort ignoring case
+ .arg(index.data(Qt::DisplayRole).toString().toLower());
+
+ setData(index, result, SortRole);
+}
+
+void PlayersListModel::setNickname(const QString &nickname) {
+ m_nickname = nickname;
+
+ // FIXME: set proper file names
+ // loadSet(m_friendsSet, "friends");
+ // loadSet(m_ignoredSet, "ignore");
+
+ for (int i = rowCount() - 1; i >= 0; --i) checkFriendIgnore(index(i));
+}
+
+bool PlayersListModel::isFriend(const QString &nickname) {
+ return m_friendsSet.contains(nickname.toLower());
+}
+
+bool PlayersListModel::isIgnored(const QString &nickname) {
+ return m_ignoredSet.contains(nickname.toLower());
+}
+
+void PlayersListModel::checkFriendIgnore(const QModelIndex &mi) {
+ setData(mi, isFriend(mi.data().toString()), Friend);
+ setData(mi, isIgnored(mi.data().toString()), Ignore);
+
+ updateIcon(mi);
+ updateSortData(mi);
+}
+
+void PlayersListModel::loadSet(QSet<QString> &set, const QString &fileName) {
+ set.clear();
+
+ QFile txt(fileName);
+ if (!txt.open(QIODevice::ReadOnly)) return;
+
+ QTextStream stream(&txt);
+ stream.setCodec("UTF-8");
+
+ while (!stream.atEnd()) {
+ QString str = stream.readLine();
+ if (str.startsWith(";") || str.isEmpty()) continue;
+
+ set.insert(str.trimmed());
+ }
+
+ txt.close();
+}
+
+void PlayersListModel::saveSet(const QSet<QString> &set,
+ const QString &fileName) {
+ qDebug("saving set");
+
+ QFile txt(fileName);
+
+ // list empty? => rather have no file for the list than an empty one
+ if (set.isEmpty()) {
+ if (txt.exists()) {
+ // try to remove file, if successful we're done here.
+ if (txt.remove()) return;
+ } else
+ // there is no file
+ return;
+ }
+
+ if (!txt.open(QIODevice::WriteOnly | QIODevice::Truncate)) return;
+
+ QTextStream stream(&txt);
+ stream.setCodec("UTF-8");
+
+ stream << "; this list is used by Hedgewars - do not edit it unless you know "
+ "what you're doing!"
+ << endl;
+
+ foreach (const QString &nick, set.values())
+ stream << nick << endl;
+
+ txt.close();
+}