Compare commits

..

No commits in common. "da780a7a92ad098dd3d5681197560e7db46ce637" and "04fecf1987c34799fcc3c021f14ae9176ceb7a49" have entirely different histories.

1 changed files with 49 additions and 61 deletions

View File

@ -5,7 +5,6 @@ import (
"github.com/bwmarrin/discordgo" "github.com/bwmarrin/discordgo"
"github.com/velour/catbase/plugins/emojy" "github.com/velour/catbase/plugins/emojy"
"regexp" "regexp"
"strings"
"github.com/velour/catbase/connectors/discord" "github.com/velour/catbase/connectors/discord"
@ -73,25 +72,26 @@ func (p *Cowboy) register() {
} }
func (p *Cowboy) makeCowboy(r bot.Request) { 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
overlays := p.c.GetMap("cowboy.overlays", defaultOverlays) overlays := p.c.GetMap("cowboy.overlays", defaultOverlays)
hat := overlays["hat"] hat := overlays["hat"]
what := r.Values["what"] i, err := cowboy(p.emojyPath, p.baseEmojyURL, hat, what)
_, err := p.mkEmojy(what, hat)
if err != nil { if err != nil {
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "Couldn't cowboify that, pardner.") 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 return
} }
log.Debug().Msgf("makeCowboy: %s", r.Values["what"])
base := p.c.Get("baseURL", "http://127.0.0.1:1337")
u := base + "/cowboy/img/hat/" + r.Values["what"]
p.b.Send(r.Conn, bot.Delete, r.Msg.Channel, r.Msg.ID) p.b.Send(r.Conn, bot.Delete, r.Msg.Channel, r.Msg.ID)
e := ":cowboy_" + what + ":" p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "", bot.ImageAttachment{
URL: u,
switch c := r.Conn.(type) { AltTxt: fmt.Sprintf("%s: %s", r.Msg.User.Name, r.Msg.Body),
case *discord.Discord: Width: i.Bounds().Max.X,
list := emojy.InvertEmojyList(c.GetEmojiList(true)) Height: i.Bounds().Max.Y,
e = strings.Trim(e, ":") })
e = fmt.Sprintf("<:%s:%s>", e, list[e])
}
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, r.Msg.User.Name+":")
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, e)
} }
func (p *Cowboy) registerCmds(d *discord.Discord) { func (p *Cowboy) registerCmds(d *discord.Discord) {
@ -142,11 +142,17 @@ func (p *Cowboy) registerCmds(d *discord.Discord) {
} }
} }
func (p *Cowboy) mkEmojy(name, overlay string) (string, error) { func (p *Cowboy) mkOverlayCB(overlay string) func(s *discordgo.Session, i *discordgo.InteractionCreate) {
return func(s *discordgo.Session, i *discordgo.InteractionCreate) {
lastEmojy := p.c.Get("cowboy.lastEmojy", "rust") lastEmojy := p.c.Get("cowboy.lastEmojy", "rust")
emojyPlugin := emojy.NewAPI(p.b) emojyPlugin := emojy.NewAPI(p.b)
list := map[string]string{} list := map[string]string{}
name := i.ApplicationCommandData().Options[0].StringValue()
if overlay == "" {
overlay = name
name = i.ApplicationCommandData().Options[1].StringValue()
}
msg := fmt.Sprintf("You asked for %s overlaid by %s", name, overlay) msg := fmt.Sprintf("You asked for %s overlaid by %s", name, overlay)
log.Debug().Msgf("got a cowboy command for %s overlaid by %s replacing %s", log.Debug().Msgf("got a cowboy command for %s overlaid by %s replacing %s",
name, overlay, lastEmojy) name, overlay, lastEmojy)
@ -154,12 +160,14 @@ func (p *Cowboy) mkEmojy(name, overlay string) (string, error) {
newEmojy, err := cowboy(p.emojyPath, p.baseEmojyURL, overlay, name) newEmojy, err := cowboy(p.emojyPath, p.baseEmojyURL, overlay, name)
if err != nil { if err != nil {
return "", err msg = err.Error()
goto resp
} }
err = emojyPlugin.RmEmojy(p.b.DefaultConnector(), lastEmojy) err = emojyPlugin.RmEmojy(p.b.DefaultConnector(), lastEmojy)
if err != nil { if err != nil {
return "", err msg = err.Error()
goto resp
} }
if overlay == "hat" { if overlay == "hat" {
@ -168,7 +176,8 @@ func (p *Cowboy) mkEmojy(name, overlay string) (string, error) {
name = emojy.SanitizeName(prefix + "_" + name) name = emojy.SanitizeName(prefix + "_" + name)
err = emojyPlugin.UploadEmojyImage(p.b.DefaultConnector(), name, newEmojy) err = emojyPlugin.UploadEmojyImage(p.b.DefaultConnector(), name, newEmojy)
if err != nil { if err != nil {
return "", err msg = err.Error()
goto resp
} }
p.c.Set("cowboy.lastEmojy", name) p.c.Set("cowboy.lastEmojy", name)
@ -177,33 +186,12 @@ func (p *Cowboy) mkEmojy(name, overlay string) (string, error) {
msg = fmt.Sprintf("You replaced %s with a new emojy %s <:%s:%s>, pardner!", msg = fmt.Sprintf("You replaced %s with a new emojy %s <:%s:%s>, pardner!",
lastEmojy, name, name, list[name]) lastEmojy, name, name, list[name])
return msg, nil resp:
}
func (p *Cowboy) mkOverlayCB(overlay string) func(s *discordgo.Session, i *discordgo.InteractionCreate) {
return func(s *discordgo.Session, i *discordgo.InteractionCreate) {
name := i.ApplicationCommandData().Options[0].StringValue()
if overlay == "" {
overlay = name
name = i.ApplicationCommandData().Options[1].StringValue()
}
msg, err := p.mkEmojy(name, overlay)
if err != nil {
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: err.Error(),
Flags: discordgo.MessageFlagsEphemeral,
},
})
return
}
s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{ s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource, Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{ Data: &discordgo.InteractionResponseData{
Content: msg, Content: msg,
Flags: discordgo.MessageFlagsEphemeral,
}, },
}) })
} }