catbase/plugins/twitter/twitter.go

package twitter

import (
	"fmt"
	"net/url"
	"strconv"
	"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
}

func New(b bot.Bot) *Twitter {
	t := Twitter{
		c: b.Config(),
		b: b,
	}
	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 {
			userKey := fmt.Sprintf("twitter.last.%s", u)
			lastTweet := int64(t.c.GetInt(userKey, 0))
			if lastTweet < tweet.Id && tweet.RetweetedStatus != nil {
				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.c.Set(userKey, strconv.FormatInt(tweet.Id, 10))
			}
		}
	}
}