🚚 move zbpctrl in

control/README.md

@@ -0,0 +1,2 @@
+# zbpctrl
+ZeroBot-Plugin 的控制库

control/cd.go

@@ -0,0 +1,94 @@
+package control
+
+import (
+	"encoding/binary"
+	"strings"
+	"time"
+
+	b14 "github.com/fumiama/go-base16384"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
+
+	"github.com/FloatTech/zbputils/math"
+	"github.com/FloatTech/zbputils/process"
+)
+
+var startTime int64
+
+func init() {
+	// 插件冲突检测 会在本群发送一条消息并在约 1s 后撤回
+	zero.OnFullMatch("插件冲突检测", zero.OnlyGroup, zero.AdminPermission, zero.OnlyToMe).SetBlock(true).FirstPriority().
+		Handle(func(ctx *zero.Ctx) {
+			tok, err := genToken()
+			if err != nil {
+				return
+			}
+			t := message.Text("●cd" + tok)
+			startTime = time.Now().Unix()
+			id := ctx.SendChain(t)
+			process.SleepAbout1sTo2s()
+			ctx.DeleteMessage(id)
+		})
+
+	zero.OnRegex("^●cd([\u4e00-\u8e00]{4})$", zero.OnlyGroup).SetBlock(true).FirstPriority().
+		Handle(func(ctx *zero.Ctx) {
+			if isValidToken(ctx.State["regex_matched"].([]string)[1]) {
+				msg := ""
+				gid := ctx.Event.GroupID
+				ForEach(func(key string, manager *Control) bool {
+					if manager.IsEnabledIn(gid) {
+						msg += "\xfe\xff" + key
+					}
+					return true
+				})
+				if len(msg) > 2 {
+					my, err := b14.UTF16be2utf8(b14.EncodeString(msg[2:]))
+					mys := "●cd●" + helper.BytesToString(my)
+					if err == nil {
+						id := ctx.SendChain(message.Text(mys))
+						process.SleepAbout1sTo2s()
+						ctx.DeleteMessage(id)
+					}
+				}
+			}
+		})
+
+	zero.OnRegex("^●cd●(([\u4e00-\u8e00]*[\u3d01-\u3d06]?))", zero.OnlyGroup).SetBlock(true).FirstPriority().
+		Handle(func(ctx *zero.Ctx) {
+			if time.Now().Unix()-startTime < 10 {
+				msg, err := b14.UTF82utf16be(helper.StringToBytes(ctx.State["regex_matched"].([]string)[1]))
+				if err == nil {
+					gid := ctx.Event.GroupID
+					for _, s := range strings.Split(b14.DecodeString(msg), "\xfe\xff") {
+						mu.RLock()
+						c, ok := managers[s]
+						mu.RUnlock()
+						if ok && c.IsEnabledIn(gid) {
+							c.Disable(gid)
+						}
+					}
+				}
+			}
+		})
+}
+
+func genToken() (tok string, err error) {
+	timebytes := make([]byte, 8)
+	binary.BigEndian.PutUint64(timebytes, uint64(time.Now().Unix()))
+	timebytes, err = b14.UTF16be2utf8(b14.Encode(timebytes[1:]))
+	if err == nil {
+		tok = helper.BytesToString(timebytes)
+	}
+	return
+}
+
+func isValidToken(tok string) (yes bool) {
+	s, err := b14.UTF82utf16be(helper.StringToBytes(tok))
+	if err == nil {
+		timebytes := make([]byte, 1, 8)
+		timebytes = append(timebytes, b14.Decode(s)...)
+		yes = math.Abs64(time.Now().Unix()-int64(binary.BigEndian.Uint64(timebytes))) < 10
+	}
+	return
+}

control/cd_test.go

@@ -0,0 +1,19 @@
+package control
+
+import "testing"
+
+func TestGenToken(t *testing.T) {
+	tok, err := genToken()
+	if err == nil {
+		t.Log(tok)
+		t.Log(isValidToken(tok))
+		t.Fail()
+	} else {
+		t.Fatal(err)
+	}
+}
+
+func TestMaru(t *testing.T) {
+	t.Log(len("\xff"))
+	t.Fail()
+}

control/register.go

@@ -0,0 +1,31 @@
+package control
+
+import (
+	zero "github.com/wdvxdr1123/ZeroBot"
+)
+
+var enmap = make(map[string]*zero.Engine)
+
+// Register 注册插件控制器
+func Register(service string, o *Options) *zero.Engine {
+	engine := zero.New()
+	engine.UsePreHandler(newctrl(service, o).Handler)
+	enmap[service] = engine
+	return engine
+}
+
+// Delete 删除插件控制器，不会删除数据
+func Delete(service string) {
+	engine, ok := enmap[service]
+	if ok {
+		engine.Delete()
+		mu.RLock()
+		_, ok = managers[service]
+		mu.RUnlock()
+		if ok {
+			mu.Lock()
+			delete(managers, service)
+			mu.Unlock()
+		}
+	}
+}