From dbdb2ed125214d7dab9dd2ef6ed03e1062b7cc27 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sat, 4 Mar 2023 09:22:55 +0800
Subject: refactor: replace config.highlight with config.no_highlight (#4)

* refactor: replace config.highlight with config.no_highlight

* update readme
---
 src/config.rs | 12 ++----------
 src/main.rs   |  2 +-
 src/repl.rs   |  6 +++---
 3 files changed, 6 insertions(+), 14 deletions(-)

(limited to 'src')

diff --git a/src/config.rs b/src/config.rs
index cce6b6a..1c6abc5 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -24,9 +24,9 @@ pub struct Config {
     /// Whether to persistently save chat messages
     #[serde(default)]
     pub save: bool,
-    /// Whether to highlight reply message
+    /// Whether to disable highlight
     #[serde(default)]
-    pub highlight: bool,
+    pub no_highlight: bool,
     /// Set proxy
     pub proxy: Option<String>,
     /// Used only for debugging
@@ -175,14 +175,6 @@ fn create_config_file(config_path: &Path) -> Result<()> {
         raw_config.push_str("save: true\n");
     }
 
-    let ans = Confirm::new("Whether to highlight reply message?")
-        .with_default(true)
-        .prompt()
-        .map_err(confirm_map_err)?;
-    if ans {
-        raw_config.push_str("highlight: true\n");
-    }
-
     std::fs::write(config_path, raw_config)
         .map_err(|err| anyhow!("Failed to write to config file, {err}"))?;
     Ok(())
diff --git a/src/main.rs b/src/main.rs
index 893850a..2d72c3e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -56,7 +56,7 @@ fn start_directive(
     let mut file = config.open_message_file()?;
     let output = client.acquire(input, role.map(|v| v.prompt))?;
     let output = output.trim();
-    if config.highlight && stdout().is_terminal() {
+    if !config.no_highlight && stdout().is_terminal() {
         let markdown_render = MarkdownRender::init()?;
         markdown_render.print(output)?;
     } else {
diff --git a/src/repl.rs b/src/repl.rs
index f8a10a3..bb33bc1 100644
--- a/src/repl.rs
+++ b/src/repl.rs
@@ -196,10 +196,10 @@ struct ReplCmdHandlerState {
 
 impl ReplCmdHandler {
     pub fn init(client: ChatGptClient, config: Arc<Config>, role: Option<Role>) -> Result<Self> {
-        let render = if config.highlight {
-            Some(Arc::new(MarkdownRender::init()?))
-        } else {
+        let render = if config.no_highlight {
             None
+        } else {
+            Some(Arc::new(MarkdownRender::init()?))
         };
         let prompt = role.map(|v| v.prompt).unwrap_or_default();
         let save_file = config.open_message_file()?;
-- 
cgit v1.2.3