Clean up 'recursive download' related code

Don't load .torrent files too early, otherwise qbt might emit a dubious error log message if the
.torrent file is invalid.
This commit is contained in:
Chocobo1 2022-09-14 15:28:44 +08:00
parent 80759f9e69
commit f8a2fbc4c5
No known key found for this signature in database
GPG key ID: 210D9C873253A68C
6 changed files with 42 additions and 37 deletions

View file

@ -4552,21 +4552,13 @@ void SessionImpl::handleTorrentFinished(TorrentImpl *const torrent)
if (const Path exportPath = finishedTorrentExportDirectory(); !exportPath.isEmpty()) if (const Path exportPath = finishedTorrentExportDirectory(); !exportPath.isEmpty())
exportTorrentFile(torrent, exportPath); exportTorrentFile(torrent, exportPath);
// Check if there are torrent files inside // Check whether it contains .torrent files
for (const Path &torrentRelpath : asConst(torrent->filePaths())) for (const Path &torrentRelpath : asConst(torrent->filePaths()))
{ {
if (torrentRelpath.hasExtension(u".torrent"_qs)) if (torrentRelpath.hasExtension(u".torrent"_qs))
{ {
const Path torrentFullpath = torrent->actualStorageLocation() / torrentRelpath; emit recursiveTorrentDownloadPossible(torrent);
if (TorrentInfo::loadFromFile(torrentFullpath)) break;
{
emit recursiveTorrentDownloadPossible(torrent);
break;
}
else
{
LogMsg(tr("Unable to load torrent. File: \"%1\"").arg(torrentFullpath.toString()), Log::CRITICAL);
}
} }
} }
@ -4854,25 +4846,32 @@ void SessionImpl::disableIPFilter()
void SessionImpl::recursiveTorrentDownload(const TorrentID &id) void SessionImpl::recursiveTorrentDownload(const TorrentID &id)
{ {
TorrentImpl *const torrent = m_torrents.value(id); const TorrentImpl *torrent = m_torrents.value(id);
if (!torrent) return; if (!torrent)
return;
for (const Path &torrentRelpath : asConst(torrent->filePaths())) for (const Path &torrentRelpath : asConst(torrent->filePaths()))
{ {
if (torrentRelpath.hasExtension(u".torrent"_qs)) if (torrentRelpath.hasExtension(u".torrent"_qs))
{ {
LogMsg(tr("Recursive download .torrent file within torrent. Source torrent: \"%1\". File: \"%2\"")
.arg(torrent->name(), torrentRelpath.toString()));
const Path torrentFullpath = torrent->savePath() / torrentRelpath; const Path torrentFullpath = torrent->savePath() / torrentRelpath;
LogMsg(tr("Recursive download .torrent file within torrent. Source torrent: \"%1\". File: \"%2\"")
.arg(torrent->name(), torrentFullpath.toString()));
AddTorrentParams params; AddTorrentParams params;
// Passing the save path along to the sub torrent file // Passing the save path along to the sub torrent file
params.savePath = torrent->savePath(); params.savePath = torrent->savePath();
const nonstd::expected<TorrentInfo, QString> loadResult = TorrentInfo::loadFromFile(torrentFullpath); const nonstd::expected<TorrentInfo, QString> loadResult = TorrentInfo::loadFromFile(torrentFullpath);
if (loadResult) if (loadResult)
{
addTorrent(loadResult.value(), params); addTorrent(loadResult.value(), params);
}
else else
LogMsg(tr("Failed to load torrent. Error: \"%1\"").arg(loadResult.error()), Log::WARNING); {
LogMsg(tr("Failed to load .torrent file within torrent. Source torrent: \"%1\". File: \"%2\". Error: \"%3\"")
.arg(torrent->name(), torrentFullpath.toString(), loadResult.error()), Log::WARNING);
}
} }
} }
} }

View file

@ -996,14 +996,14 @@ void Preferences::resolvePeerHostNames(const bool resolve)
setValue(u"Preferences/Connection/ResolvePeerHostNames"_qs, resolve); setValue(u"Preferences/Connection/ResolvePeerHostNames"_qs, resolve);
} }
bool Preferences::recursiveDownloadDisabled() const bool Preferences::isRecursiveDownloadEnabled() const
{ {
return value(u"Preferences/Advanced/DisableRecursiveDownload"_qs, false); return !value(u"Preferences/Advanced/DisableRecursiveDownload"_qs, false);
} }
void Preferences::disableRecursiveDownload(const bool disable) void Preferences::setRecursiveDownloadEnabled(const bool enable)
{ {
setValue(u"Preferences/Advanced/DisableRecursiveDownload"_qs, disable); setValue(u"Preferences/Advanced/DisableRecursiveDownload"_qs, !enable);
} }
#ifdef Q_OS_WIN #ifdef Q_OS_WIN

View file

@ -281,8 +281,8 @@ public:
void resolvePeerCountries(bool resolve); void resolvePeerCountries(bool resolve);
bool resolvePeerHostNames() const; bool resolvePeerHostNames() const;
void resolvePeerHostNames(bool resolve); void resolvePeerHostNames(bool resolve);
bool recursiveDownloadDisabled() const; bool isRecursiveDownloadEnabled() const;
void disableRecursiveDownload(bool disable = true); void setRecursiveDownloadEnabled(bool enable);
#ifdef Q_OS_WIN #ifdef Q_OS_WIN
bool neverCheckFileAssoc() const; bool neverCheckFileAssoc() const;
void setNeverCheckFileAssoc(bool check = true); void setNeverCheckFileAssoc(bool check = true);

