diff options
Diffstat (limited to 'reader/atom/atom_common.go')
-rw-r--r-- | reader/atom/atom_common.go | 10 |
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) + } + } } } |