From ff00426c2c9d7d16de9221644e4f276f7e24d74a Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 10 Mar 2023 07:08:40 +0800
Subject: chore: improve code quality, split rep/prompt.rs from repl/init.rs

---
 src/repl/mod.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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

diff --git a/src/repl/mod.rs b/src/repl/mod.rs
index 661be75..9d7d1cf 100644
--- a/src/repl/mod.rs
+++ b/src/repl/mod.rs
@@ -1,10 +1,12 @@
 mod abort;
 mod handler;
 mod init;
+mod prompt;
 
 pub use self::abort::*;
 pub use self::handler::*;
 pub use self::init::Repl;
+use self::prompt::ReplPrompt;
 
 use crate::client::ChatGptClient;
 use crate::config::SharedConfig;
@@ -33,7 +35,8 @@ pub const REPL_COMMANDS: [(&str, &str, bool); 12] = [
 impl Repl {
     pub fn run(&mut self, client: ChatGptClient, config: SharedConfig) -> Result<()> {
         let abort = AbortSignal::new();
-        let handler = ReplCmdHandler::init(client, config, abort.clone())?;
+        let handler = ReplCmdHandler::init(client, config.clone(), abort.clone())?;
+        let prompt = ReplPrompt::new(config);
         print_now!("Welcome to aichat {}\n", env!("CARGO_PKG_VERSION"));
         print_now!("Type \".help\" for more information.\n");
         let mut already_ctrlc = false;
@@ -45,7 +48,7 @@ impl Repl {
             if abort.aborted_ctrlc() && !already_ctrlc {
                 already_ctrlc = true;
             }
-            let sig = self.editor.read_line(&self.prompt);
+            let sig = self.editor.read_line(&prompt);
             match sig {
                 Ok(Signal::Success(line)) => {
                     already_ctrlc = false;
-- 
cgit v1.2.3