🎨 优化目录结构

plugin/emojimix/mix.go

@@ -0,0 +1,101 @@
+// Package emojimix 合成emoji
+package emojimix
+
+import (
+	"fmt"
+	"net/http"
+	"strconv"
+
+	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/control/order"
+	"github.com/FloatTech/zbputils/ctxext"
+	"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.Register("emojimix", order.AcquirePrio(), &control.Options{
+		DisableOnDefault: false,
+		Help: "合成emoji\n" +
+			"- [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)
+			resp1, err := http.Head(u1)
+			if err == nil {
+				resp1.Body.Close()
+				if resp1.StatusCode == http.StatusOK {
+					ctx.SendChain(message.Image(u1))
+					return
+				}
+			}
+			resp2, err := http.Head(u2)
+			if err == nil {
+				resp2.Body.Close()
+				if resp2.StatusCode == http.StatusOK {
+					ctx.SendChain(message.Image(u2))
+					return
+				}
+			}
+		})
+}
+
+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
+}