dandri/i2pd
1
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2026-08-23 20:50:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle fields list from torrent-get RPC

Browse Source
This commit is contained in:
orignal
2026-08-19 16:14:21 -04:00
parent 3740e8dc6f
commit e7009e561f
1 changed files with 64 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libi2pd_client/TorrentsRPC.cpp
+64 -20
View File
@@ -13,6 +13,7 @@
#endif
#include <boost/algorithm/hex.hpp>
#include <vector>
#include <functional>
#include "Log.h"
#include "Torrents.h"
#include "TorrentsRPC.h"
@@ -44,6 +45,7 @@ namespace torrents
std::string ResultResponse (int64_t id, boost::json::object&& result);
std::string ErrorResponse (JSONRPCErrorCode errorCode, int64_t id, std::string_view message);
int64_t GetTag (boost::json::object& jsonRequest) const;
std::string GetNewFieldName (std::string_view fieldName) const;
std::string HandleTorrrentAdd (boost::json::object&& jsonRequest);
std::string HandleTorrrentRemove (boost::json::object&& jsonRequest);
@@ -91,6 +93,21 @@ namespace torrents
return 0;
}
std::string JSONRPCHandler::GetNewFieldName (std::string_view fieldName) const
{
std::string newFieldName;
newFieldName.reserve (fieldName.length ());
for (auto it: fieldName)
if (std::isupper (it))
{
newFieldName.push_back ('_');
newFieldName.push_back (std::tolower (it));
}
else
newFieldName.push_back (it);
return newFieldName;
}
std::string JSONRPCHandler::HandleRequest (std::string_view request)
{
try
@@ -170,6 +187,7 @@ namespace torrents
return SuccessResponse (GetTag (jsonRequest), std::move (response));
}
static boost::json::value GetFieldValue (std::string_view field, std::shared_ptr<Torrent> torrent);
std::string JSONRPCHandler::HandleTorrrentGet (boost::json::object&& jsonRequest)
{
auto arguments = jsonRequest.at ("arguments").as_object ();
@@ -186,6 +204,7 @@ namespace torrents
}
else
torrentIds = m_Tunnel->GetTorrentIDs ();
auto fields = arguments.at ("fields").as_array ();
boost::json::object response;
boost::json::array torrents;
@@ -195,27 +214,16 @@ namespace torrents
if (torrent)
{
boost::json::object t;
t["id"] = id;
t["error"] = 0; // no error
t["name"] = torrent->GetName ();
t["status"] = torrent->GetStatus ();
t["is_finished"] = torrent->IsComplete ();
t["size_when_done"] = torrent->GetLength ();
t["left_until_done"] = torrent->GetLeft ();
t["eta"] = 600; // TODO
t["rate_download"] = torrent->GetDownloadRate ();
t["rate_upload"] = torrent->GetUploadRate ();
t["upload_ratio"] = 1; // TODO:
t["peers_getting_from_us"] = torrent->GetNumUploadingToPeers ();
t["peers_sending_to_us"] = torrent->GetNumDownloadingFromPeers ();
if (torrentsRequested)
for (const auto& field: fields)
{
t["piece_count"] = torrent->GetNumPieces ();
t["piece_size"] = torrent->GetNumPieces ();
t["total_size"] = torrent->GetLength ();
std::string hexHash;
boost::algorithm::hex (torrent->GetInfoHash ().begin(), torrent->GetInfoHash ().end(), std::back_inserter(hexHash));
t["hash_string"] = hexHash;
if (field == "id")
t["id"] = id;
else
{
auto fieldValue = GetFieldValue (field.as_string (), torrent);
if (!fieldValue.is_null ())
t[GetNewFieldName (field.as_string ())] = fieldValue;
}
}
torrents.push_back (t);
}
@@ -224,6 +232,42 @@ namespace torrents
return ResultResponse (GetTag (jsonRequest), std::move (response));
}
static boost::json::value GetFieldValue (std::string_view field, std::shared_ptr<Torrent> torrent)
{
const static std::map<std::string_view, std::function<boost::json::value (std::shared_ptr<Torrent>)> > fields =
{
{ "name", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetName ()); } },
{ "status", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetStatus ()); } },
{ "isFinished", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->IsComplete ()); } },
{ "sizeWhenDone", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetLength ()); } },
{ "leftUntilDone", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetLeft ()); } },
{ "rateDownload", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetDownloadRate ()); } },
{ "rateUpload", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetUploadRate ()); } },
{ "peersGettingFromUs", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetNumUploadingToPeers ()); } },
{ "peersSendingToUs", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetNumDownloadingFromPeers ()); } },
{ "pieceCount", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetNumPieces ()); } },
{ "pieceSize", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetPieceLength ()); } },
{ "totalSize", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(torrent->GetLength ()); } },
{ "hashString", [](std::shared_ptr<Torrent> torrent)
{
std::string hexHash;
boost::algorithm::hex (torrent->GetInfoHash ().begin(), torrent->GetInfoHash ().end(), std::back_inserter(hexHash));
return boost::json::value(hexHash);
}
},
{ "error", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(0); } }, // no error
{ "eta", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(0); } }, // TODO:
{ "uploadRatio", [](std::shared_ptr<Torrent> torrent) { return boost::json::value(1); } } // TODO:
};
if (torrent)
{
auto it = fields.find (field);
if (it != fields.end ())
return it->second (torrent);
}
return boost::json::value ();
}
#endif
TorrentsRPCSession::TorrentsRPCSession (TorrentsRPCServer& server, boost::asio::ip::tcp::socket&& s):