#ifndef PLAYERSLISTMODEL_H
#define PLAYERSLISTMODEL_H

#include <QAbstractListModel>
#include <QFont>
#include <QHash>
#include <QIcon>
#include <QModelIndex>
#include <QSet>

class PlayersListModel : public QAbstractListModel {
  Q_OBJECT

 public:
  enum StateFlag {
    Ready = Qt::UserRole,
    ServerAdmin = Qt::UserRole + 1,
    RoomAdmin = Qt::UserRole + 2,
    Registered = Qt::UserRole + 3,
    Friend = Qt::UserRole + 4,
    Ignore = Qt::UserRole + 5,
    InGame = Qt::UserRole + 6,
    InRoom = Qt::UserRole + 7,
    Contributor = Qt::UserRole + 8
    // if you add a role that will affect the player icon,
    // then also add it to the flags Qlist in updateIcon()!
  };

  enum SpecialRoles {
    SortRole = Qt::UserRole + 100,
    RoomFilterRole = Qt::UserRole + 101
  };

  explicit PlayersListModel(QObject *parent = 0);

  int rowCount(const QModelIndex &parent = QModelIndex()) const override;

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

  void setFlag(const QString &nickname, StateFlag flagType, bool isSet);
  bool isFlagSet(const QString &nickname, StateFlag flagType);

  bool insertRows(int row, int count,
                  const QModelIndex &parent = QModelIndex()) override;
  bool removeRows(int row, int count,
                  const QModelIndex &parent = QModelIndex()) override;

  QModelIndex nicknameIndex(const QString &nickname);

 public slots:
  void addPlayer(const QString &nickname, bool notify);
  void removePlayer(const QString &nickname, const QString &msg = QString());
  void playerJoinedRoom(const QString &nickname, bool notify);
  void playerLeftRoom(const QString &nickname);
  void resetRoomFlags();
  void setNickname(const QString &nickname);

 signals:
  void nickAdded(const QString &nick, bool notifyNick);
  void nickRemoved(const QString &nick);
  void nickAddedLobby(const QString &nick, bool notifyNick);
  void nickRemovedLobby(const QString &nick, const QString &message);

 private:
  QHash<quint32, QIcon> &m_icons();
  using DataEntry = QHash<int, QVariant>;
  QList<DataEntry> m_data;
  QSet<QString> m_friendsSet, m_ignoredSet;
  QString m_nickname;
  QFont m_fontInRoom;

  void updateIcon(const QModelIndex &index);
  void updateSortData(const QModelIndex &index);
  void loadSet(QSet<QString> &set, const QString &fileName);
  void saveSet(const QSet<QString> &set, const QString &fileName);
  void checkFriendIgnore(const QModelIndex &mi);
  bool isFriend(const QString &nickname);
  bool isIgnored(const QString &nickname);
};

#endif  // PLAYERSLISTMODEL_H