#include "MapModel.h"

MapModel::MapInfo MapModel::mapInfoFromData(const QVariant data)

{

    MapInfo mapInfo;

    mapInfo.type = Invalid;

    mapInfo.name = "";

    mapInfo.theme = "";

    mapInfo.limit = 0;

    mapInfo.scheme = "";

    mapInfo.weapons = "";

    if (data.isValid())

    {

        QList<QVariant> list = data.toList();

        if (list.size() < 1) {

            mapInfo.type = Invalid;

            return mapInfo;

        }

        mapInfo.type = (MapType)list[0].toInt();

        switch (mapInfo.type)

        {

            case GeneratedMap:

            case GeneratedMaze:

            case HandDrawnMap:

                return mapInfo;

            default:

                mapInfo.name = list[1].toString();

                mapInfo.theme = list[2].toString();

                mapInfo.limit = list[3].toInt();

                mapInfo.scheme = list[4].toString();

                mapInfo.weapons = list[5].toString();

        }

    }

    return mapInfo;

}

MapModel::MapModel(QObject *parent) :

    QAbstractListModel(parent)

{

    m_data = QList<QMap<int, QVariant> >();

}

int MapModel::rowCount(const QModelIndex &parent) const

{

    if(parent.isValid())

        return 0;

    else

        return m_data.size();

}

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

{

    if(index.column() > 0 || index.row() >= m_data.size())

        return QVariant();

    else

        return m_data.at(index.row()).value(role, QVariant());

}

void MapModel::loadMaps()

{

    beginResetModel();

    DataManager & datamgr = DataManager::instance();

    QStringList maps =

        datamgr.entryList("Maps", QDir::AllDirs | QDir::NoDotAndDotDot);

    m_data.clear();

#if QT_VERSION >= QT_VERSION_CHECK(4, 7, 0)

    m_data.reserve(maps.size());

#endif

    QMap<int, QVariant> tmp;

    QList<QVariant> mapInfo;

    // TODO: icons for these

    tmp.insert(Qt::DisplayRole, QComboBox::tr("generated map..."));

    mapInfo.append(GeneratedMap);

    tmp.insert(Qt::UserRole, mapInfo);

    m_data.append(tmp);

    tmp.insert(Qt::DisplayRole, QComboBox::tr("generated maze..."));

    mapInfo.replace(0, GeneratedMaze);

    tmp.insert(Qt::UserRole, mapInfo);

    m_data.append(tmp);

    tmp.insert(Qt::DisplayRole, QComboBox::tr("hand drawn map..."));

    mapInfo.replace(0, HandDrawnMap);

    tmp.insert(Qt::UserRole, mapInfo);

    m_data.append(tmp);

    m_nGenerators = 3;

    m_nMissions = 0;

    QFile mapLuaFile;

    QFile mapCfgFile;

    foreach (QString map, maps)

    {

        mapCfgFile.setFileName(

            datamgr.findFileForRead(QString("Maps/%1/map.cfg").arg(map)));

        mapLuaFile.setFileName(

            datamgr.findFileForRead(QString("Maps/%1/map.lua").arg(map)));

        QMap<int, QVariant> dataset;

        if (mapCfgFile.open(QFile::ReadOnly))

        {

            QString theme;

            quint32 limit = 0;

            QString scheme;

            QString weapons;

            QList<QVariant> mapInfo;

            bool isMission = mapLuaFile.exists();

            int type = isMission?MissionMap:StaticMap;

            QTextStream input(&mapCfgFile);

            input >> theme;

            input >> limit;

            input >> scheme;

            input >> weapons;

            mapInfo.push_back(type);

            mapInfo.push_back(map);

            mapInfo.push_back(theme);

            if (limit)

                mapInfo.push_back(limit);

            else

                mapInfo.push_back(18);

            if (scheme.isEmpty())

                scheme = "locked";

            scheme.replace("_", " ");

            if (weapons.isEmpty())

                weapons = "locked";

            weapons.replace("_", " ");

            mapInfo.push_back(scheme);

            mapInfo.push_back(weapons);

            if(isMission)

            {

                // TODO: icon

                map = QComboBox::tr("Mission") + ": " + map;

                m_nMissions++;

            }

            mapCfgFile.close();

            // set name

            dataset.insert(Qt::DisplayRole, map);

            // TODO

            // dataset.insert(Qt::DecorationRole, icon);

            // set mapinfo

            dataset.insert(Qt::UserRole, mapInfo);

            if (isMission) // insert missions before regular maps

                m_data.insert(m_nGenerators + m_nMissions, dataset);

            else

                m_data.append(dataset);

        }

    }

    endResetModel();

}

int MapModel::generatorCount() const

{

    return m_nGenerators;

}

int MapModel::missionCount() const

{

    return m_nMissions;

}