独立骂人插件 (#102)

* fix:vtb数据不用从zbp更新

* feat:独立骂人插件

* fix:修lint

* feat:添加关键词

Co-authored-by: Guohuiyuan <haibaraguo@yeahka.com>

plugin_curse/curse.go

@@ -0,0 +1,50 @@
+// Package curse 骂人插件(求骂,自卫)
+package curse
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/control"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/process"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/rate"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"time"
+)
+
+const (
+	prio     = 10
+	minLevel = "min"
+	maxLevel = "max"
+)
+
+var (
+	engine = control.Register("curse", &control.Options{
+		DisableOnDefault: false,
+		Help:             "骂人(求骂,自卫)\n- 骂我\n- 大力骂我\n- @bot 他妈|公交车|你妈|操|屎|去死|快死|日|逼|尼玛|艾滋|癌症|有病|戴套|啊对对对|烦你|你爹|屮|tui|cnm",
+	})
+	limit = rate.NewManager(time.Minute, 30)
+)
+
+func init() {
+	engine.OnFullMatch("骂我").SetBlock(true).SetPriority(prio).Handle(func(ctx *zero.Ctx) {
+		if !limit.Load(ctx.Event.GroupID).Acquire() {
+			return
+		}
+		process.SleepAbout1sTo2s()
+		text := getRandomCurseByLevel(minLevel).Text
+		ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(text))
+	})
+
+	engine.OnFullMatch("大力骂我").SetBlock(true).SetPriority(prio).Handle(func(ctx *zero.Ctx) {
+		if !limit.Load(ctx.Event.GroupID).Acquire() {
+			return
+		}
+		process.SleepAbout1sTo2s()
+		text := getRandomCurseByLevel(maxLevel).Text
+		ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(text))
+	})
+	engine.OnKeywordGroup([]string{"他妈", "公交车", "你妈", "操", "屎", "去死", "快死", "日", "逼", "尼玛", "艾滋", "癌症", "有病", "戴套", "啊对对对", "烦你", "你爹", "屮", "tui", "cnm"}, zero.OnlyToMe).SetBlock(true).SetPriority(prio).
+		Handle(func(ctx *zero.Ctx) {
+			text := getRandomCurseByLevel(maxLevel).Text
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(text))
+		})
+}

plugin_curse/data.go

@@ -0,0 +1,36 @@
+package curse
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/utils/file"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/process"
+	"github.com/FloatTech/ZeroBot-Plugin/utils/sql"
+	"github.com/sirupsen/logrus"
+	"os"
+)
+
+const (
+	dbpath = "data/Curse/"
+	dbfile = dbpath + "curse.db"
+)
+
+var (
+	db = &sql.Sqlite{DBPath: dbfile}
+)
+
+// 加载数据库
+func init() {
+	go func() {
+		process.SleepAbout1sTo2s()
+		_ = os.MkdirAll(dbpath, 0755)
+		_, err := file.GetLazyData(dbfile, false, true)
+		if err != nil {
+			panic(err)
+		}
+		err = db.Create("curse", &curse{})
+		if err != nil {
+			panic(err)
+		}
+		c, _ := db.Count("curse")
+		logrus.Infoln("[curse]加载", c, "条骂人语录")
+	}()
+}

plugin_curse/model.go

@@ -0,0 +1,12 @@
+package curse
+
+type curse struct {
+	ID    uint32 `db:"id"`
+	Text  string `db:"text"`
+	Level string `db:"level"`
+}
+
+func getRandomCurseByLevel(level string) (c curse) {
+	_ = db.Find("curse", &c, "where level = '"+level+"' ORDER BY RANDOM() limit 1")
+	return
+}