leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add with_stats=true to torrents list

Browse source
This commit is contained in:
Igor Katson 2024-11-06 17:19:33 +00:00
parent 30bfb3ea19
commit a90f245dc9
4 changed files with 37 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

42
crates/librqbit/src/api.rs
View file

@ -168,6 +168,12 @@ impl TorrentIdOrHash {
}
}
#[derive(Deserialize, Default)]
pub struct ApiTorrentListOpts {
#[serde(default)]
pub with_stats: bool,
}
impl Api {
pub fn new(
session: Arc<Session>,
@ -193,22 +199,32 @@ impl Api {
}
pub fn api_torrent_list(&self) -> TorrentListResponse {
self.api_torrent_list_ext(ApiTorrentListOpts { with_stats: false })
}
pub fn api_torrent_list_ext(&self, opts: ApiTorrentListOpts) -> TorrentListResponse {
let items = self.session.with_torrents(|torrents| {
torrents
.map(|(id, mgr)| TorrentDetailsResponse {
id: Some(id),
info_hash: mgr.shared().info_hash.as_string(),
name: mgr.shared().info.name.as_ref().map(|n| n.to_string()),
output_folder: mgr
.shared()
.options
.output_folder
.to_string_lossy()
.into_owned(),
.map(|(id, mgr)| {
let mut r = TorrentDetailsResponse {
id: Some(id),
info_hash: mgr.shared().info_hash.as_string(),
name: mgr.shared().info.name.as_ref().map(|n| n.to_string()),
output_folder: mgr
.shared()
.options
.output_folder
.to_string_lossy()
.into_owned(),
// These will be filled in /details and /stats endpoints
files: None,
stats: None,
// These will be filled in /details and /stats endpoints
files: None,
stats: None,
};
if opts.with_stats {
r.stats = Some(mgr.stats());
}
r
})
.collect()
});