Add get_files method on QBittorrentClient

Author: angrynode

Cargo.toml

@@ -16,7 +16,8 @@ default = [ "qbittorrent" ]
 qbittorrent = [ "reqwest" ]
 
 [dependencies]
-hightorrent = { version = "0.2", path = "../hightorrent" }
+# hightorrent = { version = "0.2", path = "../hightorrent" }
+hightorrent = { git = "https://github.com/angrynode/hightorrent" }
 tokio = { version = "1", features = [ "fs" ] }
 tokio-util = { version = "0.7" }
 async-trait = "0.1"

src/api/mod.rs

@@ -1,5 +1,5 @@
 use crate::ApiError;
-use hightorrent::{SingleTarget, Torrent, TorrentList, Tracker};
+use hightorrent::{SingleTarget, Torrent, TorrentContent, TorrentList, Tracker};
 
 mod add;
 pub use add::*;
@@ -26,4 +26,6 @@ pub trait Api: Send + Sync + for<'a> ApiAdd<'a> {
     async fn get_trackers(&self, hash: &SingleTarget) -> Result<Vec<Tracker>, ApiError>;
     async fn add_tracker(&self, hash: &SingleTarget, tracker: &str) -> Result<(), ApiError>;
     async fn remove_tracker(&self, hash: &SingleTarget, tracker: &str) -> Result<(), ApiError>;
+
+    async fn get_files(&self, hash: &SingleTarget) -> Result<Vec<TorrentContent>, ApiError>;
 }

src/qbittorrent/api.rs

@@ -1,6 +1,6 @@
 use hightorrent::{
-    InfoHash, MultiTarget, SingleTarget, ToTorrent, Torrent, TorrentID, TorrentList, Tracker,
-    TryIntoTracker,
+    InfoHash, MultiTarget, SingleTarget, ToTorrent, ToTorrentContent, Torrent,
+    TorrentContent, TorrentID, TorrentList, Tracker, TryIntoTracker,
 };
 use reqwest::multipart::Form;
 use reqwest::multipart::Part;
@@ -14,7 +14,7 @@ use std::borrow::Borrow;
 use crate::{
     api::*,
     api_error::{ApiError as Error, *},
-    qbittorrent::{QBittorrentTorrent, QBittorrentTracker},
+    qbittorrent::{QBittorrentTorrent, QBittorrentTorrentContent, QBittorrentTracker},
 };
 
 #[derive(Clone)]
@@ -296,6 +296,32 @@ impl Api for QBittorrentClient {
             })
         }
     }
+
+    async fn get_files(&self, target: &SingleTarget) -> Result<Vec<TorrentContent>, Error> {
+        let Some(id) = self.id(target).await? else {
+            return Err(Error::MissingTorrent {
+                hash: target.as_str().to_string(),
+            });
+        };
+
+        let mut form = Form::new();
+        form = form.text("hash", id.as_str().to_string());
+        let res = self
+            ._post_multipart(self._endpoint("torrents/files"), form)
+            .await?;
+
+        if res.status().is_success() {
+            let concrete: Vec<QBittorrentTorrentContent> = self._json(res).await?;
+            Ok(concrete
+                .iter()
+                .map(|t| t.to_torrent_content())
+                .collect())
+        } else {
+            Err(Error::MissingTorrent {
+                hash: target.as_str().to_string(),
+            })
+        }
+    }
 }
 
 #[async_trait]

src/qbittorrent/mod.rs

@@ -2,4 +2,4 @@ mod api;
 pub use api::QBittorrentClient;
 
 mod torrent;
-pub use torrent::{QBittorrentTorrent, QBittorrentTracker};
+pub use torrent::{QBittorrentTorrent, QBittorrentTorrentContent, QBittorrentTracker};

src/qbittorrent/torrent.rs

@@ -1,6 +1,11 @@
-use hightorrent::{InfoHash, ToTorrent, Torrent, TorrentID, Tracker, TrackerError, TryIntoTracker};
+use hightorrent::{
+    InfoHash, ToTorrent, ToTorrentContent, Torrent, TorrentContent, TorrentID, Tracker,
+    TrackerError, TryIntoTracker,
+};
 use serde::{Deserialize, Deserializer, Serialize};
 
+use std::path::PathBuf;
+
 /// Deserializes from the 'info' endpoint of QBittorrent API
 /// [See QBittorrent API docs](https://github.com/qbittorrent/qBittorrent/wiki/WebUI-API-(qBittorrent-4.1)#get-torrent-list)
 #[derive(Clone, Debug, Deserialize)]
@@ -82,3 +87,22 @@ impl PartialEq for QBittorrentTracker {
         self.url == other.url
     }
 }
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct QBittorrentTorrentContent {
+    #[serde(rename = "name")]
+    pub path: PathBuf,
+    pub size: u32,
+    pub progress: f32,
+    #[serde(default)]
+    pub is_seed: bool,
+}
+
+impl ToTorrentContent for QBittorrentTorrentContent {
+    fn to_torrent_content(&self) -> TorrentContent {
+        TorrentContent {
+            path: self.path.clone(),
+            size: self.size as u64,
+        }
+    }
+}