feat: configurable headers (#55)

Cargo.toml

@@ -31,6 +31,7 @@ rss = "2"
 reqwest = "0.12"
 regex = "1"
 thiserror = "1"
+chrono = "0.4"
 
 # Async
 tokio = { version = "1", features = ["macros", "sync"] }

README.md

@@ -78,14 +78,22 @@ headers_hide = false
 [connection]
 url = "http://CHANGE_ME:9091/transmission/rpc" # REQUIRED!
 
+# Refresh timings (in seconds)
+torrents_refresh = 5
+stats_refresh = 5
+free_space_refresh = 10
+
 # If you need username and password to authenticate:
 # username = "CHANGE_ME"
 # password = "CHANGE_ME"
 
-# Refresh timings (in seconds)
-torrents_refresh = 5
-stats_refresh = 10
-free_space_refresh = 10
+[torrents_tab]
+# Available fields:
+# Id, Name, SizeWhenDone, Progress, DownloadRate, UploadRate, DownloadDir,
+# Padding, UploadRatio, UploadedEver, AddedDate, ActivityDate, PeersConnected
+# SmallStatus
+headers = ["Name", "SizeWhenDone", "Progress", "DownloadRate", "UploadRate"]
+
 ```
 
 There's also a self-documenting keymap config located at `~/.config/rustmission/keymap.toml` with sane defaults.

rm-config/Cargo.toml

@@ -20,3 +20,4 @@ url.workspace = true
 ratatui.workspace = true
 crossterm.workspace = true
 thiserror.workspace = true
+transmission-rpc.workspace = true

rm-config/defaults/config.toml

@@ -17,11 +17,19 @@ headers_hide = false
 [connection]
 url = "http://CHANGE_ME:9091/transmission/rpc" # REQUIRED!
 
+# Refresh timings (in seconds)
+torrents_refresh = 5
+stats_refresh = 5
+free_space_refresh = 10
+
 # If you need username and password to authenticate:
 # username = "CHANGE_ME"
 # password = "CHANGE_ME"
 
-# Refresh timings (in seconds)
-torrents_refresh = 5
-stats_refresh = 10
-free_space_refresh = 10
+
+[torrents_tab]
+# Available fields:
+# Id, Name, SizeWhenDone, Progress, DownloadRate, UploadRate, DownloadDir,
+# Padding, UploadRatio, UploadedEver, AddedDate, ActivityDate, PeersConnected
+# SmallStatus
+headers = ["Name", "SizeWhenDone", "Progress", "DownloadRate", "UploadRate"]

rm-config/src/lib.rs

@@ -1,5 +1,5 @@
 pub mod keymap;
-mod main_config;
+pub mod main_config;
 mod utils;
 
 use std::path::PathBuf;
@@ -11,6 +11,7 @@ use main_config::MainConfig;
 pub struct Config {
     pub general: main_config::General,
     pub connection: main_config::Connection,
+    pub torrents_tab: main_config::TorrentsTab,
     pub keybindings: KeymapConfig,
     pub directories: Directories,
 }
@@ -33,6 +34,7 @@ impl Config {
         Ok(Self {
             general: main_config.general,
             connection: main_config.connection,
+            torrents_tab: main_config.torrents_tab,
             keybindings: keybindings.clone(),
             directories,
         })

rm-config/src/main_config.rs

@@ -1,19 +1,21 @@
-use std::{path::PathBuf, sync::OnceLock};
+use std::{io::ErrorKind, path::PathBuf, sync::OnceLock};
 
 use anyhow::Result;
-use ratatui::style::Color;
+use ratatui::{layout::Constraint, style::Color};
 use serde::{Deserialize, Serialize};
 use url::Url;
 
-use crate::utils::{self, put_config};
+use crate::utils::{self};
 
-#[derive(Serialize, Deserialize)]
+#[derive(Deserialize)]
 pub struct MainConfig {
     pub general: General,
     pub connection: Connection,
+    #[serde(default)]
+    pub torrents_tab: TorrentsTab,
 }
 
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Deserialize)]
 pub struct General {
     #[serde(default)]
     pub auto_hide: bool,
@@ -33,7 +35,7 @@ fn default_beginner_mode() -> bool {
     true
 }
 
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Deserialize)]
 pub struct Connection {
     pub username: Option<String>,
     pub password: Option<String>,
@@ -50,19 +52,103 @@ fn default_refresh() -> u64 {
     5
 }
 
+#[derive(Serialize, Deserialize, Hash, PartialEq, Eq, Clone, Copy)]
+pub enum Header {
+    Name,
+    SizeWhenDone,
+    Progress,
+    Eta,
+    DownloadRate,
+    UploadRate,
+    DownloadDir,
+    Padding,
+    UploadRatio,
+    UploadedEver,
+    Id,
+    ActivityDate,
+    AddedDate,
+    PeersConnected,
+    SmallStatus,
+}
+
+impl Header {
+    pub fn default_constraint(&self) -> Constraint {
+        match self {
+            Self::Name => Constraint::Max(70),
+            Self::SizeWhenDone => Constraint::Length(12),
+            Self::Progress => Constraint::Length(12),
+            Self::Eta => Constraint::Length(12),
+            Self::DownloadRate => Constraint::Length(12),
+            Self::UploadRate => Constraint::Length(12),
+            Self::DownloadDir => Constraint::Max(70),
+            Self::Padding => Constraint::Length(2),
+            Self::UploadRatio => Constraint::Length(6),
+            Self::UploadedEver => Constraint::Length(12),
+            Self::Id => Constraint::Length(4),
+            Self::ActivityDate => Constraint::Length(14),
+            Self::AddedDate => Constraint::Length(12),
+            Self::PeersConnected => Constraint::Length(6),
+            Self::SmallStatus => Constraint::Length(1),
+        }
+    }
+
+    pub fn header_name(&self) -> &'static str {
+        match *self {
+            Self::Name => "Name",
+            Self::SizeWhenDone => "Size",
+            Self::Progress => "Progress",
+            Self::Eta => "ETA",
+            Self::DownloadRate => "Download",
+            Self::UploadRate => "Upload",
+            Self::DownloadDir => "Directory",
+            Self::Padding => "",
+            Self::UploadRatio => "Ratio",
+            Self::UploadedEver => "Up Ever",
+            Self::Id => "Id",
+            Self::ActivityDate => "Last active",
+            Self::AddedDate => "Added",
+            Self::PeersConnected => "Peers",
+            Self::SmallStatus => "",
+        }
+    }
+}
+
+#[derive(Deserialize)]
+pub struct TorrentsTab {
+    pub headers: Vec<Header>,
+}
+
+impl Default for TorrentsTab {
+    fn default() -> Self {
+        Self {
+            headers: vec![
+                Header::Name,
+                Header::SizeWhenDone,
+                Header::Progress,
+                Header::Eta,
+                Header::DownloadRate,
+                Header::UploadRate,
+            ],
+        }
+    }
+}
+
 impl MainConfig {
     pub(crate) const FILENAME: &'static str = "config.toml";
     const DEFAULT_CONFIG: &'static str = include_str!("../defaults/config.toml");
 
     pub(crate) fn init() -> Result<Self> {
-        let Ok(config) = utils::fetch_config(Self::FILENAME) else {
-            put_config(Self::DEFAULT_CONFIG, Self::FILENAME)?;
-            // TODO: check if the user really changed the config.
-            println!("Update {:?} and start rustmission again", Self::path());
-            std::process::exit(0);
+        match utils::fetch_config::<Self>(Self::FILENAME) {
+            Ok(config) => return Ok(config),
+            Err(e) => match e {
+                utils::ConfigFetchingError::Io(e) if e.kind() == ErrorKind::NotFound => {
+                    utils::put_config::<Self>(Self::DEFAULT_CONFIG, Self::FILENAME)?;
+                    println!("Update {:?} and start rustmission again", Self::path());
+                    std::process::exit(0);
+                }
+                _ => anyhow::bail!(e),
+            },
         };
-
-        Ok(config)
     }
 
     pub(crate) fn path() -> &'static PathBuf {

rm-config/src/utils.rs

@@ -42,9 +42,9 @@ pub fn fetch_config<T: DeserializeOwned>(config_name: &str) -> Result<T, ConfigF
 pub fn put_config<T: DeserializeOwned>(
     content: &'static str,
     filename: &str,
-) -> Result<T, io::Error> {
+) -> Result<T, ConfigFetchingError> {
     let config_path = get_config_path(filename);
     let mut config_file = File::create(config_path)?;
     config_file.write_all(content.as_bytes())?;
-    Ok(toml::from_str(content).expect("default configs are correct"))
+    Ok(toml::from_str(content)?)
 }

rm-main/Cargo.toml

@@ -34,3 +34,4 @@ rss.workspace = true
 reqwest.workspace = true
 regex.workspace = true
 throbber-widgets-tui.workspace = true 
+chrono.workspace = true

rm-main/src/transmission/fetchers.rs

@@ -72,6 +72,10 @@ pub async fn torrents(ctx: app::Ctx, table_manager: Arc<Mutex<TableManager>>) {
             TorrentGetField::RateDownload,
             TorrentGetField::Status,
             TorrentGetField::DownloadDir,
+            TorrentGetField::UploadedEver,
+            TorrentGetField::ActivityDate,
+            TorrentGetField::AddedDate,
+            TorrentGetField::PeersConnected,
         ];
         let rpc_response = ctx
             .client

rm-main/src/ui/tabs/torrents/bottom_stats.rs

@@ -40,7 +40,7 @@ impl Component for BottomStats {
             let download = bytes_to_human_format(stats.download_speed);
             let upload = bytes_to_human_format(stats.upload_speed);
 
-            let mut text = format!("▼ {download} | ▲ {upload}");
+            let mut text = format!(" {download} |  {upload}");
 
             if let Some(free_space) = &*self.free_space.lock().unwrap() {
                 let free_space = bytes_to_human_format(free_space.size_bytes);

rm-main/src/ui/tabs/torrents/mod.rs

@@ -121,12 +121,10 @@ impl TorrentsTab {
             .accent_color);
 
         let table_widget = {
-            let table = Table::new(torrent_rows, table_manager_lock.widths)
+            let table = Table::new(torrent_rows, &table_manager_lock.widths)
                 .highlight_style(highlight_table_style);
             if !self.ctx.config.general.headers_hide {
-                table.header(Row::new(
-                    table_manager_lock.header().iter().map(|s| s.as_str()),
-                ))
+                table.header(Row::new(table_manager_lock.header().iter().cloned()))
             } else {
                 table
             }