View file

@ -885,27 +885,33 @@ void MainWindow::displayExecutionLogTab()
// End of keyboard shortcuts slots // End of keyboard shortcuts slots
void MainWindow::askRecursiveTorrentDownloadConfirmation(BitTorrent::Torrent *const torrent) void MainWindow::askRecursiveTorrentDownloadConfirmation(const BitTorrent::Torrent *torrent)
{ {
Preferences *const pref = Preferences::instance(); if (!Preferences::instance()->isRecursiveDownloadEnabled())
if (pref->recursiveDownloadDisabled()) return; return;
const auto torrentID = torrent->id(); const auto torrentID = torrent->id();
QMessageBox *confirmBox = new QMessageBox(QMessageBox::Question, tr("Recursive download confirmation") QMessageBox *confirmBox = new QMessageBox(QMessageBox::Question, tr("Recursive download confirmation")
, tr("The torrent '%1' contains torrent files, do you want to proceed with their download?").arg(torrent->name()) , tr("The torrent '%1' contains .torrent files, do you want to proceed with their downloads?").arg(torrent->name())
, QMessageBox::NoButton, this); , (QMessageBox::Yes | QMessageBox::No | QMessageBox::NoToAll), this);
confirmBox->setAttribute(Qt::WA_DeleteOnClose); confirmBox->setAttribute(Qt::WA_DeleteOnClose);
const QPushButton *yes = confirmBox->addButton(tr("Yes"), QMessageBox::YesRole); const QAbstractButton *yesButton = confirmBox->button(QMessageBox::Yes);
/*QPushButton *no = */ confirmBox->addButton(tr("No"), QMessageBox::NoRole); QAbstractButton *neverButton = confirmBox->button(QMessageBox::NoToAll);
const QPushButton *never = confirmBox->addButton(tr("Never"), QMessageBox::NoRole); neverButton->setText(tr("Never"));
connect(confirmBox, &QMessageBox::buttonClicked, this, [torrentID, yes, never](const QAbstractButton *button)
connect(confirmBox, &QMessageBox::buttonClicked, this
, [torrentID, yesButton, neverButton](const QAbstractButton *button)
{ {
if (button == yes) if (button == yesButton)
{
BitTorrent::Session::instance()->recursiveTorrentDownload(torrentID); BitTorrent::Session::instance()->recursiveTorrentDownload(torrentID);
if (button == never) }
Preferences::instance()->disableRecursiveDownload(); else if (button == neverButton)
{
Preferences::instance()->setRecursiveDownloadEnabled(false);
}
}); });
confirmBox->open(); confirmBox->open();
} }

View file

@ -130,7 +130,7 @@ private slots:
void reloadSessionStats(); void reloadSessionStats();
void reloadTorrentStats(const QVector<BitTorrent::Torrent *> &torrents); void reloadTorrentStats(const QVector<BitTorrent::Torrent *> &torrents);
void loadPreferences(); void loadPreferences();
void askRecursiveTorrentDownloadConfirmation(BitTorrent::Torrent *const torrent); void askRecursiveTorrentDownloadConfirmation(const BitTorrent::Torrent *torrent);
void optionsSaved(); void optionsSaved();
void toggleAlternativeSpeeds(); void toggleAlternativeSpeeds();

View file

@ -546,7 +546,7 @@ void OptionsDialog::loadDownloadsTabOptions()
m_ui->checkPreallocateAll->setChecked(session->isPreallocationEnabled()); m_ui->checkPreallocateAll->setChecked(session->isPreallocationEnabled());
m_ui->checkAppendqB->setChecked(session->isAppendExtensionEnabled()); m_ui->checkAppendqB->setChecked(session->isAppendExtensionEnabled());
m_ui->checkRecursiveDownload->setChecked(!pref->recursiveDownloadDisabled()); m_ui->checkRecursiveDownload->setChecked(pref->isRecursiveDownloadEnabled());
m_ui->comboSavingMode->setCurrentIndex(!session->isAutoTMMDisabledByDefault()); m_ui->comboSavingMode->setCurrentIndex(!session->isAutoTMMDisabledByDefault());
m_ui->comboTorrentCategoryChanged->setCurrentIndex(session->isDisableAutoTMMWhenCategoryChanged()); m_ui->comboTorrentCategoryChanged->setCurrentIndex(session->isDisableAutoTMMWhenCategoryChanged());
@ -698,7 +698,7 @@ void OptionsDialog::saveDownloadsTabOptions() const
session->setPreallocationEnabled(preAllocateAllFiles()); session->setPreallocationEnabled(preAllocateAllFiles());
session->setAppendExtensionEnabled(m_ui->checkAppendqB->isChecked()); session->setAppendExtensionEnabled(m_ui->checkAppendqB->isChecked());
pref->disableRecursiveDownload(!m_ui->checkRecursiveDownload->isChecked()); pref->setRecursiveDownloadEnabled(m_ui->checkRecursiveDownload->isChecked());
session->setAutoTMMDisabledByDefault(m_ui->comboSavingMode->currentIndex() == 0); session->setAutoTMMDisabledByDefault(m_ui->comboSavingMode->currentIndex() == 0);
session->setDisableAutoTMMWhenCategoryChanged(m_ui->comboTorrentCategoryChanged->currentIndex() == 1); session->setDisableAutoTMMWhenCategoryChanged(m_ui->comboTorrentCategoryChanged->currentIndex() == 1);