From 6d00ae1d57ed37a8d3d6d35181cb6b155dd7801c Mon Sep 17 00:00:00 2001
From: Arno <am@disconnect.de>
Date: Sun, 18 Jul 2010 12:51:36 +0200
Subject: Implemented delete files from tree view

Files can now be delete from the files tree view. When doing so, the
files will be deleted permanently and also removed from the database.
---
 filestreemodel.cpp | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

(limited to 'filestreemodel.cpp')

diff --git a/filestreemodel.cpp b/filestreemodel.cpp
index 8dc52ba..3dc5dc1 100644
--- a/filestreemodel.cpp
+++ b/filestreemodel.cpp
@@ -9,6 +9,7 @@
 #include <QLocale>
 #include <QIcon>
 #include <QFileInfo>
+#include <QFile>
 
 #include "filestreemodel.h"
 #include "smtreeitem.h"
@@ -24,6 +25,8 @@ FilesTreeModel::FilesTreeModel(QStringList &headers, QObject *parent) : SmTreeMo
 	mInsertFileQuery->prepare("INSERT INTO files(iseriespart_id, tfilename, cmd5sum, bisize, idvd, sifiletype, sifileno, siquality) VALUES(:seriespartid, :fname, :md5, :size, :dvd, :type, :fileno, :quality)");
 	mFilesQuery = new QSqlQuery(mDb);
 	mFilesQuery->prepare("SELECT tfilename, cmd5sum FROM files WHERE iseriespart_id = :id");
+	mDeleteFileQuery = new QSqlQuery(mDb);
+	mDeleteFileQuery->prepare("DELETE FROM files WHERE ifiles_id = :id");
 }
 
 FilesTreeModel::~FilesTreeModel(){
@@ -31,6 +34,7 @@ FilesTreeModel::~FilesTreeModel(){
 	delete mUpdateQualityQuery;
 	delete mInsertFileQuery;
 	delete mFilesQuery;
+	delete mDeleteFileQuery;
 	mDb = QSqlDatabase();
 }
 
@@ -231,6 +235,38 @@ bool FilesTreeModel::addFile(const QString &fullPath, int type, int quality, int
 	return false;
 }
 
+bool FilesTreeModel::deleteFile(const QModelIndex &file){
+	QString fullPath = file.data(FullPathRole).toString();
+	QFileInfo fi(fullPath);
+	if(!fi.exists()){
+		return false;
+	}
+	QVariant fileId = file.data(FilesIdRole);
+	mDb.transaction();
+	mDeleteFileQuery->bindValue(":id", fileId);
+	if(!mDeleteFileQuery->exec()){
+		mDb.rollback();
+		return false;
+	}
+	if(!QFile::remove(fullPath)){
+		mDb.rollback();
+		return false;
+	}
+	removeRows(file.row(), 1, file.parent());
+	mDb.commit();
+	return true;
+}
+
+bool FilesTreeModel::deleteFiles(const QModelIndexList &files){
+	foreach(QModelIndex i, files){
+		bool retval = deleteFile(i);
+		if(!retval){
+			return false;
+		}
+	}
+	return true;
+}
+
 void FilesTreeModel::populate(QSqlQuery &filesQuery){
 	SmTreeItem *root = new SmTreeItem(11);
 	SmTreeItem *files = new SmTreeItem(11, root);
-- 
cgit v1.2.3-70-g09d2