Vladimir Golovnev
1 year ago
committed by
GitHub
42 changed files with 931 additions and 637 deletions
@ -0,0 +1,217 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2015-2023 Vladimir Golovnev <glassez@yandex.ru> |
||||||
|
* Copyright (C) 2006 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 "addtorrentmanager.h" |
||||||
|
|
||||||
|
#include "base/bittorrent/infohash.h" |
||||||
|
#include "base/bittorrent/session.h" |
||||||
|
#include "base/bittorrent/torrentdescriptor.h" |
||||||
|
#include "base/logger.h" |
||||||
|
#include "base/net/downloadmanager.h" |
||||||
|
#include "base/preferences.h" |
||||||
|
|
||||||
|
AddTorrentManager::AddTorrentManager(IApplication *app, BitTorrent::Session *btSession, QObject *parent) |
||||||
|
: ApplicationComponent(app, parent) |
||||||
|
, m_btSession {btSession} |
||||||
|
{ |
||||||
|
Q_ASSERT(btSession); |
||||||
|
connect(btSession, &BitTorrent::Session::torrentAdded, this, &AddTorrentManager::onSessionTorrentAdded); |
||||||
|
connect(btSession, &BitTorrent::Session::addTorrentFailed, this, &AddTorrentManager::onSessionAddTorrentFailed); |
||||||
|
} |
||||||
|
|
||||||
|
BitTorrent::Session *AddTorrentManager::btSession() const |
||||||
|
{ |
||||||
|
return m_btSession; |
||||||
|
} |
||||||
|
|
||||||
|
bool AddTorrentManager::addTorrent(const QString &source, const BitTorrent::AddTorrentParams ¶ms) |
||||||
|
{ |
||||||
|
// `source`: .torrent file path, magnet URI or URL
|
||||||
|
|
||||||
|
if (source.isEmpty()) |
||||||
|
return false; |
||||||
|
|
||||||
|
if (Net::DownloadManager::hasSupportedScheme(source)) |
||||||
|
{ |
||||||
|
LogMsg(tr("Downloading torrent... Source: \"%1\"").arg(source)); |
||||||
|
const auto *pref = Preferences::instance(); |
||||||
|
// Launch downloader
|
||||||
|
Net::DownloadManager::instance()->download(Net::DownloadRequest(source).limit(pref->getTorrentFileSizeLimit()) |
||||||
|
, pref->useProxyForGeneralPurposes(), this, &AddTorrentManager::onDownloadFinished); |
||||||
|
m_downloadedTorrents[source] = params; |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(source)) |
||||||
|
{ |
||||||
|
return processTorrent(source, parseResult.value(), params); |
||||||
|
} |
||||||
|
else if (source.startsWith(u"magnet:", Qt::CaseInsensitive)) |
||||||
|
{ |
||||||
|
handleAddTorrentFailed(source, parseResult.error()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
const Path decodedPath {source.startsWith(u"file://", Qt::CaseInsensitive) |
||||||
|
? QUrl::fromEncoded(source.toLocal8Bit()).toLocalFile() : source}; |
||||||
|
auto torrentFileGuard = std::make_shared<TorrentFileGuard>(decodedPath); |
||||||
|
if (const auto loadResult = BitTorrent::TorrentDescriptor::loadFromFile(decodedPath)) |
||||||
|
{ |
||||||
|
setTorrentFileGuard(source, torrentFileGuard); |
||||||
|
return processTorrent(source, loadResult.value(), params); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
handleAddTorrentFailed(source, loadResult.error()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
bool AddTorrentManager::addTorrentToSession(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr |
||||||
|
, const BitTorrent::AddTorrentParams &addTorrentParams) |
||||||
|
{ |
||||||
|
const bool result = btSession()->addTorrent(torrentDescr, addTorrentParams); |
||||||
|
if (result) |
||||||
|
m_sourcesByInfoHash[torrentDescr.infoHash()] = source; |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::onDownloadFinished(const Net::DownloadResult &result) |
||||||
|
{ |
||||||
|
const QString &source = result.url; |
||||||
|
const BitTorrent::AddTorrentParams addTorrentParams = m_downloadedTorrents.take(source); |
||||||
|
|
||||||
|
switch (result.status) |
||||||
|
{ |
||||||
|
case Net::DownloadStatus::Success: |
||||||
|
if (const auto loadResult = BitTorrent::TorrentDescriptor::load(result.data)) |
||||||
|
processTorrent(source, loadResult.value(), addTorrentParams); |
||||||
|
else |
||||||
|
handleAddTorrentFailed(source, loadResult.error()); |
||||||
|
break; |
||||||
|
case Net::DownloadStatus::RedirectedToMagnet: |
||||||
|
if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(result.magnetURI)) |
||||||
|
processTorrent(source, parseResult.value(), addTorrentParams); |
||||||
|
else |
||||||
|
handleAddTorrentFailed(source, parseResult.error()); |
||||||
|
break; |
||||||
|
default: |
||||||
|
handleAddTorrentFailed(source, result.errorString); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::onSessionTorrentAdded(BitTorrent::Torrent *torrent) |
||||||
|
{ |
||||||
|
if (const QString source = m_sourcesByInfoHash.take(torrent->infoHash()); !source.isEmpty()) |
||||||
|
{ |
||||||
|
auto torrentFileGuard = m_guardedTorrentFiles.take(source); |
||||||
|
if (torrentFileGuard) |
||||||
|
torrentFileGuard->markAsAddedToSession(); |
||||||
|
emit torrentAdded(source, torrent); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::onSessionAddTorrentFailed(const BitTorrent::InfoHash &infoHash, const QString &reason) |
||||||
|
{ |
||||||
|
if (const QString source = m_sourcesByInfoHash.take(infoHash); !source.isEmpty()) |
||||||
|
{ |
||||||
|
auto torrentFileGuard = m_guardedTorrentFiles.take(source); |
||||||
|
if (torrentFileGuard) |
||||||
|
torrentFileGuard->setAutoRemove(false); |
||||||
|
emit addTorrentFailed(source, reason); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::handleAddTorrentFailed(const QString &source, const QString &reason) |
||||||
|
{ |
||||||
|
LogMsg(tr("Failed to add torrent. Source: \"%1\". Reason: \"%2\"").arg(source, reason), Log::WARNING); |
||||||
|
emit addTorrentFailed(source, reason); |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::handleDuplicateTorrent(const QString &source, BitTorrent::Torrent *torrent, const QString &message) |
||||||
|
{ |
||||||
|
LogMsg(tr("Detected an attempt to add a duplicate torrent. Source: %1. Existing torrent: %2. Result: %3") |
||||||
|
.arg(source, torrent->name(), message)); |
||||||
|
emit addTorrentFailed(source, message); |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::setTorrentFileGuard(const QString &source, std::shared_ptr<TorrentFileGuard> torrentFileGuard) |
||||||
|
{ |
||||||
|
m_guardedTorrentFiles.emplace(source, std::move(torrentFileGuard)); |
||||||
|
} |
||||||
|
|
||||||
|
void AddTorrentManager::releaseTorrentFileGuard(const QString &source) |
||||||
|
{ |
||||||
|
auto torrentFileGuard = m_guardedTorrentFiles.take(source); |
||||||
|
if (torrentFileGuard) |
||||||
|
torrentFileGuard->setAutoRemove(false); |
||||||
|
} |
||||||
|
|
||||||
|
bool AddTorrentManager::processTorrent(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr |
||||||
|
, const BitTorrent::AddTorrentParams &addTorrentParams) |
||||||
|
{ |
||||||
|
const BitTorrent::InfoHash infoHash = torrentDescr.infoHash(); |
||||||
|
|
||||||
|
if (BitTorrent::Torrent *torrent = btSession()->findTorrent(infoHash)) |
||||||
|
{ |
||||||
|
// a duplicate torrent is being added
|
||||||
|
|
||||||
|
const bool hasMetadata = torrentDescr.info().has_value(); |
||||||
|
if (hasMetadata) |
||||||
|
{ |
||||||
|
// Trying to set metadata to existing torrent in case if it has none
|
||||||
|
torrent->setMetadata(*torrentDescr.info()); |
||||||
|
} |
||||||
|
|
||||||
|
if (!btSession()->isMergeTrackersEnabled()) |
||||||
|
{ |
||||||
|
handleDuplicateTorrent(source, torrent, tr("Merging of trackers is disabled")); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
const bool isPrivate = torrent->isPrivate() || (hasMetadata && torrentDescr.info()->isPrivate()); |
||||||
|
if (isPrivate) |
||||||
|
{ |
||||||
|
handleDuplicateTorrent(source, torrent, tr("Trackers cannot be merged because it is a private torrent")); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// merge trackers and web seeds
|
||||||
|
torrent->addTrackers(torrentDescr.trackers()); |
||||||
|
torrent->addUrlSeeds(torrentDescr.urlSeeds()); |
||||||
|
|
||||||
|
handleDuplicateTorrent(source, torrent, tr("Trackers are merged from new source")); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
return addTorrentToSession(source, torrentDescr, addTorrentParams); |
||||||
|
} |
||||||
@ -0,0 +1,86 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2015-2023 Vladimir Golovnev <glassez@yandex.ru> |
||||||
|
* Copyright (C) 2006 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 <memory> |
||||||
|
|
||||||
|
#include <QHash> |
||||||
|
#include <QObject> |
||||||
|
|
||||||
|
#include "base/applicationcomponent.h" |
||||||
|
#include "base/bittorrent/addtorrentparams.h" |
||||||
|
#include "base/torrentfileguard.h" |
||||||
|
|
||||||
|
namespace Net |
||||||
|
{ |
||||||
|
struct DownloadResult; |
||||||
|
} |
||||||
|
|
||||||
|
namespace BitTorrent |
||||||
|
{ |
||||||
|
class Session; |
||||||
|
class TorrentDescriptor; |
||||||
|
} |
||||||
|
|
||||||
|
class AddTorrentManager : public ApplicationComponent<QObject> |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
Q_DISABLE_COPY_MOVE(AddTorrentManager) |
||||||
|
|
||||||
|
public: |
||||||
|
AddTorrentManager(IApplication *app, BitTorrent::Session *btSession, QObject *parent = nullptr); |
||||||
|
|
||||||
|
BitTorrent::Session *btSession() const; |
||||||
|
bool addTorrent(const QString &source, const BitTorrent::AddTorrentParams ¶ms = {}); |
||||||
|
|
||||||
|
signals: |
||||||
|
void torrentAdded(const QString &source, BitTorrent::Torrent *torrent); |
||||||
|
void addTorrentFailed(const QString &source, const QString &reason); |
||||||
|
|
||||||
|
protected: |
||||||
|
bool addTorrentToSession(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr |
||||||
|
, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||||
|
void handleAddTorrentFailed(const QString &source, const QString &reason); |
||||||
|
void handleDuplicateTorrent(const QString &source, BitTorrent::Torrent *torrent, const QString &message); |
||||||
|
void setTorrentFileGuard(const QString &source, std::shared_ptr<TorrentFileGuard> torrentFileGuard); |
||||||
|
void releaseTorrentFileGuard(const QString &source); |
||||||
|
|
||||||
|
private: |
||||||
|
void onDownloadFinished(const Net::DownloadResult &result); |
||||||
|
void onSessionTorrentAdded(BitTorrent::Torrent *torrent); |
||||||
|
void onSessionAddTorrentFailed(const BitTorrent::InfoHash &infoHash, const QString &reason); |
||||||
|
bool processTorrent(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr |
||||||
|
, const BitTorrent::AddTorrentParams &addTorrentParams); |
||||||
|
|
||||||
|
BitTorrent::Session *m_btSession = nullptr; |
||||||
|
QHash<QString, BitTorrent::AddTorrentParams> m_downloadedTorrents; |
||||||
|
QHash<BitTorrent::InfoHash, QString> m_sourcesByInfoHash; |
||||||
|
QHash<QString, std::shared_ptr<TorrentFileGuard>> m_guardedTorrentFiles; |
||||||
|
}; |
||||||
@ -0,0 +1,200 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2015-2023 Vladimir Golovnev <glassez@yandex.ru> |
||||||
|
* Copyright (C) 2006 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 "guiaddtorrentmanager.h" |
||||||
|
|
||||||
|
#include "base/bittorrent/session.h" |
||||||
|
#include "base/bittorrent/torrentdescriptor.h" |
||||||
|
#include "base/logger.h" |
||||||
|
#include "base/net/downloadmanager.h" |
||||||
|
#include "base/preferences.h" |
||||||
|
#include "base/torrentfileguard.h" |
||||||
|
#include "addnewtorrentdialog.h" |
||||||
|
#include "interfaces/iguiapplication.h" |
||||||
|
#include "mainwindow.h" |
||||||
|
#include "raisedmessagebox.h" |
||||||
|
|
||||||
|
GUIAddTorrentManager::GUIAddTorrentManager(IGUIApplication *app, BitTorrent::Session *session, QObject *parent) |
||||||
|
: GUIApplicationComponent(app, session, parent) |
||||||
|
{ |
||||||
|
connect(btSession(), &BitTorrent::Session::metadataDownloaded, this, &GUIAddTorrentManager::onMetadataDownloaded); |
||||||
|
} |
||||||
|
|
||||||
|
bool GUIAddTorrentManager::addTorrent(const QString &source, const BitTorrent::AddTorrentParams ¶ms, const AddTorrentOption option) |
||||||
|
{ |
||||||
|
// `source`: .torrent file path, magnet URI or URL
|
||||||
|
|
||||||
|
if (source.isEmpty()) |
||||||
|
return false; |
||||||
|
|
||||||
|
const auto *pref = Preferences::instance(); |
||||||
|
|
||||||
|
if ((option == AddTorrentOption::SkipDialog) |
||||||
|
|| ((option == AddTorrentOption::Default) && !pref->isAddNewTorrentDialogEnabled())) |
||||||
|
{ |
||||||
|
return AddTorrentManager::addTorrent(source, params); |
||||||
|
} |
||||||
|
|
||||||
|
if (Net::DownloadManager::hasSupportedScheme(source)) |
||||||
|
{ |
||||||
|
LogMsg(tr("Downloading torrent... Source: \"%1\"").arg(source)); |
||||||
|
// Launch downloader
|
||||||
|
Net::DownloadManager::instance()->download(Net::DownloadRequest(source).limit(pref->getTorrentFileSizeLimit()) |
||||||
|
, pref->useProxyForGeneralPurposes(), this, &GUIAddTorrentManager::onDownloadFinished); |
||||||
|
m_downloadedTorrents[source] = params; |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(source)) |
||||||
|
{ |
||||||
|
return processTorrent(source, parseResult.value(), params); |
||||||
|
} |
||||||
|
else if (source.startsWith(u"magnet:", Qt::CaseInsensitive)) |
||||||
|
{ |
||||||
|
handleAddTorrentFailed(source, parseResult.error()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
const Path decodedPath {source.startsWith(u"file://", Qt::CaseInsensitive) |
||||||
|
? QUrl::fromEncoded(source.toLocal8Bit()).toLocalFile() : source}; |
||||||
|
auto torrentFileGuard = std::make_shared<TorrentFileGuard>(decodedPath); |
||||||
|
if (const auto loadResult = BitTorrent::TorrentDescriptor::loadFromFile(decodedPath)) |
||||||
|
{ |
||||||
|
const BitTorrent::TorrentDescriptor &torrentDescriptor = loadResult.value(); |
||||||
|
const bool isProcessing = processTorrent(source, torrentDescriptor, params); |
||||||
|
if (isProcessing) |
||||||
|
setTorrentFileGuard(source, torrentFileGuard); |
||||||
|
return isProcessing; |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
handleAddTorrentFailed(decodedPath.toString(), loadResult.error()); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
void GUIAddTorrentManager::onDownloadFinished(const Net::DownloadResult &result) |
||||||
|
{ |
||||||
|
const QString &source = result.url; |
||||||
|
const BitTorrent::AddTorrentParams addTorrentParams = m_downloadedTorrents.take(source); |
||||||
|
|
||||||
|
switch (result.status) |
||||||
|
{ |
||||||
|
case Net::DownloadStatus::Success: |
||||||
|
if (const auto loadResult = BitTorrent::TorrentDescriptor::load(result.data)) |
||||||
|
processTorrent(source, loadResult.value(), addTorrentParams); |
||||||
|
else |
||||||
|
handleAddTorrentFailed(source, loadResult.error()); |
||||||
|
break; |
||||||
|
case Net::DownloadStatus::RedirectedToMagnet: |
||||||
|
if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(result.magnetURI)) |
||||||
|
processTorrent(source, parseResult.value(), addTorrentParams); |
||||||
|
else |
||||||
|
handleAddTorrentFailed(source, parseResult.error()); |
||||||
|
break; |
||||||
|
default: |
||||||
|
handleAddTorrentFailed(source, result.errorString); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void GUIAddTorrentManager::onMetadataDownloaded(const BitTorrent::TorrentInfo &metadata) |
||||||
|
{ |
||||||
|
Q_ASSERT(metadata.isValid()); |
||||||
|
if (!metadata.isValid()) [[unlikely]] |
||||||
|
return; |
||||||
|
|
||||||
|
for (const auto &[infoHash, dialog] : m_dialogs.asKeyValueRange()) |
||||||
|
{ |
||||||
|
if (metadata.matchesInfoHash(infoHash)) |
||||||
|
dialog->updateMetadata(metadata); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
bool GUIAddTorrentManager::processTorrent(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr, const BitTorrent::AddTorrentParams ¶ms) |
||||||
|
{ |
||||||
|
const bool hasMetadata = torrentDescr.info().has_value(); |
||||||
|
const BitTorrent::InfoHash infoHash = torrentDescr.infoHash(); |
||||||
|
|
||||||
|
// Prevent showing the dialog if download is already present
|
||||||
|
if (BitTorrent::Torrent *torrent = btSession()->findTorrent(infoHash)) |
||||||
|
{ |
||||||
|
if (hasMetadata) |
||||||
|
{ |
||||||
|
// Trying to set metadata to existing torrent in case if it has none
|
||||||
|
torrent->setMetadata(*torrentDescr.info()); |
||||||
|
} |
||||||
|
|
||||||
|
if (torrent->isPrivate() || (hasMetadata && torrentDescr.info()->isPrivate())) |
||||||
|
{ |
||||||
|
handleDuplicateTorrent(source, torrent, tr("Trackers cannot be merged because it is a private torrent")); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
bool mergeTrackers = btSession()->isMergeTrackersEnabled(); |
||||||
|
if (Preferences::instance()->confirmMergeTrackers()) |
||||||
|
{ |
||||||
|
const QMessageBox::StandardButton btn = RaisedMessageBox::question(app()->mainWindow(), tr("Torrent is already present") |
||||||
|
, tr("Torrent '%1' is already in the transfer list. Do you want to merge trackers from new source?").arg(torrent->name()) |
||||||
|
, (QMessageBox::Yes | QMessageBox::No), QMessageBox::Yes); |
||||||
|
mergeTrackers = (btn == QMessageBox::Yes); |
||||||
|
} |
||||||
|
|
||||||
|
if (mergeTrackers) |
||||||
|
{ |
||||||
|
torrent->addTrackers(torrentDescr.trackers()); |
||||||
|
torrent->addUrlSeeds(torrentDescr.urlSeeds()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
if (!hasMetadata) |
||||||
|
btSession()->downloadMetadata(torrentDescr); |
||||||
|
|
||||||
|
auto *dlg = new AddNewTorrentDialog(torrentDescr, params, app()->mainWindow()); |
||||||
|
dlg->setAttribute(Qt::WA_DeleteOnClose); |
||||||
|
m_dialogs[infoHash] = dlg; |
||||||
|
connect(dlg, &QDialog::finished, this, [this, source, infoHash, dlg](int result) |
||||||
|
{ |
||||||
|
if (dlg->isDoNotDeleteTorrentChecked()) |
||||||
|
releaseTorrentFileGuard(source); |
||||||
|
|
||||||
|
if (result == QDialog::Accepted) |
||||||
|
addTorrentToSession(source, dlg->torrentDescriptor(), dlg->addTorrentParams()); |
||||||
|
|
||||||
|
m_dialogs.remove(infoHash); |
||||||
|
}); |
||||||
|
dlg->show(); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
Loading…
Reference in new issue