flokati/modules/rss.go

87 lines
1.9 KiB
Go
Raw Normal View History

// vi:ts=4:sts=4:sw=4:noet:tw=72
2016-01-10 17:45:21 +00:00
//
// This code is mostly derived from the example code by Jim Teeuwen
// and is subject to the CC0 1.0 Universal (CC0 1.0) Public Domain Dedication
// license.
2016-02-15 23:17:42 +00:00
package modules
2016-01-10 17:45:21 +00:00
import (
"bufio"
"errors"
"fmt"
"io"
"os"
"time"
2016-03-12 12:09:25 +00:00
"code.dnix.de/an/xlog"
gorss "github.com/jteeuwen/go-pkg-rss"
2016-01-10 17:45:21 +00:00
"github.com/jteeuwen/go-pkg-xmlx"
2016-02-15 23:17:42 +00:00
"github.com/sorcix/irc"
2016-01-10 17:45:21 +00:00
)
var hideOutput = true
2016-02-15 23:17:42 +00:00
func init() {
MsgFuncs["rss"] = rssHandleMessage
RunFuncs["rss"] = rssRun
}
func rssRun() {
time.Sleep(5 * time.Second)
path := ModParams["rss-feeds"]
2016-01-10 17:45:21 +00:00
file, err := os.Open(path)
if err != nil {
2016-03-12 12:09:25 +00:00
xlog.Fatal(err.Error())
2016-01-10 17:45:21 +00:00
}
defer file.Close()
scanner := bufio.NewScanner(file)
for scanner.Scan() {
2016-02-15 23:17:42 +00:00
go rssPollFeed(scanner.Text(), 5, rssCharsetReader)
2016-01-10 17:45:21 +00:00
}
if err := scanner.Err(); err != nil {
2016-03-12 12:09:25 +00:00
xlog.Fatal(err.Error())
2016-01-10 17:45:21 +00:00
}
2016-02-15 23:17:42 +00:00
go func() {
time.Sleep(60 * time.Second)
hideOutput = false
}()
2016-01-10 17:45:21 +00:00
}
2016-02-15 23:17:42 +00:00
func rssHandleMessage(m *irc.Message) {
}
func rssPollFeed(uri string, timeout int, cr xmlx.CharsetFunc) {
feed := gorss.New(timeout, true, rssChanHandler, rssItemHandler)
2016-01-10 17:45:21 +00:00
for {
2016-03-12 12:09:25 +00:00
xlog.Info("Polling feed: %s", uri)
2016-01-10 17:45:21 +00:00
if err := feed.Fetch(uri, cr); err != nil {
2016-03-12 12:09:25 +00:00
xlog.Info("[e] %s: %s", "*", uri, err)
SayCh <- fmt.Sprintf("%s\n[RSS] Error %s: %s", "*", uri, err)
2016-01-10 17:45:21 +00:00
return
}
<-time.After(time.Duration(feed.SecondsTillUpdate() * 1e9))
}
}
2016-02-15 23:17:42 +00:00
func rssChanHandler(feed *gorss.Feed, newchannels []*gorss.Channel) {
2016-02-28 11:38:32 +00:00
//SayCh <- fmt.Sprintf("%s\n[RSS] %d new channel(s) in %s", "*", len(newchannels), feed.Url)
2016-01-10 17:45:21 +00:00
}
2016-02-15 23:17:42 +00:00
func rssItemHandler(feed *gorss.Feed, ch *gorss.Channel, newitems []*gorss.Item) {
2016-01-10 17:45:21 +00:00
if hideOutput {
return
}
for _, ni := range newitems {
SayCh <- fmt.Sprintf("%s\n[RSS] %v - %v", "*", ni.Title, ni.Links[0].Href)
2016-01-10 17:45:21 +00:00
}
}
2016-02-15 23:17:42 +00:00
func rssCharsetReader(charset string, r io.Reader) (io.Reader, error) {
2016-01-10 17:45:21 +00:00
if charset == "ISO-8859-1" || charset == "iso-8859-1" {
return r, nil
}
return nil, errors.New("Unsupported character set encoding: " + charset)
}