diff options
Diffstat (limited to 'src/config/mod.rs')
| -rw-r--r-- | src/config/mod.rs | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index 5d1172b..f7f2b2e 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -733,9 +733,10 @@ impl Config { } pub fn retrieve_role(&self, name: &str) -> Result<Role> { - let mut role = if Self::list_roles(false).contains(&name.to_string()) { - let path = Self::role_file(name)?; - let content = read_to_string(path)?; + let names = Self::list_roles(false); + let mut role = if let Some(role_name) = Role::match_name(&names, name) { + let path = Self::role_file(&role_name)?; + let content = read_to_string(&path)?; Role::new(name, &content) } else { BUILTIN_ROLES @@ -777,7 +778,9 @@ impl Config { } pub fn upsert_role(&mut self, name: &str) -> Result<()> { - let role_path = Self::role_file(name)?; + let names = Self::list_roles(false); + let role_name = Role::match_name(&names, name).unwrap_or_else(|| name.to_string()); + let role_path = Self::role_file(&role_name)?; ensure_parent_exists(&role_path)?; let editor = self.editor()?; edit_file(&editor, &role_path)?; @@ -859,7 +862,8 @@ impl Config { } pub fn has_role(name: &str) -> bool { - Self::list_roles(true).iter().any(|v| v == name) + let names = Self::list_roles(true); + Role::match_name(&names, name).is_some() } pub fn use_session(&mut self, session_name: Option<&str>) -> Result<()> { |