From 9b614600c6936c07d31db20a284ca4e66433f45e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 31 Oct 2023 10:26:44 +0800
Subject: fix: uncorrected render in command mode (#188)

---
 src/render/cmd.rs      | 10 ++++++----
 src/render/markdown.rs |  4 ++++
 2 files changed, 10 insertions(+), 4 deletions(-)

(limited to 'src/render')

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))
-- 
cgit v1.2.3