From e443905dc1f16202b29a8cdc783069ea72c2ac81 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 3 Mar 2024 15:25:21 +0800
Subject: feat: add `.clear messages` to clear session messages (#332)

---
 src/config/mod.rs | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'src/config/mod.rs')

diff --git a/src/config/mod.rs b/src/config/mod.rs
index d6ea6db..804a2fd 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -632,6 +632,13 @@ impl Config {
         Ok(())
     }
 
+    pub fn clear_session_messages(&mut self) -> Result<()> {
+        if let Some(session) = self.session.as_mut() {
+            session.clear_messgaes();
+        }
+        Ok(())
+    }
+
     pub fn list_sessions(&self) -> Vec<String> {
         let sessions_dir = match Self::sessions_dir() {
             Ok(dir) => dir,
@@ -951,10 +958,15 @@ impl State {
             .collect()
     }
 
-    pub fn can_change_role() -> Vec<Self> {
+    pub fn unable_change_role() -> Vec<Self> {
+        vec![Self::Session]
+    }
+
+    pub fn able_change_role() -> Vec<Self> {
+        let excludes: HashSet<_> = Self::unable_change_role().into_iter().collect();
         Self::all()
             .into_iter()
-            .filter(|v| *v != Self::Session)
+            .filter(|v| !excludes.contains(v))
             .collect()
     }
 
-- 
cgit v1.2.3