mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
package tarot
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
"strconv"
|
|
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
const bed = "https://gitcode.net/shudorcl/zbp-tarot/-/raw/master/"
|
|
|
|
type card struct {
|
|
Name string `json:"name"`
|
|
Info struct {
|
|
Description string `json:"description"`
|
|
ReverseDescription string `json:"reverseDescription"`
|
|
ImgURL string `json:"imgUrl"`
|
|
} `json:"info"`
|
|
}
|
|
type cardset = map[string]card
|
|
|
|
var cardMap = make(cardset, 256)
|
|
var reasons = [...]string{"您抽到的是~\n", "锵锵锵，塔罗牌的预言是~\n", "诶，让我看看您抽到了~\n"}
|
|
var position = [...]string{"正位", "逆位"}
|
|
|
|
func init() {
|
|
engine := control.Register("tarot", &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "塔罗牌\n" +
|
|
"- 抽塔罗牌\n" +
|
|
"- 抽n张塔罗牌",
|
|
// TODO 抽X张塔罗牌 解塔罗牌[牌名]
|
|
PublicDataFolder: "Tarot",
|
|
}).ApplySingle(ctxext.DefaultSingle)
|
|
|
|
engine.OnRegex(`^抽(\d{1,3}张)?塔罗牌$`, ctxext.DoOnceOnSuccess(
|
|
func(ctx *zero.Ctx) bool {
|
|
data, err := engine.GetLazyData("tarots.json", true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
err = json.Unmarshal(data, &cardMap)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
logrus.Infof("[tarot]读取%d张塔罗牌", len(cardMap))
|
|
return true
|
|
},
|
|
)).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
|
match := ctx.State["regex_matched"].([]string)[1]
|
|
n := 1
|
|
if match != "" {
|
|
var err error
|
|
n, err = strconv.Atoi(match[:len(match)-3])
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
if n <= 0 {
|
|
ctx.SendChain(message.Text("ERROR:张数必须为正"))
|
|
return
|
|
}
|
|
if n > 1 && !zero.OnlyGroup(ctx) {
|
|
ctx.SendChain(message.Text("ERROR:抽取多张仅支持群聊"))
|
|
return
|
|
}
|
|
if n > 20 {
|
|
ctx.SendChain(message.Text("ERROR:抽取张数过多"))
|
|
return
|
|
}
|
|
}
|
|
if n == 1 {
|
|
if id := ctx.Send(randTarot()); id.ID() == 0 {
|
|
ctx.SendChain(message.Text("ERROR:可能被风控了"))
|
|
}
|
|
return
|
|
}
|
|
msg := make([]message.MessageSegment, n)
|
|
for i := range msg {
|
|
msg[i] = ctxext.FakeSenderForwardNode(ctx, randTarot()...)
|
|
}
|
|
ctx.SendGroupForwardMessage(ctx.Event.GroupID, msg)
|
|
return
|
|
})
|
|
}
|
|
|
|
func randTarot() []message.MessageSegment {
|
|
i := rand.Intn(22)
|
|
p := rand.Intn(2)
|
|
card := cardMap[(strconv.Itoa(i))]
|
|
name := card.Name
|
|
var info string
|
|
if p == 0 {
|
|
info = card.Info.Description
|
|
} else {
|
|
info = card.Info.ReverseDescription
|
|
}
|
|
return []message.MessageSegment{
|
|
message.Text(reasons[rand.Intn(len(reasons))], position[p], " 的 ", name, "\n"),
|
|
message.Image(fmt.Sprintf(bed+"MajorArcana/%d.png", i)),
|
|
message.Text("\n其意义为: ", info),
|
|
}
|
|
}
|