mirror of https://github.com/velour/catbase.git
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
|
package twitter
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/url"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
|
||
|
|
"github.com/ChimeraCoder/anaconda"
|
||
|
|
|
||
|
|
"github.com/velour/catbase/bot"
|
||
|
|
"github.com/velour/catbase/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Twitter struct {
|
||
|
|
c *config.Config
|
||
|
|
b bot.Bot
|
||
|
|
api *anaconda.TwitterApi
|
||
|
|
|
||
|
|
tweetLog map[string]int64
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(b bot.Bot) *Twitter {
|
||
|
|
t := Twitter{
|
||
|
|
c: b.Config(),
|
||
|
|
b: b,
|
||
|
|
tweetLog: map[string]int64{},
|
||
|
|
}
|
||
|
|
token := t.c.GetString("twitter.accesstoken", "")
|
||
|
|
secret := t.c.GetString("twitter.accesssecret", "")
|
||
|
|
consumerKey := t.c.GetString("twitter.consumerkey", "")
|
||
|
|
consumerSecret := t.c.GetString("twitter.consumersecret", "")
|
||
|
|
//delay := t.c.GetInt("twitter.delay", 10)
|
||
|
|
|
||
|
|
// Don't create the API or start the loop without credentials
|
||
|
|
if token == "" || secret == "" || consumerKey == "" || consumerSecret == "" {
|
||
|
|
log.Error().
|
||
|
|
Str("token", token).
|
||
|
|
Str("secret", secret).
|
||
|
|
Str("consumerKey", consumerKey).
|
||
|
|
Str("consumerSecret", consumerSecret).
|
||
|
|
Msg("Could not start Twitter API")
|
||
|
|
return &t
|
||
|
|
}
|
||
|
|
|
||
|
|
t.api = anaconda.NewTwitterApiWithCredentials(token, secret, consumerKey, consumerSecret)
|
||
|
|
go t.timer(b.DefaultConnector())
|
||
|
|
|
||
|
|
return &t
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *Twitter) timer(c bot.Connector) {
|
||
|
|
checkTime := t.c.GetInt("twitter.checktime", 10)
|
||
|
|
ticker := time.NewTicker(time.Duration(checkTime) * time.Minute)
|
||
|
|
t.check(c)
|
||
|
|
for {
|
||
|
|
<-ticker.C
|
||
|
|
t.check(c)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *Twitter) check(c bot.Connector) {
|
||
|
|
log.Debug().Msg("checking twitter")
|
||
|
|
chs := t.c.GetArray("twitter.channels", []string{})
|
||
|
|
users := t.c.GetArray("twitter.users", []string{})
|
||
|
|
for _, u := range users {
|
||
|
|
vals := url.Values{"screen_name": []string{u}}
|
||
|
|
tweets, err := t.api.GetUserTimeline(vals)
|
||
|
|
if err != nil {
|
||
|
|
log.Error().Err(err).Str("user", u).Msg("error getting tweets")
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
for _, tweet := range tweets {
|
||
|
|
if last, ok := t.tweetLog[u]; !ok || last < tweet.Id {
|
||
|
|
link := fmt.Sprintf("https://twitter.com/%s/status/%s", u, tweet.IdStr)
|
||
|
|
log.Debug().Str("tweet", link).Msg("Unknown tweet")
|
||
|
|
for _, ch := range chs {
|
||
|
|
log.Debug().Str("ch", ch).Msg("Sending tweet")
|
||
|
|
t.b.Send(c, bot.Message, ch, link)
|
||
|
|
}
|
||
|
|
t.tweetLog[u] = tweet.Id
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|