🎨 优化目录结构

plugin/diana/bing.go

@@ -0,0 +1,55 @@
+// Package diana 虚拟偶像女团 A-SOUL 成员嘉然相关
+package diana
+
+import (
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	control "github.com/FloatTech/zbputils/control"
+
+	"github.com/FloatTech/zbputils/control/order"
+
+	"github.com/FloatTech/ZeroBot-Plugin/plugin/diana/data"
+)
+
+var engine = control.Register("diana", order.AcquirePrio(), &control.Options{
+	DisableOnDefault: false,
+	Help: "嘉然\n" +
+		"- 小作文\n" +
+		"- 发大病\n" +
+		"- 教你一篇小作文[作文]\n" +
+		"- [回复]查重",
+	PublicDataFolder: "Diana",
+})
+
+func init() {
+	go func() {
+		datapath := engine.DataFolder()
+		dbfile := datapath + "text.db"
+		defer order.DoneOnExit()()
+		data.LoadText(dbfile)
+	}()
+
+	// 随机发送一篇上面的小作文
+	engine.OnFullMatch("小作文").SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			// 绕过第一行发病
+			ctx.SendChain(message.Text(data.RandText()))
+		})
+	// 逆天
+	engine.OnFullMatch("发大病").SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			// 第一行是发病
+			ctx.SendChain(message.Text(data.HentaiText()))
+		})
+	// 增加小作文
+	engine.OnRegex(`^教你一篇小作文(.*)$`, zero.AdminPermission).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			err := data.AddText(ctx.State["regex_matched"].([]string)[1])
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+			} else {
+				ctx.SendChain(message.Text("记住啦!"))
+			}
+		})
+}

plugin/diana/data/text.go

@@ -0,0 +1,61 @@
+// Package data 加载位于 datapath 的小作文
+package data
+
+import (
+	"crypto/md5"
+	"encoding/binary"
+
+	sql "github.com/FloatTech/sqlite"
+	binutils "github.com/FloatTech/zbputils/binary"
+	"github.com/FloatTech/zbputils/file"
+	"github.com/sirupsen/logrus"
+)
+
+var db = sql.Sqlite{}
+
+type text struct {
+	ID   int64  `db:"id"`
+	Data string `db:"data"`
+}
+
+// LoadText 加载小作文
+func LoadText(dbfile string) {
+	_, err := file.GetLazyData(dbfile, false, false)
+	db.DBPath = dbfile
+	if err != nil {
+		panic(err)
+	}
+	err = db.Create("text", &text{})
+	if err != nil {
+		panic(err)
+	}
+	c, _ := db.Count("text")
+	logrus.Printf("[Diana]读取%d条小作文", c)
+}
+
+// AddText 添加小作文
+func AddText(txt string) error {
+	s := md5.Sum(binutils.StringToBytes(txt))
+	i := binary.LittleEndian.Uint64(s[:8])
+	return db.Insert("text", &text{ID: int64(i), Data: txt})
+}
+
+// RandText 随机小作文
+func RandText() string {
+	var t text
+	err := db.Pick("text", &t)
+	if err != nil {
+		return err.Error()
+	}
+	return t.Data
+}
+
+// HentaiText 发大病
+func HentaiText() string {
+	var t text
+	err := db.Find("text", &t, "where id = -3802576048116006195")
+	if err != nil {
+		return err.Error()
+	}
+	return t.Data
+}

plugin/diana/zhiwang.go

@@ -0,0 +1,108 @@
+// Package diana 嘉然相关
+package diana
+
+import (
+	"bytes"
+	"encoding/json"
+	"math"
+	"time"
+
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"net/http"
+	"strings"
+
+	zero "github.com/wdvxdr1123/ZeroBot"
+)
+
+type zhiwang struct {
+	Code int `json:"code"`
+	Data struct {
+		EndTime   int             `json:"end_time"`
+		Rate      float64         `json:"rate"`
+		Related   [][]interface{} `json:"related"`
+		StartTime int             `json:"start_time"`
+	} `json:"data"`
+	Message string `json:"message"`
+}
+
+// 小作文查重: 回复要查的消息 查重
+func init() {
+	engine.OnMessage(fullmatch("查重")).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			msg := ctx.Event.Message
+			if msg[0].Type == "reply" {
+				msg := ctx.GetMessage(message.NewMessageID(msg[0].Data["id"])).Elements[0].Data["text"]
+				zhiwangjson := zhiwangapi(msg)
+
+				if zhiwangjson == nil || zhiwangjson.Code != 0 {
+					ctx.SendChain(message.Text("api返回错误"))
+					return
+				}
+
+				if len(zhiwangjson.Data.Related) == 0 {
+					ctx.SendChain(message.Text("枝网没搜到，查重率为0%，我的评价是：一眼真"))
+					return
+				}
+
+				related := zhiwangjson.Data.Related[0][1].(map[string]interface{})
+				ctx.SendChain(message.Text(
+					"枝网文本复制检测报告(简洁)", "\n",
+					"查重时间: ", time.Now().Format("2006-01-02 15:04:05"), "\n",
+					"总文字复制比: ", math.Floor(zhiwangjson.Data.Rate*100), "%", "\n",
+					"相似小作文:", "\n",
+					related["content"].(string)[:102]+".....", "\n",
+					"获赞数", related["like_num"], "\n",
+					zhiwangjson.Data.Related[0][2].(string), "\n",
+					"作者: ", related["m_name"], "\n",
+					"发表时间: ", time.Unix(int64(related["ctime"].(float64)), 0).Format("2006-01-02 15:04:05"), "\n",
+					"查重结果仅作参考，请注意辨别是否为原创", "\n",
+					"数据来源: https://asoulcnki.asia/",
+				))
+			}
+		})
+}
+
+// 发起api请求并把返回body交由json库解析
+func zhiwangapi(text string) *zhiwang {
+	url := "https://asoulcnki.asia/v1/api/check"
+	post := "{\n\"text\":\"" + text + "\"\n}"
+	var jsonStr = []byte(post)
+
+	req, _ := http.NewRequest("POST", url, bytes.NewBuffer(jsonStr))
+	req.Header.Set("Content-Type", "application/json")
+	client := &http.Client{}
+
+	resp, err := client.Do(req)
+
+	if err != nil {
+		return nil
+	}
+	defer resp.Body.Close()
+
+	result := &zhiwang{}
+	if err1 := json.NewDecoder(resp.Body).Decode(result); err1 != nil {
+		return nil
+	}
+	return result
+}
+
+func fullmatch(src ...string) zero.Rule {
+	return func(ctx *zero.Ctx) bool {
+		msg := ctx.Event.Message
+		for _, elem := range msg {
+			if elem.Type == "text" {
+				text := elem.Data["text"]
+				text = strings.ReplaceAll(text, " ", "")
+				text = strings.ReplaceAll(text, "\r", "")
+				text = strings.ReplaceAll(text, "\n", "")
+				for _, s := range src {
+					if text == s {
+						return true
+					}
+				}
+			}
+		}
+		return false
+	}
+}