/*
  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.
*/

#include <QStringList>
#include <QRegExp>
#include <QList>

#include <QDebug>

#include "archiveproxy.h"
#include "moviemodel.h"

ArchiveProxy::ArchiveProxy(QObject *parent) : QSortFilterProxyModel(parent){};

void ArchiveProxy::setFilter(const QString &filter, FilterMode mode){
	mFilterMode = mode;
	switch(mFilterMode){
		case NoFilter:
			mGenreFilter = QString();
			mActorFilter = QString();
			mTitleFilter = QString();
			break;
		case GenreFilter:
			mGenreFilter = filter;
			break;
		case ActorFilter:
			mActorFilter = filter;
			break;
		case TitleFilter:
			mTitleFilter = filter.toLower();
			break;
	}
	invalidate();
}

bool ArchiveProxy::filterAcceptsRow(int sourceRow, const QModelIndex & /*sourceParent*/) const{
	if(mFilterMode == NoFilter){
		return true;
	}
	MovieModel *model = static_cast<MovieModel*>(sourceModel());
	Q_ASSERT(model != 0);
	if(mFilterMode == ActorFilter){
		QModelIndex idx = model->index(sourceRow, 0, QModelIndex());
		QStringList actors = model->data(idx, MovieModel::ActorsRole).toStringList();
		return actors.contains(mActorFilter);
	}
	if(mFilterMode == GenreFilter){
		QModelIndex idx = model->index(sourceRow, 4, QModelIndex());
		QString genre = idx.data().toString();
		return (genre == mGenreFilter);
	}
	if(mFilterMode == TitleFilter){
		QModelIndex idx = model->index(sourceRow, 0, QModelIndex());
		QString title = idx.data().toString();
		QRegExp re(mTitleFilter);
		return( re.indexIn(title) != -1);
	}
	return true;
}

void ArchiveProxy::clearFilter(){
	mFilterMode = NoFilter;
	mGenreFilter = QString();
	mActorFilter = QString();
	mTitleFilter = QString();
	invalidate();
}

bool ArchiveProxy::lessThan(const QModelIndex &left, const QModelIndex &right) const{
	if(left.column() > 0){
		return QSortFilterProxyModel::lessThan(left, right);
	}
	QString l = left.data().toString();
	QString r = right.data().toString();
	QRegExp nos("(\\d+)");
	QList<int> lnos, rnos;
	int pos = 0;
	bool dummy;
	while((pos = nos.indexIn(l, pos)) != -1){
		lnos << nos.cap(1).toInt(&dummy);	
		pos += nos.matchedLength();
	}
	pos = 0;
	while((pos = nos.indexIn(r, pos)) != -1){
		rnos << nos.cap(1).toInt(&dummy);	
		pos += nos.matchedLength();
	}
	QRegExp baseRe("(.*)\\s+\\d+");
	baseRe.indexIn(l);
	QString lbase = baseRe.cap(1);
	baseRe.indexIn(r);
	QString rbase = baseRe.cap(1);
	if((lnos.size() == 1) && (rnos.size() == 1)){
		if(lbase == rbase){
			return (lnos.at(0) < rnos.at(0));
		}
	}
	if((lnos.size() == 2) && (rnos.size() == 2)){
		if(lbase == rbase){
			return (lnos.at(1) < rnos.at(1));
		}	
	}
	return QSortFilterProxyModel::lessThan(left, right);
}