feat: provide `--prompt` for adding a prompt from cli (#62)

4 files changed, 8 insertions, 2 deletions

diff --git a/src/cli.rs b/src/cli.rs
index f2d2640..e08e9cf 100644
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -3,6 +3,9 @@ use clap::Parser;
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)]
 pub struct Cli {
+    /// Add a GPT prompt
+    #[clap(short, long)]
+    pub prompt: Option<String>,
     /// Turn off highlight
     #[clap(short = 'H', long)]
     pub no_highlight: bool,
diff --git a/src/config/mod.rs b/src/config/mod.rs
index 147d944..b438687 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -193,7 +193,7 @@ impl Config {
         Ok(())
     }
 
-    pub fn create_temp_role(&mut self, prompt: &str) -> Result<()> {
+    pub fn add_prompt(&mut self, prompt: &str) -> Result<()> {
         let role = Role::new(prompt, self.temperature);
         if let Some(conversation) = self.conversation.as_mut() {
             conversation.update_role(&role)?;
diff --git a/src/main.rs b/src/main.rs
index b198d14..5bcdf11 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -48,6 +48,9 @@ fn main() -> Result<()> {
     if cli.no_highlight {
         config.lock().highlight = false;
     }
+    if let Some(prompt) = &cli.prompt {
+        config.lock().add_prompt(prompt)?;
+    }
     let no_stream = cli.no_stream;
     let client = ChatGptClient::init(config.clone())?;
     if atty::isnt(atty::Stream::Stdin) {
diff --git a/src/repl/handler.rs b/src/repl/handler.rs
index 2263ca2..ce95855 100644
--- a/src/repl/handler.rs
+++ b/src/repl/handler.rs
@@ -75,7 +75,7 @@ impl ReplCmdHandler {
                 print_now!("\n");
             }
             ReplCmd::Prompt(prompt) => {
-                self.config.lock().create_temp_role(&prompt)?;
+                self.config.lock().add_prompt(&prompt)?;
                 print_now!("\n");
             }
             ReplCmd::ViewInfo => {