From 5343a8b653d10952ff2d838e15bcb822f5542045 Mon Sep 17 00:00:00 2001 From: yequari Date: Thu, 18 Apr 2024 11:45:00 -0700 Subject: [PATCH] get first header element --- feed/feed.go | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/feed/feed.go b/feed/feed.go index f10f2e0..23102f6 100644 --- a/feed/feed.go +++ b/feed/feed.go @@ -141,15 +141,15 @@ func getTitleAndUrl(article *html.Node) (string, string, error) { var title string var url string var header *html.Node - h1, _ := getHtmlElement(article, "h1") - h2, _ := getHtmlElement(article, "h2") - h3, _ := getHtmlElement(article, "h3") - if h1 != nil { - header = h1 - } else if h2 != nil { - header = h2 - } else if h3 != nil { - header = h3 + h1s, _ := getAllElements(article, "h1") + h2s, _ := getAllElements(article, "h2") + h3s, _ := getAllElements(article, "h3") + if len(h1s) > 0 { + header = h1s[0] + } else if len(h2s) > 0 { + header = h2s[0] + } else if len(h3s) > 0 { + header = h3s[0] } if header == nil { return "", "", nil