From 8421f23b450643ca3c66cb3f6fd21ef862a2369d Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 3 Mar 2024 09:57:50 +0800 Subject: feat: allow overriding execute/code role (#331) --- src/config/mod.rs | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'src/config/mod.rs') diff --git a/src/config/mod.rs b/src/config/mod.rs index cbf1020..5220c5d 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -285,17 +285,23 @@ impl Config { } pub fn set_execute_role(&mut self) -> Result<()> { - let role = Role::for_execute(); + let role = self + .retrieve_role(Role::EXECUTE) + .unwrap_or_else(|_| Role::for_execute()); self.set_role_obj(role) } - pub fn set_describe_role(&mut self) -> Result<()> { - let role = Role::for_describe(); + pub fn set_describe_command_role(&mut self) -> Result<()> { + let role = self + .retrieve_role(Role::DESCRIBE_COMMAND) + .unwrap_or_else(|_| Role::for_describe_command()); self.set_role_obj(role) } pub fn set_code_role(&mut self) -> Result<()> { - let role = Role::for_code(); + let role = self + .retrieve_role(Role::CODE) + .unwrap_or_else(|_| Role::for_code()); self.set_role_obj(role) } -- cgit v1.2.3