rss.go

package rss

import (
	"io"

	"github.com/luraproject/lura/v2/encoding"
	"github.com/mmcdole/gofeed"
)

func Register() error {
	return encoding.GetRegister().Register(Name, DecoderFactory)
}

// Name is the key for the rss encoding
const Name = "rss"

func DecoderFactory(_ bool) func(io.Reader, *map[string]interface{}) error { return NewDecoder() }

// NewDecoder returns the RSS decoder
func NewDecoder() func(io.Reader, *map[string]interface{}) error {
	fp := gofeed.NewParser()
	return func(r io.Reader, v *map[string]interface{}) error {
		feed, err := fp.Parse(r)
		if err != nil {
			return err
		}
		*(v) = map[string]interface{}{
			"items":       feed.Items,
			"author":      feed.Author,
			"categories":  feed.Categories,
			"custom":      feed.Custom,
			"copyright":   feed.Copyright,
			"description": feed.Description,
			"type":        feed.FeedType,
			"language":    feed.Language,
			"title":       feed.Title,
			"published":   feed.Published,
			"updated":     feed.Updated,
		}
		if feed.Image != nil {
			(*v)["img_url"] = feed.Image.URL
		}
		return nil
	}
}