Vladimir Golovnev
4 years ago
committed by
GitHub
40 changed files with 1681 additions and 1073 deletions
@ -1,187 +0,0 @@
@@ -1,187 +0,0 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2018 |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "filesystemwatcher.h" |
||||
|
||||
#include <QtGlobal> |
||||
|
||||
#include <chrono> |
||||
|
||||
#if defined(Q_OS_MACOS) || defined(Q_OS_FREEBSD) || defined(Q_OS_OPENBSD) |
||||
#include <cstring> |
||||
#include <sys/mount.h> |
||||
#include <sys/param.h> |
||||
#endif |
||||
|
||||
#include "base/algorithm.h" |
||||
#include "base/bittorrent/torrentinfo.h" |
||||
#include "base/global.h" |
||||
#include "base/logger.h" |
||||
#include "base/utils/fs.h" |
||||
|
||||
using namespace std::chrono_literals; |
||||
|
||||
namespace |
||||
{ |
||||
const std::chrono::duration WATCH_INTERVAL = 10s; |
||||
const int MAX_PARTIAL_RETRIES = 5; |
||||
} |
||||
|
||||
FileSystemWatcher::FileSystemWatcher(QObject *parent) |
||||
: QFileSystemWatcher(parent) |
||||
{ |
||||
connect(this, &QFileSystemWatcher::directoryChanged, this, &FileSystemWatcher::scanLocalFolder); |
||||
|
||||
m_partialTorrentTimer.setSingleShot(true); |
||||
connect(&m_partialTorrentTimer, &QTimer::timeout, this, &FileSystemWatcher::processPartialTorrents); |
||||
|
||||
connect(&m_watchTimer, &QTimer::timeout, this, &FileSystemWatcher::scanNetworkFolders); |
||||
} |
||||
|
||||
QStringList FileSystemWatcher::directories() const |
||||
{ |
||||
QStringList dirs = QFileSystemWatcher::directories(); |
||||
for (const QDir &dir : asConst(m_watchedFolders)) |
||||
dirs << dir.canonicalPath(); |
||||
return dirs; |
||||
} |
||||
|
||||
void FileSystemWatcher::addPath(const QString &path) |
||||
{ |
||||
if (path.isEmpty()) return; |
||||
|
||||
#if !defined Q_OS_HAIKU |
||||
const QDir dir(path); |
||||
if (!dir.exists()) return; |
||||
|
||||
// Check if the path points to a network file system or not
|
||||
if (Utils::Fs::isNetworkFileSystem(path)) |
||||
{ |
||||
// Network mode
|
||||
LogMsg(tr("Watching remote folder: \"%1\"").arg(Utils::Fs::toNativePath(path))); |
||||
m_watchedFolders << dir; |
||||
|
||||
m_watchTimer.start(WATCH_INTERVAL); |
||||
return; |
||||
} |
||||
#endif |
||||
|
||||
// Normal mode
|
||||
LogMsg(tr("Watching local folder: \"%1\"").arg(Utils::Fs::toNativePath(path))); |
||||
QFileSystemWatcher::addPath(path); |
||||
scanLocalFolder(path); |
||||
} |
||||
|
||||
void FileSystemWatcher::removePath(const QString &path) |
||||
{ |
||||
if (m_watchedFolders.removeOne(path)) |
||||
{ |
||||
if (m_watchedFolders.isEmpty()) |
||||
m_watchTimer.stop(); |
||||
return; |
||||
} |
||||
|
||||
// Normal mode
|
||||
QFileSystemWatcher::removePath(path); |
||||
} |
||||
|
||||
void FileSystemWatcher::scanLocalFolder(const QString &path) |
||||
{ |
||||
QTimer::singleShot(2000, this, [this, path]() { processTorrentsInDir(path); }); |
||||
} |
||||
|
||||
void FileSystemWatcher::scanNetworkFolders() |
||||
{ |
||||
for (const QDir &dir : asConst(m_watchedFolders)) |
||||
processTorrentsInDir(dir); |
||||
} |
||||
|
||||
void FileSystemWatcher::processPartialTorrents() |
||||
{ |
||||
QStringList noLongerPartial; |
||||
|
||||
// Check which torrents are still partial
|
||||
Algorithm::removeIf(m_partialTorrents, [&noLongerPartial](const QString &torrentPath, int &value) |
||||
{ |
||||
if (!QFile::exists(torrentPath)) |
||||
return true; |
||||
|
||||
if (BitTorrent::TorrentInfo::loadFromFile(torrentPath).isValid()) |
||||
{ |
||||
noLongerPartial << torrentPath; |
||||
return true; |
||||
} |
||||
|
||||
if (value >= MAX_PARTIAL_RETRIES) |
||||
{ |
||||
QFile::rename(torrentPath, torrentPath + ".qbt_rejected"); |
||||
return true; |
||||
} |
||||
|
||||
++value; |
||||
return false; |
||||
}); |
||||
|
||||
// Stop the partial timer if necessary
|
||||
if (m_partialTorrents.empty()) |
||||
{ |
||||
m_partialTorrentTimer.stop(); |
||||
qDebug("No longer any partial torrent."); |
||||
} |
||||
else |
||||
{ |
||||
qDebug("Still %d partial torrents after delayed processing.", m_partialTorrents.count()); |
||||
m_partialTorrentTimer.start(WATCH_INTERVAL); |
||||
} |
||||
|
||||
// Notify of new torrents
|
||||
if (!noLongerPartial.isEmpty()) |
||||
emit torrentsAdded(noLongerPartial); |
||||
} |
||||
|
||||
void FileSystemWatcher::processTorrentsInDir(const QDir &dir) |
||||
{ |
||||
QStringList torrents; |
||||
const QStringList files = dir.entryList({"*.torrent", "*.magnet"}, QDir::Files); |
||||
for (const QString &file : files) |
||||
{ |
||||
const QString fileAbsPath = dir.absoluteFilePath(file); |
||||
if (file.endsWith(".magnet", Qt::CaseInsensitive)) |
||||
torrents << fileAbsPath; |
||||
else if (BitTorrent::TorrentInfo::loadFromFile(fileAbsPath).isValid()) |
||||
torrents << fileAbsPath; |
||||
else if (!m_partialTorrents.contains(fileAbsPath)) |
||||
m_partialTorrents[fileAbsPath] = 0; |
||||
} |
||||
|
||||
if (!torrents.empty()) |
||||
emit torrentsAdded(torrents); |
||||
|
||||
if (!m_partialTorrents.empty() && !m_partialTorrentTimer.isActive()) |
||||
m_partialTorrentTimer.start(WATCH_INTERVAL); |
||||
} |
||||
@ -1,424 +0,0 @@
@@ -1,424 +0,0 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2010 Christian Kandeler, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "scanfoldersmodel.h" |
||||
|
||||
#include <QDir> |
||||
#include <QFileInfo> |
||||
#include <QStringList> |
||||
#include <QTextStream> |
||||
|
||||
#include "bittorrent/session.h" |
||||
#include "filesystemwatcher.h" |
||||
#include "global.h" |
||||
#include "preferences.h" |
||||
#include "utils/fs.h" |
||||
|
||||
struct ScanFoldersModel::PathData |
||||
{ |
||||
PathData(const QString &watchPath, const PathType &type, const QString &downloadPath) |
||||
: watchPath(watchPath) |
||||
, downloadType(type) |
||||
, downloadPath(downloadPath) |
||||
{ |
||||
if (this->downloadPath.isEmpty() && downloadType == CUSTOM_LOCATION) |
||||
downloadType = DEFAULT_LOCATION; |
||||
} |
||||
|
||||
QString watchPath; |
||||
PathType downloadType; |
||||
QString downloadPath; // valid for CUSTOM_LOCATION
|
||||
}; |
||||
|
||||
ScanFoldersModel *ScanFoldersModel::m_instance = nullptr; |
||||
|
||||
void ScanFoldersModel::initInstance() |
||||
{ |
||||
if (!m_instance) |
||||
m_instance = new ScanFoldersModel; |
||||
} |
||||
|
||||
void ScanFoldersModel::freeInstance() |
||||
{ |
||||
delete m_instance; |
||||
m_instance = nullptr; |
||||
} |
||||
|
||||
ScanFoldersModel *ScanFoldersModel::instance() |
||||
{ |
||||
return m_instance; |
||||
} |
||||
|
||||
ScanFoldersModel::ScanFoldersModel(QObject *parent) |
||||
: QAbstractListModel(parent) |
||||
, m_fsWatcher(nullptr) |
||||
{ |
||||
configure(); |
||||
connect(Preferences::instance(), &Preferences::changed, this, &ScanFoldersModel::configure); |
||||
} |
||||
|
||||
ScanFoldersModel::~ScanFoldersModel() |
||||
{ |
||||
qDeleteAll(m_pathList); |
||||
} |
||||
|
||||
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 NB_COLUMNS; |
||||
} |
||||
|
||||
QVariant ScanFoldersModel::data(const QModelIndex &index, int role) const |
||||
{ |
||||
if (!index.isValid() || (index.row() >= rowCount())) |
||||
return {}; |
||||
|
||||
const PathData *pathData = m_pathList.at(index.row()); |
||||
QVariant value; |
||||
|
||||
switch (index.column()) |
||||
{ |
||||
case WATCH: |
||||
if (role == Qt::DisplayRole) |
||||
value = Utils::Fs::toNativePath(pathData->watchPath); |
||||
break; |
||||
case DOWNLOAD: |
||||
if (role == Qt::UserRole) |
||||
{ |
||||
value = pathData->downloadType; |
||||
} |
||||
else if (role == Qt::DisplayRole) |
||||
{ |
||||
switch (pathData->downloadType) |
||||
{ |
||||
case DOWNLOAD_IN_WATCH_FOLDER: |
||||
case DEFAULT_LOCATION: |
||||
value = pathTypeDisplayName(pathData->downloadType); |
||||
break; |
||||
case CUSTOM_LOCATION: |
||||
value = pathData->downloadPath; |
||||
break; |
||||
} |
||||
} |
||||
break; |
||||
} |
||||
|
||||
return value; |
||||
} |
||||
|
||||
QVariant ScanFoldersModel::headerData(int section, Qt::Orientation orientation, int role) const |
||||
{ |
||||
if ((orientation != Qt::Horizontal) || (role != Qt::DisplayRole) || (section < 0) || (section >= columnCount())) |
||||
return {}; |
||||
|
||||
QVariant title; |
||||
|
||||
switch (section) |
||||
{ |
||||
case WATCH: |
||||
title = tr("Monitored Folder"); |
||||
break; |
||||
case DOWNLOAD: |
||||
title = tr("Override Save Location"); |
||||
break; |
||||
} |
||||
|
||||
return title; |
||||
} |
||||
|
||||
Qt::ItemFlags ScanFoldersModel::flags(const QModelIndex &index) const |
||||
{ |
||||
if (!index.isValid() || (index.row() >= rowCount())) |
||||
return QAbstractListModel::flags(index); |
||||
|
||||
Qt::ItemFlags flags; |
||||
|
||||
switch (index.column()) |
||||
{ |
||||
case WATCH: |
||||
flags = QAbstractListModel::flags(index); |
||||
break; |
||||
case DOWNLOAD: |
||||
flags = QAbstractListModel::flags(index) | Qt::ItemIsEditable; |
||||
break; |
||||
} |
||||
|
||||
return flags; |
||||
} |
||||
|
||||
bool ScanFoldersModel::setData(const QModelIndex &index, const QVariant &value, int role) |
||||
{ |
||||
if (!index.isValid() || (index.row() >= rowCount()) || (index.column() >= columnCount()) |
||||
|| (index.column() != DOWNLOAD)) |
||||
return false; |
||||
|
||||
if (role == Qt::UserRole) |
||||
{ |
||||
const auto type = static_cast<PathType>(value.toInt()); |
||||
if (type == CUSTOM_LOCATION) |
||||
return false; |
||||
|
||||
m_pathList[index.row()]->downloadType = type; |
||||
m_pathList[index.row()]->downloadPath.clear(); |
||||
emit dataChanged(index, index); |
||||
} |
||||
else if (role == Qt::DisplayRole) |
||||
{ |
||||
const QString path = value.toString(); |
||||
if (path.isEmpty()) // means we didn't pass CUSTOM_LOCATION type
|
||||
return false; |
||||
|
||||
m_pathList[index.row()]->downloadType = CUSTOM_LOCATION; |
||||
m_pathList[index.row()]->downloadPath = Utils::Fs::toNativePath(path); |
||||
emit dataChanged(index, index); |
||||
} |
||||
else |
||||
{ |
||||
return false; |
||||
} |
||||
|
||||
return true; |
||||
} |
||||
|
||||
ScanFoldersModel::PathStatus ScanFoldersModel::addPath(const QString &watchPath, const PathType &downloadType, const QString &downloadPath, bool addToFSWatcher) |
||||
{ |
||||
const QDir watchDir(watchPath); |
||||
if (!watchDir.exists()) return DoesNotExist; |
||||
if (!watchDir.isReadable()) return CannotRead; |
||||
|
||||
const QString canonicalWatchPath = watchDir.canonicalPath(); |
||||
if (findPathData(canonicalWatchPath) != -1) return AlreadyInList; |
||||
|
||||
const QDir downloadDir(downloadPath); |
||||
const QString canonicalDownloadPath = downloadDir.canonicalPath(); |
||||
|
||||
if (!m_fsWatcher) |
||||
{ |
||||
m_fsWatcher = new FileSystemWatcher(this); |
||||
connect(m_fsWatcher, &FileSystemWatcher::torrentsAdded, this, &ScanFoldersModel::addTorrentsToSession); |
||||
} |
||||
|
||||
beginInsertRows(QModelIndex(), rowCount(), rowCount()); |
||||
m_pathList << new PathData(Utils::Fs::toNativePath(canonicalWatchPath), downloadType, Utils::Fs::toNativePath(canonicalDownloadPath)); |
||||
endInsertRows(); |
||||
|
||||
// Start scanning
|
||||
if (addToFSWatcher) |
||||
m_fsWatcher->addPath(canonicalWatchPath); |
||||
return Ok; |
||||
} |
||||
|
||||
ScanFoldersModel::PathStatus ScanFoldersModel::updatePath(const QString &watchPath, const PathType &downloadType, const QString &downloadPath) |
||||
{ |
||||
const QDir watchDir(watchPath); |
||||
const QString canonicalWatchPath = watchDir.canonicalPath(); |
||||
const int row = findPathData(canonicalWatchPath); |
||||
if (row == -1) return DoesNotExist; |
||||
|
||||
const QDir downloadDir(downloadPath); |
||||
const QString canonicalDownloadPath = downloadDir.canonicalPath(); |
||||
|
||||
m_pathList.at(row)->downloadType = downloadType; |
||||
m_pathList.at(row)->downloadPath = Utils::Fs::toNativePath(canonicalDownloadPath); |
||||
|
||||
return Ok; |
||||
} |
||||
|
||||
void ScanFoldersModel::addToFSWatcher(const QStringList &watchPaths) |
||||
{ |
||||
if (!m_fsWatcher) |
||||
return; // addPath() wasn't called before this
|
||||
|
||||
for (const QString &path : watchPaths) |
||||
{ |
||||
const QDir watchDir(path); |
||||
const QString canonicalWatchPath = watchDir.canonicalPath(); |
||||
m_fsWatcher->addPath(canonicalWatchPath); |
||||
} |
||||
} |
||||
|
||||
void ScanFoldersModel::removePath(const int row, const bool removeFromFSWatcher) |
||||
{ |
||||
Q_ASSERT((row >= 0) && (row < rowCount())); |
||||
beginRemoveRows(QModelIndex(), row, row); |
||||
if (removeFromFSWatcher) |
||||
m_fsWatcher->removePath(m_pathList.at(row)->watchPath); |
||||
delete m_pathList.takeAt(row); |
||||
endRemoveRows(); |
||||
} |
||||
|
||||
bool ScanFoldersModel::removePath(const QString &path, const bool removeFromFSWatcher) |
||||
{ |
||||
const int row = findPathData(path); |
||||
if (row == -1) return false; |
||||
|
||||
removePath(row, removeFromFSWatcher); |
||||
return true; |
||||
} |
||||
|
||||
void ScanFoldersModel::removeFromFSWatcher(const QStringList &watchPaths) |
||||
{ |
||||
for (const QString &path : watchPaths) |
||||
m_fsWatcher->removePath(path); |
||||
} |
||||
|
||||
bool ScanFoldersModel::downloadInWatchFolder(const QString &filePath) const |
||||
{ |
||||
const int row = findPathData(QFileInfo(filePath).dir().path()); |
||||
Q_ASSERT(row != -1); |
||||
const PathData *data = m_pathList.at(row); |
||||
return (data->downloadType == DOWNLOAD_IN_WATCH_FOLDER); |
||||
} |
||||
|
||||
bool ScanFoldersModel::downloadInDefaultFolder(const QString &filePath) const |
||||
{ |
||||
const int row = findPathData(QFileInfo(filePath).dir().path()); |
||||
Q_ASSERT(row != -1); |
||||
const PathData *data = m_pathList.at(row); |
||||
return (data->downloadType == DEFAULT_LOCATION); |
||||
} |
||||
|
||||
QString ScanFoldersModel::downloadPathTorrentFolder(const QString &filePath) const |
||||
{ |
||||
const int row = findPathData(QFileInfo(filePath).dir().path()); |
||||
Q_ASSERT(row != -1); |
||||
const PathData *data = m_pathList.at(row); |
||||
if (data->downloadType == CUSTOM_LOCATION) |
||||
return data->downloadPath; |
||||
|
||||
return {}; |
||||
} |
||||
|
||||
int ScanFoldersModel::findPathData(const QString &path) const |
||||
{ |
||||
for (int i = 0; i < m_pathList.count(); ++i) |
||||
if (m_pathList.at(i)->watchPath == Utils::Fs::toNativePath(path)) |
||||
return i; |
||||
|
||||
return -1; |
||||
} |
||||
|
||||
void ScanFoldersModel::makePersistent() |
||||
{ |
||||
QVariantHash dirs; |
||||
|
||||
for (const PathData *pathData : asConst(m_pathList)) |
||||
{ |
||||
if (pathData->downloadType == CUSTOM_LOCATION) |
||||
dirs.insert(Utils::Fs::toUniformPath(pathData->watchPath), Utils::Fs::toUniformPath(pathData->downloadPath)); |
||||
else |
||||
dirs.insert(Utils::Fs::toUniformPath(pathData->watchPath), pathData->downloadType); |
||||
} |
||||
|
||||
Preferences::instance()->setScanDirs(dirs); |
||||
} |
||||
|
||||
void ScanFoldersModel::configure() |
||||
{ |
||||
const QVariantHash dirs = Preferences::instance()->getScanDirs(); |
||||
|
||||
for (auto i = dirs.cbegin(); i != dirs.cend(); ++i) |
||||
{ |
||||
if (i.value().type() == QVariant::Int) |
||||
addPath(i.key(), static_cast<PathType>(i.value().toInt()), QString()); |
||||
else |
||||
addPath(i.key(), CUSTOM_LOCATION, i.value().toString()); |
||||
} |
||||
} |
||||
|
||||
void ScanFoldersModel::addTorrentsToSession(const QStringList &pathList) |
||||
{ |
||||
for (const QString &file : pathList) |
||||
{ |
||||
qDebug("File %s added", qUtf8Printable(file)); |
||||
|
||||
BitTorrent::AddTorrentParams params; |
||||
if (downloadInWatchFolder(file)) |
||||
{ |
||||
params.savePath = QFileInfo(file).dir().path(); |
||||
params.useAutoTMM = false; |
||||
} |
||||
else if (!downloadInDefaultFolder(file)) |
||||
{ |
||||
params.savePath = downloadPathTorrentFolder(file); |
||||
params.useAutoTMM = false; |
||||
} |
||||
|
||||
if (file.endsWith(".magnet", Qt::CaseInsensitive)) |
||||
{ |
||||
QFile f(file); |
||||
if (f.open(QIODevice::ReadOnly | QIODevice::Text)) |
||||
{ |
||||
QTextStream str(&f); |
||||
while (!str.atEnd()) |
||||
BitTorrent::Session::instance()->addTorrent(str.readLine(), params); |
||||
|
||||
f.close(); |
||||
Utils::Fs::forceRemove(file); |
||||
} |
||||
else |
||||
{ |
||||
qDebug("Failed to open magnet file: %s", qUtf8Printable(f.errorString())); |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
const BitTorrent::TorrentInfo torrentInfo = BitTorrent::TorrentInfo::loadFromFile(file); |
||||
if (torrentInfo.isValid()) |
||||
{ |
||||
BitTorrent::Session::instance()->addTorrent(torrentInfo, params); |
||||
Utils::Fs::forceRemove(file); |
||||
} |
||||
else |
||||
{ |
||||
qDebug("Ignoring incomplete torrent file: %s", qUtf8Printable(file)); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
QString ScanFoldersModel::pathTypeDisplayName(const PathType type) |
||||
{ |
||||
switch (type) |
||||
{ |
||||
case DOWNLOAD_IN_WATCH_FOLDER: |
||||
return tr("Monitored folder"); |
||||
case DEFAULT_LOCATION: |
||||
return tr("Default save location"); |
||||
case CUSTOM_LOCATION: |
||||
return tr("Browse..."); |
||||
default: |
||||
qDebug("Invalid PathType: %d", type); |
||||
}; |
||||
return {}; |
||||
} |
||||
@ -1,111 +0,0 @@
@@ -1,111 +0,0 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2010 Christian Kandeler, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#pragma once |
||||
|
||||
#include <QAbstractListModel> |
||||
#include <QList> |
||||
#include <QtContainerFwd> |
||||
|
||||
class FileSystemWatcher; |
||||
|
||||
class ScanFoldersModel final : public QAbstractListModel |
||||
{ |
||||
Q_OBJECT |
||||
Q_DISABLE_COPY(ScanFoldersModel) |
||||
|
||||
public: |
||||
enum PathStatus |
||||
{ |
||||
Ok, |
||||
DoesNotExist, |
||||
CannotRead, |
||||
CannotWrite, |
||||
AlreadyInList |
||||
}; |
||||
|
||||
enum Column |
||||
{ |
||||
WATCH, |
||||
DOWNLOAD, |
||||
NB_COLUMNS |
||||
}; |
||||
|
||||
enum PathType |
||||
{ |
||||
DOWNLOAD_IN_WATCH_FOLDER, |
||||
DEFAULT_LOCATION, |
||||
CUSTOM_LOCATION |
||||
}; |
||||
|
||||
static void initInstance(); |
||||
static void freeInstance(); |
||||
static ScanFoldersModel *instance(); |
||||
|
||||
static QString pathTypeDisplayName(PathType type); |
||||
|
||||
int rowCount(const QModelIndex &parent = {}) const override; |
||||
int columnCount(const QModelIndex &parent = {}) const override; |
||||
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override; |
||||
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override; |
||||
Qt::ItemFlags flags(const QModelIndex &index) const override; |
||||
|
||||
// TODO: removePaths(); singular version becomes private helper functions;
|
||||
// also: remove functions should take modelindexes
|
||||
PathStatus addPath(const QString &watchPath, const PathType &downloadType, const QString &downloadPath, bool addToFSWatcher = true); |
||||
PathStatus updatePath(const QString &watchPath, const PathType &downloadType, const QString &downloadPath); |
||||
// PRECONDITION: The paths must have been added with addPath() first.
|
||||
void addToFSWatcher(const QStringList &watchPaths); |
||||
void removePath(int row, bool removeFromFSWatcher = true); |
||||
bool removePath(const QString &path, bool removeFromFSWatcher = true); |
||||
void removeFromFSWatcher(const QStringList &watchPaths); |
||||
|
||||
void makePersistent(); |
||||
|
||||
public slots: |
||||
void configure(); |
||||
|
||||
private slots: |
||||
void addTorrentsToSession(const QStringList &pathList); |
||||
|
||||
private: |
||||
explicit ScanFoldersModel(QObject *parent = nullptr); |
||||
~ScanFoldersModel(); |
||||
|
||||
virtual bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override; |
||||
bool downloadInWatchFolder(const QString &filePath) const; |
||||
bool downloadInDefaultFolder(const QString &filePath) const; |
||||
QString downloadPathTorrentFolder(const QString &filePath) const; |
||||
int findPathData(const QString &path) const; |
||||
|
||||
static ScanFoldersModel *m_instance; |
||||
struct PathData; |
||||
|
||||
QList<PathData*> m_pathList; |
||||
FileSystemWatcher *m_fsWatcher; |
||||
}; |
||||
@ -0,0 +1,658 @@
@@ -0,0 +1,658 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2021 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2010 Christian Kandeler, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "torrentfileswatcher.h" |
||||
|
||||
#include <chrono> |
||||
|
||||
#include <QtGlobal> |
||||
#include <QDir> |
||||
#include <QDirIterator> |
||||
#include <QFile> |
||||
#include <QFileSystemWatcher> |
||||
#include <QJsonArray> |
||||
#include <QJsonDocument> |
||||
#include <QJsonObject> |
||||
#include <QJsonValue> |
||||
#include <QSaveFile> |
||||
#include <QSet> |
||||
#include <QTextStream> |
||||
#include <QThread> |
||||
#include <QTimer> |
||||
#include <QVariant> |
||||
|
||||
#include "base/algorithm.h" |
||||
#include "base/bittorrent/magneturi.h" |
||||
#include "base/bittorrent/torrentcontentlayout.h" |
||||
#include "base/bittorrent/session.h" |
||||
#include "base/bittorrent/torrent.h" |
||||
#include "base/bittorrent/torrentinfo.h" |
||||
#include "base/exceptions.h" |
||||
#include "base/global.h" |
||||
#include "base/logger.h" |
||||
#include "base/profile.h" |
||||
#include "base/settingsstorage.h" |
||||
#include "base/tagset.h" |
||||
#include "base/utils/fs.h" |
||||
#include "base/utils/string.h" |
||||
|
||||
using namespace std::chrono_literals; |
||||
|
||||
const std::chrono::duration WATCH_INTERVAL = 10s; |
||||
const int MAX_FAILED_RETRIES = 5; |
||||
const QString CONF_FILE_NAME {QStringLiteral("watched_folders.json")}; |
||||
|
||||
const QString OPTION_ADDTORRENTPARAMS {QStringLiteral("add_torrent_params")}; |
||||
const QString OPTION_RECURSIVE {QStringLiteral("recursive")}; |
||||
|
||||
const QString PARAM_CATEGORY {QStringLiteral("category")}; |
||||
const QString PARAM_TAGS {QStringLiteral("tags")}; |
||||
const QString PARAM_SAVEPATH {QStringLiteral("save_path")}; |
||||
const QString PARAM_OPERATINGMODE {QStringLiteral("operating_mode")}; |
||||
const QString PARAM_STOPPED {QStringLiteral("stopped")}; |
||||
const QString PARAM_CONTENTLAYOUT {QStringLiteral("content_layout")}; |
||||
const QString PARAM_AUTOTMM {QStringLiteral("use_auto_tmm")}; |
||||
const QString PARAM_UPLOADLIMIT {QStringLiteral("upload_limit")}; |
||||
const QString PARAM_DOWNLOADLIMIT {QStringLiteral("download_limit")}; |
||||
const QString PARAM_SEEDINGTIMELIMIT {QStringLiteral("seeding_time_limit")}; |
||||
const QString PARAM_RATIOLIMIT {QStringLiteral("ratio_limit")}; |
||||
|
||||
namespace |
||||
{ |
||||
TagSet parseTagSet(const QJsonArray &jsonArr) |
||||
{ |
||||
TagSet tags; |
||||
for (const QJsonValue &jsonVal : jsonArr) |
||||
tags.insert(jsonVal.toString()); |
||||
|
||||
return tags; |
||||
} |
||||
|
||||
QJsonArray serializeTagSet(const TagSet &tags) |
||||
{ |
||||
QJsonArray arr; |
||||
for (const QString &tag : tags) |
||||
arr.append(tag); |
||||
|
||||
return arr; |
||||
} |
||||
|
||||
std::optional<bool> getOptionalBool(const QJsonObject &jsonObj, const QString &key) |
||||
{ |
||||
const QJsonValue jsonVal = jsonObj.value(key); |
||||
if (jsonVal.isUndefined() || jsonVal.isNull()) |
||||
return std::nullopt; |
||||
|
||||
return jsonVal.toBool(); |
||||
} |
||||
|
||||
template <typename Enum> |
||||
std::optional<Enum> getOptionalEnum(const QJsonObject &jsonObj, const QString &key) |
||||
{ |
||||
const QJsonValue jsonVal = jsonObj.value(key); |
||||
if (jsonVal.isUndefined() || jsonVal.isNull()) |
||||
return std::nullopt; |
||||
|
||||
return Utils::String::toEnum<Enum>(jsonVal.toString(), {}); |
||||
} |
||||
|
||||
template <typename Enum> |
||||
Enum getEnum(const QJsonObject &jsonObj, const QString &key) |
||||
{ |
||||
const QJsonValue jsonVal = jsonObj.value(key); |
||||
return Utils::String::toEnum<Enum>(jsonVal.toString(), {}); |
||||
} |
||||
|
||||
BitTorrent::AddTorrentParams parseAddTorrentParams(const QJsonObject &jsonObj) |
||||
{ |
||||
BitTorrent::AddTorrentParams params; |
||||
params.category = jsonObj.value(PARAM_CATEGORY).toString(); |
||||
params.tags = parseTagSet(jsonObj.value(PARAM_TAGS).toArray()); |
||||
params.savePath = jsonObj.value(PARAM_SAVEPATH).toString(); |
||||
params.addForced = (getEnum<BitTorrent::TorrentOperatingMode>(jsonObj, PARAM_OPERATINGMODE) == BitTorrent::TorrentOperatingMode::Forced); |
||||
params.addPaused = getOptionalBool(jsonObj, PARAM_STOPPED); |
||||
params.contentLayout = getOptionalEnum<BitTorrent::TorrentContentLayout>(jsonObj, PARAM_CONTENTLAYOUT); |
||||
params.useAutoTMM = getOptionalBool(jsonObj, PARAM_AUTOTMM); |
||||
params.uploadLimit = jsonObj.value(PARAM_UPLOADLIMIT).toInt(-1); |
||||
params.downloadLimit = jsonObj.value(PARAM_DOWNLOADLIMIT).toInt(-1); |
||||
params.seedingTimeLimit = jsonObj.value(PARAM_SEEDINGTIMELIMIT).toInt(BitTorrent::Torrent::USE_GLOBAL_SEEDING_TIME); |
||||
params.ratioLimit = jsonObj.value(PARAM_RATIOLIMIT).toDouble(BitTorrent::Torrent::USE_GLOBAL_RATIO); |
||||
|
||||
return params; |
||||
} |
||||
|
||||
QJsonObject serializeAddTorrentParams(const BitTorrent::AddTorrentParams ¶ms) |
||||
{ |
||||
QJsonObject jsonObj { |
||||
{PARAM_CATEGORY, params.category}, |
||||
{PARAM_TAGS, serializeTagSet(params.tags)}, |
||||
{PARAM_SAVEPATH, params.savePath}, |
||||
{PARAM_OPERATINGMODE, Utils::String::fromEnum(params.addForced |
||||
? BitTorrent::TorrentOperatingMode::Forced : BitTorrent::TorrentOperatingMode::AutoManaged)}, |
||||
{PARAM_UPLOADLIMIT, params.uploadLimit}, |
||||
{PARAM_DOWNLOADLIMIT, params.downloadLimit}, |
||||
{PARAM_SEEDINGTIMELIMIT, params.seedingTimeLimit}, |
||||
{PARAM_RATIOLIMIT, params.ratioLimit} |
||||
}; |
||||
|
||||
if (params.addPaused) |
||||
jsonObj[PARAM_STOPPED] = *params.addPaused; |
||||
if (params.contentLayout) |
||||
jsonObj[PARAM_CONTENTLAYOUT] = Utils::String::fromEnum(*params.contentLayout); |
||||
if (params.useAutoTMM) |
||||
jsonObj[PARAM_AUTOTMM] = *params.useAutoTMM; |
||||
|
||||
return jsonObj; |
||||
} |
||||
|
||||
TorrentFilesWatcher::WatchedFolderOptions parseWatchedFolderOptions(const QJsonObject &jsonObj) |
||||
{ |
||||
TorrentFilesWatcher::WatchedFolderOptions options; |
||||
options.addTorrentParams = parseAddTorrentParams(jsonObj.value(OPTION_ADDTORRENTPARAMS).toObject()); |
||||
options.recursive = jsonObj.value(OPTION_RECURSIVE).toBool(); |
||||
|
||||
return options; |
||||
} |
||||
|
||||
QJsonObject serializeWatchedFolderOptions(const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
return { |
||||
{OPTION_ADDTORRENTPARAMS, serializeAddTorrentParams(options.addTorrentParams)}, |
||||
{OPTION_RECURSIVE, options.recursive} |
||||
}; |
||||
} |
||||
} |
||||
|
||||
class TorrentFilesWatcher::Worker final : public QObject |
||||
{ |
||||
Q_OBJECT |
||||
Q_DISABLE_COPY(Worker) |
||||
|
||||
public: |
||||
Worker(); |
||||
|
||||
public slots: |
||||
void setWatchedFolder(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options); |
||||
void removeWatchedFolder(const QString &path); |
||||
|
||||
signals: |
||||
void magnetFound(const BitTorrent::MagnetUri &magnetURI, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||
void torrentFound(const BitTorrent::TorrentInfo &torrentInfo, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||
|
||||
private: |
||||
void onTimeout(); |
||||
void processWatchedFolder(const QString &path); |
||||
void processFolder(const QString &path, const QString &watchedFolderPath, const TorrentFilesWatcher::WatchedFolderOptions &options); |
||||
void processFailedTorrents(); |
||||
void addWatchedFolder(const QString &watchedFolderID, const TorrentFilesWatcher::WatchedFolderOptions &options); |
||||
void updateWatchedFolder(const QString &watchedFolderID, const TorrentFilesWatcher::WatchedFolderOptions &options); |
||||
|
||||
QFileSystemWatcher *m_watcher = nullptr; |
||||
QTimer *m_watchTimer = nullptr; |
||||
QHash<QString, TorrentFilesWatcher::WatchedFolderOptions> m_watchedFolders; |
||||
QSet<QString> m_watchedByTimeoutFolders; |
||||
|
||||
// Failed torrents
|
||||
QTimer *m_retryTorrentTimer = nullptr; |
||||
QHash<QString, QHash<QString, int>> m_failedTorrents; |
||||
}; |
||||
|
||||
TorrentFilesWatcher *TorrentFilesWatcher::m_instance = nullptr; |
||||
|
||||
void TorrentFilesWatcher::initInstance() |
||||
{ |
||||
if (!m_instance) |
||||
m_instance = new TorrentFilesWatcher; |
||||
} |
||||
|
||||
void TorrentFilesWatcher::freeInstance() |
||||
{ |
||||
delete m_instance; |
||||
m_instance = nullptr; |
||||
} |
||||
|
||||
TorrentFilesWatcher *TorrentFilesWatcher::instance() |
||||
{ |
||||
return m_instance; |
||||
} |
||||
|
||||
TorrentFilesWatcher::TorrentFilesWatcher(QObject *parent) |
||||
: QObject {parent} |
||||
, m_ioThread {new QThread(this)} |
||||
, m_asyncWorker {new TorrentFilesWatcher::Worker} |
||||
{ |
||||
connect(m_asyncWorker, &TorrentFilesWatcher::Worker::magnetFound, this, &TorrentFilesWatcher::onMagnetFound); |
||||
connect(m_asyncWorker, &TorrentFilesWatcher::Worker::torrentFound, this, &TorrentFilesWatcher::onTorrentFound); |
||||
|
||||
m_asyncWorker->moveToThread(m_ioThread); |
||||
m_ioThread->start(); |
||||
|
||||
load(); |
||||
} |
||||
|
||||
TorrentFilesWatcher::~TorrentFilesWatcher() |
||||
{ |
||||
m_ioThread->quit(); |
||||
m_ioThread->wait(); |
||||
delete m_asyncWorker; |
||||
} |
||||
|
||||
QString TorrentFilesWatcher::makeCleanPath(const QString &path) |
||||
{ |
||||
if (path.isEmpty()) |
||||
throw InvalidArgument(tr("Watched folder path cannot be empty.")); |
||||
|
||||
const QDir dir {path}; |
||||
if (dir.isRelative()) |
||||
throw InvalidArgument(tr("Watched folder path cannot be relative.")); |
||||
|
||||
return dir.canonicalPath(); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::load() |
||||
{ |
||||
QFile confFile {QDir(specialFolderLocation(SpecialFolder::Config)).absoluteFilePath(CONF_FILE_NAME)}; |
||||
if (!confFile.exists()) |
||||
{ |
||||
loadLegacy(); |
||||
return; |
||||
} |
||||
|
||||
if (!confFile.open(QFile::ReadOnly)) |
||||
{ |
||||
LogMsg(tr("Couldn't load Watched Folders configuration from %1. Error: %2") |
||||
.arg(confFile.fileName(), confFile.errorString()), Log::WARNING); |
||||
return; |
||||
} |
||||
|
||||
QJsonParseError jsonError; |
||||
const QJsonDocument jsonDoc = QJsonDocument::fromJson(confFile.readAll(), &jsonError); |
||||
if (jsonError.error != QJsonParseError::NoError) |
||||
{ |
||||
LogMsg(tr("Couldn't parse Watched Folders configuration from %1. Error: %2") |
||||
.arg(confFile.fileName(), jsonError.errorString()), Log::WARNING); |
||||
return; |
||||
} |
||||
|
||||
if (!jsonDoc.isObject()) |
||||
{ |
||||
LogMsg(tr("Couldn't load Watched Folders configuration from %1. Invalid data format.") |
||||
.arg(confFile.fileName()), Log::WARNING); |
||||
return; |
||||
} |
||||
|
||||
const QJsonObject jsonObj = jsonDoc.object(); |
||||
for (auto it = jsonObj.constBegin(); it != jsonObj.constEnd(); ++it) |
||||
{ |
||||
const QString &watchedFolder = it.key(); |
||||
const WatchedFolderOptions options = parseWatchedFolderOptions(it.value().toObject()); |
||||
try |
||||
{ |
||||
doSetWatchedFolder(watchedFolder, options); |
||||
} |
||||
catch (const InvalidArgument &err) |
||||
{ |
||||
LogMsg(err.message(), Log::WARNING); |
||||
} |
||||
} |
||||
} |
||||
|
||||
void TorrentFilesWatcher::loadLegacy() |
||||
{ |
||||
const auto dirs = SettingsStorage::instance()->loadValue<QVariantHash>("Preferences/Downloads/ScanDirsV2"); |
||||
|
||||
for (auto i = dirs.cbegin(); i != dirs.cend(); ++i) |
||||
{ |
||||
const QString watchedFolder = i.key(); |
||||
BitTorrent::AddTorrentParams params; |
||||
if (i.value().type() == QVariant::Int) |
||||
{ |
||||
if (i.value().toInt() == 0) |
||||
{ |
||||
params.savePath = watchedFolder; |
||||
params.useAutoTMM = false; |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
const QString customSavePath = i.value().toString(); |
||||
params.savePath = customSavePath; |
||||
params.useAutoTMM = false; |
||||
} |
||||
|
||||
try |
||||
{ |
||||
doSetWatchedFolder(watchedFolder, {params, false}); |
||||
} |
||||
catch (const InvalidArgument &err) |
||||
{ |
||||
LogMsg(err.message(), Log::WARNING); |
||||
} |
||||
} |
||||
|
||||
store(); |
||||
SettingsStorage::instance()->removeValue("Preferences/Downloads/ScanDirsV2"); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::store() const |
||||
{ |
||||
QJsonObject jsonObj; |
||||
for (auto it = m_watchedFolders.cbegin(); it != m_watchedFolders.cend(); ++it) |
||||
{ |
||||
const QString &watchedFolder = it.key(); |
||||
const WatchedFolderOptions &options = it.value(); |
||||
jsonObj[watchedFolder] = serializeWatchedFolderOptions(options); |
||||
} |
||||
|
||||
const QByteArray data = QJsonDocument(jsonObj).toJson(); |
||||
|
||||
QSaveFile confFile {QDir(specialFolderLocation(SpecialFolder::Config)).absoluteFilePath(CONF_FILE_NAME)}; |
||||
if (!confFile.open(QIODevice::WriteOnly) || (confFile.write(data) != data.size()) || !confFile.commit()) |
||||
{ |
||||
LogMsg(tr("Couldn't store Watched Folders configuration to %1. Error: %2") |
||||
.arg(confFile.fileName(), confFile.errorString()), Log::WARNING); |
||||
} |
||||
} |
||||
|
||||
QHash<QString, TorrentFilesWatcher::WatchedFolderOptions> TorrentFilesWatcher::folders() const |
||||
{ |
||||
return m_watchedFolders; |
||||
} |
||||
|
||||
void TorrentFilesWatcher::setWatchedFolder(const QString &path, const WatchedFolderOptions &options) |
||||
{ |
||||
doSetWatchedFolder(path, options); |
||||
store(); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::doSetWatchedFolder(const QString &path, const WatchedFolderOptions &options) |
||||
{ |
||||
const QString cleanPath = makeCleanPath(path); |
||||
m_watchedFolders[cleanPath] = options; |
||||
|
||||
QMetaObject::invokeMethod(m_asyncWorker, [this, path, options]() |
||||
{ |
||||
m_asyncWorker->setWatchedFolder(path, options); |
||||
}); |
||||
|
||||
emit watchedFolderSet(cleanPath, options); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::removeWatchedFolder(const QString &path) |
||||
{ |
||||
const QString cleanPath = makeCleanPath(path); |
||||
if (m_watchedFolders.remove(cleanPath)) |
||||
{ |
||||
QMetaObject::invokeMethod(m_asyncWorker, [this, cleanPath]() |
||||
{ |
||||
m_asyncWorker->removeWatchedFolder(cleanPath); |
||||
}); |
||||
|
||||
emit watchedFolderRemoved(cleanPath); |
||||
|
||||
store(); |
||||
} |
||||
} |
||||
|
||||
void TorrentFilesWatcher::onMagnetFound(const BitTorrent::MagnetUri &magnetURI |
||||
, const BitTorrent::AddTorrentParams &addTorrentParams) |
||||
{ |
||||
BitTorrent::Session::instance()->addTorrent(magnetURI, addTorrentParams); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::onTorrentFound(const BitTorrent::TorrentInfo &torrentInfo |
||||
, const BitTorrent::AddTorrentParams &addTorrentParams) |
||||
{ |
||||
BitTorrent::Session::instance()->addTorrent(torrentInfo, addTorrentParams); |
||||
} |
||||
|
||||
TorrentFilesWatcher::Worker::Worker() |
||||
: m_watcher {new QFileSystemWatcher(this)} |
||||
, m_watchTimer {new QTimer(this)} |
||||
, m_retryTorrentTimer {new QTimer(this)} |
||||
{ |
||||
connect(m_watcher, &QFileSystemWatcher::directoryChanged, this, &Worker::processWatchedFolder); |
||||
connect(m_watchTimer, &QTimer::timeout, this, &Worker::onTimeout); |
||||
|
||||
connect(m_retryTorrentTimer, &QTimer::timeout, this, &Worker::processFailedTorrents); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::onTimeout() |
||||
{ |
||||
for (const QString &path : asConst(m_watchedByTimeoutFolders)) |
||||
processWatchedFolder(path); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::setWatchedFolder(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
if (m_watchedFolders.contains(path)) |
||||
updateWatchedFolder(path, options); |
||||
else |
||||
addWatchedFolder(path, options); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::removeWatchedFolder(const QString &path) |
||||
{ |
||||
m_watchedFolders.remove(path); |
||||
|
||||
m_watcher->removePath(path); |
||||
m_watchedByTimeoutFolders.remove(path); |
||||
if (m_watchedByTimeoutFolders.isEmpty()) |
||||
m_watchTimer->stop(); |
||||
|
||||
m_failedTorrents.remove(path); |
||||
if (m_failedTorrents.isEmpty()) |
||||
m_retryTorrentTimer->stop(); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::processWatchedFolder(const QString &path) |
||||
{ |
||||
const TorrentFilesWatcher::WatchedFolderOptions options = m_watchedFolders.value(path); |
||||
processFolder(path, path, options); |
||||
|
||||
if (!m_failedTorrents.empty() && !m_retryTorrentTimer->isActive()) |
||||
m_retryTorrentTimer->start(WATCH_INTERVAL); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::processFolder(const QString &path, const QString &watchedFolderPath |
||||
, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
const QDir watchedDir {watchedFolderPath}; |
||||
|
||||
QDirIterator dirIter {path, {"*.torrent", "*.magnet"}, QDir::Files}; |
||||
while (dirIter.hasNext()) |
||||
{ |
||||
const QString filePath = dirIter.next(); |
||||
BitTorrent::AddTorrentParams addTorrentParams = options.addTorrentParams; |
||||
if (path != watchedFolderPath) |
||||
{ |
||||
const QString subdirPath = watchedDir.relativeFilePath(path); |
||||
addTorrentParams.savePath = QDir::cleanPath(QDir(addTorrentParams.savePath).filePath(subdirPath)); |
||||
} |
||||
|
||||
if (filePath.endsWith(QLatin1String(".magnet"), Qt::CaseInsensitive)) |
||||
{ |
||||
QFile file {filePath}; |
||||
if (file.open(QIODevice::ReadOnly | QIODevice::Text)) |
||||
{ |
||||
QTextStream str {&file}; |
||||
while (!str.atEnd()) |
||||
emit magnetFound(BitTorrent::MagnetUri(str.readLine()), addTorrentParams); |
||||
|
||||
file.close(); |
||||
Utils::Fs::forceRemove(filePath); |
||||
} |
||||
else |
||||
{ |
||||
LogMsg(tr("Failed to open magnet file: %1").arg(file.errorString())); |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
const auto torrentInfo = BitTorrent::TorrentInfo::loadFromFile(filePath); |
||||
if (torrentInfo.isValid()) |
||||
{ |
||||
emit torrentFound(torrentInfo, addTorrentParams); |
||||
Utils::Fs::forceRemove(filePath); |
||||
} |
||||
else |
||||
{ |
||||
if (!m_failedTorrents.value(path).contains(filePath)) |
||||
{ |
||||
m_failedTorrents[path][filePath] = 0; |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
if (options.recursive) |
||||
{ |
||||
QDirIterator dirIter {path, (QDir::Dirs | QDir::NoDot | QDir::NoDotDot)}; |
||||
while (dirIter.hasNext()) |
||||
{ |
||||
const QString folderPath = dirIter.next(); |
||||
// Skip processing of subdirectory that is explicitly set as watched folder
|
||||
if (!m_watchedFolders.contains(folderPath)) |
||||
processFolder(folderPath, watchedFolderPath, options); |
||||
} |
||||
} |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::processFailedTorrents() |
||||
{ |
||||
// Check which torrents are still partial
|
||||
Algorithm::removeIf(m_failedTorrents, [this](const QString &watchedFolderPath, QHash<QString, int> &partialTorrents) |
||||
{ |
||||
const QDir dir {watchedFolderPath}; |
||||
const TorrentFilesWatcher::WatchedFolderOptions options = m_watchedFolders.value(watchedFolderPath); |
||||
Algorithm::removeIf(partialTorrents, [this, &dir, &options](const QString &torrentPath, int &value) |
||||
{ |
||||
if (!QFile::exists(torrentPath)) |
||||
return true; |
||||
|
||||
const auto torrentInfo = BitTorrent::TorrentInfo::loadFromFile(torrentPath); |
||||
if (torrentInfo.isValid()) |
||||
{ |
||||
BitTorrent::AddTorrentParams addTorrentParams = options.addTorrentParams; |
||||
const QString exactDirPath = QFileInfo(torrentPath).canonicalPath(); |
||||
if (exactDirPath != dir.path()) |
||||
{ |
||||
const QString subdirPath = dir.relativeFilePath(exactDirPath); |
||||
addTorrentParams.savePath = QDir(addTorrentParams.savePath).filePath(subdirPath); |
||||
} |
||||
|
||||
emit torrentFound(torrentInfo, addTorrentParams); |
||||
Utils::Fs::forceRemove(torrentPath); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
if (value >= MAX_FAILED_RETRIES) |
||||
{ |
||||
LogMsg(tr("Rejecting failed torrent file: %1").arg(torrentPath)); |
||||
QFile::rename(torrentPath, torrentPath + ".qbt_rejected"); |
||||
return true; |
||||
} |
||||
|
||||
++value; |
||||
return false; |
||||
}); |
||||
|
||||
if (partialTorrents.isEmpty()) |
||||
return true; |
||||
|
||||
return false; |
||||
}); |
||||
|
||||
// Stop the partial timer if necessary
|
||||
if (m_failedTorrents.empty()) |
||||
m_retryTorrentTimer->stop(); |
||||
else |
||||
m_retryTorrentTimer->start(WATCH_INTERVAL); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::addWatchedFolder(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
#if !defined Q_OS_HAIKU |
||||
// Check if the path points to a network file system or not
|
||||
if (Utils::Fs::isNetworkFileSystem(path)) |
||||
{ |
||||
m_watchedByTimeoutFolders.insert(path); |
||||
} |
||||
else |
||||
#endif |
||||
if (options.recursive) |
||||
{ |
||||
m_watchedByTimeoutFolders.insert(path); |
||||
if (!m_watchTimer->isActive()) |
||||
m_watchTimer->start(WATCH_INTERVAL); |
||||
} |
||||
else |
||||
{ |
||||
m_watcher->addPath(path); |
||||
QTimer::singleShot(2000, this, [this, path]() { processWatchedFolder(path); }); |
||||
} |
||||
|
||||
m_watchedFolders[path] = options; |
||||
|
||||
LogMsg(tr("Watching folder: \"%1\"").arg(Utils::Fs::toNativePath(path))); |
||||
} |
||||
|
||||
void TorrentFilesWatcher::Worker::updateWatchedFolder(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
const bool recursiveModeChanged = (m_watchedFolders[path].recursive != options.recursive); |
||||
#if !defined Q_OS_HAIKU |
||||
if (recursiveModeChanged && !Utils::Fs::isNetworkFileSystem(path)) |
||||
#else |
||||
if (recursiveModeChanged) |
||||
#endif |
||||
{ |
||||
if (options.recursive) |
||||
{ |
||||
m_watcher->removePath(path); |
||||
|
||||
m_watchedByTimeoutFolders.insert(path); |
||||
if (!m_watchTimer->isActive()) |
||||
m_watchTimer->start(WATCH_INTERVAL); |
||||
} |
||||
else |
||||
{ |
||||
m_watchedByTimeoutFolders.remove(path); |
||||
if (m_watchedByTimeoutFolders.isEmpty()) |
||||
m_watchTimer->stop(); |
||||
|
||||
m_watcher->addPath(path); |
||||
QTimer::singleShot(2000, this, [this, path]() { processWatchedFolder(path); }); |
||||
} |
||||
} |
||||
|
||||
m_watchedFolders[path] = options; |
||||
} |
||||
|
||||
#include "torrentfileswatcher.moc" |
||||
@ -0,0 +1,96 @@
@@ -0,0 +1,96 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2021 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2010 Christian Kandeler, Christophe Dumez <chris@qbittorrent.org> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#pragma once |
||||
|
||||
#include <QHash> |
||||
|
||||
#include "base/bittorrent/addtorrentparams.h" |
||||
|
||||
class QThread; |
||||
|
||||
namespace BitTorrent |
||||
{ |
||||
class MagnetUri; |
||||
} |
||||
|
||||
/*
|
||||
* Watches the configured directories for new .torrent files in order |
||||
* to add torrents to BitTorrent session. Supports Network File System |
||||
* watching (NFS, CIFS) on Linux and Mac OS. |
||||
*/ |
||||
class TorrentFilesWatcher final : public QObject |
||||
{ |
||||
Q_OBJECT |
||||
Q_DISABLE_COPY(TorrentFilesWatcher) |
||||
|
||||
public: |
||||
struct WatchedFolderOptions |
||||
{ |
||||
BitTorrent::AddTorrentParams addTorrentParams; |
||||
bool recursive = false; |
||||
}; |
||||
|
||||
static void initInstance(); |
||||
static void freeInstance(); |
||||
static TorrentFilesWatcher *instance(); |
||||
|
||||
static QString makeCleanPath(const QString &path); |
||||
|
||||
QHash<QString, WatchedFolderOptions> folders() const; |
||||
void setWatchedFolder(const QString &path, const WatchedFolderOptions &options); |
||||
void removeWatchedFolder(const QString &path); |
||||
|
||||
signals: |
||||
void watchedFolderSet(const QString &path, const WatchedFolderOptions &options); |
||||
void watchedFolderRemoved(const QString &path); |
||||
|
||||
private slots: |
||||
void onMagnetFound(const BitTorrent::MagnetUri &magnetURI, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||
void onTorrentFound(const BitTorrent::TorrentInfo &torrentInfo, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||
|
||||
private: |
||||
explicit TorrentFilesWatcher(QObject *parent = nullptr); |
||||
~TorrentFilesWatcher() override; |
||||
|
||||
void load(); |
||||
void loadLegacy(); |
||||
void store() const; |
||||
|
||||
void doSetWatchedFolder(const QString &path, const WatchedFolderOptions &options); |
||||
|
||||
static TorrentFilesWatcher *m_instance; |
||||
|
||||
QHash<QString, WatchedFolderOptions> m_watchedFolders; |
||||
|
||||
QThread *m_ioThread = nullptr; |
||||
|
||||
class Worker; |
||||
Worker *m_asyncWorker = nullptr; |
||||
}; |
||||
@ -1,118 +0,0 @@
@@ -1,118 +0,0 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2015 sledgehammer999 <hammered999@gmail.com> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "scanfoldersdelegate.h" |
||||
|
||||
#include <QComboBox> |
||||
#include <QDebug> |
||||
#include <QFileDialog> |
||||
#include <QTreeView> |
||||
|
||||
#include "base/bittorrent/session.h" |
||||
#include "base/scanfoldersmodel.h" |
||||
|
||||
ScanFoldersDelegate::ScanFoldersDelegate(QObject *parent, QTreeView *foldersView) |
||||
: QStyledItemDelegate(parent) |
||||
, m_folderView(foldersView) |
||||
{ |
||||
} |
||||
|
||||
void ScanFoldersDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const |
||||
{ |
||||
auto *combobox = static_cast<QComboBox*>(editor); |
||||
// Set combobox index
|
||||
if (index.data(Qt::UserRole).toInt() == ScanFoldersModel::CUSTOM_LOCATION) |
||||
combobox->setCurrentIndex(4); // '4' is the index of the item after the separator in the QComboBox menu
|
||||
else |
||||
combobox->setCurrentIndex(index.data(Qt::UserRole).toInt()); |
||||
} |
||||
|
||||
QWidget *ScanFoldersDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &, const QModelIndex &index) const |
||||
{ |
||||
if (index.column() != ScanFoldersModel::DOWNLOAD) return nullptr; |
||||
|
||||
auto *editor = new QComboBox(parent); |
||||
|
||||
editor->setFocusPolicy(Qt::StrongFocus); |
||||
editor->addItem(ScanFoldersModel::pathTypeDisplayName(ScanFoldersModel::DOWNLOAD_IN_WATCH_FOLDER)); |
||||
editor->addItem(ScanFoldersModel::pathTypeDisplayName(ScanFoldersModel::DEFAULT_LOCATION)); |
||||
editor->addItem(ScanFoldersModel::pathTypeDisplayName(ScanFoldersModel::CUSTOM_LOCATION)); |
||||
if (index.data(Qt::UserRole).toInt() == ScanFoldersModel::CUSTOM_LOCATION) |
||||
{ |
||||
editor->insertSeparator(3); |
||||
editor->addItem(index.data().toString()); |
||||
} |
||||
|
||||
connect(editor, qOverload<int>(&QComboBox::currentIndexChanged) |
||||
, this, &ScanFoldersDelegate::comboboxIndexChanged); |
||||
return editor; |
||||
} |
||||
|
||||
void ScanFoldersDelegate::comboboxIndexChanged(int index) |
||||
{ |
||||
if (index == ScanFoldersModel::CUSTOM_LOCATION) |
||||
{ |
||||
auto *w = static_cast<QWidget *>(sender()); |
||||
if (w && w->parentWidget()) |
||||
w->parentWidget()->setFocus(); |
||||
} |
||||
} |
||||
|
||||
void ScanFoldersDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const |
||||
{ |
||||
auto *combobox = static_cast<QComboBox*>(editor); |
||||
int value = combobox->currentIndex(); |
||||
|
||||
switch (value) |
||||
{ |
||||
case ScanFoldersModel::DOWNLOAD_IN_WATCH_FOLDER: |
||||
case ScanFoldersModel::DEFAULT_LOCATION: |
||||
model->setData(index, value, Qt::UserRole); |
||||
break; |
||||
|
||||
case ScanFoldersModel::CUSTOM_LOCATION: |
||||
model->setData( |
||||
index, |
||||
QFileDialog::getExistingDirectory( |
||||
nullptr, tr("Select save location"), |
||||
index.data(Qt::UserRole).toInt() == ScanFoldersModel::CUSTOM_LOCATION ? |
||||
index.data().toString() : |
||||
BitTorrent::Session::instance()->defaultSavePath()), |
||||
Qt::DisplayRole); |
||||
break; |
||||
|
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
|
||||
void ScanFoldersDelegate::updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option, const QModelIndex &) const |
||||
{ |
||||
qDebug("UpdateEditor Geometry called"); |
||||
editor->setGeometry(option.rect); |
||||
} |
||||
@ -0,0 +1,153 @@
@@ -0,0 +1,153 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2021 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "watchedfolderoptionsdialog.h" |
||||
|
||||
#include <QDir> |
||||
#include <QPushButton> |
||||
|
||||
#include "base/bittorrent/session.h" |
||||
#include "base/global.h" |
||||
#include "base/settingsstorage.h" |
||||
#include "base/utils/fs.h" |
||||
#include "ui_watchedfolderoptionsdialog.h" |
||||
#include "utils.h" |
||||
|
||||
#define SETTINGS_KEY(name) "WatchedFolderOptionsDialog/" name |
||||
|
||||
WatchedFolderOptionsDialog::WatchedFolderOptionsDialog( |
||||
const TorrentFilesWatcher::WatchedFolderOptions &watchedFolderOptions, QWidget *parent) |
||||
: QDialog {parent} |
||||
, m_ui {new Ui::WatchedFolderOptionsDialog} |
||||
, m_savePath {watchedFolderOptions.addTorrentParams.savePath} |
||||
, m_storeDialogSize {SETTINGS_KEY("DialogSize")} |
||||
{ |
||||
m_ui->setupUi(this); |
||||
|
||||
m_ui->savePath->setMode(FileSystemPathEdit::Mode::DirectorySave); |
||||
m_ui->savePath->setDialogCaption(tr("Choose save path")); |
||||
|
||||
connect(m_ui->comboTTM, qOverload<int>(&QComboBox::currentIndexChanged), this, &WatchedFolderOptionsDialog::onTMMChanged); |
||||
connect(m_ui->categoryComboBox, qOverload<int>(&QComboBox::currentIndexChanged), this, &WatchedFolderOptionsDialog::onCategoryChanged); |
||||
|
||||
m_ui->checkBoxRecursive->setChecked(watchedFolderOptions.recursive); |
||||
populateSavePathComboBox(); |
||||
|
||||
const auto *session = BitTorrent::Session::instance(); |
||||
const BitTorrent::AddTorrentParams &torrentParams = watchedFolderOptions.addTorrentParams; |
||||
m_ui->startTorrentCheckBox->setChecked(!torrentParams.addPaused.value_or(session->isAddTorrentPaused())); |
||||
m_ui->comboTTM->setCurrentIndex(torrentParams.useAutoTMM.value_or(!session->isAutoTMMDisabledByDefault())); |
||||
m_ui->contentLayoutComboBox->setCurrentIndex( |
||||
static_cast<int>(torrentParams.contentLayout.value_or(session->torrentContentLayout()))); |
||||
|
||||
// Load categories
|
||||
QStringList categories = session->categories().keys(); |
||||
std::sort(categories.begin(), categories.end(), Utils::Compare::NaturalLessThan<Qt::CaseInsensitive>()); |
||||
|
||||
if (!torrentParams.category.isEmpty()) |
||||
m_ui->categoryComboBox->addItem(torrentParams.category); |
||||
m_ui->categoryComboBox->addItem(""); |
||||
|
||||
for (const QString &category : asConst(categories)) |
||||
{ |
||||
if (category != torrentParams.category) |
||||
m_ui->categoryComboBox->addItem(category); |
||||
} |
||||
|
||||
loadState(); |
||||
|
||||
m_ui->buttonBox->button(QDialogButtonBox::Ok)->setFocus(); |
||||
} |
||||
|
||||
WatchedFolderOptionsDialog::~WatchedFolderOptionsDialog() |
||||
{ |
||||
saveState(); |
||||
delete m_ui; |
||||
} |
||||
|
||||
TorrentFilesWatcher::WatchedFolderOptions WatchedFolderOptionsDialog::watchedFolderOptions() const |
||||
{ |
||||
TorrentFilesWatcher::WatchedFolderOptions watchedFolderOptions; |
||||
watchedFolderOptions.recursive = m_ui->checkBoxRecursive->isChecked(); |
||||
|
||||
BitTorrent::AddTorrentParams ¶ms = watchedFolderOptions.addTorrentParams; |
||||
params.useAutoTMM = (m_ui->comboTTM->currentIndex() == 1); |
||||
if (!*params.useAutoTMM) |
||||
params.savePath = m_ui->savePath->selectedPath(); |
||||
params.category = m_ui->categoryComboBox->currentText();; |
||||
params.addPaused = !m_ui->startTorrentCheckBox->isChecked(); |
||||
params.contentLayout = static_cast<BitTorrent::TorrentContentLayout>(m_ui->contentLayoutComboBox->currentIndex()); |
||||
|
||||
return watchedFolderOptions; |
||||
} |
||||
|
||||
void WatchedFolderOptionsDialog::loadState() |
||||
{ |
||||
Utils::Gui::resize(this, m_storeDialogSize); |
||||
} |
||||
|
||||
void WatchedFolderOptionsDialog::saveState() |
||||
{ |
||||
m_storeDialogSize = size(); |
||||
} |
||||
|
||||
void WatchedFolderOptionsDialog::onCategoryChanged(const int index) |
||||
{ |
||||
Q_UNUSED(index); |
||||
|
||||
const QString category = m_ui->categoryComboBox->currentText(); |
||||
if (m_ui->comboTTM->currentIndex() == 1) |
||||
{ |
||||
const QString savePath = BitTorrent::Session::instance()->categorySavePath(category); |
||||
m_ui->savePath->setSelectedPath(Utils::Fs::toNativePath(savePath)); |
||||
} |
||||
} |
||||
|
||||
void WatchedFolderOptionsDialog::populateSavePathComboBox() |
||||
{ |
||||
const QString defSavePath {BitTorrent::Session::instance()->defaultSavePath()}; |
||||
m_ui->savePath->setSelectedPath(!m_savePath.isEmpty() ? m_savePath : defSavePath); |
||||
} |
||||
|
||||
void WatchedFolderOptionsDialog::onTMMChanged(const int index) |
||||
{ |
||||
if (index != 1) |
||||
{ // 0 is Manual mode and 1 is Automatic mode. Handle all non 1 values as manual mode.
|
||||
populateSavePathComboBox(); |
||||
m_ui->groupBoxSavePath->setEnabled(true); |
||||
m_ui->savePath->blockSignals(false); |
||||
} |
||||
else |
||||
{ |
||||
m_ui->groupBoxSavePath->setEnabled(false); |
||||
m_ui->savePath->blockSignals(true); |
||||
m_savePath = m_ui->savePath->selectedPath(); |
||||
const QString savePath = BitTorrent::Session::instance()->categorySavePath(m_ui->categoryComboBox->currentText()); |
||||
m_ui->savePath->setSelectedPath(savePath); |
||||
} |
||||
} |
||||
@ -0,0 +1,315 @@
@@ -0,0 +1,315 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?> |
||||
<ui version="4.0"> |
||||
<class>WatchedFolderOptionsDialog</class> |
||||
<widget class="QDialog" name="WatchedFolderOptionsDialog"> |
||||
<property name="geometry"> |
||||
<rect> |
||||
<x>0</x> |
||||
<y>0</y> |
||||
<width>462</width> |
||||
<height>306</height> |
||||
</rect> |
||||
</property> |
||||
<property name="windowTitle"> |
||||
<string>Watched Folder Options</string> |
||||
</property> |
||||
<layout class="QVBoxLayout" name="verticalLayout_3"> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="horizontalLayout_6"> |
||||
<item> |
||||
<widget class="QCheckBox" name="checkBoxRecursive"> |
||||
<property name="toolTip"> |
||||
<string><html><head/><body><p>Will watch the folder and all its subfolders. In Manual torrent management mode it will also add subfolder name to the selected Save path.</p></body></html></string> |
||||
</property> |
||||
<property name="text"> |
||||
<string>Recursive mode</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="horizontalSpacer_2"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>40</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<widget class="QSplitter" name="splitter"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="childrenCollapsible"> |
||||
<bool>false</bool> |
||||
</property> |
||||
<widget class="QFrame" name="torrentoptionsFrame"> |
||||
<layout class="QVBoxLayout" name="mainlayout_addui"> |
||||
<property name="leftMargin"> |
||||
<number>0</number> |
||||
</property> |
||||
<property name="topMargin"> |
||||
<number>0</number> |
||||
</property> |
||||
<property name="rightMargin"> |
||||
<number>0</number> |
||||
</property> |
||||
<property name="bottomMargin"> |
||||
<number>0</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QGroupBox" name="groupBoxParameters"> |
||||
<property name="title"> |
||||
<string>Torrent parameters</string> |
||||
</property> |
||||
<layout class="QVBoxLayout" name="verticalLayout_2"> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="managementLayout"> |
||||
<item> |
||||
<widget class="QLabel" name="labelTorrentManagementMode"> |
||||
<property name="text"> |
||||
<string>Torrent Management Mode:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QComboBox" name="comboTTM"> |
||||
<property name="toolTip"> |
||||
<string>Automatic mode means that various torrent properties(eg save path) will be decided by the associated category</string> |
||||
</property> |
||||
<item> |
||||
<property name="text"> |
||||
<string>Manual</string> |
||||
</property> |
||||
</item> |
||||
<item> |
||||
<property name="text"> |
||||
<string>Automatic</string> |
||||
</property> |
||||
</item> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="horizontalSpacer"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>20</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<widget class="QGroupBox" name="groupBoxSavePath"> |
||||
<property name="title"> |
||||
<string>Save at</string> |
||||
</property> |
||||
<layout class="QVBoxLayout" name="verticalLayout"> |
||||
<item> |
||||
<widget class="FileSystemPathLineEdit" name="savePath" native="true"/> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="categoryLayout"> |
||||
<item> |
||||
<widget class="QLabel" name="labelCategory"> |
||||
<property name="text"> |
||||
<string>Category:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QComboBox" name="categoryComboBox"> |
||||
<property name="sizePolicy"> |
||||
<sizepolicy hsizetype="Expanding" vsizetype="Fixed"> |
||||
<horstretch>0</horstretch> |
||||
<verstretch>0</verstretch> |
||||
</sizepolicy> |
||||
</property> |
||||
<property name="editable"> |
||||
<bool>true</bool> |
||||
</property> |
||||
<property name="insertPolicy"> |
||||
<enum>QComboBox::InsertAtTop</enum> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="horizontalSpacer_7"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>40</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<layout class="QVBoxLayout" name="verticalLayout_4"> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="horizontalLayout_3"> |
||||
<item> |
||||
<widget class="QCheckBox" name="startTorrentCheckBox"> |
||||
<property name="text"> |
||||
<string>Start torrent</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="horizontalSpacer_3"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>40</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="horizontalLayout_2"> |
||||
<item> |
||||
<widget class="QLabel" name="contentLayoutLabel"> |
||||
<property name="text"> |
||||
<string>Content layout:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QComboBox" name="contentLayoutComboBox"> |
||||
<property name="currentIndex"> |
||||
<number>0</number> |
||||
</property> |
||||
<item> |
||||
<property name="text"> |
||||
<string>Original</string> |
||||
</property> |
||||
</item> |
||||
<item> |
||||
<property name="text"> |
||||
<string>Create subfolder</string> |
||||
</property> |
||||
</item> |
||||
<item> |
||||
<property name="text"> |
||||
<string>Don't create subfolder</string> |
||||
</property> |
||||
</item> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="horizontalSpacer_4"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>40</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer name="verticalSpacer"> |
||||
<property name="orientation"> |
||||
<enum>Qt::Vertical</enum> |
||||
</property> |
||||
<property name="sizeHint" stdset="0"> |
||||
<size> |
||||
<width>20</width> |
||||
<height>40</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
<item> |
||||
<layout class="QHBoxLayout" name="horizontalLayout"> |
||||
<item> |
||||
<widget class="QDialogButtonBox" name="buttonBox"> |
||||
<property name="standardButtons"> |
||||
<set>QDialogButtonBox::Cancel|QDialogButtonBox::Ok</set> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
<customwidgets> |
||||
<customwidget> |
||||
<class>FileSystemPathLineEdit</class> |
||||
<extends>QWidget</extends> |
||||
<header>gui/fspathedit.h</header> |
||||
<container>1</container> |
||||
</customwidget> |
||||
</customwidgets> |
||||
<resources/> |
||||
<connections> |
||||
<connection> |
||||
<sender>buttonBox</sender> |
||||
<signal>accepted()</signal> |
||||
<receiver>WatchedFolderOptionsDialog</receiver> |
||||
<slot>accept()</slot> |
||||
<hints> |
||||
<hint type="sourcelabel"> |
||||
<x>928</x> |
||||
<y>855</y> |
||||
</hint> |
||||
<hint type="destinationlabel"> |
||||
<x>157</x> |
||||
<y>274</y> |
||||
</hint> |
||||
</hints> |
||||
</connection> |
||||
<connection> |
||||
<sender>buttonBox</sender> |
||||
<signal>rejected()</signal> |
||||
<receiver>WatchedFolderOptionsDialog</receiver> |
||||
<slot>reject()</slot> |
||||
<hints> |
||||
<hint type="sourcelabel"> |
||||
<x>928</x> |
||||
<y>855</y> |
||||
</hint> |
||||
<hint type="destinationlabel"> |
||||
<x>286</x> |
||||
<y>274</y> |
||||
</hint> |
||||
</hints> |
||||
</connection> |
||||
</connections> |
||||
</ui> |
||||
@ -0,0 +1,178 @@
@@ -0,0 +1,178 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2021 Vladimir Golovnev <glassez@yandex.ru> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include "watchedfoldersmodel.h" |
||||
|
||||
#include <QDir> |
||||
|
||||
#include "base/exceptions.h" |
||||
#include "base/global.h" |
||||
#include "base/utils/fs.h" |
||||
|
||||
WatchedFoldersModel::WatchedFoldersModel(TorrentFilesWatcher *fsWatcher, QObject *parent) |
||||
: QAbstractListModel {parent} |
||||
, m_fsWatcher {fsWatcher} |
||||
, m_watchedFolders {m_fsWatcher->folders().keys()} |
||||
, m_watchedFoldersOptions {m_fsWatcher->folders()} |
||||
{ |
||||
connect(m_fsWatcher, &TorrentFilesWatcher::watchedFolderSet, this, &WatchedFoldersModel::onFolderSet); |
||||
connect(m_fsWatcher, &TorrentFilesWatcher::watchedFolderRemoved, this, &WatchedFoldersModel::onFolderRemoved); |
||||
} |
||||
|
||||
int WatchedFoldersModel::rowCount(const QModelIndex &parent) const |
||||
{ |
||||
return parent.isValid() ? 0 : m_watchedFolders.count(); |
||||
} |
||||
|
||||
int WatchedFoldersModel::columnCount(const QModelIndex &parent) const |
||||
{ |
||||
Q_UNUSED(parent); |
||||
return 1; |
||||
} |
||||
|
||||
QVariant WatchedFoldersModel::data(const QModelIndex &index, const int role) const |
||||
{ |
||||
if (!index.isValid() || (index.row() >= rowCount()) || (index.column() >= columnCount())) |
||||
return {}; |
||||
|
||||
if (role == Qt::DisplayRole) |
||||
return Utils::Fs::toNativePath(m_watchedFolders.at(index.row())); |
||||
|
||||
return {}; |
||||
} |
||||
|
||||
QVariant WatchedFoldersModel::headerData(const int section, const Qt::Orientation orientation, const int role) const |
||||
{ |
||||
if ((orientation != Qt::Horizontal) || (role != Qt::DisplayRole) |
||||
|| (section < 0) || (section >= columnCount())) |
||||
{ |
||||
return {}; |
||||
} |
||||
|
||||
return tr("Watched Folder"); |
||||
} |
||||
|
||||
bool WatchedFoldersModel::removeRows(const int row, const int count, const QModelIndex &parent) |
||||
{ |
||||
if (parent.isValid() || (row < 0) || (row >= rowCount()) |
||||
|| (count <= 0) || ((row + count) > rowCount())) |
||||
{ |
||||
return false; |
||||
} |
||||
|
||||
const int firstRow = row; |
||||
const int lastRow = row + (count - 1); |
||||
|
||||
beginRemoveRows(parent, firstRow, lastRow); |
||||
for (int i = firstRow; i <= lastRow; ++i) |
||||
{ |
||||
const QString folderPath = m_watchedFolders.takeAt(i); |
||||
m_watchedFoldersOptions.remove(folderPath); |
||||
m_deletedFolders.insert(folderPath); |
||||
} |
||||
endRemoveRows(); |
||||
|
||||
return true; |
||||
} |
||||
|
||||
void WatchedFoldersModel::addFolder(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
if (path.isEmpty()) |
||||
throw InvalidArgument(tr("Watched folder path cannot be empty.")); |
||||
|
||||
const QDir watchDir {path}; |
||||
const QString canonicalWatchPath = watchDir.canonicalPath(); |
||||
if (m_watchedFoldersOptions.contains(canonicalWatchPath)) |
||||
throw RuntimeError(tr("Folder '%1' is already in watch list.").arg(path)); |
||||
if (!watchDir.exists()) |
||||
throw RuntimeError(tr("Folder '%1' doesn't exist.").arg(path)); |
||||
if (!watchDir.isReadable()) |
||||
throw RuntimeError(tr("Folder '%1' isn't readable.").arg(path)); |
||||
|
||||
m_deletedFolders.remove(canonicalWatchPath); |
||||
|
||||
beginInsertRows(QModelIndex(), rowCount(), rowCount()); |
||||
m_watchedFolders.append(canonicalWatchPath); |
||||
m_watchedFoldersOptions[canonicalWatchPath] = options; |
||||
endInsertRows(); |
||||
} |
||||
|
||||
TorrentFilesWatcher::WatchedFolderOptions WatchedFoldersModel::folderOptions(const int row) const |
||||
{ |
||||
Q_ASSERT((row >= 0) && (row < rowCount())); |
||||
|
||||
const QString folderPath = m_watchedFolders.at(row); |
||||
return m_watchedFoldersOptions[folderPath]; |
||||
} |
||||
|
||||
void WatchedFoldersModel::setFolderOptions(const int row, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
Q_ASSERT((row >= 0) && (row < rowCount())); |
||||
|
||||
const QString folderPath = m_watchedFolders.at(row); |
||||
m_watchedFoldersOptions[folderPath] = options; |
||||
} |
||||
|
||||
void WatchedFoldersModel::apply() |
||||
{ |
||||
const QSet<QString> deletedFolders {m_deletedFolders}; |
||||
// We have to clear `m_deletedFolders` for optimization reason, otherwise
|
||||
// it will be cleared one element at a time in `onFolderRemoved()` handler
|
||||
m_deletedFolders.clear(); |
||||
for (const QString &path : deletedFolders) |
||||
m_fsWatcher->removeWatchedFolder(path); |
||||
|
||||
for (const QString &path : asConst(m_watchedFolders)) |
||||
m_fsWatcher->setWatchedFolder(path, m_watchedFoldersOptions.value(path)); |
||||
} |
||||
|
||||
void WatchedFoldersModel::onFolderSet(const QString &path, const TorrentFilesWatcher::WatchedFolderOptions &options) |
||||
{ |
||||
if (!m_watchedFoldersOptions.contains(path)) |
||||
{ |
||||
m_deletedFolders.remove(path); |
||||
|
||||
beginInsertRows(QModelIndex(), rowCount(), rowCount()); |
||||
m_watchedFolders.append(path); |
||||
m_watchedFoldersOptions[path] = options; |
||||
endInsertRows(); |
||||
} |
||||
else |
||||
{ |
||||
m_watchedFoldersOptions[path] = options; |
||||
} |
||||
} |
||||
|
||||
void WatchedFoldersModel::onFolderRemoved(const QString &path) |
||||
{ |
||||
const int row = m_watchedFolders.indexOf(path); |
||||
if (row >= 0) |
||||
removeRows(row, 1); |
||||
|
||||
m_deletedFolders.remove(path); |
||||
} |
||||
Loading…
Reference in new issue