优化代码结构 & fix job regex

Author: fumiama

go.mod

@@ -7,8 +7,9 @@ require (
 	github.com/FloatTech/AnimeAPI v1.5.1-0.20220901132657-2585bbc03bf6
 	github.com/FloatTech/floatbox v0.0.0-20220822040527-f059031fec44
 	github.com/FloatTech/sqlite v0.3.3
+	github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b
 	github.com/FloatTech/zbpctrl v1.4.1-0.20220715042842-93f081cb0133
-	github.com/FloatTech/zbputils v1.5.1-0.20220826103123-0c73a585e38f
+	github.com/FloatTech/zbputils v1.5.1-0.20220906100116-30c0892066d6
 	github.com/RomiChan/syncx v0.0.0-20220404072119-d7ea0ae15a4c
 	github.com/antchfx/htmlquery v1.2.5
 	github.com/corona10/goimagehash v1.0.3
@@ -36,7 +37,6 @@ require (
 )
 
 require (
-	github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b // indirect
 	github.com/RomiChan/websocket v1.4.3-0.20220123145318-307a86b127bc // indirect
 	github.com/antchfx/xpath v1.2.1 // indirect
 	github.com/cheekybits/genny v1.0.0 // indirect

go.sum

@@ -21,8 +21,8 @@ github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b h1:tvciXWq2nuvTbFeJG
 github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b/go.mod h1:fHZFWGquNXuHttu9dUYoKuNbm3dzLETnIOnm1muSfDs=
 github.com/FloatTech/zbpctrl v1.4.1-0.20220715042842-93f081cb0133 h1:nP9NI4I+vtwAbiU7wCJwjuzCzMZ/yJYg8h3667HGnv0=
 github.com/FloatTech/zbpctrl v1.4.1-0.20220715042842-93f081cb0133/go.mod h1:72BnjyBwQWUC8mqM9dPk5ZrjxXCilQCVp+jfgHATNdw=
-github.com/FloatTech/zbputils v1.5.1-0.20220826103123-0c73a585e38f h1:R13GKhRZfB42nsCEqek/ZlGbKBopQOZFYwBU9qSCfhg=
-github.com/FloatTech/zbputils v1.5.1-0.20220826103123-0c73a585e38f/go.mod h1:ZT91eCYR6y1HbLRTO5EB7o8K7qEPbCsX7vjhYIuMijc=
+github.com/FloatTech/zbputils v1.5.1-0.20220906100116-30c0892066d6 h1:9+PCM0UzbtrpYSzkvy+AeVy6bpprmVd6BUs17Njao04=
+github.com/FloatTech/zbputils v1.5.1-0.20220906100116-30c0892066d6/go.mod h1:ZT91eCYR6y1HbLRTO5EB7o8K7qEPbCsX7vjhYIuMijc=
 github.com/PuerkitoBio/goquery v1.5.1/go.mod h1:GsLWisAFVj4WgDibEWF4pvYnkVQBpKBKeU+7zCJoLcc=
 github.com/RomiChan/syncx v0.0.0-20220404072119-d7ea0ae15a4c h1:cNPOdTNiVwxLpROLjXCgbIPvdkE+BwvxDvgmdYmWx6Q=
 github.com/RomiChan/syncx v0.0.0-20220404072119-d7ea0ae15a4c/go.mod h1:KqZzu7slNKROh3TSYEH/IUMG6f4M+1qubZ5e52QypsE=

plugin/antiabuse/anti.go

@@ -2,13 +2,15 @@
 package antiabuse
 
 import (
-	"fmt"
+	"strconv"
 	"strings"
 	"time"
 
+	"github.com/FloatTech/floatbox/binary"
 	fcext "github.com/FloatTech/floatbox/ctxext"
 	ctrl "github.com/FloatTech/zbpctrl"
 	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/img/text"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
 )
@@ -19,6 +21,7 @@ func init() {
 		Help:              "违禁词检测",
 		PrivateDataFolder: "anti_abuse",
 	})
+
 	onceRule := fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
 		managers = ctx.State["managers"].(*ctrl.Control[*zero.Ctx]).Manager
 		db.DBPath = engine.DataFolder() + "anti_abuse.db"
@@ -32,39 +35,62 @@ func init() {
 			ctx.SendChain(message.Text("create table error: ", err))
 			return false
 		}
-		err = recoverWord()
-		if err != nil {
-			ctx.SendChain(message.Text("recover data error: ", err))
+		return true
+	})
+
+	engine.OnMessage(onceRule, zero.OnlyGroup, func(ctx *zero.Ctx) bool {
+		if !ctx.Event.IsToMe {
+			return true
+		}
+		uid := ctx.Event.UserID
+		gid := ctx.Event.GroupID
+		grp := strconv.FormatInt(gid, 36)
+		msg := strings.ReplaceAll(ctx.MessageString(), "\n", "")
+		msg = strings.ReplaceAll(msg, "\r", "")
+		msg = strings.ReplaceAll(msg, "\t", "")
+		msg = strings.ReplaceAll(msg, ";", "")
+		mu.RLock()
+		defer mu.RUnlock()
+		if db.CanFind(grp, "WHERE instr('"+msg+"', word)>=0") {
+			if err := managers.DoBlock(uid); err == nil {
+				cache.Set(uid, struct{}{})
+				ctx.SetGroupBan(gid, uid, 4*3600)
+				ctx.SendChain(message.Text("检测到违禁词,已封禁/屏蔽4小时"))
+			} else {
+				ctx.SendChain(message.Text("block user error:", err))
+			}
 			return false
 		}
 		return true
 	})
