From 3c16aff59145134885f32056fa3be9bd6bed2cce Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 2 Mar 2024 21:11:28 +0800
Subject: feat: add `-c/--code` to generate only code (#327)

---
 src/config/mod.rs | 5 +++++
 1 file changed, 5 insertions(+)

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

diff --git a/src/config/mod.rs b/src/config/mod.rs
index f8d97c2..6b99483 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -288,6 +288,11 @@ impl Config {
         self.set_role_obj(role)
     }
 
+    pub fn set_code_role(&mut self) -> Result<()> {
+        let role = Role::for_code();
+        self.set_role_obj(role)
+    }
+
     pub fn set_role_obj(&mut self, role: Role) -> Result<()> {
         if let Some(session) = self.session.as_mut() {
             session.update_role(Some(role.clone()))?;
-- 
cgit v1.2.3