summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-10-30 10:07:01 +0800
committerGitHub <noreply@github.com>2023-10-30 10:07:01 +0800
commitb276dfedd2d13de9918b9b87a3d543e7b5eca95e (patch)
tree33e1f864ef38e45ebc297e15ce519ee8105b8757 /src/main.rs
parent632a6fcc0bfb5b33858cd0fa3876e43ce2073cb3 (diff)
downloadaichat-b276dfedd2d13de9918b9b87a3d543e7b5eca95e.tar.gz
feat: support textwrap (#171)
* feat: support textwrap * improve cmd_render_stream * done cmd_render_stream * add `-w` alias to `--wrap` * done repl_render_stream * add `config.wrap_code` * remove cached config.wrap_width * fix unxpected duplicate lines on kitty * refactor markdown render * improve render * fix test
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs13
1 files changed, 6 insertions, 7 deletions
diff --git a/src/main.rs b/src/main.rs
index d2660a8..f551e9c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -47,6 +47,9 @@ fn main() -> Result<()> {
println!("{sessions}");
exit(0);
}
+ if let Some(wrap) = &cli.wrap {
+ config.write().set_wrap(wrap)?;
+ }
if cli.light_theme {
config.write().light_theme = true;
}
@@ -115,14 +118,10 @@ fn start_directive(
}
config.read().maybe_print_send_tokens(input);
let output = if no_stream {
- let (highlight, light_theme) = config.read().get_render_options();
+ let render_options = config.read().get_render_options();
let output = client.send_message(input)?;
- if highlight {
- let mut markdown_render = MarkdownRender::new(light_theme);
- println!("{}", markdown_render.render_block(&output).trim());
- } else {
- println!("{}", output.trim());
- }
+ let mut markdown_render = MarkdownRender::init(render_options)?;
+ println!("{}", markdown_render.render(&output).trim());
output
} else {
let wg = WaitGroup::new();