Fix mangled newlines in some code blocks

Because some tokens can contain newlines and not only comment tokens, I
removed the comments specific code to handle newlines in a more generic
way.

ui/messages/html/parser.go

@@ -305,27 +305,19 @@ func (parser *htmlParser) syntaxHighlight(text, language string) Entity {
 
 	var children []Entity
 	for _, token := range tokens {
-		if token.Value == "\n" {
+		lines := strings.SplitAfter(token.Value, "\n")
-			children = append(children, NewBreakEntity())
+		for _, line := range lines {
-
+			line_len := len(line)
-		} else if token.Type.String() == "CommentSingle" {
+			if line_len == 0 {
-			children = append(children, tokenToTextEntity(style, &token))
+				continue
-			children = append(children, NewBreakEntity())
-
-		} else if token.Type.String() == "CommentMultiline" {
-			lines := strings.Split(token.Value, "\n")
-			for i, line := range lines {
-				t := token.Clone()
-				t.Value = line
-				children = append(children, tokenToTextEntity(style, &t))
-
-				if i < len(lines)-1 {
-					children = append(children, NewBreakEntity())
-				}
 			}
+			t := token.Clone()
+			t.Value = line
+			children = append(children, tokenToTextEntity(style, &t))
 
-		} else {
+			if line[line_len-1:] == "\n" {
-			children = append(children, tokenToTextEntity(style, &token))
+				children = append(children, NewBreakEntity())
+			}
 		}
 	}