refactor: allow `.edit role` in empty session with role (#1031)

1 files changed, 11 insertions, 1 deletions

diff --git a/src/config/mod.rs b/src/config/mod.rs
index c98c81d..2eecf62 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -915,7 +915,17 @@ impl Config {
     }
 
     pub fn edit_role(&mut self) -> Result<()> {
-        if let Some(name) = self.role.as_ref().map(|v| v.name().to_string()) {
+        if let Some(session) = self.session.as_ref() {
+            if let Some(name) = session.role_name().map(|v| v.to_string()) {
+                if session.is_empty() {
+                    self.upsert_role(&name)
+                } else {
+                    bail!("Cannot perform this operation because you are in a non-empty session")
+                }
+            } else {
+                bail!("No role")
+            }
+        } else if let Some(name) = self.role.as_ref().map(|v| v.name().to_string()) {
             self.upsert_role(&name)
         } else {
             bail!("No role")