Function init

plugin/vote/vote.go:22–129

()

)

func init() {
	engine := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
		DisableOnDefault:  false,
		Brief:             "投票",
		Help:              "- 投票选择可口可乐还是百事可乐\n",
		PrivateDataFolder: "vote",
	})

	engine.OnPrefix("投票选择", zero.OnlyGroup, getPara).SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			rawOptions := strings.Split(ctx.State["args"].(string), "还是")
			options := make([]string, 0, 16)
			for count, option := range rawOptions {
				options = append(options, strconv.Itoa(count)+". "+option)
			}
			if len(options) == 0 || len(options) == 1 {
				ctx.SendChain(message.Text("投票的选项太少, 退出投票"))
				return
			}
			ctx.SendChain(message.Text("投票开始, 设定的投票时间到或者60秒内无人投票就停止投票\n请按序号投票:\n", strings.Join(options, "\n")))
			voteMap := make(map[int]int, 0)
			repeatMap := make(map[int64]int, 0)
			mode := ctx.State["vote_paras"].([2]int)[0]
			voteDuration := ctx.State["vote_paras"].([2]int)[1]
			next := zero.NewFutureEvent("message", 999, false, zero.RegexRule(`^\d{1,2}$`),
				zero.OnlyGroup, zero.CheckGroup(ctx.Event.GroupID))
			recv, cancel := next.Repeat()
			defer cancel()
			after := time.NewTimer(time.Duration(voteDuration) * time.Second)
		EXIT:
			for {
				select {
				case <-time.After(60 * time.Second):
					ctx.SendChain(message.Text("投票结束"))
					break EXIT
				case <-after.C:
					ctx.SendChain(message.Text("投票结束"))
					break EXIT
				case c := <-recv:
					uid := c.Event.UserID
					choose, err := strconv.Atoi(c.Event.Message.String())
					if err != nil {
						ctx.SendChain(message.Text("ERROR:", err))
					}
					if _, ok := repeatMap[uid]; !ok && choose >= 0 && choose < len(options) {
						voteMap[choose]++
					}
					if mode == 2 {
						repeatMap[uid] = 1
					}
				}
			}
			vc := rankByVote(voteMap)
			if len(vc) > 20 {
				vc = vc[:20]
			}
			// 绘图
			b, err := os.ReadFile(text.FontFile)