/*
  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 <QSqlDatabase>
#include <QAbstractItemModel>
#include <QStringList>
#include <QMessageBox>
#include <QSettings>
#include <QDir>
#include <QFile>
#include <QDataStream>
#include <QFileInfo>
#include <QTemporaryFile>
#include <QProcess>

#include <QX11Info>
#include <X11/Xlib.h>
#include <X11/extensions/Xfixes.h>

#include "smglobals.h"
#include "seriestreemodel.h"
#include "filestreemodel.h"
#include "mappingtablemodel.h"
#include "pictureviewer.h"

SmGlobals *SmGlobals::mInstance = 0;

SmGlobals::~SmGlobals(){
	foreach(QAbstractItemModel *model, mModels.values()){
		delete model;
	}
	delete mFrameCache;
}

SmGlobals *SmGlobals::instance(){
	if(!mInstance){
		mInstance = new SmGlobals;
		if(!QSqlDatabase::contains("treedb")){
			QSettings s;
			QSqlDatabase db = QSqlDatabase::addDatabase("QPSQL", "treedb");
			db.setHostName(s.value("database/hostname").toString());
			db.setUserName(s.value("database/dbuser").toString());
			db.setPassword(s.value("database/dbpass").toString());
			db.setDatabaseName(s.value("database/dbname").toString());
			if(!db.open()){
				QMessageBox::critical(0, tr("Error"), tr("Could not open database. Please configure it and restart program."));
			}
		}
	}
	return mInstance;
}

QAbstractItemModel *SmGlobals::model(const QString &which){
	if(which == "SeriesModel"){
		if(!mModels.contains(which)){
			QStringList headers = QStringList() << tr("Series") << tr("Series ID") << tr("Series part ID") << tr("Part") << tr("Type");
			SeriesTreeModel *model = new SeriesTreeModel(headers);
			mModels.insert(which, model);
		}
	}else if(which == "FilesModel"){
		if(!mModels.contains(which)){
			QStringList headers = QStringList() << tr("Name") << tr("Part") << tr("Size") << tr("Qual.") << tr("Dvd") << tr("Full Path") << tr("Size (int)") << tr("Type") << tr("Md5 sum") << tr("Series part ID") << tr("File ID") << tr("Seriespart") << tr("Display name") << tr("Dur./Size") << tr("Series name");
			FilesTreeModel *model = new FilesTreeModel(headers);
			mModels.insert(which, model);
		}
	}else if(which == "actors"){
		if(!mModels.contains(which)){
			QStringList headers = QStringList() << tr("Actor") << tr("Id");
			MappingTableModel *model = new MappingTableModel(headers, "actors");
			mModels.insert(which, model);
		}
	}else if(which == "genres"){
		if(!mModels.contains(which)){
			QStringList headers = QStringList() << tr("Genre") << tr("Id");
			MappingTableModel *model = new MappingTableModel(headers, "genres");
			mModels.insert(which, model);
		}
	}
	return mModels.contains(which) ? mModels.value(which) : 0;
}

PictureViewer *SmGlobals::pictureViewer() {
	if(!mPictureViewer){
		mPictureViewer = new PictureViewer;
	}
	return mPictureViewer;
}

SmGlobals::FrameCache *SmGlobals::frameCache() {
	if(!mFrameCache){
		mFrameCache = new SmGlobals::FrameCache;
	}
	return mFrameCache;
}

QSize SmGlobals::cursorSize() {
	if(!mCursorSize.isValid()){
		XFixesCursorImage *curImage = XFixesGetCursorImage(QX11Info::display());
		mCursorSize = QSize(curImage->width, curImage->height);
		XFree(curImage);
	}
	return mCursorSize;
}

SmGlobals::SmGlobals() : mPictureViewer(0), mFrameCache(0){
	mIcons.insert("Dildo", ":/dildo.png");
	mIcons.insert("Dick to left", ":/back_dick.png");
	mIcons.insert("Dick pointing up", ":/up_dick.png");
	mIcons.insert("Chastity belt", ":/chastity_belt.png");
	mIcons.insert("Clitoris", ":/clitoris.png");
	mIcons.insert("Gaping ass", ":/gaping_ass.png");
	mIcons.insert("Nipple pointing up", ":/nipple_up.png");
	mIcons.insert("Bald pussy", ":/bald_pussy.png");
	mIcons.insert("Prince Albert", ":/prince_albert.png");
	mIcons.insert("Diaper", ":/diaper.png");
	mIcons.insert("High heels", ":/higheels.png");
	mIcons.insert("Ball gag", ":/ball_gag.png");
	mIcons.insert("French Maid Dress", ":/french_maid_dress.png");
	mIcons.insert("Shackles", ":/shackles.png");
	mIcons.insert("Steel collar", ":/steel_collar.png");
	mIcons.insert("Butt plug", ":/butt_plug.png");
	mIcons.insert("Hourglass figure", ":/hourglass_figure.png");
	mIcons.insert("Big ass", ":/big_ass.png");
	mIcons.insert("Big tit", ":/big_tit.png");
	mIcons.insert("Bizarre amputee", ":/bizarre_amputee.png");
	mIcons.insert("Spreading pants", ":/spreadingpants.png");
}

//FrameCache
SmGlobals::FrameCache::FrameCache(QObject *parent) : QObject(parent), mMagic(0xDEADBEEF), mCacheSubDir(".frameCache"), mCacheFileName("cache") {
	readConfig();
	readCache();
}

SmGlobals::FrameCache::~FrameCache(){
	QFile outfile(mCacheFile);
	outfile.open(QIODevice::WriteOnly | QIODevice::Truncate);
	QDataStream ds(&outfile);
	ds << (qint32)mMagic;
	for(QHash<QPair<QString, QString>, QString>::const_iterator it = mFrameCache.constBegin(); it != mFrameCache.constEnd(); ++it){
		ds << it.key().first << it.key().second << it.value();
	}
	outfile.close();
}

void SmGlobals::FrameCache::readConfig(){
	QSettings s;
	QString archive = s.value("paths/archivedir").toString();
	if(archive.isEmpty()){
		return;
	}
	QDir archiveDir(archive);
	if(!archiveDir.exists(mCacheSubDir)){
		archiveDir.mkdir(mCacheSubDir);
	}
	mCacheDir = QString("%1/%2").arg(archive).arg(mCacheSubDir);
	mCacheFile = QString("%1/%2").arg(mCacheDir).arg(mCacheFileName);
	mWhen = s.value("ui/grabframe", "00:00:00").toString();
	mFfMpegPath = s.value("paths/ffmpeg").toString();
}

const QPixmap SmGlobals::FrameCache::entry(const QString &sourcePath, const QString &when){
	QString realWhen = when;
	if(realWhen.isEmpty()){
		realWhen = mWhen;
	}
	const QPair<QString, QString> source(sourcePath, realWhen);
	if(!mFrameCache.contains(source)){
		grabFrame(sourcePath, realWhen);
	}
	return QPixmap(mFrameCache.value(source));
}

void SmGlobals::FrameCache::grabFrame(const QString &sourceFile, QString when){
	QFileInfo sourceInfo(sourceFile);
	if(!sourceInfo.exists()){
		return;
	}
	QString tmpTemplate = QString("%1/%2_%3-XXXXXX.png").arg(mCacheDir).arg(sourceInfo.fileName()).arg(when);
	QTemporaryFile tempFile(tmpTemplate);
	tempFile.setAutoRemove(false);
	if(tempFile.open()){
		QStringList ffMpegArgs = QStringList() << "-vframes" << "1" << "-ss" << when << "-i" << sourceFile << "-y" << tempFile.fileName();
		QProcess ffmpeg;
		ffmpeg.start(mFfMpegPath, ffMpegArgs);
		if(!ffmpeg.waitForStarted()){
			return;
		}
		ffmpeg.waitForFinished();
		QPair<QString, QString> pair(sourceFile, when);
		mFrameCache.insert(pair, tempFile.fileName());
	}
}

void SmGlobals::FrameCache::readCache(){
	QFile cache(mCacheFile);
	cache.open(QIODevice::ReadOnly);
	QDataStream ds(&cache);
	qint32 magic;
	ds >> magic;
	if(magic != mMagic){
		return;
	}
	while(!ds.atEnd()){
		QString source, pos, cacheFile;
		ds >> source >> pos >> cacheFile;
		QPair<QString, QString> pair(source, pos);
		mFrameCache.insert(pair, cacheFile);
	}
}