diff options
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/mod.rs | 16 | ||||
| -rw-r--r-- | src/config/session.rs | 9 |
2 files changed, 23 insertions, 2 deletions
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() } diff --git a/src/config/session.rs b/src/config/session.rs index cbfe1c2..076fa27 100644 --- a/src/config/session.rs +++ b/src/config/session.rs @@ -258,6 +258,15 @@ impl Session { Ok(()) } + pub fn clear_messgaes(&mut self) { + if self.messages.is_empty() { + return; + } + self.messages.clear(); + self.data_urls.clear(); + self.dirty = true; + } + pub fn echo_messages(&self, input: &Input) -> String { let messages = self.build_emssages(input); serde_yaml::to_string(&messages).unwrap_or_else(|_| "Unable to echo message".into()) |