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

meme: fix another crash

This commit is contained in:
Chris Sexton 2020-05-01 10:39:42 -04:00 committed by Chris Sexton
parent 699532d789
commit 3ad15c5f3e
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
plugins/meme/meme.go
View File

@ -69,6 +69,7 @@ func (p *MemePlugin) message(c bot.Connector, kind bot.Kind, message msg.Message
func (p *MemePlugin) help(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool { func (p *MemePlugin) help(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
formats := p.c.GetMap("meme.memes", defaultFormats) formats := p.c.GetMap("meme.memes", defaultFormats)
msg := "Use `/meme [format] [text]` to create a meme.\nI know the following formats:" msg := "Use `/meme [format] [text]` to create a meme.\nI know the following formats:"
msg += "\n`[format]` can be a URL"
for k := range formats { for k := range formats {
msg += "\n" + k msg += "\n" + k
} }
@ -106,7 +107,12 @@ func (p *MemePlugin) registerWeb(c bot.Connector) {
top, bottom = parts[0], parts[1] top, bottom = parts[0], parts[1]
} }
id := p.genMeme(format, top, bottom) id, err := p.genMeme(format, top, bottom)
if err != nil {
msg := fmt.Sprintf("Hey %s, I couldn't download that image you asked for.", from)
p.bot.Send(c, bot.Message, channel, msg)
return
}
baseURL := p.c.Get("BaseURL", `https://catbase.velour.ninja`) baseURL := p.c.Get("BaseURL", `https://catbase.velour.ninja`)
u, _ := url.Parse(baseURL) u, _ := url.Parse(baseURL)
u.Path = path.Join(u.Path, "meme", "img", id) u.Path = path.Join(u.Path, "meme", "img", id)
@ -146,17 +152,19 @@ func (p *MemePlugin) registerWeb(c bot.Connector) {
}) })
} }
func DownloadTemplate(u *url.URL) image.Image { func DownloadTemplate(u *url.URL) (image.Image, error) {
res, err := http.Get(u.String()) res, err := http.Get(u.String())
if err != nil { if err != nil {
log.Error().Msgf("template from %s failed because of %v", u.String(), err) log.Error().Msgf("template from %s failed because of %v", u.String(), err)
return nil, err
} }
defer res.Body.Close() defer res.Body.Close()
image, _, err := image.Decode(res.Body) image, _, err := image.Decode(res.Body)
if err != nil { if err != nil {
log.Error().Msgf("Could not decode %v because of %v", u, err) log.Error().Msgf("Could not decode %v because of %v", u, err)
return nil, err
} }
return image return image, nil
} }
var defaultFormats = map[string]string{ var defaultFormats = map[string]string{
@ -169,7 +177,7 @@ var defaultFormats = map[string]string{
"raptor": "Philosoraptor.jpg", "raptor": "Philosoraptor.jpg",
} }
func (p *MemePlugin) genMeme(meme, top, bottom string) string { func (p *MemePlugin) genMeme(meme, top, bottom string) (string, error) {
fontSizes := []float64{48, 36, 24, 16, 12} fontSizes := []float64{48, 36, 24, 16, 12}
fontSize := fontSizes[0] fontSize := fontSizes[0]
@ -190,7 +198,11 @@ func (p *MemePlugin) genMeme(meme, top, bottom string) string {
log.Debug().Msgf("Attempting to download url: %s", u.String()) log.Debug().Msgf("Attempting to download url: %s", u.String())
img := DownloadTemplate(u) img, err := DownloadTemplate(u)
if err != nil {
log.Debug().Msgf("failed to download image: %s", err)
return "", err
}
r := img.Bounds() r := img.Bounds()
w := r.Dx() w := r.Dx()
@ -259,5 +271,5 @@ func (p *MemePlugin) genMeme(meme, top, bottom string) string {
log.Debug().Msgf("Saved to %s\n", path) log.Debug().Msgf("Saved to %s\n", path)
return path return path, nil
} }