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

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,

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)?;

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) {

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 => {