diff --git a/README.md b/README.md index 61517fd3..b660522e 100644 --- a/README.md +++ b/README.md @@ -185,6 +185,9 @@ zerobot [-h] [-t token] [-u url] [-n nickname] [-p prefix] [-d|w] [-g 监听地 - [x] @BOT 创建频道 xxx - [x] 跳入(频道)海中 - [x] 注：不显式限制时，私聊发送可在所有群抽到，群聊发送仅可在本群抽到，默认频道为 global +- **猜单词** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_wordle"` + - [x] 个人猜单词 + - [x] 团队猜单词 - **bilibili** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin_bilibili"` - [x] >vup info [名字 | uid] - [x] >user info [名字 | uid] diff --git a/main.go b/main.go index ce36bd51..2b38fba0 100644 --- a/main.go +++ b/main.go @@ -98,6 +98,7 @@ import ( _ "github.com/FloatTech/ZeroBot-Plugin/plugin_translation" // 翻译 _ "github.com/FloatTech/ZeroBot-Plugin/plugin_vtb_quotation" // vtb语录 _ "github.com/FloatTech/ZeroBot-Plugin/plugin_wangyiyun" // 网易云音乐热评 + _ "github.com/FloatTech/ZeroBot-Plugin/plugin_wordle" // 猜单词 // _ "github.com/FloatTech/ZeroBot-Plugin/plugin_wtf" // 鬼东西 // _ "github.com/FloatTech/ZeroBot-Plugin/plugin_bilibili_push" // b站推送 diff --git a/plugin_wordle/wordle.go b/plugin_wordle/wordle.go index 1a6aa787..eac7892a 100644 --- a/plugin_wordle/wordle.go +++ b/plugin_wordle/wordle.go @@ -5,7 +5,9 @@ import ( "errors" "image/color" "math/rand" + "sort" "strings" + "sync" "time" "github.com/FloatTech/zbputils/binary" @@ -39,13 +41,14 @@ var colors = [...]color.RGBA{ {219, 219, 219, 255}, } -var words []wordpack +var words []string func init() { en := control.Register("wordle", order.AcquirePrio(), &control.Options{ DisableOnDefault: false, Help: "猜单词\n" + - "- 开始猜单词", + "- 个人猜单词" + + "- 团队猜单词", PublicDataFolder: "Wordle", }) go func() { @@ -53,79 +56,99 @@ func init() { if err != nil { panic(err) } - words = loadwords(data) + var wordpacks = loadwords(data) + words = make([]string, 0, len(wordpacks)) + for i := range wordpacks { + words = append(words, wordpacks[i].String()) + } + sort.Strings(words) }() - en.OnFullMatch("猜单词").SetBlock(true).Limit(ctxext.LimitByUser). + var room = sync.Map{} + en.OnRegex(`(个人|团队)猜单词`, zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser). Handle(func(ctx *zero.Ctx) { + if ongoing, ok := room.Load(ctx.Event.GroupID); ok && ongoing.(bool) { + ctx.SendChain(message.ReplyWithMessage(ctx.Event.MessageID, + message.Text("已经有正在进行的游戏..."))...) + return + } + room.Store(ctx.Event.GroupID, true) game := newWordleGame() _, img, _ := game("") - ctx.SendChain(message.Image("base64://"+binary.BytesToString(img)), message.Text("请发送单词")) - // 没有图片就索取 - next := zero.NewFutureEvent("message", 999, false, zero.RegexRule(`^[A-Z]{5}$|^[a-z]{5}$`)) + typ := ctx.State["regex_matched"].([]string)[1] + ctx.SendChain(message.ReplyWithMessage(ctx.Event.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("你有6次机会猜出单词，单词长度为5，请发送单词"))...) + var next *zero.FutureEvent + if typ == "个人" { + next = zero.NewFutureEvent("message", 999, false, zero.RegexRule(`^[A-Z]{5}$|^[a-z]{5}$`), zero.OnlyGroup, zero.CheckUser(ctx.Event.UserID)) + } else { + next = zero.NewFutureEvent("message", 999, false, zero.RegexRule(`^[A-Z]{5}$|^[a-z]{5}$`), zero.OnlyGroup) + } recv, cancel := next.Repeat() - var msg message.Message defer func() { cancel() - ctx.Send(msg) + room.Store(ctx.Event.GroupID, false) }() for { select { case <-time.After(time.Second * 120): + ctx.SendChain(message.ReplyWithMessage(ctx.Event.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("猜单词超时，游戏结束..."))...) return case e := <-recv: win, img, err := game(e.Message.String()) - msg = []message.MessageSegment{message.Image("base64://" + binary.BytesToString(img))} - switch err { - case nil: - if win { - msg = append(msg, message.Text("你赢了")) - return - } - case errLengthNotEnough: - msg = append(msg, message.Text("单词长度错误")) - case errUnknownWord: - msg = append(msg, message.Text("不存在这样的单词")) - case errTimesRunOut: - msg = append(msg, message.Text("你输了")) + switch { + case win: + ctx.SendChain(message.ReplyWithMessage(e.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("太棒了，你猜出来了！"))...) return + case err == errTimesRunOut: + ctx.SendChain(message.ReplyWithMessage(e.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("游戏结束..."))...) + return + case err == errLengthNotEnough: + ctx.SendChain(message.ReplyWithMessage(e.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("单词长度错误"))...) + case err == errUnknownWord: + ctx.SendChain(message.ReplyWithMessage(e.MessageID, + message.Image("base64://"+binary.BytesToString(img)), + message.Text("你确定存在这样的单词吗？"))...) + default: + ctx.SendChain(message.ReplyWithMessage(e.MessageID, + message.Image("base64://"+binary.BytesToString(img)))...) } - ctx.Send(msg) } } }) } func newWordleGame() func(string) (bool, []byte, error) { - onhandpack := words[rand.Intn(len(words))] - onhand := onhandpack.String() + onhand := words[rand.Intn(len(words))] record := make([]string, 0, len(onhand)+1) return func(s string) (win bool, base64Image []byte, err error) { if s != "" { s = strings.ToLower(s) - sp := pack(s) - if onhandpack == sp { + if onhand == s { win = true } else { if len(s) != len(onhand) { err = errLengthNotEnough return } - i := 0 - for ; i < len(words); i++ { - if words[i] == sp { - break - } - } - if i >= len(words) || words[i] != sp { + i := sort.SearchStrings(words, s) + if i >= len(words) || words[i] != s { err = errUnknownWord return } } + record = append(record, s) if len(record) >= cap(record) { err = errTimesRunOut - return } - record = append(record, s) } var side = 20 ctx := gg.NewContext((side+2)*5+26, (side+2)*6+26) @@ -154,7 +177,7 @@ func newWordleGame() func(string) (bool, []byte, error) { } } } - base64Image, err = writer.ToBase64(ctx.Image()) + base64Image, _ = writer.ToBase64(ctx.Image()) return } }