#ifndef GAMESMODEL_H
#define GAMESMODEL_H

#include <QAbstractTableModel>
#include <QSortFilterProxyModel>
#include <QList>
#include <QSet>
#include "gametypemap.h"
#include "pb/serverinfo_game.pb.h"

class ServerInfo_User;

class GamesModel : public QAbstractTableModel {
	Q_OBJECT
private:
	QList<ServerInfo_Game> gameList;
	QMap<int, QString> rooms;
	QMap<int, GameTypeMap> gameTypes;
public:
	GamesModel(const QMap<int, QString> &_rooms, const QMap<int, GameTypeMap> &_gameTypes, QObject *parent = 0);
	int rowCount(const QModelIndex &parent = QModelIndex()) const { return parent.isValid() ? 0 : gameList.size(); }
	int columnCount(const QModelIndex &/*parent*/ = QModelIndex()) const { return 8; }
	QVariant data(const QModelIndex &index, int role) const;
	QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
	
	const ServerInfo_Game &getGame(int row);
	void updateGameList(const ServerInfo_Game &game);
	const QMap<int, GameTypeMap> &getGameTypes() { return gameTypes; }
};

class GamesProxyModel : public QSortFilterProxyModel {
	Q_OBJECT
private:
	ServerInfo_User *ownUser;
	bool unavailableGamesVisible;
	QString gameNameFilter, creatorNameFilter;
	QSet<int> gameTypeFilter;
	int maxPlayersFilterMin, maxPlayersFilterMax;
public:
	GamesProxyModel(QObject *parent = 0, ServerInfo_User *_ownUser = 0);
	
	bool getUnavailableGamesVisible() const { return unavailableGamesVisible; }
	void setUnavailableGamesVisible(bool _unavailableGamesVisible);
	QString getGameNameFilter() const { return gameNameFilter; }
	void setGameNameFilter(const QString &_gameNameFilter);
	QString getCreatorNameFilter() const { return creatorNameFilter; }
	void setCreatorNameFilter(const QString &_creatorNameFilter);
	QSet<int> getGameTypeFilter() const { return gameTypeFilter; }
	void setGameTypeFilter(const QSet<int> &_gameTypeFilter);
	int getMaxPlayersFilterMin() const { return maxPlayersFilterMin; }
	int getMaxPlayersFilterMax() const { return maxPlayersFilterMax; }
	void setMaxPlayersFilter(int _maxPlayersFilterMin, int _maxPlayersFilterMax);
	void resetFilterParameters();
protected:
	bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const;
};

#endif
