#include <QtCore/QDebug>
#include <QtCore/QDataStream>
#include <QtCore/QDirIterator>
#include "smartpen.h"
#include "paperreplay.h"

namespace
{
bool readUtfString(QDataStream &stream, QString &s)
{
	quint16 len;
	stream >> len;
	QByteArray buffer(len, Qt::Uninitialized);
	if (stream.readRawData(buffer.data(), len) != len) {
		return false;
	}
	s = QString::fromUtf8(buffer);
	return true;
}
}

PaperReplay::PaperReplay(QObject *parent) :
    QObject(parent)
{
}

PaperReplay::Session::Session() : d(new SessionData)
{
}

PaperReplay::Session::~Session()
{
}

bool PaperReplay::open(const QString &path, const QString &penSerial, quint64 notebookGuid)
{
	_dir.setPath(path + QString("/userdata/%1/Paper Replay/99/%2/sessions")
	             .arg(penSerial)
	             .arg(qulonglong(notebookGuid), notebookGuid == 0 ? 1 : 16, 16, QLatin1Char('0')));
	if (!_dir.exists()) {
		qDebug() << "Cannot open paper replay:" << _dir.absolutePath() << "does not exist";
		return false;
	}

	QDirIterator iter(_dir.path(), QStringList("PRS-*"), QDir::Dirs | QDir::NoDotAndDotDot);
	while (iter.hasNext()) {
		QDir sessionDir(iter.next());
		bool ok;
		quint64 sessionId = sessionDir.dirName().mid(4).toULongLong(&ok, 16);
		if (!ok) {
			qWarning() << "Invalid session identifier:" << sessionDir.dirName();
		}
		Session &session = _sessions[sessionId];
		if (!parseSessionInfo(session.d, sessionDir.filePath("session.info"))) {
			qWarning() << "Could not parse:" << sessionDir.absoluteFilePath("session.info");
		}
		if (!parseSessionPages(session.d, sessionDir.filePath("session.pages"))) {
			qWarning() << "Could not parse:" << sessionDir.absoluteFilePath("session.pages");
		}
	}

	return true;
}

void PaperReplay::close()
{
	_sessions.clear();
	_byPage.clear();
	_dir.setPath(QString());
}

QList<PaperReplay::Session> PaperReplay::sessions() const
{
	return _sessions.values();
}

QList<PaperReplay::Session> PaperReplay::sessions(quint64 pageAddress) const
{
	QList<Session> sessions;
	QMultiMap<quint64, quint64>::const_iterator it = _byPage.constFind(pageAddress);
	while (it != _byPage.end() && it.key() == pageAddress) {
		sessions.append(_sessions[it.value()]);
	}
	return sessions;
}

bool PaperReplay::parseSessionInfo(SessionData *session, const QString &path)
{
	QFile f(path);
	if (f.open(QIODevice::ReadOnly)) {
		return parseSessionInfo(session, &f);
	} else {
		return false;
	}
}

bool PaperReplay::parseSessionInfo(SessionData *session, QIODevice *dev)
{
	unsigned char magic[2];
	if (dev->read(reinterpret_cast<char*>(magic), 2) != 2 ||
	        magic[0] != 0xFA || magic[1] != 0xCE) {
		qWarning() << "Invalid magic";
		return false;
	}

	char version = 0;
	if (!dev->getChar(&version)) {
		qWarning() << "Short read while getting version number";
		return false;
	}

	switch (version) {
	case 3:
		return parseSessionInfoV3(session, dev);
	default:
		qWarning() << "Unknown version:" << version;
		return false;
	}
}

bool PaperReplay::parseSessionInfoV3(SessionData *session, QIODevice *dev)
{
	QDataStream s(dev);
	if (s.skipRawData(5) != 5) return false;

	quint64 startTime, endTime, creationTime;
	QString name;

	s >> startTime >> endTime >> creationTime;
	if (!readUtfString(s, name)) {
		return false;
	}

	session->name = name;
	session->start = Smartpen::fromPenTime(startTime);
	session->end = Smartpen::fromPenTime(endTime);

	qDebug() << "Session:" << name << session->start << session->end;

	quint16 num_clips;
	s >> num_clips;

	Q_ASSERT(num_clips == 1); // TODO: We do not yet know how to handle this scenario

	for (uint i = 0; i < num_clips; ++i) {
		QString file;
		if (!readUtfString(s, file)) {
			return false;
		}
		s >> startTime >> endTime;

		qDebug() << " Clip:" << file << Smartpen::fromPenTime(startTime) << Smartpen::fromPenTime(endTime);
		session->file = file;
	}

	quint16 num_strokes;
	s >> num_strokes;

	for (uint i = 0; i < num_strokes; ++i) {
		quint64 a, b, c;
		quint32 d;
		QString nbGuid;
		quint8 f, g;
		s >> a >> b >> c >> d;
		if (!readUtfString(s, nbGuid)) {
			return false;
		}
		s >> f >> g;
		qDebug() << " Stroke:" << a << b << c << d << nbGuid << f << g;
	}

	return true;
}

bool PaperReplay::parseSessionPages(SessionData *session, const QString &path)
{
	QFile f(path);
	if (f.open(QIODevice::ReadOnly)) {
		return parseSessionPages(session, &f);
	} else {
		return false;
	}
}

bool PaperReplay::parseSessionPages(SessionData *session, QIODevice *dev)
{
	unsigned char magic[2];
	if (dev->read(reinterpret_cast<char*>(magic), 2) != 2 ||
	        magic[0] != 0xCA || magic[1] != 0xBE) {
		qWarning() << "Invalid magic";
		return false;
	}

	char version = 0;
	if (!dev->getChar(&version)) {
		qWarning() << "Short read while getting version number";
		return false;
	}

	switch (version) {
	case 1:
		return parseSessionPagesV1(session, dev);
	default:
		qWarning() << "Unknown version:" << version;
		return false;
	}
}

bool PaperReplay::parseSessionPagesV1(SessionData *session, QIODevice *dev)
{
	QDataStream s(dev);
	if (s.skipRawData(1) != 1) return false;

	quint16 num_pages = 0;
	s >> num_pages;
	session->pages.reserve(session->pages.size() + num_pages);

	for (uint i = 0; i < num_pages; ++i) {
		quint64 address, time;
		s >> address >> time;

		session->pages.append(address);
		qDebug() << " Page:" << address << time << Smartpen::fromPenTime(time);
	}

	return true;
}