From d9b8eabf23ae96fb395a8059c1e08840609499f4 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 28 Apr 2024 19:18:56 +0800
Subject: feat: add `--prompt` option (#454)

---
 src/main.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index f3d3846..f691b6f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -80,7 +80,9 @@ async fn main() -> Result<()> {
     if cli.dry_run {
         config.write().dry_run = true;
     }
-    if let Some(name) = &cli.role {
+    if let Some(prompt) = &cli.prompt {
+        config.write().set_prompt(prompt)?;
+    } else if let Some(name) = &cli.role {
         config.write().set_role(name)?;
     } else if cli.execute {
         config.write().set_role(SHELL_ROLE)?;
-- 
cgit v1.2.3