From 112ccea89aa4b0abdc2588e1cff02b99e9bd088d Mon Sep 17 00:00:00 2001
From: Chris Sexton <chrissexton@users.noreply.github.com>
Date: Fri, 22 Jul 2022 17:06:55 -0400
Subject: [PATCH] cowboy: add cache

---
 plugins/cowboy/cowboy.go       | 20 +++++++++++++++++++-
 plugins/cowboy/cowboy_image.go | 25 +++++++++++++++++++++++--
 2 files changed, 42 insertions(+), 3 deletions(-)

diff --git a/plugins/cowboy/cowboy.go b/plugins/cowboy/cowboy.go
index b224d33..856dcd1 100644
--- a/plugins/cowboy/cowboy.go
+++ b/plugins/cowboy/cowboy.go
@@ -2,9 +2,10 @@ package cowboy
 
 import (
 	"fmt"
-	"github.com/velour/catbase/connectors/discord"
 	"regexp"
 
+	"github.com/velour/catbase/connectors/discord"
+
 	"github.com/rs/zerolog/log"
 	"github.com/velour/catbase/bot"
 	"github.com/velour/catbase/config"
@@ -38,6 +39,15 @@ func New(b bot.Bot) *Cowboy {
 
 func (p *Cowboy) register() {
 	tbl := bot.HandlerTable{
+		{
+			Kind: bot.Message, IsCmd: true,
+			Regex: regexp.MustCompile(`(?i)^:cowboy_clear_cache:$`),
+			Handler: func(r bot.Request) bool {
+				cowboyClearCache()
+				p.b.Send(r.Conn, bot.Ephemeral, r.Msg.Channel, r.Msg.User.ID, ":cowboy_cache_cleared:")
+				return true
+			},
+		},
 		{
 			Kind: bot.Message, IsCmd: false,
 			Regex: regexp.MustCompile(`(?i)^:cowboy_(?P<what>.+):$`),
@@ -51,6 +61,14 @@ func (p *Cowboy) register() {
 }
 
 func (p *Cowboy) makeCowboy(r bot.Request) {
+	what := r.Values["what"]
+	// This'll add the image to the cowboy_cache before discord tries to access it over http
+	_, err := cowboy(p.c, p.emojyPath, p.baseEmojyURL, what)
+	if err != nil {
+		log.Error().Err(err).Msg(":cowboy_fail:")
+		p.b.Send(r.Conn, bot.Ephemeral, r.Msg.Channel, r.Msg.User.ID, "Hey cowboy, that image wasn't there.")
+		return
+	}
 	log.Debug().Msgf("makeCowboy: %s", r.Values["what"])
 	base := p.c.Get("baseURL", "http://127.0.0.1:1337")
 	u := base + "/cowboy/img/" + r.Values["what"]
diff --git a/plugins/cowboy/cowboy_image.go b/plugins/cowboy/cowboy_image.go
index 8647137..5c27884 100644
--- a/plugins/cowboy/cowboy_image.go
+++ b/plugins/cowboy/cowboy_image.go
@@ -9,12 +9,19 @@ import (
 	"math"
 	"os"
 	"path"
+	"sync"
 
 	"github.com/nfnt/resize"
+	"github.com/rs/zerolog/log"
 	"github.com/velour/catbase/config"
 	"github.com/velour/catbase/plugins/emojy"
 )
 
+var (
+	cowboyCache = map[string][]byte{}
+	cowboyMutex = sync.Mutex{}
+)
+
 func getEmojy(emojyPath, baseEmojyURL, name string) (image.Image, error) {
 	files, _, err := emojy.AllFiles(emojyPath, baseEmojyURL)
 	if err != nil {
@@ -66,11 +73,18 @@ func cowboyifyImage(c *config.Config, emojyPath string, input image.Image) (imag
 }
 
 func cowboy(c *config.Config, emojyPath, baseEmojyURL, name string) ([]byte, error) {
-	emojy, err := getEmojy(emojyPath, baseEmojyURL, name)
+	cowboyMutex.Lock()
+	defer cowboyMutex.Unlock()
+	if img, ok := cowboyCache[name]; ok {
+		log.Debug().Msgf(":cowboy_using_cached_image: %s", name)
+		return img, nil
+	}
+	log.Debug().Msgf(":cowboy_generating_image: %s", name)
+	emjy, err := getEmojy(emojyPath, baseEmojyURL, name)
 	if err != nil {
 		return nil, err
 	}
-	img, err := cowboyifyImage(c, emojyPath, emojy)
+	img, err := cowboyifyImage(c, emojyPath, emjy)
 	if err != nil {
 		return nil, err
 	}
@@ -79,5 +93,12 @@ func cowboy(c *config.Config, emojyPath, baseEmojyURL, name string) ([]byte, err
 	if err != nil {
 		return nil, err
 	}
+	cowboyCache[name] = w.Bytes()
 	return w.Bytes(), nil
 }
+
+func cowboyClearCache() {
+	cowboyMutex.Lock()
+	defer cowboyMutex.Unlock()
+	cowboyCache = map[string][]byte{}
+}