conf.go

package main

import (
	"time"

	"github.com/dlepex/newsmaker/news"
)

type config struct {
	RTick     duration            `toml:"rotation_tick"`
	MuteHours *[2]int             `toml:"mute_hours"`
	Filters   []*filterConf       `toml:"filters"`
	Sources   map[string]*srcConf `toml:"src"`
	Pubs      map[string]*pubConf `toml:"pub"`
}

type filterConf struct {
	Cond    string   `toml:"cond"`
	Sources []string `toml:"sources"`
	Pubs    []string `toml:"pubs"`
}

type srcConf struct {
	CD    duration `toml:"cd"`
	Links []string `toml:"links"`
	Categ []string `toml:"categ"`
}

type pubConf struct {
	SendPause duration `toml:"send_pause"`
	GetURL    string   `toml:"get_url"`
	Template  string   `toml:"template"` // optional go template (Item struct fields)
}

func (c *config) newPipeline() (pl *news.Pipeline, ers []error) {
	pl = news.NewPipelineDefault()
	globalMuteHours := news.DayInterval{}
	if c.MuteHours != nil {
		globalMuteHours = news.DayHoursFromTo(c.MuteHours[0], c.MuteHours[1])
	}
	check := func(e error) bool {
		if e == nil {
			return true
		}
		ers = append(ers, e)
		return false
	}
	for n, c := range c.Pubs {
		pub, err := c.toPub(n)
		if check(err) {
			check(pl.AddPublisher(pub))
		}
	}
	for n, c := range c.Sources {
		src, err := c.toSource(n, globalMuteHours)
		if check(err) {
			check(pl.AddSource(src))
		}
	}
	for _, c := range c.Filters {
		check(pl.AddFilter(c.toFilter()))
	}
	if len(ers) != 0 {
		pl = nil
	}

	return
}

func (c *filterConf) toFilter() *news.Filter {
	return &news.Filter{Cond: c.Cond, Sources: c.Sources, Pubs: c.Pubs}
}

func (c *srcConf) toSource(n string, muteHours news.DayInterval) (news.Source, error) {
	return news.NewFeedSrc(news.FeedSrcParams{
		SourceInfo: news.SourceInfo{
			Name:         n,
			Categories:   c.Categ,
			Cooldown:     c.CD.Duration,
			MuteInterval: muteHours,
		},
		Links: c.Links,
	})
}

func (c *pubConf) toPub(n string) (news.Pub, error) {
	params := &news.HTTPPubParams{
		PubInfo: news.PubInfo{
			Name: n,
		},
		Link:  c.GetURL,
		Pause: c.SendPause.Duration,
	}
	tpl := c.Template
	if tpl == "" {
		tpl = "*{{.Title}}* {{.DateFmt}} \n{{.Src.Name}} {{.Link}}" // telegram friendly template by default
	}
	params.ItemStringer = news.NewItemTemplateStringer(tpl)
	return news.NewHTTPPub(params), nil
}

type duration struct {
	time.Duration
}

func (d *duration) UnmarshalText(text []byte) error {
	var err error
	d.Duration, err = time.ParseDuration(string(text))
	return err
}