🎨 优化目录结构

plugin/wtf/main.go

@@ -0,0 +1,63 @@
+// Package wtf 鬼东西
+package wtf
+
+import (
+	"fmt"
+	"strconv"
+
+	control "github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"github.com/FloatTech/zbputils/control/order"
+)
+
+func init() {
+	en := control.Register("wtf", order.AcquirePrio(), &control.Options{
+		DisableOnDefault: false,
+		Help:             "鬼东西\n- 鬼东西列表\n- 查询鬼东西[序号][@xxx]",
+	})
+	en.OnFullMatch("鬼东西列表").SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			s := ""
+			for i, w := range table {
+				s += fmt.Sprintf("%02d. %s\n", i, w.name)
+			}
+			ctx.SendChain(message.Text(s))
+		})
+	en.OnRegex(`^查询鬼东西(\d*)`, zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
+		Handle(func(ctx *zero.Ctx) {
+			// 调用接口
+			i, err := strconv.Atoi(ctx.State["regex_matched"].([]string)[1])
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
+			}
+			w := newWtf(i)
+			if w == nil {
+				ctx.SendChain(message.Text("没有这项内容！"))
+				return
+			}
+			// 获取名字
+			var name string
+			var secondname string
+			if len(ctx.Event.Message) > 1 && ctx.Event.Message[1].Type == "at" {
+				qq, _ := strconv.ParseInt(ctx.Event.Message[1].Data["qq"], 10, 64)
+				secondname = ctx.GetGroupMemberInfo(ctx.Event.GroupID, qq, false).Get("nickname").Str
+			}
+			name = ctx.Event.Sender.NickName
+			var text string
+			if secondname != "" {
+				text, err = w.predict(name, secondname)
+			} else {
+				text, err = w.predict(name)
+			}
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
+			}
+			// TODO: 可注入
+			ctx.Send(text)
+		})
+}