Move show_spaces option to spans::LinesIterator

src/utils/lines/simple/lines_iterator.rs

@@ -1,6 +1,4 @@
-use super::{prefix, Row};
-use unicode_segmentation::UnicodeSegmentation;
-use unicode_width::UnicodeWidthStr;
+use super::Row;
 use utils::lines::spans;
 use utils::span::{IndexedSpan, SpannedText};
 
@@ -13,10 +11,6 @@ pub struct LinesIterator<'a> {
 
     /// Available width. Don't output lines wider than that.
     width: usize,
-
-    /// If `true`, keep a blank cell at the end of lines
-    /// when a whitespace or newline should be.
-    show_spaces: bool,
 }
 
 struct DummySpannedText<'a> {
@@ -50,21 +44,17 @@ impl<'a> LinesIterator<'a> {
     pub fn new(content: &'a str, width: usize) -> Self {
         let iter =
             spans::LinesIterator::new(DummySpannedText::new(content), width);
-        let show_spaces = false;
-        LinesIterator {
-            iter,
-            width,
-            show_spaces,
-        }
+        LinesIterator { iter, width }
     }
 
     /// Leave a blank cell at the end of lines.
     ///
     /// Unless a word had to be truncated, in which case
     /// it takes the entire width.
-    pub fn show_spaces(mut self) -> Self {
-        self.show_spaces = true;
-        self
+    pub fn show_spaces(self) -> Self {
+        let iter = self.iter.show_spaces();
+        let width = self.width;
+        LinesIterator { iter, width }
     }
 }
 

src/utils/lines/spans/lines_iterator.rs

@@ -2,7 +2,7 @@ use super::chunk::{Chunk, ChunkPart};
 use super::chunk_iterator::ChunkIterator;
 use super::prefix::prefix;
 use super::row::Row;
-use super::segment::{Segment};
+use super::segment::Segment;
 use super::segment_merge_iterator::SegmentMergeIterator;
 use std::iter::Peekable;
 use std::rc::Rc;
@@ -26,6 +26,10 @@ where
     /// If a chunk wouldn't fit, we had to cut it in pieces.
     /// This is how far in the current chunk we are.
     chunk_offset: ChunkPart,
+
+    /// If `true`, keep a blank cell at the end of lines
+    /// when a whitespace or newline should be.
+    show_spaces: bool,
 }
 
 impl<S> LinesIterator<S>
@@ -41,8 +45,18 @@ where
             source,
             width,
             chunk_offset: ChunkPart::default(),
+            show_spaces: false,
         }
     }
+
+    /// Leave a blank cell at the end of lines.
+    ///
+    /// Unless a word had to be truncated, in which case
+    /// it can take the entire width.
+    pub fn show_spaces(mut self) -> Self {
+        self.show_spaces = true;
+        self
+    }
 }
 
 impl<S> Iterator for LinesIterator<S>
@@ -53,9 +67,17 @@ where
 
     fn next(&mut self) -> Option<Row> {
         // Let's build a beautiful row.
+        let allowed_width = if self.show_spaces {
+            // Remove 1 from the available space, if possible.
+            // But only for regular words.
+            // If we have to split a chunk, forget about that.
+            self.width.saturating_sub(1)
+        } else {
+            self.width
+        };
 
         let mut chunks =
-            prefix(&mut self.iter, self.width, &mut self.chunk_offset);
+            prefix(&mut self.iter, allowed_width, &mut self.chunk_offset);
 
         if chunks.is_empty() {
             // Desperate action to make something fit:
@@ -71,7 +93,8 @@ where
 
                     // Try to fit part of it?
                     let source = self.source.as_ref();
-                    let graphemes = chunk.segments.iter().flat_map(move |seg| {
+                    let graphemes =
+                        chunk.segments.iter().flat_map(move |seg| {
                             let mut offset = seg.start;
 
                             let text = seg.resolve_plain(source);