Christophe Dumez
15 years ago
13 changed files with 600 additions and 269 deletions
@ -0,0 +1,191 @@
@@ -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 @@
@@ -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