Christophe Dumez
15 years ago
13 changed files with 600 additions and 269 deletions
@ -0,0 +1,191 @@ |
|||||||
|
/*
|
||||||
|
* 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) |
||||||
|
{ } |
||||||
|
|
||||||
|
ScanFoldersModel::~ScanFoldersModel() { } |
||||||
|
|
||||||
|
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(); |
||||||
|
|
||||||
|
const QSharedPointer<PathData> &pathData = m_pathList.at(index.row()); |
||||||
|
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()); |
||||||
|
m_pathList << QSharedPointer<PathData>(new PathData(canonicalPath)); |
||||||
|
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) { |
||||||
|
const QSharedPointer<PathData> &pathData = m_pathList.at(i); |
||||||
|
if (pathData->path == path) |
||||||
|
return i; |
||||||
|
} |
||||||
|
|
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
void ScanFoldersModel::makePersistent(QSettings &settings) { |
||||||
|
QStringList paths; |
||||||
|
QList<QVariant> downloadInFolderInfo; |
||||||
|
foreach (const QSharedPointer<PathData> &pathData, m_pathList) { |
||||||
|
paths << pathData->path; |
||||||
|
downloadInFolderInfo << pathData->downloadAtPath; |
||||||
|
} |
||||||
|
settings.setValue(QString::fromUtf8("ScanDirs"), paths); |
||||||
|
settings.setValue(QString::fromUtf8("DownloadInScanDirs"), downloadInFolderInfo); |
||||||
|
} |
||||||
|
|
||||||
|
ScanFoldersModel *ScanFoldersModel::m_instance = 0; |
||||||
@ -0,0 +1,82 @@ |
|||||||
|
/*
|
||||||
|
* 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 |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef SCANNEDFOLDERSMODEL_H |
||||||
|
#define SCANNEDFOLDERSMODEL_H |
||||||
|
|
||||||
|
#include <QAbstractTableModel> |
||||||
|
#include <QList> |
||||||
|
#include <QSharedPointer> |
||||||
|
#include <QStringList> |
||||||
|
|
||||||
|
class FileSystemWatcher; |
||||||
|
class QSettings; |
||||||
|
|
||||||
|
class ScanFoldersModel : public QAbstractTableModel { |
||||||
|
Q_OBJECT |
||||||
|
Q_DISABLE_COPY(ScanFoldersModel) |
||||||
|
|
||||||
|
public: |
||||||
|
enum PathStatus { Ok, DoesNotExist, CannotRead, CannotWrite, AlreadyInList }; |
||||||
|
static ScanFoldersModel *instance(QObject *parent = 0); |
||||||
|
~ScanFoldersModel(); |
||||||
|
|
||||||
|
virtual int rowCount(const QModelIndex & parent = QModelIndex()) const; |
||||||
|
virtual int columnCount(const QModelIndex &parent = QModelIndex()) const; |
||||||
|
virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const; |
||||||
|
virtual QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const; |
||||||
|
virtual Qt::ItemFlags flags(const QModelIndex &index) const; |
||||||
|
|
||||||
|
// TODO: removePaths(); singular version becomes private helper functions;
|
||||||
|
// also: remove functions should take modelindexes
|
||||||
|
PathStatus addPath(const QString &path); |
||||||
|
void removePath(int row); |
||||||
|
bool removePath(const QString &path); |
||||||
|
PathStatus setDownloadAtPath(int row, bool downloadAtPath); |
||||||
|
|
||||||
|
bool downloadInTorrentFolder(const QString &filePath) const; |
||||||
|
void makePersistent(QSettings &settings); |
||||||
|
|
||||||
|
signals: |
||||||
|
// The absolute paths of new torrent files in the scanned directories.
|
||||||
|
void torrentsAdded(QStringList &pathList); |
||||||
|
|
||||||
|
private: |
||||||
|
explicit ScanFoldersModel(QObject *parent); |
||||||
|
virtual bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole); |
||||||
|
static ScanFoldersModel *m_instance; |
||||||
|
class PathData; |
||||||
|
int findPathData(const QString &path) const; |
||||||
|
|
||||||
|
QList<QSharedPointer<PathData> > m_pathList; |
||||||
|
FileSystemWatcher *m_fsWatcher; |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // SCANNEDFOLDERSMODEL_H
|
||||||
Loading…
Reference in new issue