serverlistmodel.h

//------------------------------------------------------------------------------
// serverlistmodel.h
//------------------------------------------------------------------------------
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
// 02110-1301, USA.
//
//------------------------------------------------------------------------------
// Copyright (C) 2009 "Zalewa" <zalewapl@gmail.com>
//------------------------------------------------------------------------------
#ifndef __SERVER_LIST_MODEL_H_
#define __SERVER_LIST_MODEL_H_

#include "serverapi/serverptr.h"
#include <QStandardItemModel>

class EnginePlugin;
class Server;
class ServerListSortFilterProxyModel;
class ServerList;

class ServerListModel : public QStandardItemModel
{
        Q_OBJECT

        friend class ServerList;
        friend class ServerListProxyModel;

        public:
                enum ServerGroup
                {
                        SG_NORMAL = 200,
                        SG_WAIT = 175,
                        SG_BANNED = 150,
                        SG_TIMEOUT = 125,
                        SG_BAD = 100,
                        SG_FIRST_QUERY = 50,
                };

                enum ServerListDataTypes
                {
                        // Pointer to the server structure is always stored in the first column
                        SLDT_POINTER_TO_SERVER_STRUCTURE = Qt::UserRole+1,
                        SLDT_SORT = Qt::UserRole+2
                };

                ServerListModel(ServerList* parent);

                int addServer(ServerPtr server);
                QList<ServerPtr> customServers() const;

                int findServerOnTheList(const Server* server);

                ServerList* handler() { return parentHandler; }

                QList<ServerPtr> nonSpecialServers() const;
                QList<ServerPtr> servers() const;
                QList<ServerPtr> serversForPlugin(const EnginePlugin *plugin) const;

                void redraw(int row);
                void redrawAll();

                void removeServer(const ServerPtr &server);

                void updateFlag(int row, bool force = true);

                int updateServer(int row, ServerPtr server);

                ServerPtr serverFromList(int rowIndex) const;
                ServerPtr serverFromList(const QModelIndex&) const;

                void setRefreshing(ServerPtr server);

        signals:
                void allRowsContentChanged();
                void rowContentChanged(int row);

        private:
                void prepareHeaders();
                ServerGroup serverGroup(int row);

                QVariant columnSortData(int row, int column);

                ServerList* parentHandler;
};

#endif