Fix EOF rendering

yuin · Oct 12, 2024 · 41273a4 · 41273a4
1 parent d80ac93
commit 41273a4
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/_test/extra.txt b/_test/extra.txt
@@ -821,7 +821,7 @@ text" /></p>
 </blockquote>
 //= = = = = = = = = = = = = = = = = = = = = = = =//
 
-66: EOF should be rendered as a newline with an unclosed block
+66: EOF should be rendered as a newline with an unclosed block(w/ TAB)
 //- - - - - - - - -//
 > ```
 > 0
@@ -831,3 +831,14 @@ text" /></p>
 </code></pre>
 </blockquote>
 //= = = = = = = = = = = = = = = = = = = = = = = =//
+
+67: EOF should be rendered as a newline with an unclosed block
+//- - - - - - - - -//
+> ```
+> 0
+//- - - - - - - - -//
+<blockquote>
+<pre><code> 0
+</code></pre>
+</blockquote>
+//= = = = = = = = = = = = = = = = = = = = = = = =//
diff --git a/text/segment.go b/text/segment.go
@@ -44,12 +44,14 @@ func NewSegmentPadding(start, stop, n int) Segment {
 
 // Value returns a value of the segment.
 func (t *Segment) Value(buffer []byte) []byte {
+ var result []byte
  if t.Padding == 0 {
- return buffer[t.Start:t.Stop]
+ result = buffer[t.Start:t.Stop]
+ } else {
+ result = make([]byte, 0, t.Padding+t.Stop-t.Start+1)
+ result = append(result, bytes.Repeat(space, t.Padding)...)
+ result = append(result, buffer[t.Start:t.Stop]...)
  }
- result := make([]byte, 0, t.Padding+t.Stop-t.Start+1)
- result = append(result, bytes.Repeat(space, t.Padding)...)
- result = append(result, buffer[t.Start:t.Stop]...)
  if t.EOB && len(result) > 0 && result[len(result)-1] != '\n' {
  result = append(result, '\n')
  }