2010-02-28 18:15:00 +03:00
|
|
|
/*
|
|
|
|
* Bittorrent Client using Qt4 and libtorrent.
|
|
|
|
* Copyright (C) 2010 Christian Kandeler, Christophe Dumez
|
|
|
|
*
|
|
|
|
* 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 program; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
*
|
|
|
|
* In addition, as a special exception, the copyright holders give permission to
|
|
|
|
* link this program with the OpenSSL project's "OpenSSL" library (or with
|
|
|
|
* modified versions of it that use the same license as the "OpenSSL" library),
|
|
|
|
* and distribute the linked executables. You must obey the GNU General Public
|
|
|
|
* License in all respects for all of the code used other than "OpenSSL". If you
|
|
|
|
* modify file(s), you may extend this exception to your version of the file(s),
|
|
|
|
* but you are not obligated to do so. If you do not wish to do so, delete this
|
|
|
|
* exception statement from your version.
|
|
|
|
*
|
|
|
|
* Contact : chris@qbittorrent.org
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "scannedfoldersmodel.h"
|
|
|
|
|
|
|
|
#include "filesystemwatcher.h"
|
|
|
|
|
|
|
|
#include <QDir>
|
|
|
|
#include <QFileInfo>
|
|
|
|
#include <QSettings>
|
|
|
|
#include <QString>
|
|
|
|
#include <QTemporaryFile>
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
const int PathColumn = 0;
|
|
|
|
const int DownloadAtTorrentColumn = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
class ScanFoldersModel::PathData {
|
|
|
|
public:
|
|
|
|
PathData(const QString &path) : path(path), downloadAtPath(false) {}
|
|
|
|
const QString path;
|
|
|
|
bool downloadAtPath;
|
|
|
|
};
|
|
|
|
|
|
|
|
ScanFoldersModel *ScanFoldersModel::instance(QObject *parent) {
|
|
|
|
Q_ASSERT(!parent != !m_instance);
|
|
|
|
if (!m_instance)
|
|
|
|
m_instance = new ScanFoldersModel(parent);
|
|
|
|
return m_instance;
|
|
|
|
}
|
|
|
|
|
|
|
|
ScanFoldersModel::ScanFoldersModel(QObject *parent) :
|
|
|
|
QAbstractTableModel(parent), m_fsWatcher(0)
|
|
|
|
{ }
|
|
|
|
|
2010-02-28 19:23:59 +03:00
|
|
|
ScanFoldersModel::~ScanFoldersModel() {
|
|
|
|
qDeleteAll(m_pathList);
|
|
|
|
}
|
2010-02-28 18:15:00 +03:00
|
|
|
|
|
|
|
int ScanFoldersModel::rowCount(const QModelIndex &parent) const {
|
|
|
|
return parent.isValid() ? 0 : m_pathList.count();
|
|
|
|
}
|
|
|
|
|
|
|
|
int ScanFoldersModel::columnCount(const QModelIndex &parent) const {
|
|
|
|
Q_UNUSED(parent);
|
|
|
|
return 2;
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant ScanFoldersModel::data(const QModelIndex &index, int role) const {
|
|
|
|
if (!index.isValid() || index.row() >= rowCount())
|
|
|
|
return QVariant();
|
|
|
|
|
2010-02-28 19:00:52 +03:00
|
|
|
const PathData* pathData = m_pathList.at(index.row());
|
2010-02-28 18:15:00 +03:00
|
|
|
if (index.column() == PathColumn && role == Qt::DisplayRole)
|
|
|
|
return pathData->path;
|
|
|
|
if (index.column() == DownloadAtTorrentColumn && role == Qt::CheckStateRole)
|
|
|
|
return pathData->downloadAtPath ? Qt::Checked : Qt::Unchecked;
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant ScanFoldersModel::headerData(int section, Qt::Orientation orientation, int role) const {
|
|
|
|
if (orientation != Qt::Horizontal || role != Qt::DisplayRole || section < 0 || section >= columnCount())
|
|
|
|
return QVariant();
|
|
|
|
|
|
|
|
if (section == PathColumn)
|
|
|
|
return tr("Watched Folder");
|
|
|
|
return tr("Download here");
|
|
|
|
}
|
|
|
|
|
|
|
|
Qt::ItemFlags ScanFoldersModel::flags(const QModelIndex &index) const {
|
|
|
|
if (!index.isValid() || index.row() >= rowCount() || index.column() != DownloadAtTorrentColumn)
|
|
|
|
return QAbstractTableModel::flags(index);
|
|
|
|
return QAbstractTableModel::flags(index) | Qt::ItemIsUserCheckable;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ScanFoldersModel::setData(const QModelIndex &index, const QVariant &value, int role) {
|
|
|
|
if (!index.isValid() || index.row() >= rowCount() || index.column() > DownloadAtTorrentColumn || role != Qt::CheckStateRole)
|
|
|
|
return false;
|
|
|
|
Q_ASSERT(index.column() == DownloadAtTorrentColumn);
|
|
|
|
m_pathList[index.row()]->downloadAtPath = (value.toInt() == Qt::Checked);
|
|
|
|
emit dataChanged(index, index);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
ScanFoldersModel::PathStatus ScanFoldersModel::addPath(const QString &path) {
|
|
|
|
QDir dir(path);
|
|
|
|
if (!dir.exists())
|
|
|
|
return DoesNotExist;
|
|
|
|
if (!dir.isReadable())
|
|
|
|
return CannotRead;
|
|
|
|
const QString &canonicalPath = dir.canonicalPath();
|
|
|
|
if (findPathData(canonicalPath) != -1)
|
|
|
|
return AlreadyInList;
|
|
|
|
if (!m_fsWatcher) {
|
|
|
|
m_fsWatcher = new FileSystemWatcher(this);
|
|
|
|
connect(m_fsWatcher, SIGNAL(torrentsAdded(QStringList&)), this, SIGNAL(torrentsAdded(QStringList&)));
|
|
|
|
}
|
|
|
|
beginInsertRows(QModelIndex(), rowCount(), rowCount());
|
2010-02-28 19:00:52 +03:00
|
|
|
m_pathList << new PathData(canonicalPath);
|
2010-02-28 18:15:00 +03:00
|
|
|
endInsertRows();
|
|
|
|
m_fsWatcher->addPath(canonicalPath);
|
|
|
|
return Ok;
|
|
|
|
}
|
|
|
|
|
|
|
|
void ScanFoldersModel::removePath(int row) {
|
|
|
|
Q_ASSERT(row >= 0 && row < rowCount());
|
|
|
|
beginRemoveRows(QModelIndex(), row, row);
|
|
|
|
m_fsWatcher->removePath(m_pathList.at(row)->path);
|
|
|
|
m_pathList.removeAt(row);
|
|
|
|
endRemoveRows();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ScanFoldersModel::removePath(const QString &path) {
|
|
|
|
const int row = findPathData(path);
|
|
|
|
if (row == -1)
|
|
|
|
return false;
|
|
|
|
removePath(row);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
ScanFoldersModel::PathStatus ScanFoldersModel::setDownloadAtPath(int row, bool downloadAtPath) {
|
|
|
|
Q_ASSERT(row >= 0 && row < rowCount());
|
|
|
|
|
|
|
|
bool &oldValue = m_pathList[row]->downloadAtPath;
|
|
|
|
if (oldValue != downloadAtPath) {
|
|
|
|
if (downloadAtPath) {
|
|
|
|
QTemporaryFile testFile(m_pathList[row]->path + "/tmpFile");
|
|
|
|
if (!testFile.open())
|
|
|
|
return CannotWrite;
|
|
|
|
}
|
|
|
|
oldValue = downloadAtPath;
|
|
|
|
const QModelIndex &changedIndex = index(row, DownloadAtTorrentColumn);
|
|
|
|
emit dataChanged(changedIndex, changedIndex);
|
|
|
|
}
|
|
|
|
return Ok;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ScanFoldersModel::downloadInTorrentFolder(const QString &filePath) const {
|
|
|
|
const int row = findPathData(QFileInfo(filePath).dir().path());
|
|
|
|
Q_ASSERT(row != -1);
|
|
|
|
return m_pathList.at(row)->downloadAtPath;
|
|
|
|
}
|
|
|
|
|
|
|
|
int ScanFoldersModel::findPathData(const QString &path) const {
|
|
|
|
for (int i = 0; i < m_pathList.count(); ++i) {
|
2010-02-28 19:00:52 +03:00
|
|
|
const PathData* pathData = m_pathList.at(i);
|
2010-02-28 18:15:00 +03:00
|
|
|
if (pathData->path == path)
|
|
|
|
return i;
|
|
|
|
}
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
void ScanFoldersModel::makePersistent(QSettings &settings) {
|
|
|
|
QStringList paths;
|
|
|
|
QList<QVariant> downloadInFolderInfo;
|
2010-02-28 19:00:52 +03:00
|
|
|
foreach (const PathData* pathData, m_pathList) {
|
2010-02-28 18:15:00 +03:00
|
|
|
paths << pathData->path;
|
|
|
|
downloadInFolderInfo << pathData->downloadAtPath;
|
|
|
|
}
|
|
|
|
settings.setValue(QString::fromUtf8("ScanDirs"), paths);
|
|
|
|
settings.setValue(QString::fromUtf8("DownloadInScanDirs"), downloadInFolderInfo);
|
|
|
|
}
|
|
|
|
|
|
|
|
ScanFoldersModel *ScanFoldersModel::m_instance = 0;
|