cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

beers: host your own images

This commit is contained in:
Chris Sexton 2020-05-25 14:42:56 -04:00 committed by Chris Sexton
parent 645a533f49
commit b13385774e
4 changed files with 199 additions and 90 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bot/interfaces.go
View File

@ -39,7 +39,7 @@ type Kind int
type Callback func(Connector, Kind, msg.Message, ...interface{}) bool
type CallbackMap map[string]map[Kind][]Callback
// Bot interface serves to allow mocking of the actual bot
// b interface serves to allow mocking of the actual bot
type Bot interface {
// Config allows access to the bot's configuration system
Config() *config.Config

283
plugins/beers/beers.go
View File

@ -3,21 +3,29 @@
package beers
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"image"
"image/png"
"io/ioutil"
"math/rand"
"net/http"
"net/url"
"path"
"strconv"
"strings"
"time"
"github.com/google/uuid"
"github.com/jmoiron/sqlx"
"github.com/nfnt/resize"
"github.com/rs/zerolog/log"
"github.com/velour/catbase/bot"
"github.com/velour/catbase/bot/msg"
"github.com/velour/catbase/config"
"github.com/velour/catbase/plugins/counter"
)
@ -25,9 +33,12 @@ import (
const itemName = ":beer:"
var cachedImages = map[string][]byte{}
type BeersPlugin struct {
Bot bot.Bot
db *sqlx.DB
b bot.Bot
c *config.Config
db *sqlx.DB
untapdCache map[int]bool
}
@ -52,16 +63,22 @@ func New(b bot.Bot) *BeersPlugin {
log.Fatal().Err(err)
}
p := &BeersPlugin{
Bot: b,
db: b.DB(),
b: b,
c: b.Config(),
db: b.DB(),
untapdCache: make(map[int]bool),
}
for _, channel := range b.Config().GetArray("Untappd.Channels", []string{}) {
go p.untappdLoop(b.DefaultConnector(), channel)
}
b.Register(p, bot.Message, p.message)
b.Register(p, bot.Help, p.help)
p.registerWeb()
for _, channel := range p.c.GetArray("Untappd.Channels", []string{}) {
go p.untappdLoop(b.DefaultConnector(), channel)
}
return p
}
@ -88,13 +105,13 @@ func (p *BeersPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messag
count, err := strconv.Atoi(parts[2])
if err != nil {
// if it's not a number, maybe it's a nick!
p.Bot.Send(c, bot.Message, channel, "Sorry, that didn't make any sense.")
p.b.Send(c, bot.Message, channel, "Sorry, that didn't make any sense.")
}
if count < 0 {
// you can't be negative
msg := fmt.Sprintf("Sorry %s, you can't have negative beers!", nick)
p.Bot.Send(c, bot.Message, channel, msg)
p.b.Send(c, bot.Message, channel, msg)
return true
}
if parts[1] == "+=" {
@ -108,14 +125,14 @@ func (p *BeersPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messag
p.randomReply(c, channel)
}
} else {
p.Bot.Send(c, bot.Message, channel, "I don't know your math.")
p.b.Send(c, bot.Message, channel, "I don't know your math.")
}
} else if len(parts) == 2 {
if p.doIKnow(parts[1]) {
p.reportCount(c, parts[1], channel, false)
} else {
msg := fmt.Sprintf("Sorry, I don't know %s.", parts[1])
p.Bot.Send(c, bot.Message, channel, msg)
p.b.Send(c, bot.Message, channel, msg)
}
} else if len(parts) == 1 {
p.reportCount(c, nick, channel, true)
@ -139,7 +156,7 @@ func (p *BeersPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messag
channel := message.Channel
if len(parts) < 2 {
p.Bot.Send(c, bot.Message, channel, "You must also provide a user name.")
p.b.Send(c, bot.Message, channel, "You must also provide a user name.")
} else if len(parts) == 3 {
chanNick = parts[2]
} else if len(parts) == 4 {
@ -164,7 +181,7 @@ func (p *BeersPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messag
log.Error().Err(err).Msgf("Error registering untappd")
}
if count > 0 {
p.Bot.Send(c, bot.Message, channel, "I'm already watching you.")
p.b.Send(c, bot.Message, channel, "I'm already watching you.")
return true
}
_, err = p.db.Exec(`insert into untappd (
@ -180,11 +197,11 @@ func (p *BeersPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messag
)
if err != nil {
log.Error().Err(err).Msgf("Error registering untappd")
p.Bot.Send(c, bot.Message, channel, "I can't see.")
p.b.Send(c, bot.Message, channel, "I can't see.")
return true
}
p.Bot.Send(c, bot.Message, channel, "I'll be watching you.")
p.b.Send(c, bot.Message, channel, "I'll be watching you.")
p.checkUntappd(c, channel)
@ -207,7 +224,7 @@ func (p *BeersPlugin) help(c bot.Connector, kind bot.Kind, message msg.Message,
msg := "Beers: imbibe by using either beers +=,=,++ or with the !imbibe/drink " +
"commands. I'll keep a count of how many beers you've had and then if you want " +
"to reset, just !puke it all up!"
p.Bot.Send(c, bot.Message, message.Channel, msg)
p.b.Send(c, bot.Message, message.Channel, msg)
return true
}
@ -247,13 +264,13 @@ func (p *BeersPlugin) reportCount(c bot.Connector, nick, channel string, himself
msg = fmt.Sprintf("You've had %d beers so far, %s.", beers, nick)
}
}
p.Bot.Send(c, bot.Message, channel, msg)
p.b.Send(c, bot.Message, channel, msg)
}
func (p *BeersPlugin) puke(c bot.Connector, user string, channel string) {
p.setBeers(user, 0)
msg := fmt.Sprintf("Ohhhhhh, and a reversal of fortune for %s!", user)
p.Bot.Send(c, bot.Message, channel, msg)
p.b.Send(c, bot.Message, channel, msg)
}
func (p *BeersPlugin) doIKnow(nick string) bool {
@ -268,7 +285,7 @@ func (p *BeersPlugin) doIKnow(nick string) bool {
// Sends random affirmation to the channel. This could be better (with a datastore for sayings)
func (p *BeersPlugin) randomReply(c bot.Connector, channel string) {
replies := []string{"ZIGGY! ZAGGY!", "HIC!", "Stay thirsty, my friend!"}
p.Bot.Send(c, bot.Message, channel, replies[rand.Intn(len(replies))])
p.b.Send(c, bot.Message, channel, replies[rand.Intn(len(replies))])
}
type checkin struct {
@ -326,7 +343,7 @@ type Beers struct {
}
func (p *BeersPlugin) pullUntappd() ([]checkin, error) {
token := p.Bot.Config().Get("Untappd.Token", "NONE")
token := p.c.Get("Untappd.Token", "NONE")
if token == "NONE" || token == "" {
return []checkin{}, fmt.Errorf("No untappd token")
}
@ -361,7 +378,7 @@ func (p *BeersPlugin) pullUntappd() ([]checkin, error) {
}
func (p *BeersPlugin) checkUntappd(c bot.Connector, channel string) {
token := p.Bot.Config().Get("Untappd.Token", "NONE")
token := p.c.Get("Untappd.Token", "NONE")
if token == "NONE" {
log.Info().
Msg(`Set config value "untappd.token" if you wish to enable untappd`)
@ -398,81 +415,158 @@ func (p *BeersPlugin) checkUntappd(c bot.Connector, channel string) {
continue
}
venue := ""
switch v := checkin.Venue.(type) {
case map[string]interface{}:
venue = " at " + v["venue_name"].(string)
}
beerName := checkin.Beer["beer_name"].(string)
breweryName := checkin.Brewery["brewery_name"].(string)
user, ok := userMap[checkin.User.User_name]
if !ok {
continue
}
log.Debug().
Msgf("user.chanNick: %s, user.untappdUser: %s, checkin.User.User_name: %s",
user.chanNick, user.untappdUser, checkin.User.User_name)
p.addBeers(user.chanNick, 1)
drunken := p.getBeers(user.chanNick)
msg := fmt.Sprintf("%s just drank %s by %s%s, bringing his drunkeness to %d",
user.chanNick, beerName, breweryName, venue, drunken)
if checkin.Rating_score > 0 {
msg = fmt.Sprintf("%s. Rating: %.2f", msg, checkin.Rating_score)
}
if checkin.Checkin_comment != "" {
msg = fmt.Sprintf("%s -- %s",
msg, checkin.Checkin_comment)
}
args := []interface{}{
channel,
msg,
}
if checkin.Badges.Count > 0 {
for _, b := range checkin.Badges.Items {
args = append(args, bot.ImageAttachment{
URL: b.BadgeImage.Sm,
AltTxt: b.BadgeName,
})
}
}
if checkin.Media.Count > 0 {
if strings.Contains(checkin.Media.Items[0].Photo.Photo_img_lg, "photos-processing") {
continue
}
args = append(args, bot.ImageAttachment{
URL: checkin.Media.Items[0].Photo.Photo_img_lg,
AltTxt: "Here's a photo",
})
} else if !p.untapdCache[checkin.Checkin_id] {
// Mark checkin as "seen" but not complete, continue to next checkin
log.Debug().Msgf("Deferring checkin: %#v", checkin)
p.untapdCache[checkin.Checkin_id] = true
continue
} else {
// We've seen this checkin, so unmark and accept that there's no media
delete(p.untapdCache, checkin.Checkin_id)
}
user.lastCheckin = checkin.Checkin_id
_, err := p.db.Exec(`update untappd set
lastCheckin = ?
where id = ?`, user.lastCheckin, user.id)
if err != nil {
log.Error().Err(err).Msg("UPDATE ERROR!")
}
log.Debug().
Int("checkin_id", checkin.Checkin_id).
Str("msg", msg).
Msg("checkin")
p.Bot.Send(c, bot.Message, args...)
p.sendCheckin(c, channel, user, checkin)
}
}
func (p *BeersPlugin) sendCheckin(c bot.Connector, channel string, user untappdUser, checkin checkin) {
venue := ""
switch v := checkin.Venue.(type) {
case map[string]interface{}:
venue = " at " + v["venue_name"].(string)
}
beerName := checkin.Beer["beer_name"].(string)
breweryName := checkin.Brewery["brewery_name"].(string)
log.Debug().
Msgf("user.chanNick: %s, user.untappdUser: %s, checkin.User.User_name: %s",
user.chanNick, user.untappdUser, checkin.User.User_name)
p.addBeers(user.chanNick, 1)
drunken := p.getBeers(user.chanNick)
msg := fmt.Sprintf("%s just drank %s by %s%s, bringing his drunkeness to %d",
user.chanNick, beerName, breweryName, venue, drunken)
if checkin.Rating_score > 0 {
msg = fmt.Sprintf("%s. Rating: %.2f", msg, checkin.Rating_score)
}
if checkin.Checkin_comment != "" {
msg = fmt.Sprintf("%s -- %s",
msg, checkin.Checkin_comment)
}
args := []interface{}{
channel,
msg,
}
if checkin.Badges.Count > 0 {
for _, b := range checkin.Badges.Items {
args = append(args, bot.ImageAttachment{
URL: b.BadgeImage.Sm,
AltTxt: b.BadgeName,
})
}
}
if checkin.Media.Count > 0 {
if strings.Contains(checkin.Media.Items[0].Photo.Photo_img_lg, "photos-processing") {
return
}
mediaURL := p.getMedia(checkin.Media.Items[0].Photo.Photo_img_lg)
args = append(args, bot.ImageAttachment{
URL: mediaURL,
AltTxt: "Here's a photo",
})
} else if !p.untapdCache[checkin.Checkin_id] {
// Mark checkin as "seen" but not complete, continue to next checkin
log.Debug().Msgf("Deferring checkin: %#v", checkin)
p.untapdCache[checkin.Checkin_id] = true
return
} else {
// We've seen this checkin, so unmark and accept that there's no media
delete(p.untapdCache, checkin.Checkin_id)
}
user.lastCheckin = checkin.Checkin_id
_, err := p.db.Exec(`update untappd set
lastCheckin = ?
where id = ?`, user.lastCheckin, user.id)
if err != nil {
log.Error().Err(err).Msg("UPDATE ERROR!")
}
log.Debug().
Int("checkin_id", checkin.Checkin_id).
Str("msg", msg).
Interface("args", args).
Msg("checkin")
p.b.Send(c, bot.Message, args...)
}
func (p *BeersPlugin) getMedia(src string) string {
u, err := url.Parse(src)
if err != nil {
return src
}
img, err := downloadMedia(u)
if err != nil {
return src
}
r := img.Bounds()
w := r.Dx()
h := r.Dy()
maxSz := p.c.GetFloat64("maxImgSz", 750.0)
if w > h {
scale := maxSz / float64(w)
w = int(float64(w) * scale)
h = int(float64(h) * scale)
} else {
scale := maxSz / float64(h)
w = int(float64(w) * scale)
h = int(float64(h) * scale)
}
log.Debug().Msgf("trynig to resize to %v, %v", w, h)
img = resize.Resize(uint(w), uint(h), img, resize.Lanczos3)
r = img.Bounds()
w = r.Dx()
h = r.Dy()
log.Debug().Msgf("resized to %v, %v", w, h)
buf := bytes.Buffer{}
err = png.Encode(&buf, img)
if err != nil {
return src
}
baseURL := p.c.Get("BaseURL", `https://catbase.velour.ninja`)
id := uuid.New().String()
cachedImages[id] = buf.Bytes()
u, _ = url.Parse(baseURL)
u.Path = path.Join(u.Path, "beers", "img", id)
log.Debug().Msgf("New image at %s", u)
return u.String()
}
func downloadMedia(u *url.URL) (image.Image, error) {
res, err := http.Get(u.String())
if err != nil {
log.Error().Msgf("template from %s failed because of %v", u.String(), err)
return nil, err
}
defer res.Body.Close()
image, _, err := image.Decode(res.Body)
if err != nil {
log.Error().Msgf("Could not decode %v because of %v", u, err)
return nil, err
}
return image, nil
}
func (p *BeersPlugin) untappdLoop(c bot.Connector, channel string) {
frequency := p.Bot.Config().GetInt("Untappd.Freq", 120)
frequency := p.c.GetInt("Untappd.Freq", 120)
if frequency == 0 {
return
}
@ -484,3 +578,18 @@ func (p *BeersPlugin) untappdLoop(c bot.Connector, channel string) {
p.checkUntappd(c, channel)
}
}
func (p *BeersPlugin) registerWeb() {
http.HandleFunc("/beers/img/", p.img)
}
func (p *BeersPlugin) img(w http.ResponseWriter, r *http.Request) {
_, file := path.Split(r.URL.Path)
id := file
if img, ok := cachedImages[id]; ok {
w.Write(img)
} else {
w.WriteHeader(404)
w.Write([]byte("not found"))
}
}

2
plugins/meme/meme.go
View File

@ -348,7 +348,7 @@ func (p *MemePlugin) genMeme(meme, top, bottom, bully string) (string, error) {
w := r.Dx()
h := r.Dy()
maxSz := 750.0
maxSz := p.c.GetFloat64("maxImgSz", 750.0)
if w > h {
scale := maxSz / float64(w)

2
plugins/reaction/reaction.go
View File

@ -62,7 +62,7 @@ func (p *ReactionPlugin) message(c bot.Connector, kind bot.Kind, message msg.Mes
return false
}
// Bot will always react if a message contains a check word
// b will always react if a message contains a check word
// Note that reactions must not be enclosed in :
func (p *ReactionPlugin) checkReactions(c bot.Connector, m msg.Message) {
checkWords := p.config.GetArray("reaction.checkwords", []string{})