55 lines
1.8 KiB
Rust
55 lines
1.8 KiB
Rust
|
use transmission_rpc::TransClient;
|
||
|
|
use transmission_rpc::types::{Result, BasicAuth, Torrent, Torrents, Id};
|
||
|
|
use log::{info};
|
||
|
|
|
||
|
|
pub struct App {
|
||
|
|
client: TransClient,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl App {
|
||
|
|
pub fn new(url: &String, username: &String, password: &String) -> Self {
|
||
|
|
let auth = BasicAuth { user: username.clone(), password: password.clone() };
|
||
|
|
let client = TransClient::with_auth(url.as_str(), auth);
|
||
|
|
Self {
|
||
|
|
client
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn run(&self) -> Result<()> {
|
||
|
|
let response = self.client.torrent_get(None, None).await;
|
||
|
|
match response {
|
||
|
|
Ok(res) => self.scan_files(res.arguments).await.unwrap(),
|
||
|
|
Err(e) => println!("{}", e.to_string())
|
||
|
|
}
|
||
|
|
|
||
|
|
return Ok(());
|
||
|
|
}
|
||
|
|
async fn scan_files(&self, torrents: Torrents<Torrent>) -> Result<()> {
|
||
|
|
let mut torrents_to_remove = Vec::new();
|
||
|
|
|
||
|
|
for torrent in torrents.torrents {
|
||
|
|
match (torrent.id, torrent.is_stalled) {
|
||
|
|
(Some(id), Some(true)) => {
|
||
|
|
info!("processing: (id: {}, name: {})", id, torrent.name.unwrap_or("name could not be found from torrent".to_string()));
|
||
|
|
torrents_to_remove.push(id);
|
||
|
|
}
|
||
|
|
_ => {
|
||
|
|
info!("did not fit criteria (everything is good for this torrent)");
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if torrents_to_remove.len() != 0 {
|
||
|
|
info!("cleaning up torrents");
|
||
|
|
self.clean_up_torrents(torrents_to_remove).await;
|
||
|
|
} else {
|
||
|
|
info!("nothing to cleanup")
|
||
|
|
}
|
||
|
|
|
||
|
|
return Ok(());
|
||
|
|
}
|
||
|
|
async fn clean_up_torrents(&self, torrent_ids: Vec<i64>) {
|
||
|
|
self.client.torrent_remove(torrent_ids.iter().map(|ti| { Id::Id(*ti) }).collect(), true).await;
|
||
|
|
}
|
||
|
|
}
|