mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
// Package choose 选择困难症帮手
|
|
package choose
|
|
|
|
import (
|
|
"math/rand"
|
|
"strconv"
|
|
"strings"
|
|
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
func init() {
|
|
engine := control.Register("choose", &ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "选择困难症帮手",
|
|
Help: "例: 选择可口可乐还是百事可乐\n" +
|
|
"选择肯德基还是麦当劳还是必胜客",
|
|
})
|
|
engine.OnPrefix("选择").SetBlock(true).Handle(handle)
|
|
}
|
|
func handle(ctx *zero.Ctx) {
|
|
rawOptions := strings.Split(ctx.State["args"].(string), "还是")
|
|
var options = make([]string, 0)
|
|
for count, option := range rawOptions {
|
|
options = append(options, strconv.Itoa(count+1)+", "+option)
|
|
}
|
|
result := rawOptions[rand.Intn(len(rawOptions))]
|
|
name := ctx.Event.Sender.NickName
|
|
ctx.SendChain(message.Text("> ", name, "\n",
|
|
"你的选项有:", "\n",
|
|
strings.Join(options, "\n"), "\n",
|
|
"你最终会选: ", result,
|
|
))
|
|
}
|