|
|
@ -1,6 +1,6 @@ |
|
|
|
/*
|
|
|
|
/*
|
|
|
|
* Bittorrent Client using Qt and libtorrent. |
|
|
|
* Bittorrent Client using Qt and libtorrent. |
|
|
|
* Copyright (C) 2017 Vladimir Golovnev <glassez@yandex.ru> |
|
|
|
* Copyright (C) 2017-2023 Vladimir Golovnev <glassez@yandex.ru> |
|
|
|
* |
|
|
|
* |
|
|
|
* This program is free software; you can redistribute it and/or |
|
|
|
* This program is free software; you can redistribute it and/or |
|
|
|
* modify it under the terms of the GNU General Public License |
|
|
|
* modify it under the terms of the GNU General Public License |
|
|
@ -133,6 +133,8 @@ AutoDownloader::AutoDownloader() |
|
|
|
|
|
|
|
|
|
|
|
load(); |
|
|
|
load(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
connect(Session::instance(), &Session::feedURLChanged, this, &AutoDownloader::handleFeedURLChanged); |
|
|
|
|
|
|
|
|
|
|
|
m_processingTimer->setSingleShot(true); |
|
|
|
m_processingTimer->setSingleShot(true); |
|
|
|
connect(m_processingTimer, &QTimer::timeout, this, &AutoDownloader::process); |
|
|
|
connect(m_processingTimer, &QTimer::timeout, this, &AutoDownloader::process); |
|
|
|
|
|
|
|
|
|
|
@ -331,22 +333,28 @@ void AutoDownloader::setDownloadRepacks(const bool enabled) |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::process() |
|
|
|
void AutoDownloader::process() |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (m_processingQueue.isEmpty()) return; // processing was disabled
|
|
|
|
if (m_processingQueue.isEmpty()) // processing was disabled
|
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
|
|
processJob(m_processingQueue.takeFirst()); |
|
|
|
processJob(m_processingQueue.takeFirst()); |
|
|
|
if (!m_processingQueue.isEmpty()) |
|
|
|
if (!m_processingQueue.isEmpty()) |
|
|
|
|
|
|
|
{ |
|
|
|
// Schedule to process the next torrent (if any)
|
|
|
|
// Schedule to process the next torrent (if any)
|
|
|
|
m_processingTimer->start(); |
|
|
|
m_processingTimer->start(); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::handleTorrentDownloadFinished(const QString &url) |
|
|
|
void AutoDownloader::handleTorrentDownloadFinished(const QString &url) |
|
|
|
{ |
|
|
|
{ |
|
|
|
const auto job = m_waitingJobs.take(url); |
|
|
|
const auto job = m_waitingJobs.take(url); |
|
|
|
if (!job) return; |
|
|
|
if (!job) |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
|
|
if (Feed *feed = Session::instance()->feedByURL(job->feedURL)) |
|
|
|
if (Feed *feed = Session::instance()->feedByURL(job->feedURL)) |
|
|
|
|
|
|
|
{ |
|
|
|
if (Article *article = feed->articleByGUID(job->articleData.value(Article::KeyId).toString())) |
|
|
|
if (Article *article = feed->articleByGUID(job->articleData.value(Article::KeyId).toString())) |
|
|
|
article->markAsRead(); |
|
|
|
article->markAsRead(); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::handleTorrentDownloadFailed(const QString &url) |
|
|
|
void AutoDownloader::handleTorrentDownloadFailed(const QString &url) |
|
|
@ -361,6 +369,34 @@ void AutoDownloader::handleNewArticle(const Article *article) |
|
|
|
addJobForArticle(article); |
|
|
|
addJobForArticle(article); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::handleFeedURLChanged(Feed *feed, const QString &oldURL) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
for (AutoDownloadRule &rule : m_rules) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
if (const auto i = rule.feedURLs().indexOf(oldURL); i >= 0) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
auto feedURLs = rule.feedURLs(); |
|
|
|
|
|
|
|
feedURLs.replace(i, feed->url()); |
|
|
|
|
|
|
|
rule.setFeedURLs(feedURLs); |
|
|
|
|
|
|
|
m_dirty = true; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for (QSharedPointer<ProcessingJob> job : asConst(m_processingQueue)) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
if (job->feedURL == oldURL) |
|
|
|
|
|
|
|
job->feedURL = feed->url(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
for (QSharedPointer<ProcessingJob> job : asConst(m_waitingJobs)) |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
if (job->feedURL == oldURL) |
|
|
|
|
|
|
|
job->feedURL = feed->url(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
store(); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::setRule_impl(const AutoDownloadRule &rule) |
|
|
|
void AutoDownloader::setRule_impl(const AutoDownloadRule &rule) |
|
|
|
{ |
|
|
|
{ |
|
|
|
m_rules.insert(rule.name(), rule); |
|
|
|
m_rules.insert(rule.name(), rule); |
|
|
@ -369,9 +405,10 @@ void AutoDownloader::setRule_impl(const AutoDownloadRule &rule) |
|
|
|
void AutoDownloader::addJobForArticle(const Article *article) |
|
|
|
void AutoDownloader::addJobForArticle(const Article *article) |
|
|
|
{ |
|
|
|
{ |
|
|
|
const QString torrentURL = article->torrentUrl(); |
|
|
|
const QString torrentURL = article->torrentUrl(); |
|
|
|
if (m_waitingJobs.contains(torrentURL)) return; |
|
|
|
if (m_waitingJobs.contains(torrentURL)) |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
|
|
QSharedPointer<ProcessingJob> job(new ProcessingJob); |
|
|
|
auto job = QSharedPointer<ProcessingJob>::create(); |
|
|
|
job->feedURL = article->feed()->url(); |
|
|
|
job->feedURL = article->feed()->url(); |
|
|
|
job->articleData = article->data(); |
|
|
|
job->articleData = article->data(); |
|
|
|
m_processingQueue.append(job); |
|
|
|
m_processingQueue.append(job); |
|
|
@ -383,9 +420,12 @@ void AutoDownloader::processJob(const QSharedPointer<ProcessingJob> &job) |
|
|
|
{ |
|
|
|
{ |
|
|
|
for (AutoDownloadRule &rule : m_rules) |
|
|
|
for (AutoDownloadRule &rule : m_rules) |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (!rule.isEnabled()) continue; |
|
|
|
if (!rule.isEnabled()) |
|
|
|
if (!rule.feedURLs().contains(job->feedURL)) continue; |
|
|
|
continue; |
|
|
|
if (!rule.accepts(job->articleData)) continue; |
|
|
|
if (!rule.feedURLs().contains(job->feedURL)) |
|
|
|
|
|
|
|
continue; |
|
|
|
|
|
|
|
if (!rule.accepts(job->articleData)) |
|
|
|
|
|
|
|
continue; |
|
|
|
|
|
|
|
|
|
|
|
m_dirty = true; |
|
|
|
m_dirty = true; |
|
|
|
storeDeferred(); |
|
|
|
storeDeferred(); |
|
|
@ -423,12 +463,18 @@ void AutoDownloader::load() |
|
|
|
QFile rulesFile {(m_fileStorage->storageDir() / Path(RULES_FILE_NAME)).data()}; |
|
|
|
QFile rulesFile {(m_fileStorage->storageDir() / Path(RULES_FILE_NAME)).data()}; |
|
|
|
|
|
|
|
|
|
|
|
if (!rulesFile.exists()) |
|
|
|
if (!rulesFile.exists()) |
|
|
|
|
|
|
|
{ |
|
|
|
loadRulesLegacy(); |
|
|
|
loadRulesLegacy(); |
|
|
|
|
|
|
|
} |
|
|
|
else if (rulesFile.open(QFile::ReadOnly)) |
|
|
|
else if (rulesFile.open(QFile::ReadOnly)) |
|
|
|
|
|
|
|
{ |
|
|
|
loadRules(rulesFile.readAll()); |
|
|
|
loadRules(rulesFile.readAll()); |
|
|
|
|
|
|
|
} |
|
|
|
else |
|
|
|
else |
|
|
|
|
|
|
|
{ |
|
|
|
LogMsg(tr("Couldn't read RSS AutoDownloader rules from %1. Error: %2") |
|
|
|
LogMsg(tr("Couldn't read RSS AutoDownloader rules from %1. Error: %2") |
|
|
|
.arg(rulesFile.fileName(), rulesFile.errorString()), Log::CRITICAL); |
|
|
|
.arg(rulesFile.fileName(), rulesFile.errorString()), Log::CRITICAL); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::loadRules(const QByteArray &data) |
|
|
|
void AutoDownloader::loadRules(const QByteArray &data) |
|
|
@ -460,7 +506,8 @@ void AutoDownloader::loadRulesLegacy() |
|
|
|
|
|
|
|
|
|
|
|
void AutoDownloader::store() |
|
|
|
void AutoDownloader::store() |
|
|
|
{ |
|
|
|
{ |
|
|
|
if (!m_dirty) return; |
|
|
|
if (!m_dirty) |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
|
|
m_dirty = false; |
|
|
|
m_dirty = false; |
|
|
|
m_savingTimer.stop(); |
|
|
|
m_savingTimer.stop(); |
|
|
@ -486,7 +533,8 @@ bool AutoDownloader::isProcessingEnabled() const |
|
|
|
void AutoDownloader::resetProcessingQueue() |
|
|
|
void AutoDownloader::resetProcessingQueue() |
|
|
|
{ |
|
|
|
{ |
|
|
|
m_processingQueue.clear(); |
|
|
|
m_processingQueue.clear(); |
|
|
|
if (!isProcessingEnabled()) return; |
|
|
|
if (!isProcessingEnabled()) |
|
|
|
|
|
|
|
return; |
|
|
|
|
|
|
|
|
|
|
|
for (Article *article : asConst(Session::instance()->rootFolder()->articles())) |
|
|
|
for (Article *article : asConst(Session::instance()->rootFolder()->articles())) |
|
|
|
{ |
|
|
|
{ |
|
|
|