Introduce atomExtensionsWithKey

Necoro · Mar 5, 2023 · 96ac374 · 96ac374
1 parent 9eb41f5
commit 96ac374
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/translator.go b/translator.go
@@ -105,16 +105,13 @@ func (t *DefaultRSSTranslator) translateFeedLink(rss *rss.Feed) (link string) {
 }
 
 func (t *DefaultRSSTranslator) translateFeedFeedLink(rss *rss.Feed) (link string) {
-	atomExtensions := t.extensionsForKeys([]string{"atom", "atom10", "atom03"}, rss.Extensions)
-	for _, ex := range atomExtensions {
-		if links, ok := ex["link"]; ok {
-			for _, l := range links {
-				if l.Attrs["rel"] == "self" {
-					link = l.Attrs["href"]
-				}
-			}
+	t.atomExtensionsWithKey(rss, "link", func(l ext.Extension) bool {
+		if l.Attrs["rel"] == "self" {
+			link = l.Attrs["href"]
+			return true
 		}
-	}
+		return false
+	})
 	return
 }
 
@@ -467,6 +464,19 @@ func (t *DefaultRSSTranslator) extensionsForKeys(keys []string, extensions ext.E
 	return
 }
 
+func (t *DefaultRSSTranslator) atomExtensionsWithKey(rss *rss.Feed, tag string, f func(ext.Extension) bool) {
+	atomExtensions := t.extensionsForKeys([]string{"atom", "atom10", "atom03"}, rss.Extensions)
+	for _, ex := range atomExtensions {
+		if exts, ok := ex[tag]; ok {
+			for _, e := range exts {
+				if f(e) {
+					return
+				}
+			}
+		}
+	}
+}
+
 func (t *DefaultRSSTranslator) firstEntry(entries []string) (value string) {
 	if entries == nil {
 		return