Fix line computation

badoualy · Mar 22, 2023 · 67b1618 · 67b1618
1 parent 320a765
commit 67b1618
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/story-editor/src/main/java/com/github/badoualy/storyeditor/util/TextLayoutResult.kt b/story-editor/src/main/java/com/github/badoualy/storyeditor/util/TextLayoutResult.kt
@@ -6,14 +6,11 @@ internal fun TextLayoutResult.getLines(): String {
     val input = layoutInput.text.toString()
     if (input.isEmpty()) return ""
 
-    return buildString {
-        var currentIndex = 0
-
-        repeat(lineCount) { line ->
-            val lineEnd = getLineEnd(line)
-            appendLine(input.substring(currentIndex, lineEnd))
-
-            currentIndex = lineEnd
-        }
-    }.trim()
+    return (0 until lineCount).joinToString("\n") { line ->
+        val lineContent = input.substring(
+            getLineStart(line),
+            getLineEnd(line, visibleEnd = true)
+        )
+        lineContent
+    }
 }