🎨 优化目录结构

plugin/juejuezi/juejuezi.go

@@ -0,0 +1,71 @@
+// Package juejuezi 绝绝子
+package juejuezi
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+
+	control "github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	log "github.com/sirupsen/logrus"
+	"github.com/tidwall/gjson"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
+
+	"github.com/FloatTech/zbputils/control/order"
+)
+
+const (
+	juejueziURL = "https://www.offjuan.com/api/juejuezi/text"
+	referer     = "https://juejuezi.offjuan.com/"
+	ua          = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"
+)
+
+func init() {
+	control.Register("juejuezi", order.AcquirePrio(), &control.Options{
+		DisableOnDefault: false,
+		Help: "绝绝子生成器\n" +
+			"- 喝奶茶绝绝子 | 绝绝子吃饭",
+	}).OnRegex("[\u4E00-\u9FA5]{0,10}绝绝子[\u4E00-\u9FA5]{0,10}").SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+		toDealStr := []rune(strings.ReplaceAll(ctx.ExtractPlainText(), "绝绝子", ""))
+		switch len(toDealStr) {
+		case 0, 1:
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("不要只输入绝绝子"))
+		case 2:
+			data, err := juejuezi(string(toDealStr[0]), string(toDealStr[1]))
+			if err != nil {
+				ctx.SendChain(message.Text(err))
+			}
+			ctx.SendChain(message.Text(gjson.Get(helper.BytesToString(data), "text").String()))
+		default:
+			params := ctx.GetWordSlices(string(toDealStr)).Get("slices").Array()
+			data, err := juejuezi(params[0].String(), params[1].String())
+			if err != nil {
+				ctx.SendChain(message.Text(err))
+			}
+			ctx.SendChain(message.Text(gjson.Get(helper.BytesToString(data), "text").String()))
+		}
+	})
+}
+
+func juejuezi(verb, noun string) (data []byte, err error) {
+	juejueziStr := fmt.Sprintf("{\"verb\":\"%s\",\"noun\":\"%s\"}", verb, noun)
+	client := &http.Client{}
+	// 提交请求
+	request, err := http.NewRequest("POST", juejueziURL, strings.NewReader(juejueziStr))
+	if err != nil {
+		log.Errorln("[juejuezi]:", err)
+	}
+	request.Header.Add("Referer", referer)
+	request.Header.Add("User-Agent", ua)
+	response, err := client.Do(request)
+	if err != nil {
+		log.Errorln("[juejuezi]:", err)
+	}
+	data, err = io.ReadAll(response.Body)
+	response.Body.Close()
+	return
+}