Some work about adaptive color scheme for Web UI (PR #19901)
http://[316:c51a:62a3:8b9::4]/d4708/qBittorrent/src/branch/adaptive-webui
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
682 lines
23 KiB
682 lines
23 KiB
/* |
|
* 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 <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/io.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_SKIPCHECKING {QStringLiteral("skip_checking")}; |
|
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.skipChecking = jsonObj.value(PARAM_SKIPCHECKING).toBool(); |
|
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_SKIPCHECKING, params.skipChecking}, |
|
{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_MOVE(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 scheduleWatchedFolderProcessing(const QString &path); |
|
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.")); |
|
|
|
if (QDir::isRelativePath(path)) |
|
throw InvalidArgument(tr("Watched folder path cannot be relative.")); |
|
|
|
return QDir::cleanPath(path); |
|
} |
|
|
|
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 QString path = QDir(specialFolderLocation(SpecialFolder::Config)).absoluteFilePath(CONF_FILE_NAME); |
|
const QByteArray data = QJsonDocument(jsonObj).toJson(); |
|
const nonstd::expected<void, QString> result = Utils::IO::saveToFile(path, data); |
|
if (!result) |
|
{ |
|
LogMsg(tr("Couldn't store Watched Folders configuration to %1. Error: %2") |
|
.arg(path, result.error()), 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::scheduleWatchedFolderProcessing); |
|
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::scheduleWatchedFolderProcessing(const QString &path) |
|
{ |
|
QTimer::singleShot(2000, this, [this, path]() |
|
{ |
|
processWatchedFolder(path); |
|
}); |
|
} |
|
|
|
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); |
|
if (addTorrentParams.useAutoTMM) |
|
{ |
|
addTorrentParams.category = addTorrentParams.category.isEmpty() |
|
? subdirPath : (addTorrentParams.category + QLatin1Char('/') + subdirPath); |
|
} |
|
else |
|
{ |
|
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 nonstd::expected<BitTorrent::TorrentInfo, QString> result = BitTorrent::TorrentInfo::loadFromFile(filePath); |
|
if (result) |
|
{ |
|
emit torrentFound(result.value(), 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 nonstd::expected<BitTorrent::TorrentInfo, QString> result = BitTorrent::TorrentInfo::loadFromFile(torrentPath); |
|
if (result) |
|
{ |
|
BitTorrent::AddTorrentParams addTorrentParams = options.addTorrentParams; |
|
const QString exactDirPath = QFileInfo(torrentPath).canonicalPath(); |
|
if (exactDirPath != dir.path()) |
|
{ |
|
const QString subdirPath = dir.relativeFilePath(exactDirPath); |
|
if (addTorrentParams.useAutoTMM) |
|
{ |
|
addTorrentParams.category = addTorrentParams.category.isEmpty() |
|
? subdirPath : (addTorrentParams.category + QLatin1Char('/') + subdirPath); |
|
} |
|
else |
|
{ |
|
addTorrentParams.savePath = QDir(addTorrentParams.savePath).filePath(subdirPath); |
|
} |
|
} |
|
|
|
emit torrentFound(result.value(), 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) || options.recursive) |
|
#else |
|
if (options.recursive) |
|
#endif |
|
{ |
|
m_watchedByTimeoutFolders.insert(path); |
|
if (!m_watchTimer->isActive()) |
|
m_watchTimer->start(WATCH_INTERVAL); |
|
} |
|
else |
|
{ |
|
m_watcher->addPath(path); |
|
scheduleWatchedFolderProcessing(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); |
|
scheduleWatchedFolderProcessing(path); |
|
} |
|
} |
|
|
|
m_watchedFolders[path] = options; |
|
} |
|
|
|
#include "torrentfileswatcher.moc"
|
|
|