yacreader/YACReaderLibrary/comic_vine/model/volume_comics_model.cpp

#include "volume_comics_model.h"
#include "qnaturalsorting.h"


#include <QtScript>

bool lessThan(const QList<QString> & left, const QList<QString> & right)
{
	if ((left.count() > 0) && (right.count() > 0))
		return naturalSortLessThanCI(left.at(0),right.at(0));
	else
		return true;
}

VolumeComicsModel::VolumeComicsModel(QObject * parent) :
	JSONModel(parent),numExtraRows(0)
{
}

void VolumeComicsModel::load(const QString & json)
{
	QScriptEngine engine;
	QScriptValue sc;
	sc = engine.evaluate("(" + json + ")");

	if (!sc.property("error").isValid() && sc.property("error").toString() != "OK")
	{
		qDebug("Error detected");
	}
	else
	{
		QScriptValueIterator it(sc.property("results"));
		//bool test;
		QScriptValue resultsValue;
		while (it.hasNext()) {
			it.next();
            if(it.flags() & QScriptValue::SkipInEnumeration)
                continue;
			resultsValue = it.value();
			QString issueNumber = resultsValue.property("issue_number").toString();
            QScriptValue propertyName = resultsValue.property("name");
            QString name = propertyName.isNull() ? "-" : propertyName.toString();
			QString coverURL = resultsValue.property("image").property("medium_url").toString();
			QString id = resultsValue.property("id").toString();
			QStringList l;
			l << issueNumber << name << coverURL << id;
            _data.push_back(l);
		}

		qSort(_data.begin(),_data.end(),lessThan);
    }
}

/*void VolumeComicsModel::load(const QStringList &jsonList)
{
    foreach (QString json, jsonList) {
        load(json);
    }
}*/

QModelIndex VolumeComicsModel::parent(const QModelIndex &index) const
{
	Q_UNUSED(index)
	return QModelIndex(); //no parent
}

int VolumeComicsModel::rowCount(const QModelIndex &parent) const
{
	Q_UNUSED(parent)
	return _data.count() + numExtraRows;
}

int VolumeComicsModel::columnCount(const QModelIndex &parent) const
{
	Q_UNUSED(parent)
	if(_data.isEmpty())
		return 0;
	else
		return 2;
}

QVariant VolumeComicsModel::data(const QModelIndex &index, int role) const
{
	if (!index.isValid())
		return QVariant();

	int row = index.row();
	int column = index.column();

	if (role == Qt::DecorationRole)
	{
			return QVariant();
	}
	if (role == Qt::TextAlignmentRole)
	{
		switch(column)//TODO obtener esto de la query
		{
		case ISSUE:
			return QVariant(Qt::AlignRight | Qt::AlignVCenter);
		case TITLE:
			return QVariant(Qt::AlignLeft | Qt::AlignVCenter);
		}
	}

	if(role != Qt::DisplayRole)
		return QVariant();

	if(row<_data.count())
		return _data[row][column];
	else
		return QVariant();
}

Qt::ItemFlags VolumeComicsModel::flags(const QModelIndex &index) const
{
	if (!index.isValid())
		return 0;
	return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
}

QVariant VolumeComicsModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	if (orientation == Qt::Horizontal && role == Qt::DisplayRole)
	{
		switch(section)//TODO obtener esto de la query
		{
		case ISSUE:
			return QVariant(QString("issue"));
		case TITLE:
			return QVariant(QString(tr("title")));
		}
	}

	if (orientation == Qt::Horizontal && role == Qt::TextAlignmentRole)
	{
		switch(section)//TODO obtener esto de la query
		{
		case ISSUE:
			return QVariant(Qt::AlignRight | Qt::AlignVCenter);
		case TITLE:
			return QVariant(Qt::AlignLeft | Qt::AlignVCenter);
		}
	}

	return QVariant();
}

QModelIndex VolumeComicsModel::index(int row, int column, const QModelIndex &parent) const
{
	if (!hasIndex(row, column, parent))
		return QModelIndex();

	return createIndex(row, column);
}

QString VolumeComicsModel::getComicId(const QModelIndex &index) const
{
	int row = index.row();
	if(row >= _data.count())
		return "";
	return _data[row][ID];
}

QString VolumeComicsModel::getComicId(int row) const
{
	if(row >= _data.count())
		return "";
	return _data[row][ID];
}

QString VolumeComicsModel::getCoverURL(const QModelIndex &index) const
{
	return _data[index.row()][COVER_URL];
}

void VolumeComicsModel::addExtraRows(int numRows)
{
	numExtraRows = numRows;
}