101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"local1/logger"
|
|
"local3/rssmon2/config"
|
|
"local3/rssmon2/monitor"
|
|
"local3/rssmon2/rss"
|
|
"local3/rssmon2/server"
|
|
"local3/rssmon2/store"
|
|
"time"
|
|
)
|
|
|
|
const nsForFeeds = "FEEDS"
|
|
|
|
func main() {
|
|
config := config.New()
|
|
logger.Log("Starting with", config)
|
|
|
|
var sclient store.Client
|
|
var err error
|
|
if sclient, err = store.NewBolt(config.DBPath); err != nil {
|
|
panic(err)
|
|
}
|
|
defer sclient.Close()
|
|
|
|
allFeeds := make(map[string]*rss.Feed)
|
|
mon, err := monitor.New(func(url string) {
|
|
feed, ok := allFeeds[url]
|
|
if !ok {
|
|
f, err := rss.New(url, "", "")
|
|
if err != nil {
|
|
logger.Log("cannot identify unknown feed triggered in monitor: %q: %v", url, err)
|
|
return
|
|
}
|
|
b, err := sclient.Get(nsForFeeds, f.ID())
|
|
if err != nil {
|
|
logger.Log("cannot get unknown feed triggered in monitor: %q: %v", url, err)
|
|
return
|
|
}
|
|
feed, err = rss.Deserialize(b)
|
|
if err != nil {
|
|
logger.Log("cannot deserialize feed triggered in monitor: %q: %v", url, err)
|
|
return
|
|
}
|
|
}
|
|
items, err := allFeeds[url].Update()
|
|
if err != nil {
|
|
logger.Log("can't update old RSS %q: %v", url, err)
|
|
return
|
|
}
|
|
b, err := feed.Serialize()
|
|
if err != nil {
|
|
logger.Log("can't serialize to save RSS %q: %v", url, err)
|
|
return
|
|
}
|
|
if err := sclient.Set(nsForFeeds, feed.ID(), b); err != nil {
|
|
logger.Log("can't save RSS %q.%q: %v", nsForFeeds, feed.ID(), err)
|
|
return
|
|
}
|
|
logger.Log("Saved feed", feed)
|
|
for i := range items {
|
|
b, err := items[i].Serialize()
|
|
if err != nil {
|
|
logger.Log("can't save rss item %q.%q: %v", url, items[i].Link, err)
|
|
return
|
|
}
|
|
if err := sclient.Set(feed.ID(), items[i].ID(), b); err != nil {
|
|
logger.Log("can't save rss item %q.%q: %v", feed.ID(), items[i].ID(), err)
|
|
return
|
|
}
|
|
logger.Log("Saved feed item", feed.ID(), items[i].ID(), items[i])
|
|
}
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if err := mon.Start(); err != nil {
|
|
panic(err)
|
|
}
|
|
defer mon.Stop()
|
|
|
|
server, err := server.New(config.Port, func(url string, itemFilter, contentFilter string, interval time.Duration) {
|
|
feed, err := rss.New(url, itemFilter, contentFilter)
|
|
if err != nil {
|
|
logger.Log("can't create new RSS %q: %v", url, err)
|
|
return
|
|
}
|
|
allFeeds[url] = feed
|
|
if err := mon.Submit(url, interval); err != nil {
|
|
logger.Log("Cannot accept new feed %q: %v", url, err)
|
|
}
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if err := server.Serve(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|