Allow application/xhtml+xml links as comments URL in Atom replies

1 files changed, 7 insertions, 3 deletions

diff --git a/reader/atom/atom_common.go b/reader/atom/atom_common.go
index 85e6b29..511777b 100644
--- a/reader/atom/atom_common.go
+++ b/reader/atom/atom_common.go
@@ -57,10 +57,14 @@ func (a atomLinks) firstLinkWithRelation(relation string) string {
 	return ""
 }
 
-func (a atomLinks) firstLinkWithRelationAndType(relation, contentType string) string {
+func (a atomLinks) firstLinkWithRelationAndType(relation string, contentTypes ...string) string {
 	for _, link := range a {
-		if strings.ToLower(link.Rel) == relation && strings.ToLower(link.Type) == contentType {
-			return strings.TrimSpace(link.URL)
+		if strings.ToLower(link.Rel) == relation {
+			for _, contentType := range contentTypes {
+				if strings.ToLower(link.Type) == contentType {
+					return strings.TrimSpace(link.URL)
+				}
+			}
 		}
 	}