mirror of https://github.com/velour/catbase.git
51 lines
922 B
Go
51 lines
922 B
Go
|
package cowboy
|
||
|
|
|
||
|
|
import (
|
||
|
|
"regexp"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
"github.com/velour/catbase/bot"
|
||
|
|
"github.com/velour/catbase/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Cowboy struct {
|
||
|
|
b bot.Bot
|
||
|
|
c *config.Config
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(b bot.Bot) *Cowboy {
|
||
|
|
c := Cowboy{
|
||
|
|
b: b,
|
||
|
|
c: b.Config(),
|
||
|
|
}
|
||
|
|
c.register()
|
||
|
|
c.registerWeb()
|
||
|
|
return &c
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Cowboy) register() {
|
||
|
|
tbl := bot.HandlerTable{
|
||
|
|
{
|
||
|
|
Kind: bot.Message, IsCmd: false,
|
||
|
|
Regex: regexp.MustCompile(`(?i)^:cowboy_(?P<what>.+):$`),
|
||
|
|
Handler: func(r bot.Request) bool {
|
||
|
|
p.makeCowboy(r)
|
||
|
|
return true
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
p.b.RegisterTable(p, tbl)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Cowboy) makeCowboy(r bot.Request) {
|
||
|
|
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"]
|
||
|
|
p.b.Send(r.Conn, bot.Message, r.Msg.Channel, "", bot.ImageAttachment{
|
||
|
|
URL: u,
|
||
|
|
AltTxt: r.Msg.Body,
|
||
|
|
Width: 64,
|
||
|
|
Height: 64,
|
||
|
|
})
|
||
|
|
}
|