truncate message body if over 100 chars

ulyssa · Mar 17, 2024 · d93315a · d93315a
1 parent 33641cf
commit d93315a
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/notifications.rs b/src/notifications.rs
@@ -11,6 +11,7 @@ use matrix_sdk::{
     },
     Client,
 };
+use unicode_segmentation::UnicodeSegmentation;
 
 use crate::{
     base::{AsyncProgramStore, IambError, IambResult},
@@ -146,7 +147,8 @@ pub async fn parse_notification(
         &event,
         sender_name,
         room.is_direct().await.map_err(IambError::from)?,
-    );
+    )
+    .map(truncate);
     return Ok((sender_name.to_string(), body, server_ts));
 }
 
@@ -208,3 +210,13 @@ pub fn event_notification_body(
         _ => None,
     }
 }
+
+fn truncate(s: String) -> String {
+    static MAX_LENGTH: usize = 100;
+    if s.graphemes(true).count() > MAX_LENGTH {
+        let truncated: String = s.graphemes(true).take(MAX_LENGTH).collect();
+        truncated + "..."
+    } else {
+        s.to_string()
+    }
+}