Add service selection to hydrus file wrapper operations

Signed-off-by: trivernis <trivernis@protonmail.com>

src/wrapper/hydrus_file.rs

@@ -239,12 +239,32 @@ impl HydrusFile {
         DeleteFilesBuilder::new(self.client.clone()).add_file(self.id.clone())
     }
 
-    /// Undeletes the file
+    /// Undeletes the file for the given service or all services
-    pub async fn undelete(&mut self) -> Result<()> {
+    /// if `FileServiceSelection::none` is passed
+    pub async fn undelete(&mut self, service_selection: FileServiceSelection) -> Result<()> {
         let hash = self.hash().await?;
         self.metadata = None;
         self.client
-            .undelete_files(FileSelection::by_hash(hash), FileServiceSelection::none())
+            .undelete_files(FileSelection::by_hash(hash), service_selection)
+            .await
+    }
+
+    /// Archives the file in all passed file services or all configured services
+    /// if no selection is passed
+    pub async fn archive(&mut self, service_selection: FileServiceSelection) -> Result<()> {
+        let hash = self.hash().await?;
+        self.metadata = None;
+        self.client
+            .archive_files(FileSelection::by_hash(hash), service_selection)
+            .await
+    }
+
+    /// Unarchives the file for the given services
+    pub async fn unarchive(&mut self, service_selection: FileServiceSelection) -> Result<()> {
+        let hash = self.hash().await?;
+        self.metadata = None;
+        self.client
+            .unarchive_files(FileSelection::by_hash(hash), service_selection)
             .await
     }
 

tests/wrapper/test_files.rs

@@ -117,9 +117,8 @@ async fn it_deletes() {
     let mut file = get_file().await;
     file.delete()
         .reason("I just don't like that file")
-        .service(ServiceName::all_local_files().into())
         .run()
         .await
         .unwrap();
-    file.undelete().await.unwrap();
+    file.undelete(ServiceName::my_files().into()).await.unwrap();
 }