meme: add configurable memes

plugins/meme/meme.go

@@ -4,7 +4,6 @@ import (
 	"bytes"
 	"encoding/json"
 	"fmt"
-	"html/template"
 	"image"
 	"image/color"
 	"image/draw"
@@ -13,7 +12,6 @@ import (
 	"net/url"
 	"path"
 	"regexp"
-	"sort"
 	"strings"
 	"time"
 
@@ -40,6 +38,12 @@ type cachedImage struct {
 	repr    []byte
 }
 
+type memeText struct {
+	XPerc float64 `json:"x"`
+	YPerc float64 `json:"y"`
+	Text  string  `json:"t"`
+}
+
 var horizon = 24 * 7
 
 type cachedImages map[string]*cachedImage
@@ -92,6 +96,8 @@ func (p *MemePlugin) help(c bot.Connector, kind bot.Kind, message msg.Message, a
 	msg += fmt.Sprintf("\nor a format from the list of %d pre-made memes listed on the website", len(formats))
 	msg += fmt.Sprintf("\nHead over to %s/meme to view and add new meme formats", webRoot)
 	msg += "\nYou can use `_` as a placeholder for empty text and a newline (|| on Discord) to separate top vs bottom."
+	msg += "\nOR you can configure the text yourself with JSON. Send a code quoted message like:"
+	msg += "```/meme doge `[{\"x\": 0.2, \"y\": 0.1, \"t\": \"such image\"},{\"x\": 0.7, \"y\": 0.5, \"t\": \"much meme\"},{\"x\": 0.4, \"y\": 0.8, \"t\": \"wow\"}]` ```"
 	p.bot.Send(c, bot.Message, message.Channel, msg)
 	return true
 }
@@ -120,101 +126,6 @@ type ByName struct{ webResps }
 
 func (s ByName) Less(i, j int) bool { return s.webResps[i].Name < s.webResps[j].Name }
 
-func (p *MemePlugin) all(w http.ResponseWriter, r *http.Request) {
-	memes := p.c.GetMap("meme.memes", defaultFormats)
-	values := webResps{}
-	for n, u := range memes {
-
-		realURL, err := url.Parse(u)
-		if err != nil || realURL.Scheme == "" {
-			realURL, err = url.Parse("https://imgflip.com/s/meme/" + u)
-			if err != nil {
-				values = append(values, webResp{n, "404.png"})
-				log.Error().Err(err).Msgf("invalid URL")
-				continue
-			}
-		}
-		values = append(values, webResp{n, realURL.String()})
-	}
-	sort.Sort(ByName{values})
-
-	out, err := json.Marshal(values)
-	if err != nil {
-		w.WriteHeader(500)
-		log.Error().Err(err).Msgf("could not serve all memes route")
-		return
-	}
-	w.Write(out)
-}
-
-func mkCheckError(w http.ResponseWriter) func(error) bool {
-	return func(err error) bool {
-		if err != nil {
-			log.Error().Err(err).Msgf("meme failed")
-			w.WriteHeader(500)
-			e, _ := json.Marshal(err)
-			w.Write(e)
-			return true
-		}
-		return false
-	}
-}
-
-func (p *MemePlugin) rmMeme(w http.ResponseWriter, r *http.Request) {
-	if r.Method != http.MethodDelete {
-		w.WriteHeader(405)
-		fmt.Fprintf(w, "Incorrect HTTP method")
-		return
-	}
-	checkError := mkCheckError(w)
-	decoder := json.NewDecoder(r.Body)
-	values := webResp{}
-	err := decoder.Decode(&values)
-	if checkError(err) {
-		return
-	}
-	formats := p.c.GetMap("meme.memes", defaultFormats)
-	delete(formats, values.Name)
-	err = p.c.SetMap("meme.memes", formats)
-	checkError(err)
-}
-
-func (p *MemePlugin) addMeme(w http.ResponseWriter, r *http.Request) {
-	if r.Method != http.MethodPost {
-		w.WriteHeader(405)
-		fmt.Fprintf(w, "Incorrect HTTP method")
-		return
-	}
-	checkError := mkCheckError(w)
-	decoder := json.NewDecoder(r.Body)
-	values := webResp{}
-	err := decoder.Decode(&values)
-	if checkError(err) {
-		return
-	}
-	formats := p.c.GetMap("meme.memes", defaultFormats)
-	formats[values.Name] = values.URL
-	err = p.c.SetMap("meme.memes", formats)
-	checkError(err)
-}
-
-func (p *MemePlugin) webRoot(w http.ResponseWriter, r *http.Request) {
-	var tpl = template.Must(template.New("factoidIndex").Parse(string(memeIndex)))
-	tpl.Execute(w, struct{ Nav []bot.EndPoint }{p.bot.GetWebNavigation()})
-}
-
-func (p *MemePlugin) img(w http.ResponseWriter, r *http.Request) {
-	_, file := path.Split(r.URL.Path)
-	id := file
-	if img, ok := p.images[id]; ok {
-		w.Write(img.repr)
-	} else {
-		w.WriteHeader(404)
-		w.Write([]byte("not found"))
-	}
-	p.images.cleanup()
-}
-
 func (p *MemePlugin) bully(c bot.Connector, format, id string) image.Image {
 	bullyIcon := ""
 
@@ -262,6 +173,18 @@ func (p *MemePlugin) sendMeme(c bot.Connector, channel, channelName, msgID strin
 	log.Debug().Strs("parts", parts).Msgf("Meme:\n%+v", text)
 
 	go func() {
+		var config []memeText
+
+		message = strings.TrimPrefix(message, "`")
+		message = strings.TrimSuffix(message, "`")
+
+		err := json.Unmarshal([]byte(message), &config)
+		if err == nil {
+			message = ""
+			for _, c := range config {
+				message += c.Text + "\n"
+			}
+		} else {
 			top, bottom := "", message
 			if strings.Contains(message, "||") {
 				parts = strings.Split(message, "||")
@@ -278,9 +201,17 @@ func (p *MemePlugin) sendMeme(c bot.Connector, channel, channelName, msgID strin
 				message = top
 			}
 
+			topPos := p.bot.Config().GetFloat64("meme.top", 0.05)
+			bottomPos := p.bot.Config().GetFloat64("meme.bottom", 0.95)
+			config = []memeText{
+				{Text: top, XPerc: 0.5, YPerc: topPos},
+				{Text: bottom, XPerc: 0.5, YPerc: bottomPos},
+			}
+		}
+
 		bullyImg := p.bully(c, format, from.ID)
 
-		id, w, h, err := p.genMeme(format, top, bottom, bullyImg)
+		id, w, h, err := p.genMeme(format, bullyImg, config)
 		if err != nil {
 			msg := fmt.Sprintf("Hey %v, I couldn't download that image you asked for.", from.Name)
 			p.bot.Send(c, bot.Message, channel, msg)
@@ -374,10 +305,46 @@ var defaultFormats = map[string]string{
 	"raptor": "Philosoraptor.jpg",
 }
 
-func (p *MemePlugin) genMeme(meme, top, bottom string, bully image.Image) (string, int, int, error) {
+func (p *MemePlugin) findFontSize(config []memeText, w, h int, sizes []float64) float64 {
-	fontSizes := []float64{48, 36, 24, 16, 12}
+	fontSize := 12.0
-	fontSize := fontSizes[0]
 
+	m := gg.NewContext(w, h)
+	fontLocation := p.c.Get("meme.font", "impact.ttf")
+
+	longestStr, longestW := "", 0.0
+
+	for _, s := range config {
+		err := m.LoadFontFace(fontLocation, 12) // problem
+		if err != nil {
+			log.Error().Err(err).Msg("could not load font")
+			return fontSize
+		}
+
+		w, _ := m.MeasureString(s.Text)
+		if w > longestW {
+			longestStr = s.Text
+			longestW = w
+		}
+	}
+
+	for _, sz := range sizes {
+		err := m.LoadFontFace(fontLocation, sz) // problem
+		if err != nil {
+			log.Error().Err(err).Msg("could not load font")
+			return fontSize
+		}
+
+		topW, _ := m.MeasureString(longestStr)
+		if topW < float64(w) {
+			fontSize = sz
+			break
+		}
+	}
+	return fontSize
+}
+
+func (p *MemePlugin) genMeme(meme string, bully image.Image, config []memeText) (string, int, int, error) {
+	fontSizes := []float64{48, 36, 24, 16, 12}
 	formats := p.c.GetMap("meme.memes", defaultFormats)
 
 	path := uuid.New().String()
@@ -405,6 +372,8 @@ func (p *MemePlugin) genMeme(meme, top, bottom string, bully image.Image) (strin
 	w := r.Dx()
 	h := r.Dy()
 
+	// /meme2 5guys [{"x": 0.1, "y": 0.1, "t": "test"}]
+
 	maxSz := p.c.GetFloat64("maxImgSz", 750.0)
 
 	if w > h {
@@ -431,25 +400,7 @@ func (p *MemePlugin) genMeme(meme, top, bottom string, bully image.Image) (strin
 	m := gg.NewContext(w, h)
 	m.DrawImage(img, 0, 0)
 	fontLocation := p.c.Get("meme.font", "impact.ttf")
-	for _, sz := range fontSizes {
+	m.LoadFontFace(fontLocation, p.findFontSize(config, w, h, fontSizes))
-		err = m.LoadFontFace(fontLocation, sz) // problem
-		if err != nil {
-			log.Error().Err(err).Msg("could not load font")
-		}
-		topW, _ := m.MeasureString(top)
-		botW, _ := m.MeasureString(bottom)
-		if topW < float64(w) && botW < float64(w) {
-			fontSize = sz
-			break
-		}
-	}
-
-	if top == "_" {
-		top = ""
-	}
-	if bottom == "_" {
-		bottom = ""
-	}
 
 	// Apply black stroke
 	m.SetHexColor("#000")
@@ -460,18 +411,21 @@ func (p *MemePlugin) genMeme(meme, top, bottom string, bully image.Image) (strin
 			if dx*dx+dy*dy >= strokeSize*strokeSize {
 				continue
 			}
-			x := float64(w/2 + dx)
+			for _, c := range config {
-			y := float64(h) - fontSize + float64(dy)
+				x := float64(w)*c.XPerc + float64(dx)
-			y0 := fontSize + float64(dy)
+				y := float64(h)*c.YPerc + float64(dy)
-			m.DrawStringAnchored(top, x, y0, 0.5, 0.5)
+				m.DrawStringAnchored(c.Text, x, y, 0.5, 0.5)
-			m.DrawStringAnchored(bottom, x, y, 0.5, 0.5)
+			}
 		}
 	}
 
 	// Apply white fill
 	m.SetHexColor("#FFF")
-	m.DrawStringAnchored(top, float64(w)/2, fontSize, 0.5, 0.5)
+	for _, c := range config {
-	m.DrawStringAnchored(bottom, float64(w)/2, float64(h)-fontSize, 0.5, 0.5)
+		x := float64(w) * c.XPerc
+		y := float64(h) * c.YPerc
+		m.DrawStringAnchored(c.Text, x, y, 0.5, 0.5)
+	}
 
 	i := bytes.Buffer{}
 	png.Encode(&i, m.Image())

plugins/meme/webHandlers.go

@@ -0,0 +1,110 @@
+package meme
+
+import (
+	"encoding/json"
+	"fmt"
+	"html/template"
+	"net/http"
+	"net/url"
+	"path"
+	"sort"
+
+	"github.com/rs/zerolog/log"
+
+	"github.com/velour/catbase/bot"
+)
+
+func (p *MemePlugin) all(w http.ResponseWriter, r *http.Request) {
+	memes := p.c.GetMap("meme.memes", defaultFormats)
+	values := webResps{}
+	for n, u := range memes {
+
+		realURL, err := url.Parse(u)
+		if err != nil || realURL.Scheme == "" {
+			realURL, err = url.Parse("https://imgflip.com/s/meme/" + u)
+			if err != nil {
+				values = append(values, webResp{n, "404.png"})
+				log.Error().Err(err).Msgf("invalid URL")
+				continue
+			}
+		}
+		values = append(values, webResp{n, realURL.String()})
+	}
+	sort.Sort(ByName{values})
+
+	out, err := json.Marshal(values)
+	if err != nil {
+		w.WriteHeader(500)
+		log.Error().Err(err).Msgf("could not serve all memes route")
+		return
+	}
+	w.Write(out)
+}
+
+func mkCheckError(w http.ResponseWriter) func(error) bool {
+	return func(err error) bool {
+		if err != nil {
+			log.Error().Err(err).Msgf("meme failed")
+			w.WriteHeader(500)
+			e, _ := json.Marshal(err)
+			w.Write(e)
+			return true
+		}
+		return false
+	}
+}
+
+func (p *MemePlugin) rmMeme(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodDelete {
+		w.WriteHeader(405)
+		fmt.Fprintf(w, "Incorrect HTTP method")
+		return
+	}
+	checkError := mkCheckError(w)
+	decoder := json.NewDecoder(r.Body)
+	values := webResp{}
+	err := decoder.Decode(&values)
+	if checkError(err) {
+		return
+	}
+	formats := p.c.GetMap("meme.memes", defaultFormats)
+	delete(formats, values.Name)
+	err = p.c.SetMap("meme.memes", formats)
+	checkError(err)
+}
+
+func (p *MemePlugin) addMeme(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodPost {
+		w.WriteHeader(405)
+		fmt.Fprintf(w, "Incorrect HTTP method")
+		return
+	}
+	checkError := mkCheckError(w)
+	decoder := json.NewDecoder(r.Body)
+	values := webResp{}
+	err := decoder.Decode(&values)
+	if checkError(err) {
+		return
+	}
+	formats := p.c.GetMap("meme.memes", defaultFormats)
+	formats[values.Name] = values.URL
+	err = p.c.SetMap("meme.memes", formats)
+	checkError(err)
+}
+
+func (p *MemePlugin) webRoot(w http.ResponseWriter, r *http.Request) {
+	var tpl = template.Must(template.New("factoidIndex").Parse(string(memeIndex)))
+	tpl.Execute(w, struct{ Nav []bot.EndPoint }{p.bot.GetWebNavigation()})
+}
+
+func (p *MemePlugin) img(w http.ResponseWriter, r *http.Request) {
+	_, file := path.Split(r.URL.Path)
+	id := file
+	if img, ok := p.images[id]; ok {
+		w.Write(img.repr)
+	} else {
+		w.WriteHeader(404)
+		w.Write([]byte("not found"))
+	}
+	p.images.cleanup()
+}