path: root/seriestreewidget.cpp

/*
  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 <QHBoxLayout>
#include <QVBoxLayout>
#include <QPushButton>
#include <QLineEdit>
#include <QLabel>
#include <QSortFilterProxyModel>
#include <QRegExp>
#include <QContextMenuEvent>
#include <QMenu>
#include <QAction>
#include <QMessageBox>
#include <QSettings>
#include <QFileDialog>
#include <QSettings>
#include <QFileInfo>
#include <QComboBox>
#include <QFile>

#include "seriestreewidget.h"
#include "smtreemodel.h"
#include "smglobals.h"
#include "seriestreemodel.h"
#include "filestreemodel.h"
#include "helper.h"

SeriesTreeWidget::SeriesTreeWidget(QWidget *parent) : QWidget(parent){
	//filter bar
	QLabel *l1 = new QLabel(tr("&Filter:"));
	mFilterEdit = new QLineEdit;
	l1->setBuddy(mFilterEdit);
	mFilter = new QPushButton(tr("Filter"));
	QHBoxLayout *filterLayout = new QHBoxLayout;
	connect(mFilter, SIGNAL(clicked()), this, SLOT(filter()));
	connect(mFilterEdit, SIGNAL(returnPressed()), this, SLOT(filter()));
	filterLayout->addWidget(l1);
	filterLayout->addWidget(mFilterEdit);
	filterLayout->addWidget(mFilter);

	//the view
	mView = new SeriesTreeView;
	mProxy = new SeriesTreeSortModel(this);
	mModel = static_cast<SeriesTreeModel*>(SmGlobals::instance()->model("SeriesModel"));
	mProxy->setSourceModel(mModel);
	mView->setModel(mProxy);
	mView->setSortingEnabled(true);
	for(int i = 1; i < 5 ;++i){
		mView->setColumnHidden(i, true);
	}
	mView->resizeColumnToContents(0);
	mView->setSelectionMode(QAbstractItemView::ExtendedSelection);
	connect(mModel, SIGNAL(needResort()), this, SLOT(resort()));
	connect(mView, SIGNAL(expanded(QModelIndex)), this, SLOT(itemExpanded(QModelIndex)));
	connect(mView, SIGNAL(collapsed(QModelIndex)), this, SLOT(itemCollaped(QModelIndex)));

	//cover dialog
	mCoverDialog = new AddCoverDialog(this);

	//layout
	QVBoxLayout *mainLayout = new QVBoxLayout;
	mainLayout->addLayout(filterLayout);
	mainLayout->addWidget(mView);
	setLayout(mainLayout);

}

void SeriesTreeWidget::newSeries(){
	QList<QVariant> data;
	data << tr("<New series>") << QVariant() << QVariant() << QVariant() << SeriesTreeModel::NewSeries;
	if(mModel->addRow(data, QModelIndex())){
		QModelIndex newRow = mModel->index(mModel->rowCount(QModelIndex()) - 1, 0, QModelIndex());
		if(newRow.isValid()){
			QModelIndex proxyIndex = mProxy->mapFromSource(newRow);
			mView->selectionModel()->setCurrentIndex(proxyIndex, QItemSelectionModel::ClearAndSelect);
			mView->edit(proxyIndex);
		}
	}
}

void SeriesTreeWidget::seriesAdded(QString seriesName, int seriesPart, bool resort){
	if(resort){
		mProxy->invalidate();
	}
	QModelIndex destIdx = mModel->findValue(seriesName);
	if(destIdx.isValid()){
		QModelIndex seriesPartIdx = mModel->findValue(seriesPart, destIdx, SeriesTreeModel::SeriesPart);
		if(seriesPartIdx.isValid()){
			destIdx = seriesPartIdx;
		}
	}
	if(destIdx.isValid()){
		QModelIndex proxyIndex = mProxy->mapFromSource(destIdx);
		mView->selectionModel()->setCurrentIndex(proxyIndex, QItemSelectionModel::ClearAndSelect);
	}
}

void SeriesTreeWidget::deleteFromSeries(){
	QModelIndexList selected = mView->selectionModel()->selectedRows();
	if(selected.isEmpty()){
		return;
	}
	QList<QPersistentModelIndex> series;
	QList<QPersistentModelIndex> parts;
	foreach(QModelIndex s, selected){
		QModelIndex real = mProxy->mapToSource(s);
		if(real.data(SeriesTreeModel::TypeRole).toInt() == SeriesTreeModel::Series || real.data(SeriesTreeModel::TypeRole).toInt() == SeriesTreeModel::NewSeries){
			series << QPersistentModelIndex(real);
		}
		if(real.data(SeriesTreeModel::TypeRole).toInt() == SeriesTreeModel::Part){
			parts << QPersistentModelIndex(real);
		}
	}
	QList<QPersistentModelIndex> removeParts;
	foreach(QPersistentModelIndex s, parts){
		if(series.contains(s.parent())){
			removeParts << s;
		}
	}
	foreach(QPersistentModelIndex s, removeParts){
		parts.removeAll(s);
	}
	QFileInfoList files;
	foreach(QPersistentModelIndex s, series){
		files.append(mModel->findFiles(s));
	}

	QString message;
	if(!files.isEmpty()){
		message = QString(tr("<p>This operation will delete %1 file(s) permanently:</p>")).arg(files.count());
		message.append(tr("<ul>"));
		int ctr = 0;
		foreach(QFileInfo f, files){
			++ctr;
			QString li = QString(tr("<li>%1</li>")).arg(f.absoluteFilePath());
			message.append(li);
			if(ctr == 20){
				QString elide = QString(tr("<li>(%1 more files)</li>")).arg(files.count() - ctr);
				message.append(elide);
				break;
			}
		}
		message.append(tr("</ul>"));
		message.append(tr("<p>Continue?</p>"));
	}else{
		message = QString(tr("<p>This operation will delete no files. Continue?</p>"));
	}
	int retval = QMessageBox::critical(this, tr("Delete Series"), message, QMessageBox::Yes | QMessageBox::No, QMessageBox::No);
	if(retval == QMessageBox::Yes){
		foreach(QPersistentModelIndex s, series){
			mModel->deleteFromSeries(s);
		}
		foreach(QPersistentModelIndex s, parts){
			mModel->deleteFromSeries(s);
		}
	}
}

void SeriesTreeWidget::readSettings(){
	QSettings s;
	QStringList expanded = s.value("archive/expanded").toStringList();
	foreach(QString s, expanded){
		QModelIndex idx = mProxy->mapFromSource(mModel->findValue(s));
		mView->expand(idx);
	}
	int sortOrder = s.value("archive/sortorder", Qt::DescendingOrder).toInt();
	mView->sortByColumn(0, static_cast<Qt::SortOrder>(sortOrder));
	QString selectedSeries = s.value("archive/selectedseries").toString();
	QModelIndex seriesIdx = mModel->findValue(selectedSeries);
	if(seriesIdx.isValid()){
		QModelIndex real = mProxy->mapFromSource(seriesIdx);
		mView->selectionModel()->setCurrentIndex(real, QItemSelectionModel::SelectCurrent);
	}
}

void SeriesTreeWidget::writeSettings(){
	QSettings s;
	s.setValue("archive/expanded", mExpandedItems);
	s.setValue("archive/sortorder", mProxy->sortOrder());
	QString selected = mView->selectionModel()->currentIndex().data(SeriesTreeModel::NameRole).toString();
	s.setValue("archive/selectedseries", selected);
}

void SeriesTreeWidget::expandCurrent(){
	QModelIndexList selected = mView->selectionModel()->selectedRows();
	if(selected.isEmpty()){
		return;
	}
	QModelIndex idx = selected.at(0);
	mView->expand(idx);
}

void SeriesTreeWidget::filter(){
	mProxy->setFilterRegExp(mFilterEdit->text());
}

void SeriesTreeWidget::resort(){
	mView->sortByColumn(0, mProxy->sortOrder());
	mView->scrollTo(mView->selectionModel()->currentIndex(), QAbstractItemView::PositionAtCenter);
}

void SeriesTreeWidget::itemExpanded(const QModelIndex &what){
	mExpandedItems << what.data(SeriesTreeModel::NameRole).toString();
}

void SeriesTreeWidget::itemCollaped(const QModelIndex &what){
	QString itemName = what.data(SeriesTreeModel::NameRole).toString();
	if(mExpandedItems.contains(itemName)){
		mExpandedItems.removeAll(itemName);
	}
}

void SeriesTreeWidget::addCover(){
	QModelIndexList selected = mView->selectionModel()->selectedRows();
	if(selected.isEmpty()){
		return;
	}
	if(selected.count() > 1){
		QMessageBox::critical(this, tr("Error"), tr("Please don't select more than one item when adding covers."));
		return;
	}
	QModelIndex real = mProxy->mapToSource(selected.at(0));
	if(real.data(SeriesTreeModel::TypeRole).toInt() != SeriesTreeModel::Part){
		QMessageBox::critical(this, tr("Error"), tr("Please select a series part when adding covers."));
		return;
	}
	int retval = mCoverDialog->exec();
	if(retval == QDialog::Accepted){
		QString fileName = mCoverDialog->file();
		QFileInfo fi(fileName);
		if(fi.exists()){
			FilesTreeModel *filesModel = static_cast<FilesTreeModel*>(SmGlobals::instance()->model("FilesModel"));
			int seriesPartId = real.data(SeriesTreeModel::SeriesPartIdRole).toInt();
			int type = mCoverDialog->fileType();
			if(filesModel->addFile(fileName, type, 0, -1, seriesPartId, -1)){
				QString md5sum = Helper::md5Sum(fileName);
				QString targetFile = Helper::createArchivePath(fileName, md5sum);
				QFile::rename(fileName, targetFile);
				filesModel->setIds(QList<int>() << seriesPartId);
				emit filesReload();
			}
		}
	}
}

SeriesTreeView::SeriesTreeView(QWidget *parent) : QTreeView(parent) {}

void SeriesTreeView::contextMenuEvent(QContextMenuEvent *e){
	QMenu contextMenu(this);
	foreach(QAction *a, actions()){
		contextMenu.addAction(a);
	}
	contextMenu.exec(e->globalPos());
}

SeriesTreeSortModel::SeriesTreeSortModel(QObject *parent) : QSortFilterProxyModel(parent) {}

bool SeriesTreeSortModel::lessThan(const QModelIndex &left, const QModelIndex &right) const{
	if(left.data(SeriesTreeModel::TypeRole).toInt() == SeriesTreeModel::Part && right.data(SeriesTreeModel::TypeRole).toInt() == SeriesTreeModel::Part){
		return left.data(SeriesTreeModel::SeriesPartRole).toInt() < right.data(SeriesTreeModel::SeriesPartRole).toInt();
	}
	return QSortFilterProxyModel::lessThan(left, right);
}

AddCoverDialog::AddCoverDialog(QWidget *parent, Qt::WindowFlags f) : QDialog(parent, f){
	//File selection
	QLabel *l1 = new QLabel(tr("Select file"));
	mFile = new QLineEdit;
	QPushButton *mSelectFile = new QPushButton(tr("Browse..."));
	connect(mSelectFile, SIGNAL(clicked()), this, SLOT(selectFile()));
	QHBoxLayout *fileLayout = new QHBoxLayout;
	fileLayout->addWidget(l1);
	fileLayout->addWidget(mFile);
	fileLayout->addWidget(mSelectFile);

	//file type
	mFilesModel = static_cast<FilesTreeModel*>(SmGlobals::instance()->model("FilesModel"));
	QHash<int, QString> coverTypes = mFilesModel->coverTypes();
	mFileType = new QComboBox;
	QStringList types;
	foreach(QString t, coverTypes.values()){
		types << t;
	}
	connect(mFileType, SIGNAL(currentIndexChanged(QString)), this, SLOT(typeChanged(QString)));
	mFileType->addItems(types);
	QHBoxLayout *typesLayout = new QHBoxLayout;
	typesLayout->setAlignment(Qt::AlignCenter);
	typesLayout->addWidget(mFileType);

	//buttons
	mOk = new QPushButton(tr("Ok"));
	connect(mOk, SIGNAL(clicked()), this, SLOT(accept()));
	mCancel = new QPushButton(tr("Cancel"));
	connect(mCancel, SIGNAL(clicked()), this, SLOT(reject()));
	QHBoxLayout *buttonLayout = new QHBoxLayout;
	buttonLayout->addStretch();
	buttonLayout->addWidget(mOk);
	buttonLayout->addWidget(mCancel);

	//main layout
	QVBoxLayout *mainLayout = new QVBoxLayout;
	mainLayout->addLayout(fileLayout);
	mainLayout->addLayout(typesLayout);
	mainLayout->addLayout(buttonLayout);
	setLayout(mainLayout);
}

const QString AddCoverDialog::file() const {
	return mFile->text();
}

void AddCoverDialog::selectFile(){
	QString startDir = mLastOpenedDir;
	if(startDir.isEmpty()){
		QSettings s;
		startDir = s.value("paths/selecteddir").toString();
	}
	QString file = QFileDialog::getOpenFileName(this, tr("Select cover"), startDir);
	mFile->setText(file);
	QFileInfo fi(file);
	mLastOpenedDir = fi.absolutePath();
}

void AddCoverDialog::typeChanged(const QString &type){
	QHash<int, QString>::const_iterator it = mFilesModel->coverTypes().constBegin();
	while(it != mFilesModel->coverTypes().constEnd()){
		if(it.value() == type){
			mCurrentType = it.key();
			return;
		}
		++it;
	}
}