From cfb6ce6958751da5e13683ea07f42a5dbc4eae01 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sat, 11 Mar 2023 18:53:36 +0800 Subject: feat: support light theme (#65) There are two ways to enable the light theme: - add `light_theme: true` to config.yaml - use `AICHAT_LIGHT_THEME=true` env var --- src/render/repl.rs | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'src/render/repl.rs') diff --git a/src/render/repl.rs b/src/render/repl.rs index d7ccbc1..04dee73 100644 --- a/src/render/repl.rs +++ b/src/render/repl.rs @@ -16,11 +16,15 @@ use std::{ }; use unicode_width::UnicodeWidthStr; -pub fn repl_render_stream(rx: Receiver, abort: SharedAbortSignal) -> Result<()> { +pub fn repl_render_stream( + rx: Receiver, + light_theme: bool, + abort: SharedAbortSignal, +) -> Result<()> { enable_raw_mode()?; let mut stdout = io::stdout(); - let ret = repl_render_stream_inner(rx, abort, &mut stdout); + let ret = repl_render_stream_inner(rx, light_theme, abort, &mut stdout); disable_raw_mode()?; @@ -29,13 +33,14 @@ pub fn repl_render_stream(rx: Receiver, abort: SharedAbortSign fn repl_render_stream_inner( rx: Receiver, + light_theme: bool, abort: SharedAbortSignal, writer: &mut Stdout, ) -> Result<()> { let mut last_tick = Instant::now(); let tick_rate = Duration::from_millis(100); let mut buffer = String::new(); - let mut markdown_render = MarkdownRender::new(); + let mut markdown_render = MarkdownRender::new(light_theme); let terminal_columns = terminal::size()?.0; loop { if abort.aborted() { -- cgit v1.2.3