refactor title work

Author: raphamorim

docs/docs/config/title.md

@@ -7,14 +7,16 @@ Rio allows you to configure window and tabs title through configuration via temp
 
 - `content` - Configure window title using template
 
-  - Default: `{{ TITLE || PROGRAM }}`
+  - Default: `{{ title || program }}`
 
 - `placeholder` - Configure initial title
 
   - Default: `▲`
 
 ## content
 
+Note: Variables are not case sensitive.
+
 Possible options:
 
 - `TITLE`: terminal title via OSC sequences for setting terminal title
@@ -37,7 +39,7 @@ Result: `fish - .../Documents/a/rio`.
 
 ```toml
 [title]
-content = "{{ PROGRAM }} ({{COLUMNS}}x{{LINES}})"
+content = "{{ program }} ({{columns}}x{{lines}})"
 ```
 
 Result: `fish (85x23)`.

frontends/rioterm/src/context/mod.rs

@@ -1,11 +1,12 @@
 pub mod grid;
 pub mod renderable;
-mod title;
+pub mod title;
 
 use crate::ansi::CursorShape;
 use crate::context::grid::ContextDimension;
 use crate::context::grid::ContextGrid;
 use crate::context::grid::Delta;
+use crate::context::title::{update_title, ContextManagerTitles};
 use crate::event::sync::FairMutex;
 use crate::event::RioEvent;
 use crate::ime::Ime;
@@ -21,7 +22,6 @@ use rio_backend::event::WindowId;
 use rio_backend::selection::SelectionRange;
 use rio_backend::sugarloaf::{font::SugarloafFont, Object, SugarloafErrors};
 use std::borrow::Cow;
-use std::collections::HashMap;
 use std::error::Error;
 use std::sync::Arc;
 use std::time::{Duration, Instant};
@@ -130,43 +130,7 @@ pub struct ContextManagerConfig {
     pub is_native: bool,
     pub should_update_titles: bool,
     pub split_color: [f32; 4],
-}
-
-pub struct ContextManagerTitles {
-    last_title_update: Option<Instant>,
-    pub titles: HashMap<usize, [String; 3]>,
-    pub key: String,
-}
-
-impl ContextManagerTitles {
-    pub fn new(
-        idx: usize,
-        program: String,
-        terminal_title: String,
-        path: String,
-    ) -> ContextManagerTitles {
-        ContextManagerTitles {
-            key: format!("{}{}{};", idx, program, terminal_title),
-            titles: HashMap::from([(idx, [program, terminal_title, path])]),
-            last_title_update: None,
-        }
-    }
-
-    #[inline]
-    pub fn set_key_val(
-        &mut self,
-        idx: usize,
-        program: String,
-        terminal_title: String,
-        path: String,
-    ) {
-        self.titles.insert(idx, [program, terminal_title, path]);
-    }
-
-    #[inline]
-    pub fn set_key(&mut self, key: String) {
-        self.key = key;
-    }
+    pub title: rio_backend::config::title::Title,
 }
 
 pub struct ContextManager<T: EventListener> {
@@ -236,7 +200,7 @@ pub fn create_mock_context<
         is_native: false,
         should_update_titles: false,
         use_current_path: false,
-        split_color: [0., 0., 0., 0.],
+        ..ContextManagerConfig::default()
     };
     ContextManager::create_context(
         (&Cursor::default(), false),
@@ -408,7 +372,6 @@ impl<T: EventListener + Clone + std::marker::Send + 'static> ContextManager<T> {
         let titles = ContextManagerTitles::new(
             0,
             String::from("tab"),
-            String::new(),
             ctx_config.working_dir.clone().unwrap_or_default(),
         );
 
@@ -462,7 +425,7 @@ impl<T: EventListener + Clone + std::marker::Send + 'static> ContextManager<T> {
             is_native: false,
             should_update_titles: false,
             use_current_path: false,
-            split_color: [0., 0., 0., 0.],
+            ..ContextManagerConfig::default()
         };
         let initial_context = ContextManager::create_context(
             (&Cursor::default(), false),
@@ -474,8 +437,7 @@ impl<T: EventListener + Clone + std::marker::Send + 'static> ContextManager<T> {
             &config,
         )?;
 
-        let titles =
-            ContextManagerTitles::new(0, String::new(), String::new(), String::new());
+        let titles = ContextManagerTitles::new(0, String::new(), String::new());
 
         Ok(ContextManager {
             current_index: 0,
@@ -693,64 +655,29 @@ impl<T: EventListener + Clone + std::marker::Send + 'static> ContextManager<T> {
             .unwrap_or(true)
         {
             self.titles.last_title_update = Some(Instant::now());
-            #[cfg(unix)]
-            {
-                let mut id = String::default();
-                for (i, context) in self.contexts.iter_mut().enumerate() {
-                    let program = teletypewriter::foreground_process_name(
-                        *context.current().main_fd,
-                        context.current().shell_pid,
-                    );
-
-                    let path = teletypewriter::foreground_process_path(
-                        *context.current().main_fd,
-                        context.current().shell_pid,
-                    )
-                    .map(|p| p.to_string_lossy().to_string())
-                    .unwrap_or_default();
-
-                    let terminal_title = {
-                        let terminal = context.current().terminal.lock();
-                        terminal.title.to_string()
-                    };
-
-                    let window_title = if terminal_title.is_empty() {
-                        program.to_owned()
-                    } else {
-                        format!("{} ({})", terminal_title, program)
-                    };
-
-                    if cfg!(target_os = "macos") {
-                        self.event_proxy.send_event(
-                            RioEvent::TitleWithSubtitle(window_title, path.clone()),
-                            self.window_id,
-                        );
-                    } else {
-                        self.event_proxy
-                            .send_event(RioEvent::Title(window_title), self.window_id);
-                    }
+            let mut id = String::default();
+            for (i, context) in self.contexts.iter_mut().enumerate() {
+                let content = update_title(&self.config.title.content, context.current());
+
+                #[cfg(unix)]
+                let path = teletypewriter::foreground_process_path(
+                    *context.current().main_fd,
+                    context.current().shell_pid,
+                )
+                .map(|p| p.to_string_lossy().to_string())
+                .unwrap_or_default();
 
-                    id.push_str(&format!("{}{}{};", i, program, terminal_title));
-                    self.titles.set_key_val(i, program, terminal_title, path);
-                }
-                self.titles.set_key(id);
-            }
+                #[cfg(not(unix))]
+                let path = String::default();
 
-            #[cfg(not(unix))]
-            {
-                let mut id = String::from("");
-                for (i, _context) in self.contexts.iter().enumerate() {
-                    let program = self.config.shell.program.to_owned();
-                    id.push_str(&format!("{}{}{};", i, program, String::default()));
-                    self.titles.set_key_val(
-                        i,
-                        program,
-                        String::default(),
-                        String::default(),
-                    );
-                }
-                self.titles.set_key(id);
+                self.event_proxy
+                    .send_event(RioEvent::Title(content.to_owned()), self.window_id);
+
+                id.push_str(&format!("{}{};", i, content));
+                self.titles.set_key_val(i, content, path);
             }
+
+            self.titles.set_key(id);
         }
     }
 
@@ -985,6 +912,7 @@ impl<T: EventListener + Clone + std::marker::Send + 'static> ContextManager<T> {
             should_update_titles: !(config.navigation.is_collapsed_mode()
                 && config.navigation.color_automation.is_empty()),
             split_color: config.colors.split,
+            title: config.title,
         };
 
         self.acc_current_route += 1;