Better detection if a text starts with html or not

1 files changed, 12 insertions, 3 deletions

diff --git a/internal/feed/mail.go b/internal/feed/mail.go
index 0a8bd10..f77987e 100644
--- a/internal/feed/mail.go
+++ b/internal/feed/mail.go
@@ -16,6 +16,7 @@ import (
 	"github.com/emersion/go-message"
 	"github.com/emersion/go-message/mail"
 	"github.com/gabriel-vasile/mimetype"
+	"golang.org/x/net/html"
 
 	"github.com/Necoro/feed2imap-go/internal/feed/template"
 	"github.com/Necoro/feed2imap-go/internal/http"
@@ -240,6 +241,13 @@ func getBody(content, description string, bodyCfg config.Body) string {
 	}
 }
 
+func startsWithText(str string) bool {
+	reader := strings.NewReader(str)
+	tokenizer := html.NewTokenizerFragment(reader, "")
+
+	return tokenizer.Next() == html.TextToken
+}
+
 func (item *item) buildBody() {
 	feed := item.feed
 	feedUrl, err := url.Parse(feed.Url)
@@ -248,6 +256,9 @@ func (item *item) buildBody() {
 	}
 
 	body := getBody(item.Content, item.Description, feed.Body)
+	if body != "" && startsWithText(body) {
+		body = "<br />" + body
+	}
 
 	if !feed.InclImages {
 		item.Body = body
@@ -257,9 +268,7 @@ func (item *item) buildBody() {
 	doc, err := goquery.NewDocumentFromReader(strings.NewReader(body))
 	if err != nil {
 		log.Errorf("Feed %s: Item %s: Error while parsing html content: %s", feed.Name, item.Link, err)
-		if body != "" {
-			item.Body = "<br />" + body
-		}
+		item.Body = body
 		return
 	}