帮助选择困难症

plugin_choose/choose.go

@@ -0,0 +1,35 @@
+package choose
+
+import (
+	"math/rand"
+	"strconv"
+	"strings"
+
+	"github.com/FloatTech/ZeroBot-Plugin/control"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+)
+
+func init() {
+	engine := control.Register("choose", &control.Options{
+		DisableOnDefault: false,
+		Help: "choose\n" +
+			"- 选择可口可乐还是百事可乐\n" +
+			"- 选择肯德基还是麦当劳还是必胜客\n",
+	})
+	engine.OnPrefix("选择").SetBlock(true).FirstPriority().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,
+	))
+}