Fix leading and trailing whitespaces issue

tneotia · tneotia · commit fbfb50503172 · 2021-03-06T20:01:15.000-05:00
diff --git a/lib/html_parser.dart b/lib/html_parser.dart
@@ -173,7 +173,7 @@ class HtmlParser extends StatelessWidget {
         return EmptyContentElement();
       }
     } else if (node is dom.Text) {
-      return TextContentElement(text: node.text, style: Style());
+      return TextContentElement(text: node.text, style: Style(), element: node.parent, node: node);
     } else {
       return EmptyContentElement();
     }
@@ -465,14 +465,18 @@ class HtmlParser extends StatelessWidget {
     }
 
     if (tree is TextContentElement) {
-      if (wpc.data && tree.text!.startsWith(' ')) {
+      int index = -1;
+      if ((tree.element?.nodes.length ?? 0) > 1) {
+        index = tree.element?.nodes.indexWhere((element) => element == tree.node) ?? -1;
+      }
+      if (index < 1 && tree.text!.startsWith(' ')
+          && tree.element?.localName != "br") {
         tree.text = tree.text!.replaceFirst(' ', '');
       }
-
-      if (tree.text!.endsWith(' ') || tree.text!.endsWith('\n')) {
-        wpc.data = true;
-      } else {
-        wpc.data = false;
+      if (index == (tree.element?.nodes.length ?? 1) - 1
+          && (tree.text!.endsWith(' ') || tree.text!.endsWith('\n'))
+          && tree.element?.localName != "br") {
+        tree.text = tree.text!.trimRight();
       }
     }
 
diff --git a/lib/src/replaced_element.dart b/lib/src/replaced_element.dart
@@ -44,11 +44,14 @@ abstract class ReplacedElement extends StyledElement {
 /// [TextContentElement] is a [ContentElement] with plaintext as its content.
 class TextContentElement extends ReplacedElement {
   String? text;
+  dom.Node? node;
 
   TextContentElement({
     required Style style,
     required this.text,
-  }) : super(name: "[text]", style: style);
+    this.node,
+    dom.Element? element,
+  }) : super(name: "[text]", style: style, node: element);
 
   @override
   String toString() {
@@ -290,6 +293,8 @@ ReplacedElement parseReplacedElement(
       return TextContentElement(
         text: "\n",
         style: Style(whiteSpace: WhiteSpace.PRE),
+        element: element,
+        node: element
       );
     case "iframe":
       return IframeContentElement(

Original file line number	Diff line number	Diff line change
`@@ -173,7 +173,7 @@ class HtmlParser extends StatelessWidget {`
`173`	`173`	`return EmptyContentElement();`
`174`	`174`	`}`
`175`	`175`	`} else if (node is dom.Text) {`
`176`		`- return TextContentElement(text: node.text, style: Style());`
	`176`	`+ return TextContentElement(text: node.text, style: Style(), element: node.parent, node: node);`
`177`	`177`	`} else {`
`178`	`178`	`return EmptyContentElement();`
`179`	`179`	`}`
`@@ -465,14 +465,18 @@ class HtmlParser extends StatelessWidget {`
`465`	`465`	`}`
`466`	`466`
`467`	`467`	`if (tree is TextContentElement) {`
`468`		`- if (wpc.data && tree.text!.startsWith(' ')) {`
	`468`	`+ int index = -1;`
	`469`	`+ if ((tree.element?.nodes.length ?? 0) > 1) {`
	`470`	`+ index = tree.element?.nodes.indexWhere((element) => element == tree.node) ?? -1;`
	`471`	`+ }`
	`472`	`+ if (index < 1 && tree.text!.startsWith(' ')`
	`473`	`+ && tree.element?.localName != "br") {`
`469`	`474`	`tree.text = tree.text!.replaceFirst(' ', '');`
`470`	`475`	`}`
`471`		`-`
`472`		`- if (tree.text!.endsWith(' ') \|\| tree.text!.endsWith('\n')) {`
`473`		`- wpc.data = true;`
`474`		`- } else {`
`475`		`- wpc.data = false;`
	`476`	`+ if (index == (tree.element?.nodes.length ?? 1) - 1`
	`477`	`+ && (tree.text!.endsWith(' ') \|\| tree.text!.endsWith('\n'))`
	`478`	`+ && tree.element?.localName != "br") {`
	`479`	`+ tree.text = tree.text!.trimRight();`
`476`	`480`	`}`
`477`	`481`	`}`
`478`	`482`