-	engine.OnMessage(onceRule, zero.OnlyGroup, banRule)
+
 	engine.OnCommand("添加违禁词", zero.OnlyGroup, zero.AdminPermission, onceRule).Handle(
 		func(ctx *zero.Ctx) {
 			args := ctx.State["args"].(string)
 			if err := insertWord(ctx.Event.GroupID, args); err != nil {
-				ctx.SendChain(message.Text("error:", err))
+				ctx.SendChain(message.Text("ERROR: ", err))
 			} else {
-				ctx.SendChain(message.Text(fmt.Sprintf("添加违禁词 %s 成功", args)))
+				ctx.SendChain(message.Text("添加违禁词 ", args, " 成功"))
 			}
 		})
+
 	engine.OnCommand("删除违禁词", zero.OnlyGroup, zero.AdminPermission, onceRule).Handle(
 		func(ctx *zero.Ctx) {
 			args := ctx.State["args"].(string)
 			if err := deleteWord(ctx.Event.GroupID, args); err != nil {
-				ctx.SendChain(message.Text("error:", err))
+				ctx.SendChain(message.Text("ERROR: ", err))
 			} else {
-				ctx.SendChain(message.Text(fmt.Sprintf("删除违禁词 %s 成功", args)))
+				ctx.SendChain(message.Text("删除违禁词 ", args, " 成功"))
 			}
 		})
+
 	engine.OnCommand("查看违禁词", zero.OnlyGroup, onceRule).Handle(
 		func(ctx *zero.Ctx) {
-			if set, ok := wordMap[ctx.Event.GroupID]; !ok {
-				ctx.SendChain(message.Text("本群无违禁词"))
-			} else {
-				ctx.SendChain(message.Text("本群违禁词有:", strings.Join(set.ToSlice(), " |")))
+			b, err := text.RenderToBase64(listWords(ctx.Event.GroupID), text.FontFile, 400, 20)
+			if err != nil {
+				ctx.SendChain(message.Text("ERROR: ", err))
+				return
 			}
+			ctx.SendChain(message.Text("本群违禁词有:\n"), message.Image("base64://"+binary.BytesToString(b)))
 		})
-
 }

plugin/antiabuse/db.go

@@ -0,0 +1,76 @@
+package antiabuse
+
+import (
+	"errors"
+	"strconv"
+	"strings"
+	"sync"
+	"time"
+	"unsafe"
+
+	sqlite "github.com/FloatTech/sqlite"
+	"github.com/FloatTech/ttl"
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+)
+
+var managers *ctrl.Manager[*zero.Ctx] // managers lazy load
+var db = &sqlite.Sqlite{}
+var mu sync.RWMutex
+
+func onDel(uid int64, _ struct{}) {
+	if managers == nil {
+		return
+	}
+	if err := managers.DoUnblock(uid); err != nil {
+		logrus.Error("do unblock error:", err)
+	}
+}
+
+var cache = ttl.NewCacheOn(4*time.Hour, [4]func(int64, struct{}){nil, nil, onDel, nil})
+
+type banWord struct {
+	Word string `db:"word"`
+}
+
+var nilban = &banWord{}
+
+func insertWord(gid int64, word string) error {
+	grp := strconv.FormatInt(gid, 36)
+	mu.Lock()
+	defer mu.Unlock()
+	err := db.Create(grp, nilban)
+	if err != nil {
+		return err
+	}
+	return db.Insert(grp, (*banWord)(unsafe.Pointer(&word)))
+}
+
+func deleteWord(gid int64, word string) error {
+	grp := strconv.FormatInt(gid, 36)
+	mu.Lock()
+	defer mu.Unlock()
+	if n, _ := db.Count(grp); n == 0 {
+		return errors.New("本群还没有违禁词~")
+	}
+	return db.Del(grp, "WHRER word='"+word+"'")
+}
+
+func listWords(gid int64) string {
+	grp := strconv.FormatInt(gid, 36)
+	word := ""
+	ptr := (*banWord)(unsafe.Pointer(&word))
+	sb := strings.Builder{}
+	mu.Lock()
+	defer mu.Unlock()
+	_ = db.FindFor(grp, ptr, "", func() error {
+		sb.WriteString(word)
+		sb.WriteString(" |")
+		return nil
+	})
+	if sb.Len() <= 2 {
+		return ""
+	}
+	return sb.String()[:sb.Len()-2]
+}