movie.cpp

/*
 * movie.cpp
 * Copyright (C) 2008  Michail Pishchagin
 *
 * 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 library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "movie.h"

#include <QDebug>
#include <QDomElement>
#include <QDomDocument>
#include <QMapIterator>
#include <QRegExp>

#include "xmpp_xmlcommon.h"
#include "httprequest.h"
#include "afishahelpers.h"

Movie::Movie()
	: QObject()
{
}

Movie::~Movie()
{
}

QString Movie::name() const
{
	return name_;
}

QString Movie::id() const
{
	return id_;
}

QStringList Movie::sortTimes(const QStringList& times) const
{
	QStringList result = times;
	result.sort();

	QStringList tmp;
	foreach(QString t, result) {
		QStringList tt = t.split(":");
		int hour = tt[0].toInt();
		if (hour <= 6)
			tmp += t;
	}
	foreach(QString t, tmp) {
		result.removeAll(t);
	}

	result += tmp;
	return result;
}

QStringList Movie::times() const
{
	QStringList result;
	QMapIterator<QString, QStringList> it(cinemas_);
	while (it.hasNext()) {
		it.next();
		foreach(QString t, it.value()) {
			if (!result.contains(t))
				result << t;
		}
	}
	return sortTimes(result);
}

QStringList Movie::timesForCinema(const QString& cinemaId)
{
	if (cinemas_.contains(cinemaId)) {
		return cinemas_[cinemaId];
	}
	return QStringList();
}

QMap<QString, QStringList> Movie::cinemasForTimes(const QStringList& _times)
{
	QStringList times = sortTimes(_times);
	QMap<QString, QStringList> result;

	QMapIterator<QString, QStringList> it(cinemas_);
	while (it.hasNext()) {
		it.next();
		foreach(QString t, it.value()) {
			if (times.contains(t)) {
				if (result.contains(it.key()))
					result[it.key()] += t;
				else
					result[it.key()] = QStringList() << t;
			}
		}
	}

	return result;
}

void Movie::initFromData(const QString& xml)
{
	QDomDocument doc;
	if (!doc.setContent(xml))
		return;

	initFromXml(doc.documentElement());
}

void Movie::initFromXml(const QDomElement& e)
{
	Q_ASSERT(e.tagName() == "movie");
	clear();

	id_ = XMLHelper::subTagText(e, "id");
	name_ = XMLHelper::subTagText(e, "name");

	for (QDomNode n = e.firstChild(); !n.isNull(); n = n.nextSibling()) {
		QDomElement ee = n.toElement();
		if (ee.isNull())
			continue;

		if (ee.tagName() == "cinema") {
			QString name = ee.attribute("name");
			QString id = ee.attribute("id");
			QStringList times;

			for (QDomNode nn = ee.firstChild(); !nn.isNull(); nn = nn.nextSibling()) {
				QDomElement eee = nn.toElement();
				if (eee.isNull())
					continue;

				if (eee.tagName() == "time") {
					// protection against 'Расписание сеансов появится позднее'
					QRegExp timeRegExp("^\\d\\d:\\d\\d$");
					if (timeRegExp.exactMatch(eee.text())) {
						times << eee.text();
					}
				}
			}

			cinemas_[id] = times;
		}
	}

	if (e.attribute("detailed").isEmpty()) {
		// if (id_ == "190306" || id_ == "171644" || id_ == "181518") {
		// 	updateFromWeb();
		// }
	}
	else {
		// qWarning() << id_ << name_;
		// qWarning() << cinemas_;
	}

	emit dataChanged();
}

void Movie::updateFromWeb()
{
	Q_ASSERT(!id_.isEmpty());
	if (request_)
		delete request_;
	request_ = new HttpRequest(QString("%1_movie_%2")
	                           .arg(AfishaHelpers::currentDate())
	                           .arg(id_), this);
	connect(request_, SIGNAL(finished()), SLOT(requestFinished()));

	// in order to get times even for movies with single cinema, we should fetch with '&page=1'
	request_->request(":queries/movie.xq",
	                  QString("http://afisha.yandex.ru/event.xml?event_id=%2&date=%1&page=1")
	                  .arg(AfishaHelpers::currentDate())
	                  .arg(id_));
}

void Movie::clear()
{
	id_ = QString();
	name_ = QString();
	cinemas_.clear();
	lastUpdatedAt_ = QDateTime::currentDateTime();
}

void Movie::requestFinished()
{
	if (request_.isNull()) {
		return;
	}

	if (!request_->error()) {
		initFromData(request_->result());
	}

	delete request_;
}