feature: show hostname in title

Cargo.toml

@@ -94,6 +94,7 @@ tui = "0.19.0"
 typed-builder = "0.10.0"
 unicode-segmentation = "1.10.0"
 unicode-width = "0.1.10"
+gethostname = "0.4.1"
 
 [target.'cfg(unix)'.dependencies]
 libc = "0.2.124"

sample_configs/default_config.toml

@@ -61,6 +61,8 @@
 #mem_as_value = false
 # Show tree mode by default in the processes widget.
 #tree = false
+# Set terminal name to hostname
+#title_to_hostname = false
 # Shows an indicator in table widgets tracking where in the list you are.
 #show_table_scroll_position = false
 # Show processes as their commands by default in the process widget.

src/bin/main.rs

@@ -27,7 +27,7 @@ use bottom::{
 use crossterm::{
     event::{EnableBracketedPaste, EnableMouseCapture},
     execute,
-    terminal::{enable_raw_mode, EnterAlternateScreen},
+    terminal::{enable_raw_mode, EnterAlternateScreen, SetTitle},
 };
 use tui::{backend::CrosstermBackend, Terminal};
 
@@ -123,8 +123,14 @@ fn main() -> Result<()> {
         stdout_val,
         EnterAlternateScreen,
         EnableMouseCapture,
-        EnableBracketedPaste
+        EnableBracketedPaste,
     )?;
+
+    let use_terminal_name = get_use_terminal_name(&matches, &config);
+    if use_terminal_name.0 {
+        execute!(stdout_val, SetTitle(use_terminal_name.1),)?;
+    }
+
     enable_raw_mode()?;
 
     let mut terminal = Terminal::new(CrosstermBackend::new(stdout_val))?;

src/clap.rs

@@ -338,6 +338,12 @@ use CPU (3) as the default instead.
         .help("The amount in ms changed upon zooming.")
         .long_help("The amount of time in milliseconds changed when zooming in/out. The minimum is 1s (1000), and defaults to 15s (15000).");
 
+    let title = Arg::new("title")
+        .long("title")
+        .takes_value(true)
+        .value_name("Title")
+        .help("Sets the title of the current terminal.");
+
     let tree = Arg::new("tree")
         .short('T')
         .long("tree")
@@ -410,6 +416,7 @@ use CPU (3) as the default instead.
         .arg(rate)
         .arg(regex)
         .arg(time_delta)
+        .arg(title)
         .arg(tree)
         .arg(network_use_bytes)
         .arg(network_use_log)

src/options.rs

@@ -31,6 +31,7 @@ use crate::{
 pub mod layout_options;
 
 use anyhow::{Context, Result};
+use gethostname::gethostname;
 
 #[derive(Clone, Debug, Default, Deserialize, Serialize)]
 pub struct Config {
@@ -92,6 +93,7 @@ pub struct ConfigFlags {
     pub network_use_log: Option<bool>,
     pub network_use_binary_prefix: Option<bool>,
     pub enable_gpu_memory: Option<bool>,
+    pub title_to_hostname: Option<bool>,
     #[serde(with = "humantime_serde")]
     #[serde(default)]
     pub retention: Option<Duration>,
@@ -732,6 +734,26 @@ pub fn get_app_use_regex(matches: &ArgMatches, config: &Config) -> bool {
     false
 }
 
+pub fn get_use_terminal_name(matches: &ArgMatches, config: &Config) -> (bool, String) {
+    if matches.is_present("title") {
+        return if let Some(custom_name) = matches.value_of("title") {
+            (true, String::from(custom_name))
+        } else {
+            (false, String::from(""))
+        };
+    } else if let Some(flags) = &config.flags {
+        if let Some(title_to_hostname) = flags.title_to_hostname {
+            if title_to_hostname {
+                return match gethostname().into_string() {
+                    Ok(hostname) => (title_to_hostname, format!("btm ({})", hostname)),
+                    Err(_) => (false, String::from("")),
+                };
+            }
+        }
+    }
+    (false, String::from(""))
+}
+
 fn get_hide_time(matches: &ArgMatches, config: &Config) -> bool {
     if matches.is_present("hide_time") {
         return true;