Color rooms in the buflist based on notifications/highlights

src/buffers/log.rs

@@ -98,4 +98,8 @@ impl Buffer for LogBuffer {
         }),
     )
   }
+
+  fn unread_notification_counts(&self) -> matrix_sdk::sync::UnreadNotificationsCount {
+    Default::default()
+  }
 }

src/buffers/mod.rs

@@ -119,6 +119,8 @@ pub trait Buffer: Send + Sync + memuse::DynamicUsage {
   /// This should return immediately, not waiting for anything to be loaded.
   fn request_back_pagination(&self, num: u16) {}
 
+  fn unread_notification_counts(&self) -> matrix_sdk::sync::UnreadNotificationsCount;
+
   fn dynamic_usage(&self) -> usize {
     memuse::DynamicUsage::dynamic_usage(self)
   }

src/buffers/room.rs

@@ -32,6 +32,7 @@ use matrix_sdk::room::ParentSpace;
 use matrix_sdk::ruma::events::space::child::SpaceChildEventContent;
 use matrix_sdk::ruma::events::SyncStateEvent;
 use matrix_sdk::ruma::{OwnedRoomId, RoomId};
+use matrix_sdk::sync::UnreadNotificationsCount;
 use matrix_sdk::{Client, DisplayName, Room, RoomInfo};
 use matrix_sdk_ui::timeline::{
   BackPaginationStatus, PaginationOptions, RoomExt, Timeline, TimelineItem, TimelineItemKind,
@@ -561,6 +562,22 @@ impl Buffer for RoomBuffer {
       .back_pagination_request
       .fetch_max(num, Ordering::Relaxed);
   }
+
+  fn unread_notification_counts(&self) -> UnreadNotificationsCount {
+    let mut acc = UnreadNotificationsCount::default();
+    for buf in &self.buffers {
+      let Some(room) = buf.client.get_room(&self.room_id) else {
+        continue;
+      };
+      let UnreadNotificationsCount {
+        highlight_count,
+        notification_count,
+      } = room.unread_notification_counts();
+      acc.highlight_count += highlight_count;
+      acc.notification_count += notification_count;
+    }
+    acc
+  }
 }
 
 async fn compute_room_info(

src/components/buflist.rs

@@ -23,6 +23,7 @@ use tokio::sync::OnceCell;
 use super::Component;
 use crate::{
   action::Action,
+  buffers::{Buffer, Buffers},
   config::{Config, KeyBindings},
 };
 
@@ -44,7 +45,7 @@ impl Buflist {
     }
   }
 
-  pub fn num_columns(&mut self, area: Rect, buffers: &crate::buffers::Buffers) -> u16 {
+  pub fn num_columns(&mut self, area: Rect, buffers: &Buffers) -> u16 {
     let borders_height = 2;
     buffers
       .len()
@@ -56,6 +57,25 @@ impl Buflist {
   pub fn borders_width(&self) -> u16 {
     2
   }
+
+  pub fn get_room_name_style(&self, buf: &dyn Buffer) -> Style {
+    use matrix_sdk::sync::UnreadNotificationsCount;
+    let config = &self.config.style.buflist;
+    match buf.unread_notification_counts() {
+      UnreadNotificationsCount {
+        highlight_count: 0,
+        notification_count: 0,
+      } => *config.uneventful,
+      UnreadNotificationsCount {
+        highlight_count: 0,
+        notification_count: _,
+      } => *config.notification,
+      UnreadNotificationsCount {
+        highlight_count: _,
+        notification_count: _,
+      } => *config.highlight,
+    }
+  }
 }
 
 impl Component for Buflist {
@@ -166,10 +186,13 @@ impl Component for Buflist {
         stack.push(buf.id());
         let buf_number = format!("{}.", i + 1);
         let mut base_style = Style::default();
+        let mut name_style = self.get_room_name_style(buf);
         if i == buffers.active_index() {
           base_style = base_style.on_blue();
+          name_style = name_style.on_blue();
         } else {
           base_style = base_style.bg(Color::Reset);
+          name_style = name_style.bg(Color::Reset);
         }
         let tree_pad = " ".repeat(stack.len() - 1);
         let buf_number_style = base_style.green();
@@ -180,7 +203,7 @@ impl Component for Buflist {
               Span::styled(left_pad, base_style),
               Span::styled(buf_number, buf_number_style),
               Span::styled(tree_pad, base_style),
-              Span::styled(buf.short_name(), base_style),
+              Span::styled(buf.short_name(), name_style),
               Span::styled(right_pad.clone(), base_style),
             ]
             .into(),

src/config.rs

@@ -63,6 +63,20 @@ pub struct LayoutConfig {
   pub backlog: BacklogLayoutConfig,
 }
 
+#[derive(Clone, Debug, Deserialize)]
+pub struct BuflistStylesConfig {
+  pub highlight: StyleConfig,
+  pub notification: StyleConfig,
+  // TODO: pub unread_message: StyleConfig,
+  // TODO: pub unread_event: StyleConfig,
+  pub uneventful: StyleConfig,
+}
+
+#[derive(Clone, Debug, Deserialize)]
+pub struct StylesConfig {
+  pub buflist: BuflistStylesConfig,
+}
+
 #[derive(Clone, Debug, Deserialize)]
 pub struct Config {
   #[serde(default, flatten)]
@@ -71,9 +85,8 @@ pub struct Config {
   #[serde(default)]
   pub keybindings: KeyBindings,
   pub mouse: MouseConfig,
+  pub style: StylesConfig,
   pub layout: LayoutConfig,
-  #[serde(default)]
-  pub styles: Styles,
 }
 
 impl Config {
@@ -310,28 +323,17 @@ pub fn parse_key_sequence(raw: &str) -> Result<Vec<KeyEvent>, String> {
   sequences.into_iter().map(parse_key_event).collect()
 }
 
-#[derive(Clone, Debug, Default, Deref, DerefMut)]
-pub struct Styles(pub HashMap<Mode, HashMap<String, Style>>);
+#[derive(Clone, Debug, Deref)]
+pub struct StyleConfig(pub Style);
 
-impl<'de> Deserialize<'de> for Styles {
+impl<'de> Deserialize<'de> for StyleConfig {
   fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
   where
     D: Deserializer<'de>,
   {
-    let parsed_map = HashMap::<Mode, HashMap<String, String>>::deserialize(deserializer)?;
-
-    let styles = parsed_map
-      .into_iter()
-      .map(|(mode, inner_map)| {
-        let converted_inner_map = inner_map
-          .into_iter()
-          .map(|(str, style)| (str, parse_style(&style)))
-          .collect();
-        (mode, converted_inner_map)
-      })
-      .collect();
-
-    Ok(Styles(styles))
+    Ok(StyleConfig(parse_style(&String::deserialize(
+      deserializer,
+    )?)))
   }
 }
 

src/default_config.toml

@@ -24,3 +24,16 @@ penultimate_right_overflow = true
 # Minimum width of the main chat history area and text input.
 # This takes precedence over the buflist settings.
 min_width = 30
+
+# Style of rooms in the buflist
+[style.buflist]
+# When your name is mentioned
+highlight = "bold yellow"
+# On other types of notifications
+notification = "bold green"
+# TODO: When no notification, but there is a new message
+# unread_message = "red"
+# TODO: When no new message, but something changed (someone joined, left, topic changed, ...)
+# unread_event = "bold"
+# If nothing happened since you last looked at the room
+uneventful = ""