traxor/src/app/torrent.rs

use color_eyre::eyre::Result;
use std::{collections::HashSet, fmt::Debug};
use transmission_rpc::{
    types::{Torrent, TorrentGetField},
    TransClient,
};
use url::Url;

/// List of torrents.
pub struct Torrents {
    /// Constructs a new instance of [`Torrents`].
    pub client: TransClient,
    pub torrents: Vec<Torrent>,
    pub selected: HashSet<i64>,
    pub fields: Option<Vec<TorrentGetField>>,
}

impl Torrents {
    /// Constructs a new instance of [`Torrents`].
    pub fn new() -> Result<Torrents> {
        let url = Url::parse("http://localhost:9091/transmission/rpc")?;
        Ok(Self {
            client: TransClient::new(url),
            torrents: Vec::new(),
            selected: HashSet::new(),
            fields: None,
        })
    }

    /// Returns the number of [`Torrent`]s in [`Torrents`]
    pub fn len(&self) -> usize {
        self.torrents.len()
    }

    /// Sets `self.fields`
    pub fn set_fields(&mut self, fields: Option<Vec<TorrentGetField>>) -> &mut Self {
        self.fields = fields;
        self
    }

    /// Sets
    pub fn url(&mut self, url: &str) -> Result<&mut Self> {
        self.client = TransClient::new(Url::parse(url)?);
        Ok(self)
    }

    /// Updates [`Torrent`] values.
    pub async fn update(&mut self) -> Result<&mut Self> {
        self.torrents = self
            .client
            .torrent_get(self.fields.clone(), None)
            .await
            .map_err(|e| color_eyre::eyre::eyre!("Transmission RPC error: {}", e.to_string()))?
            .arguments
            .torrents;
        Ok(self)
    }
}

impl Debug for Torrents {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let fields: Vec<String> = match &self.fields {
            Some(fields) => fields.iter().map(|field| field.to_str()).collect(),
            None => vec![String::from("None")],
        };
        write!(
            f,
            "fields:
        {:?};\n\ntorrents: {:?}",
            fields, self.torrents
        )
    }
}