mirror of https://github.com/velour/catbase.git
93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
|
package rss
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/velour/catbase/bot"
|
||
|
|
"github.com/velour/catbase/bot/msg"
|
||
|
|
|
||
|
|
"github.com/mmcdole/gofeed"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RSSPlugin struct {
|
||
|
|
Bot bot.Bot
|
||
|
|
cache map[string]*cacheItem
|
||
|
|
shelfLife time.Duration
|
||
|
|
}
|
||
|
|
|
||
|
|
type cacheItem struct {
|
||
|
|
key string
|
||
|
|
data string
|
||
|
|
expiration time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(bot bot.Bot) *RSSPlugin {
|
||
|
|
return &RSSPlugin{
|
||
|
|
Bot: bot,
|
||
|
|
cache: map[string]*cacheItem{},
|
||
|
|
shelfLife: time.Minute * 20,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *RSSPlugin) Message(message msg.Message) bool {
|
||
|
|
tokens := strings.Fields(message.Body)
|
||
|
|
numTokens := len(tokens)
|
||
|
|
|
||
|
|
if numTokens == 2 && strings.ToLower(tokens[0]) == "rss" {
|
||
|
|
if data, ok := p.cache[strings.ToLower(tokens[1])]; ok && time.Now().Before(data.expiration) {
|
||
|
|
log.Printf("rss cache hit")
|
||
|
|
p.Bot.SendMessage(message.Channel, data.data)
|
||
|
|
return true
|
||
|
|
} else {
|
||
|
|
fp := gofeed.NewParser()
|
||
|
|
feed, err := fp.ParseURL(tokens[1])
|
||
|
|
if err != nil {
|
||
|
|
p.Bot.SendMessage(message.Channel, fmt.Sprintf("RSS error: %s", err.Error()))
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
response := feed.Title
|
||
|
|
for _, item := range feed.Items {
|
||
|
|
response += fmt.Sprintf("\n%s", item.Title)
|
||
|
|
}
|
||
|
|
|
||
|
|
p.cache[strings.ToLower(tokens[1])] = &cacheItem{
|
||
|
|
key: strings.ToLower(tokens[1]),
|
||
|
|
data: response,
|
||
|
|
expiration: time.Now().Add(p.shelfLife),
|
||
|
|
}
|
||
|
|
|
||
|
|
p.Bot.SendMessage(message.Channel, response)
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *RSSPlugin) LoadData() {
|
||
|
|
// This bot has no data to load
|
||
|
|
}
|
||
|
|
|
||
|
|
// Help responds to help requests. Every plugin must implement a help function.
|
||
|
|
func (p *RSSPlugin) Help(channel string, parts []string) {
|
||
|
|
p.Bot.SendMessage(channel, "try '!rss http://rss.cnn.com/rss/edition.rss'")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Empty event handler because this plugin does not do anything on event recv
|
||
|
|
func (p *RSSPlugin) Event(kind string, message msg.Message) bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Handler for bot's own messages
|
||
|
|
func (p *RSSPlugin) BotMessage(message msg.Message) bool {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Register any web URLs desired
|
||
|
|
func (p *RSSPlugin) RegisterWeb() *string {
|
||
|
|
return nil
|
||
|
|
}
|