mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
121 lines
2.8 KiB
Go
121 lines
2.8 KiB
Go
// Package emojimix 合成emoji
|
|
package emojimix
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/FloatTech/floatbox/binary"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/fumiama/terasu/http2"
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
const bed = "https://www.gstatic.com/android/keyboard/emojikitchen/%d/u%x/u%x_u%x.png"
|
|
|
|
func init() {
|
|
control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "合成emoji",
|
|
Help: "- [emoji][emoji]",
|
|
}).OnMessage(match).SetBlock(true).Limit(ctxext.LimitByUser).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
r := ctx.State["emojimix"].([]rune)
|
|
logrus.Debugln("[emojimix] match:", r)
|
|
r1, r2 := r[0], r[1]
|
|
u1 := fmt.Sprintf(bed, emojis[r1], r1, r1, r2)
|
|
u2 := fmt.Sprintf(bed, emojis[r2], r2, r2, r1)
|
|
logrus.Debugln("[emojimix] u1:", u1)
|
|
logrus.Debugln("[emojimix] u2:", u2)
|
|
send := func(resp *http.Response) {
|
|
buf := bytes.NewBuffer(make([]byte, 0, 65536))
|
|
buf.WriteString("base64://")
|
|
enc := base64.NewEncoder(base64.StdEncoding, buf)
|
|
_, err := io.Copy(enc, resp.Body)
|
|
if err != nil {
|
|
logrus.Warnln("[emojimix] copy err:", err)
|
|
return
|
|
}
|
|
err = enc.Close()
|
|
if err != nil {
|
|
logrus.Warnln("[emojimix] close enc err:", err)
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image(binary.BytesToString(buf.Bytes()), ctx.Event.Message.String()))
|
|
}
|
|
resp, err := http2.Get(u1)
|
|
if err == nil {
|
|
send(resp)
|
|
_ = resp.Body.Close()
|
|
return
|
|
}
|
|
logrus.Warnln("[emojimix] http get u1 err:", err)
|
|
resp, err = http2.Get(u2)
|
|
if err == nil {
|
|
send(resp)
|
|
_ = resp.Body.Close()
|
|
return
|
|
}
|
|
logrus.Warnln("[emojimix] http get u2 err:", err)
|
|
})
|
|
}
|
|
|
|
func match(ctx *zero.Ctx) bool {
|
|
logrus.Debugln("[emojimix] msg:", ctx.Event.Message)
|
|
if len(ctx.Event.Message) == 2 {
|
|
r1 := face2emoji(ctx.Event.Message[0])
|
|
if _, ok := emojis[r1]; !ok {
|
|
return false
|
|
}
|
|
r2 := face2emoji(ctx.Event.Message[1])
|
|
if _, ok := emojis[r2]; !ok {
|
|
return false
|
|
}
|
|
ctx.State["emojimix"] = []rune{r1, r2}
|
|
return true
|
|
}
|
|
|
|
r := []rune(ctx.Event.RawMessage)
|
|
logrus.Debugln("[emojimix] raw msg:", ctx.Event.RawMessage)
|
|
if len(r) == 2 {
|
|
if _, ok := emojis[r[0]]; !ok {
|
|
return false
|
|
}
|
|
if _, ok := emojis[r[1]]; !ok {
|
|
return false
|
|
}
|
|
ctx.State["emojimix"] = r
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func face2emoji(face message.MessageSegment) rune {
|
|
if face.Type == "text" {
|
|
r := []rune(face.Data["text"])
|
|
if len(r) != 1 {
|
|
return 0
|
|
}
|
|
return r[0]
|
|
}
|
|
if face.Type != "face" {
|
|
return 0
|
|
}
|
|
id, err := strconv.Atoi(face.Data["id"])
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
if r, ok := qqface[id]; ok {
|
|
return r
|
|
}
|
|
return 0
|
|
}
|