Add WebAPI for downloading torrent metadata

Signed-off-by: Thomas Piccirello <[email protected]>
qbittorrent · Sep 30, 2024 · dee0d8c · dee0d8c
1 parent 92d45ea
commit dee0d8c
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 0 deletions.
diff --git a/src/base/bittorrent/torrentdescriptor.cpp b/src/base/bittorrent/torrentdescriptor.cpp
@@ -143,6 +143,22 @@ catch (const lt::system_error &err)
     return nonstd::make_unexpected(QString::fromLocal8Bit(err.what()));
 }
 
+nonstd::expected<QByteArray, QString> BitTorrent::TorrentDescriptor::saveToBuffer() const
+try
+{
+    const lt::entry torrentEntry = lt::write_torrent_file(m_ltAddTorrentParams);
+    // usually torrent size should be smaller than 1 MB,
+    // however there are >100 MB v2/hybrid torrent files out in the wild
+    QByteArray buffer;
+    buffer.reserve(1024 * 1024);
+    lt::bencode(std::back_inserter(buffer), torrentEntry);
+    return buffer;
+}
+catch (const lt::system_error &err)
+{
+    return nonstd::make_unexpected(QString::fromLocal8Bit(err.what()));
+}
+
 BitTorrent::TorrentDescriptor::TorrentDescriptor(lt::add_torrent_params ltAddTorrentParams)
     : m_ltAddTorrentParams {std::move(ltAddTorrentParams)}
 {

diff --git a/src/base/bittorrent/torrentdescriptor.h b/src/base/bittorrent/torrentdescriptor.h
@@ -69,6 +69,7 @@ namespace BitTorrent
         static nonstd::expected<TorrentDescriptor, QString> loadFromFile(const Path &path) noexcept;
         static nonstd::expected<TorrentDescriptor, QString> parse(const QString &str) noexcept;
         nonstd::expected<void, QString> saveToFile(const Path &path) const;
+        nonstd::expected<QByteArray, QString> saveToBuffer() const;
 
         const lt::add_torrent_params &ltAddTorrentParams() const;
 

diff --git a/src/webui/api/torrentscontroller.cpp b/src/webui/api/torrentscontroller.cpp
@@ -1922,6 +1922,35 @@ void TorrentsController::parseMetadataAction()
     setResult(result);
 }
 
+void TorrentsController::saveMetadataAction()
+{
+    requireParams({u"source"_s});
+
+    const QString sourceParam = params()[u"source"_s].trimmed();
+    if (sourceParam.isEmpty())
+        throw APIError(APIErrorType::BadParams, tr("Must specify URI or hash"));
+
+    const QString source = QUrl::fromPercentEncoding(sourceParam.toLatin1());
+    if (const auto iter = m_torrentSource.find(source); iter != m_torrentSource.end())
+    {
+        const BitTorrent::InfoHash &infoHash = iter.value();
+        if (isMetadataDownloaded(infoHash))
+        {
+            const BitTorrent::TorrentDescriptor &torrentDescr = m_torrentMetadata[infoHash];
+            const nonstd::expected<QByteArray, QString> result = torrentDescr.saveToBuffer();
+            if (!result)
+                throw APIError(APIErrorType::Conflict, tr("Unable to export torrent metadata. Error: %1").arg(result.error()));
+
+            setResult(result.value(), u"application/x-bittorrent"_s, (infoHash.toTorrentID().toString() + u".torrent"));
+            return;
+        }
+
+        throw APIError(APIErrorType::Conflict, tr("Metadata is not yet available"));
+    }
+
+    throw APIError(APIErrorType::NotFound);
+}
+
 void TorrentsController::onDownloadFinished(const Net::DownloadResult &result)
 {
     const QString &source = result.url;

diff --git a/src/webui/api/torrentscontroller.h b/src/webui/api/torrentscontroller.h
@@ -106,6 +106,7 @@ private slots:
     void setSSLParametersAction();
     void fetchMetadataAction();
     void parseMetadataAction();
+    void saveMetadataAction();
 
 private:
     void onMetadataDownloaded(const BitTorrent::TorrentInfo &info);