Enable customizing the save statistics time interval
[qBittorrent.git] / src / base / addtorrentmanager.cpp
bloba094dbd85acc15cffdf70b6f4059856a577532b0
1 /*
2 * Bittorrent Client using Qt and libtorrent.
3 * Copyright (C) 2015-2023 Vladimir Golovnev <glassez@yandex.ru>
4 * Copyright (C) 2006 Christophe Dumez <chris@qbittorrent.org>
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20 * In addition, as a special exception, the copyright holders give permission to
21 * link this program with the OpenSSL project's "OpenSSL" library (or with
22 * modified versions of it that use the same license as the "OpenSSL" library),
23 * and distribute the linked executables. You must obey the GNU General Public
24 * License in all respects for all of the code used other than "OpenSSL". If you
25 * modify file(s), you may extend this exception to your version of the file(s),
26 * but you are not obligated to do so. If you do not wish to do so, delete this
27 * exception statement from your version.
30 #include "addtorrentmanager.h"
32 #include "base/bittorrent/infohash.h"
33 #include "base/bittorrent/session.h"
34 #include "base/bittorrent/torrentdescriptor.h"
35 #include "base/logger.h"
36 #include "base/net/downloadmanager.h"
37 #include "base/preferences.h"
39 AddTorrentManager::AddTorrentManager(IApplication *app, BitTorrent::Session *btSession, QObject *parent)
40 : ApplicationComponent(app, parent)
41 , m_btSession {btSession}
43 Q_ASSERT(btSession);
44 connect(btSession, &BitTorrent::Session::torrentAdded, this, &AddTorrentManager::onSessionTorrentAdded);
45 connect(btSession, &BitTorrent::Session::addTorrentFailed, this, &AddTorrentManager::onSessionAddTorrentFailed);
48 BitTorrent::Session *AddTorrentManager::btSession() const
50 return m_btSession;
53 bool AddTorrentManager::addTorrent(const QString &source, const BitTorrent::AddTorrentParams &params)
55 // `source`: .torrent file path, magnet URI or URL
57 if (source.isEmpty())
58 return false;
60 if (Net::DownloadManager::hasSupportedScheme(source))
62 LogMsg(tr("Downloading torrent... Source: \"%1\"").arg(source));
63 const auto *pref = Preferences::instance();
64 // Launch downloader
65 Net::DownloadManager::instance()->download(Net::DownloadRequest(source).limit(pref->getTorrentFileSizeLimit())
66 , pref->useProxyForGeneralPurposes(), this, &AddTorrentManager::onDownloadFinished);
67 m_downloadedTorrents[source] = params;
68 return true;
71 if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(source))
73 return processTorrent(source, parseResult.value(), params);
75 else if (source.startsWith(u"magnet:", Qt::CaseInsensitive))
77 handleAddTorrentFailed(source, parseResult.error());
78 return false;
81 const Path decodedPath {source.startsWith(u"file://", Qt::CaseInsensitive)
82 ? QUrl::fromEncoded(source.toLocal8Bit()).toLocalFile() : source};
83 auto torrentFileGuard = std::make_shared<TorrentFileGuard>(decodedPath);
84 if (const auto loadResult = BitTorrent::TorrentDescriptor::loadFromFile(decodedPath))
86 setTorrentFileGuard(source, torrentFileGuard);
87 return processTorrent(source, loadResult.value(), params);
89 else
91 handleAddTorrentFailed(source, loadResult.error());
92 return false;
95 return false;
98 bool AddTorrentManager::addTorrentToSession(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr
99 , const BitTorrent::AddTorrentParams &addTorrentParams)
101 const bool result = btSession()->addTorrent(torrentDescr, addTorrentParams);
102 if (result)
103 m_sourcesByInfoHash[torrentDescr.infoHash()] = source;
105 return result;
108 void AddTorrentManager::onDownloadFinished(const Net::DownloadResult &result)
110 const QString &source = result.url;
111 const BitTorrent::AddTorrentParams addTorrentParams = m_downloadedTorrents.take(source);
113 switch (result.status)
115 case Net::DownloadStatus::Success:
116 if (const auto loadResult = BitTorrent::TorrentDescriptor::load(result.data))
117 processTorrent(source, loadResult.value(), addTorrentParams);
118 else
119 handleAddTorrentFailed(source, loadResult.error());
120 break;
121 case Net::DownloadStatus::RedirectedToMagnet:
122 if (const auto parseResult = BitTorrent::TorrentDescriptor::parse(result.magnetURI))
123 processTorrent(source, parseResult.value(), addTorrentParams);
124 else
125 handleAddTorrentFailed(source, parseResult.error());
126 break;
127 default:
128 handleAddTorrentFailed(source, result.errorString);
132 void AddTorrentManager::onSessionTorrentAdded(BitTorrent::Torrent *torrent)
134 if (const QString source = m_sourcesByInfoHash.take(torrent->infoHash()); !source.isEmpty())
136 auto torrentFileGuard = m_guardedTorrentFiles.take(source);
137 if (torrentFileGuard)
138 torrentFileGuard->markAsAddedToSession();
139 emit torrentAdded(source, torrent);
143 void AddTorrentManager::onSessionAddTorrentFailed(const BitTorrent::InfoHash &infoHash, const QString &reason)
145 if (const QString source = m_sourcesByInfoHash.take(infoHash); !source.isEmpty())
147 auto torrentFileGuard = m_guardedTorrentFiles.take(source);
148 if (torrentFileGuard)
149 torrentFileGuard->setAutoRemove(false);
150 emit addTorrentFailed(source, reason);
154 void AddTorrentManager::handleAddTorrentFailed(const QString &source, const QString &reason)
156 LogMsg(tr("Failed to add torrent. Source: \"%1\". Reason: \"%2\"").arg(source, reason), Log::WARNING);
157 emit addTorrentFailed(source, reason);
160 void AddTorrentManager::handleDuplicateTorrent(const QString &source, BitTorrent::Torrent *torrent, const QString &message)
162 LogMsg(tr("Detected an attempt to add a duplicate torrent. Source: %1. Existing torrent: %2. Result: %3")
163 .arg(source, torrent->name(), message));
164 emit addTorrentFailed(source, message);
167 void AddTorrentManager::setTorrentFileGuard(const QString &source, std::shared_ptr<TorrentFileGuard> torrentFileGuard)
169 m_guardedTorrentFiles.emplace(source, std::move(torrentFileGuard));
172 void AddTorrentManager::releaseTorrentFileGuard(const QString &source)
174 auto torrentFileGuard = m_guardedTorrentFiles.take(source);
175 if (torrentFileGuard)
176 torrentFileGuard->setAutoRemove(false);
179 bool AddTorrentManager::processTorrent(const QString &source, const BitTorrent::TorrentDescriptor &torrentDescr
180 , const BitTorrent::AddTorrentParams &addTorrentParams)
182 const BitTorrent::InfoHash infoHash = torrentDescr.infoHash();
184 if (BitTorrent::Torrent *torrent = btSession()->findTorrent(infoHash))
186 // a duplicate torrent is being added
188 const bool hasMetadata = torrentDescr.info().has_value();
189 if (hasMetadata)
191 // Trying to set metadata to existing torrent in case if it has none
192 torrent->setMetadata(*torrentDescr.info());
195 if (!btSession()->isMergeTrackersEnabled())
197 handleDuplicateTorrent(source, torrent, tr("Merging of trackers is disabled"));
198 return false;
201 const bool isPrivate = torrent->isPrivate() || (hasMetadata && torrentDescr.info()->isPrivate());
202 if (isPrivate)
204 handleDuplicateTorrent(source, torrent, tr("Trackers cannot be merged because it is a private torrent"));
205 return false;
208 // merge trackers and web seeds
209 torrent->addTrackers(torrentDescr.trackers());
210 torrent->addUrlSeeds(torrentDescr.urlSeeds());
212 handleDuplicateTorrent(source, torrent, tr("Trackers are merged from new source"));
213 return false;
216 return addTorrentToSession(source, torrentDescr, addTorrentParams);