国际象棋

go.mod

@@ -12,7 +12,7 @@ require (
 	github.com/FloatTech/sqlite v1.6.2
 	github.com/FloatTech/ttl v0.0.0-20220715042055-15612be72f5b
 	github.com/FloatTech/zbpctrl v1.5.3-0.20230514154630-b74e6fcca380
-	github.com/FloatTech/zbputils v1.6.2-0.20230728081122-94d4d957f3bf
+	github.com/FloatTech/zbputils v1.6.2-0.20230831134542-28c5ba506758
 	github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e
 	github.com/RomiChan/websocket v1.4.3-0.20220227141055-9b2c6168c9c5
 	github.com/antchfx/htmlquery v1.2.5
@@ -31,6 +31,7 @@ require (
 	github.com/jozsefsallai/gophersauce v1.0.1
 	github.com/lithammer/fuzzysearch v1.1.5
 	github.com/mroth/weightedrand v1.0.0
+	github.com/notnil/chess v1.9.0
 	github.com/pkg/errors v0.9.1
 	github.com/quic-go/quic-go v0.38.1
 	github.com/shirou/gopsutil/v3 v3.23.1
@@ -46,6 +47,7 @@ require (
 )
 
 require (
+	github.com/ajstarks/svgo v0.0.0-20200320125537-f189e35d30ca // indirect
 	github.com/antchfx/xpath v1.2.1 // indirect
 	github.com/ericpauley/go-quantize v0.0.0-20200331213906-ae555eb2afa4 // indirect
 	github.com/faiface/beep v1.1.0 // indirect

go.sum

@@ -20,11 +20,15 @@ github.com/FloatTech/zbpctrl v1.5.3-0.20230514154630-b74e6fcca380 h1:qmwoT8xVaND
 github.com/FloatTech/zbpctrl v1.5.3-0.20230514154630-b74e6fcca380/go.mod h1:gkGC1C1eEUd/Ld/ja68zas5j2ZktIZCdnj2FMaM+Au0=
 github.com/FloatTech/zbputils v1.6.2-0.20230728081122-94d4d957f3bf h1:PwH9aMnmN+m204cVIqUrI3e7nsdQi/IGW012Fjzb1bs=
 github.com/FloatTech/zbputils v1.6.2-0.20230728081122-94d4d957f3bf/go.mod h1:JRnGR7EGeEQgxOs+c0rZAhrS9Es2BTcGHdIDHXIPRzQ=
+github.com/FloatTech/zbputils v1.6.2-0.20230831134542-28c5ba506758 h1:z0hhIwGN8ifKExa6xkujZwAQwJNU6AnELt+/A6nAdcY=
+github.com/FloatTech/zbputils v1.6.2-0.20230831134542-28c5ba506758/go.mod h1:JRnGR7EGeEQgxOs+c0rZAhrS9Es2BTcGHdIDHXIPRzQ=
 github.com/PuerkitoBio/goquery v1.5.1/go.mod h1:GsLWisAFVj4WgDibEWF4pvYnkVQBpKBKeU+7zCJoLcc=
 github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e h1:wR3MXQ3VbUlPKOOUwLOYgh/QaJThBTYtsl673O3lqSA=
 github.com/RomiChan/syncx v0.0.0-20221202055724-5f842c53020e/go.mod h1:vD7Ra3Q9onRtojoY5sMCLQ7JBgjUsrXDnDKyFxqpf9w=
 github.com/RomiChan/websocket v1.4.3-0.20220227141055-9b2c6168c9c5 h1:bBmmB7he0iVN4m5mcehfheeRUEer/Avo4ujnxI3uCqs=
 github.com/RomiChan/websocket v1.4.3-0.20220227141055-9b2c6168c9c5/go.mod h1:0UcFaCkhp6vZw6l5Dpq0Dp673CoF9GdvA8lTfst0GiU=
+github.com/ajstarks/svgo v0.0.0-20200320125537-f189e35d30ca h1:kWzLcty5V2rzOqJM7Tp/MfSX0RMSI1x4IOLApEefYxA=
+github.com/ajstarks/svgo v0.0.0-20200320125537-f189e35d30ca/go.mod h1:K08gAheRH3/J6wwsYMMT4xOr94bZjxIelGM0+d/wbFw=
 github.com/andybalholm/cascadia v1.1.0/go.mod h1:GsXiBklL0woXo1j/WYWtSYYC4ouU9PqHO0sqidkEA4Y=
 github.com/antchfx/htmlquery v1.2.5 h1:1lXnx46/1wtv1E/kzmH8vrfMuUKYgkdDBA9pIdMJnk4=
 github.com/antchfx/htmlquery v1.2.5/go.mod h1:2MCVBzYVafPBmKbrmwB9F5xdd+IEgRY61ci2oOsOQVw=
@@ -152,6 +156,8 @@ github.com/mroth/weightedrand v1.0.0 h1:V8JeHChvl2MP1sAoXq4brElOcza+jxLkRuwvtQu8
 github.com/mroth/weightedrand v1.0.0/go.mod h1:3p2SIcC8al1YMzGhAIoXD+r9olo/g/cdJgAD905gyNE=
 github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ=
 github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
+github.com/notnil/chess v1.9.0 h1:YMxR5kUVjtwcuFptGU0/3q7eG3MSHQNbg0VUekvRKV0=
+github.com/notnil/chess v1.9.0/go.mod h1:cRuJUIBFq9Xki05TWHJxHYkC+fFpq45IWwk94DdlCrA=
 github.com/onsi/ginkgo/v2 v2.9.5 h1:+6Hr4uxzP4XIUyAkg61dWBw8lb/gc4/X5luuxN/EC+Q=
 github.com/onsi/ginkgo/v2 v2.9.5/go.mod h1:tvAoo1QUJwNEU2ITftXTpR7R1RbCzoZUOs3RonqW57k=
 github.com/onsi/gomega v1.27.6 h1:ENqfyGeS5AX/rlXDd/ETokDz93u0YufY1Pgxuy/PvWE=

main.go

@@ -74,6 +74,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/bilibili"         // b站相关
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/book_review"      // 哀伤雪刃吧推书记录
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/cangtoushi"       // 藏头诗
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/chess"            // 国际象棋
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/choose"           // 选择困难症帮手
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/chouxianghua"     // 说抽象话
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/chrev"            // 英文字符翻转

plugin/chess/chess.go

@@ -0,0 +1,165 @@
+// Package chess 国际象棋
+package chess
+
+import (
+	"fmt"
+	"os"
+	"path"
+	"strconv"
+	"strings"
+	"time"
+
+	ctrl "github.com/FloatTech/zbpctrl"
+	"github.com/FloatTech/zbputils/control"
+	"github.com/FloatTech/zbputils/ctxext"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/extension/single"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+const helpString = `- 参与/创建一盘游戏：「下棋」(chess)
+- 参与/创建一盘盲棋：「盲棋」(blind)
+- 投降认输：「认输」 (resign)
+- 请求、接受和棋：「和棋」 (draw)
+- 走棋：!Nxf3 中英文感叹号均可，格式请参考“代数记谱法”(Algebraic notation)
+- 中断对局：「中断」 (abort)（仅群主/管理员有效）
+- 查看等级分排行榜：「排行榜」(ranking)
+- 查看自己的等级分：「等级分」(rate)
+- 清空等级分：「清空等级分 QQ号」(.clean.rate) （仅超管有效）`
+
+var (
+	limit       = ctxext.NewLimiterManager(time.Microsecond*2500, 1)
+	tempFileDir string
+	engine      = control.Register("chess", &ctrl.Options[*zero.Ctx]{
+		DisableOnDefault:  false,
+		Brief:             "国际象棋",
+		Help:              helpString,
+		PrivateDataFolder: "chess",
+	}).ApplySingle(single.New(
+		single.WithKeyFn(func(ctx *zero.Ctx) int64 { return ctx.Event.GroupID }),
+		single.WithPostFn[int64](func(ctx *zero.Ctx) {
+			ctx.Send(
+				message.ReplyWithMessage(ctx.Event.MessageID,
+					message.Text("有操作正在执行, 请稍后再试..."),
+				),
+			)
+		}),
+	))
+)
+
+func init() {
+	// 初始化临时文件夹
+	tempFileDir = path.Join(engine.DataFolder(), "temp")
+	err := os.MkdirAll(tempFileDir, 0750)
+	if err != nil {
+		panic(err)
+	}
+	// 初始化数据库
+	dbFilePath := engine.DataFolder() + "chess.db"
+	initDatabase(dbFilePath)
+	// 注册指令
+	engine.OnFullMatchGroup([]string{"下棋", "chess"}, zero.OnlyGroup).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.Sender == nil {
+				return
+			}
+			userUin := ctx.Event.UserID
+			userName := ctx.Event.Sender.NickName
+			groupCode := ctx.Event.GroupID
+			if replyMessage := game(groupCode, userUin, userName); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"认输", "resign"}, zero.OnlyGroup).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			userUin := ctx.Event.UserID
+			groupCode := ctx.Event.GroupID
+			if replyMessage := resign(groupCode, userUin); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"和棋", "draw"}, zero.OnlyGroup).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			userUin := ctx.Event.UserID
+			groupCode := ctx.Event.GroupID
+			if replyMessage := draw(groupCode, userUin); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"中断", "abort"}, zero.OnlyGroup, zero.AdminPermission).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			groupCode := ctx.Event.GroupID
+			if replyMessage := abort(groupCode); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"盲棋", "blind"}, zero.OnlyGroup).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.Sender == nil {
+				return
+			}
+			userUin := ctx.Event.UserID
+			userName := ctx.Event.Sender.NickName
+			groupCode := ctx.Event.GroupID
+			if replyMessage := blindfold(groupCode, userUin, userName); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnRegex("^[!|！]([0-8]|[R|N|B|Q|K|O|a-h|x]|[-|=|+])+$", zero.OnlyGroup).
+		SetBlock(true).
+		Limit(limit.LimitByGroup).
+		Handle(func(ctx *zero.Ctx) {
+			userUin := ctx.Event.UserID
+			groupCode := ctx.Event.GroupID
+			userMsgStr := ctx.State["regex_matched"].([]string)[0]
+			moveStr := strings.TrimPrefix(strings.TrimPrefix(userMsgStr, "！"), "!")
+			if replyMessage := play(userUin, groupCode, moveStr); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"排行榜", "ranking"}).
+		SetBlock(true).
+		Limit(limit.LimitByUser).
+		Handle(func(ctx *zero.Ctx) {
+			if replyMessage := ranking(); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnFullMatchGroup([]string{"等级分", "rate"}).
+		SetBlock(true).
+		Limit(limit.LimitByUser).
+		Handle(func(ctx *zero.Ctx) {
+			if ctx.Event.Sender == nil {
+				return
+			}
+			userUin := ctx.Event.UserID
+			userName := ctx.Event.Sender.NickName
+			if replyMessage := rate(userUin, userName); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+	engine.OnPrefixGroup([]string{"清空等级分", ".clean.rate"}, zero.SuperUserPermission).
+		SetBlock(true).
+		Limit(limit.LimitByUser).
+		Handle(func(ctx *zero.Ctx) {
+			args := ctx.State["args"].(string)
+			playerUin, err := strconv.ParseInt(strings.TrimSpace(args), 10, 64)
+			if err != nil || playerUin <= 0 {
+				ctx.Send(fmt.Sprintf("解析失败「%s」不是正确的 QQ 号。", args))
+				return
+			}
+			if replyMessage := cleanUserRate(playerUin); len(replyMessage) >= 1 {
+				ctx.Send(replyMessage)
+			}
+		})
+}