ZeroBot-Plugin/plugin/plugin_wordle/wordle.go

// Package wordle 猜单词
package wordle

import (
	"errors"
	"image/color"
	"math/rand"
	"sort"
	"strings"
	"time"

	"github.com/FloatTech/zbputils/control"
	"github.com/FloatTech/zbputils/control/order"
	"github.com/FloatTech/zbputils/ctxext"
	"github.com/FloatTech/zbputils/file"
	"github.com/FloatTech/zbputils/img/writer"
	"github.com/fogleman/gg"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/extension/single"
	"github.com/wdvxdr1123/ZeroBot/message"
)

var (
	errLengthNotEnough = errors.New("length not enough")
	errUnknownWord     = errors.New("unknown word")
	errTimesRunOut     = errors.New("times run out")
)

const (
	match = iota
	exist
	notexist
	undone
)

var colors = [...]color.RGBA{
	{125, 166, 108, 255},
	{199, 183, 96, 255},
	{123, 123, 123, 255},
	{219, 219, 219, 255},
}

var words []string
var questions []string

func init() {
	en := control.Register("wordle", order.AcquirePrio(), &control.Options{
		DisableOnDefault: false,
		Help: "猜单词\n" +
			"- 个人猜单词" +
			"- 团队猜单词",
		PublicDataFolder: "Wordle",
	}).ApplySingle(single.New(
		single.WithKeyFn(func(ctx *zero.Ctx) interface{} { return ctx.Event.GroupID }),
		single.WithPostFn(func(ctx *zero.Ctx) {
			ctx.Send(
				message.ReplyWithMessage(ctx.Event.MessageID,
					message.Text("已经有正在进行的游戏..."),
				),
			)
		}),
	))
	go func() {
		questionsdata, err := file.GetLazyData(en.DataFolder()+"questions.bin", true, true)
		if err != nil {
			panic(err)
		}
		questionspacks := loadwords(questionsdata)
		questions = make([]string, len(questionspacks))
		for i := range questionspacks {
			questions[i] = questionspacks[i].String()
		}
		wordsdata, err := file.GetLazyData(en.DataFolder()+"words.bin", true, true)
		if err != nil {
			panic(err)
		}
		wordpacks := loadwords(wordsdata)
		words = make([]string, len(wordpacks))
		for i := range wordpacks {
			words[i] = wordpacks[i].String()
		}
		sort.Strings(words)
	}()
	en.OnRegex(`(个人|团队)猜单词`, zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByUser).
		Handle(func(ctx *zero.Ctx) {
			target := questions[rand.Intn(len(questions))]
			game := newWordleGame(target)
			_, img, cl, _ := game("")
			typ := ctx.State["regex_matched"].([]string)[1]
			ctx.Send(
				message.ReplyWithMessage(ctx.Event.MessageID,
					message.ImageBytes(img),
					message.Text("你有6次机会猜出单词，单词长度为5，请发送单词"),
				),
			)
			cl()
			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, zero.CheckGroup(ctx.Event.GroupID))
			}
			var win bool
			var err error
			for {
				select {
				case <-time.After(time.Second * 120):
					ctx.Send(
						message.ReplyWithMessage(ctx.Event.MessageID,
							message.Text("猜单词超时，游戏结束...答案是: ", target),
						),
					)
					return
				case e := <-next.Next():
					win, img, cl, err = game(e.Message.String())
					switch {
					case win:
						ctx.Send(
							message.ReplyWithMessage(e.MessageID,
								message.ImageBytes(img),
								message.Text("太棒了，你猜出来了！"),
							),
						)
						cl()
						return
					case err == errTimesRunOut:
						ctx.Send(
							message.ReplyWithMessage(e.MessageID,
								message.ImageBytes(img),
								message.Text("游戏结束...答案是: ", target),
							),
						)
						cl()
						return
					case err == errLengthNotEnough:
						ctx.Send(
							message.ReplyWithMessage(e.MessageID,
								message.Text("单词长度错误"),
							),
						)
					case err == errUnknownWord:
						ctx.Send(
							message.ReplyWithMessage(e.MessageID,
								message.Text("你确定存在这样的单词吗？"),
							),
						)
					default:
						ctx.Send(
							message.ReplyWithMessage(e.MessageID,
								message.ImageBytes(img),
							),
						)
						cl()
					}
				}
			}
		})
}

func newWordleGame(target string) func(string) (bool, []byte, func(), error) {
	record := make([]string, 0, len(target)+1)
	return func(s string) (win bool, data []byte, cl func(), err error) {
		if s != "" {
			s = strings.ToLower(s)
			if target == s {
				win = true
			} else {
				if len(s) != len(target) {
					err = errLengthNotEnough
					return
				}
				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
			}
		}
		var side = 20
		ctx := gg.NewContext((side+2)*5+26, (side+2)*6+26)
		ctx.SetColor(color.RGBA{255, 255, 255, 255})
		ctx.Clear()
		for i := 0; i < len(target)+1; i++ {
			for j := 0; j < len(target); j++ {
				if len(record) > i {
					ctx.DrawRectangle(float64(10+j*(side+4)), float64(10+i*(side+4)), float64(side), float64(side))
					switch {
					case record[i][j] == target[j]:
						ctx.SetColor(colors[match])
					case strings.IndexByte(target, record[i][j]) != -1:
						ctx.SetColor(colors[exist])
					default:
						ctx.SetColor(colors[notexist])
					}
					ctx.Fill()
					ctx.SetColor(color.RGBA{255, 255, 255, 255})
					ctx.DrawString(strings.ToUpper(string(record[i][j])), float64(10+j*(side+4)+7), float64(10+i*(side+4)+15))
				} else {
					ctx.DrawRectangle(float64(10+j*(side+4)+1), float64(10+i*(side+4)+1), float64(side-2), float64(side-2))
					ctx.SetLineWidth(1)
					ctx.SetColor(colors[undone])
					ctx.Stroke()
				}
			}
		}
		data, cl = writer.ToBytes(ctx.Image())
		return
	}
}