diff options
| author | sigoden <sigoden@gmail.com> | 2023-10-31 10:26:44 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-31 10:26:44 +0800 |
| commit | 9b614600c6936c07d31db20a284ca4e66433f45e (patch) | |
| tree | bb40c914389681307a7069576b7ea472778338af /src/render | |
| parent | 680670e834ff57cf6eb15c373870d1fd478f8edb (diff) | |
| download | aichat-9b614600c6936c07d31db20a284ca4e66433f45e.tar.gz | |
fix: uncorrected render in command mode (#188)
Diffstat (limited to 'src/render')
| -rw-r--r-- | src/render/cmd.rs | 10 | ||||
| -rw-r--r-- | src/render/markdown.rs | 4 |
2 files changed, 10 insertions, 4 deletions
diff --git a/src/render/cmd.rs b/src/render/cmd.rs index 781d376..57b6256 100644 --- a/src/render/cmd.rs +++ b/src/render/cmd.rs @@ -43,10 +43,12 @@ pub fn cmd_render_stream( let output = render.render(&input); let output = &output[col..]; let (_, tail) = split_line_tail(output); - if output.contains('\n') { - col = display_width(tail); - } else { - col += display_width(output); + if render.wrap_width().is_some() { + if output.contains('\n') { + col = display_width(tail); + } else { + col += display_width(output); + } } print_now!("{}", output); } diff --git a/src/render/markdown.rs b/src/render/markdown.rs index 8513a43..8ae0a09 100644 --- a/src/render/markdown.rs +++ b/src/render/markdown.rs @@ -71,6 +71,10 @@ impl MarkdownRender { ) } + pub(crate) const fn wrap_width(&self) -> Option<u16> { + self.wrap_width + } + pub fn render(&mut self, text: &str) -> String { text.split('\n') .map(|line| self.render_line_mut(line)) |