diff options
| author | sigoden <sigoden@gmail.com> | 2023-10-30 10:07:01 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-30 10:07:01 +0800 |
| commit | b276dfedd2d13de9918b9b87a3d543e7b5eca95e (patch) | |
| tree | 33e1f864ef38e45ebc297e15ce519ee8105b8757 /src/main.rs | |
| parent | 632a6fcc0bfb5b33858cd0fa3876e43ce2073cb3 (diff) | |
| download | aichat-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.rs | 13 |
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(); |
