mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
359 lines
11 KiB
Go
359 lines
11 KiB
Go
// Package tarot 塔罗牌
|
|
package tarot
|
|
|
|
import (
|
|
"encoding/json"
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/FloatTech/floatbox/binary"
|
|
fcext "github.com/FloatTech/floatbox/ctxext"
|
|
"github.com/FloatTech/floatbox/file"
|
|
"github.com/FloatTech/floatbox/process"
|
|
"github.com/FloatTech/floatbox/web"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/img/pool"
|
|
"github.com/FloatTech/zbputils/img/text"
|
|
"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 cardInfo struct {
|
|
Description string `json:"description"`
|
|
ReverseDescription string `json:"reverseDescription"`
|
|
ImgURL string `json:"imgUrl"`
|
|
}
|
|
type card struct {
|
|
Name string `json:"name"`
|
|
cardInfo `json:"info"`
|
|
}
|
|
|
|
type formation struct {
|
|
CardsNum int `json:"cards_num"`
|
|
IsCut bool `json:"is_cut"`
|
|
Represent [][]string `json:"represent"`
|
|
}
|
|
type cardSet = map[string]card
|
|
|
|
var (
|
|
cardMap = make(cardSet, 80)
|
|
infoMap = make(map[string]cardInfo, 80)
|
|
formationMap = make(map[string]formation, 10)
|
|
majorArcanaName = make([]string, 0, 80)
|
|
formationName = make([]string, 0, 10)
|
|
)
|
|
|
|
func init() {
|
|
engine := control.Register("tarot", &ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "塔罗牌",
|
|
Help: "- 抽[塔罗牌|大阿卡纳|小阿卡纳]\n" +
|
|
"- 抽n张[塔罗牌|大阿卡纳|小阿卡纳]\n" +
|
|
"- 解塔罗牌[牌名]\n" +
|
|
"- [塔罗|大阿卡纳|小阿卡纳|混合]牌阵[圣三角|时间之流|四要素|五牌阵|吉普赛十字|马蹄|六芒星]",
|
|
PublicDataFolder: "Tarot",
|
|
}).ApplySingle(ctxext.DefaultSingle)
|
|
|
|
cache := engine.DataFolder() + "cache"
|
|
_ = os.RemoveAll(cache)
|
|
err := os.MkdirAll(cache, 0755)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
getTarot := fcext.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
|
|
}
|
|
for _, card := range cardMap {
|
|
infoMap[card.Name] = card.cardInfo
|
|
}
|
|
for i := 0; i < 22; i++ {
|
|
majorArcanaName = append(majorArcanaName, cardMap[strconv.Itoa(i)].Name)
|
|
}
|
|
logrus.Infof("[tarot]读取%d张塔罗牌", len(cardMap))
|
|
formation, err := engine.GetLazyData("formation.json", true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return false
|
|
}
|
|
err = json.Unmarshal(formation, &formationMap)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return false
|
|
}
|
|
for k := range formationMap {
|
|
formationName = append(formationName, k)
|
|
}
|
|
logrus.Infof("[tarot]读取%d组塔罗牌阵", len(formationMap))
|
|
return true
|
|
})
|
|
engine.OnRegex(`^抽(\d{1,2}张)?((塔罗牌|大阿(尔)?卡纳)|小阿(尔)?卡纳)$`, getTarot).SetBlock(true).Limit(ctxext.LimitByGroup).Handle(func(ctx *zero.Ctx) {
|
|
match := ctx.State["regex_matched"].([]string)[1]
|
|
cardType := ctx.State["regex_matched"].([]string)[2]
|
|
n := 1
|
|
reasons := [...]string{"您抽到的是~\n", "锵锵锵，塔罗牌的预言是~\n", "诶，让我看看您抽到了~\n"}
|
|
position := [...]string{"『正位』", "『逆位』"}
|
|
reverse := [...]string{"", "Reverse/"}
|
|
start := 0
|
|
length := 22
|
|
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 strings.Contains(cardType, "小") {
|
|
start = 22
|
|
length = 55
|
|
}
|
|
if n == 1 {
|
|
i := rand.Intn(length) + start
|
|
p := rand.Intn(2)
|
|
card := cardMap[strconv.Itoa(i)]
|
|
name := card.Name
|
|
description := card.Description
|
|
if p == 1 {
|
|
description = card.ReverseDescription
|
|
}
|
|
imgurl := bed + reverse[p] + card.ImgURL
|
|
imgname := ""
|
|
if p == 1 {
|
|
imgname = reverse[p][:len(reverse[p])-1] + name
|
|
} else {
|
|
imgname = name
|
|
}
|
|
imgpath := cache + "/" + imgname + ".png"
|
|
err := pool.SendImageFromPool("pool"+imgname, imgpath, func() error {
|
|
data, err := web.RequestDataWith(web.NewTLS12Client(), imgurl, "GET", "gitcode.net", web.RandUA(), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f, err := os.Create(imgpath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
return os.WriteFile(f.Name(), data, 0755)
|
|
}, ctxext.Send(ctx), ctxext.GetMessage(ctx))
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
process.SleepAbout1sTo2s()
|
|
ctx.SendChain(message.Text(reasons[rand.Intn(len(reasons))], position[p], "的『", name, "』\n其释义为: ", description))
|
|
return
|
|
}
|
|
msg := make(message.Message, n)
|
|
randomIntMap := make(map[int]int, 30)
|
|
for i := range msg {
|
|
j := rand.Intn(length)
|
|
_, ok := randomIntMap[j]
|
|
for ok {
|
|
j = rand.Intn(length)
|
|
_, ok = randomIntMap[j]
|
|
}
|
|
randomIntMap[j] = 0
|
|
p := rand.Intn(2)
|
|
card := cardMap[strconv.Itoa(j+start)]
|
|
name := card.Name
|
|
description := card.Description
|
|
if p == 1 {
|
|
description = card.ReverseDescription
|
|
}
|
|
imgurl := bed + reverse[p] + card.ImgURL
|
|
tarotmsg := message.Message{message.Text(reasons[rand.Intn(len(reasons))], position[p], "的『", name, "』\n")}
|
|
var imgmsg message.MessageSegment
|
|
var err error
|
|
if p == 1 {
|
|
imgmsg, err = poolimg(ctx, imgurl, reverse[p][:len(reverse[p])-1]+name, cache)
|
|
} else {
|
|
imgmsg, err = poolimg(ctx, imgurl, name, cache)
|
|
}
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
tarotmsg = append(tarotmsg, imgmsg)
|
|
tarotmsg = append(tarotmsg, message.Text("\n其释义为: ", description))
|
|
msg[i] = ctxext.FakeSenderForwardNode(ctx, tarotmsg...)
|
|
}
|
|
if id := ctx.Send(msg).ID(); id == 0 {
|
|
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
|
}
|
|
})
|
|
|
|
engine.OnRegex(`^解塔罗牌\s?(.*)`, getTarot).SetBlock(true).Limit(ctxext.LimitByGroup).Handle(func(ctx *zero.Ctx) {
|
|
match := ctx.State["regex_matched"].([]string)[1]
|
|
info, ok := infoMap[match]
|
|
if ok {
|
|
imgurl := bed + info.ImgURL
|
|
var tarotmsg message.Message
|
|
imgmsg, err := poolimg(ctx, imgurl, match, cache)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
tarotmsg = append(tarotmsg, imgmsg)
|
|
tarotmsg = append(tarotmsg, message.Text("\n", match, "的含义是~\n『正位』:", info.Description, "\n『逆位』:", info.ReverseDescription))
|
|
if id := ctx.Send(tarotmsg).ID(); id == 0 {
|
|
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
|
}
|
|
return
|
|
}
|
|
var build strings.Builder
|
|
build.WriteString("塔罗牌列表\n大阿尔卡纳:\n")
|
|
build.WriteString(strings.Join(majorArcanaName[:7], " "))
|
|
build.WriteString("\n")
|
|
build.WriteString(strings.Join(majorArcanaName[7:14], " "))
|
|
build.WriteString("\n")
|
|
build.WriteString(strings.Join(majorArcanaName[14:22], " "))
|
|
build.WriteString("\n小阿尔卡纳:\n[圣杯|星币|宝剑|权杖] [0-10|侍从|骑士|王后|国王]")
|
|
txt := build.String()
|
|
cardList, err := text.RenderToBase64(txt, text.FontFile, 420, 20)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Text("没有找到", match, "噢~"), message.Image("base64://"+binary.BytesToString(cardList)))
|
|
})
|
|
engine.OnRegex(`^((塔罗|大阿(尔)?卡纳)|小阿(尔)?卡纳|混合)牌阵\s?(.*)`, getTarot).SetBlock(true).Limit(ctxext.LimitByGroup).Handle(func(ctx *zero.Ctx) {
|
|
cardType := ctx.State["regex_matched"].([]string)[1]
|
|
match := ctx.State["regex_matched"].([]string)[5]
|
|
info, ok := formationMap[match]
|
|
position := [...]string{"『正位』", "『逆位』"}
|
|
reverse := [...]string{"", "Reverse/"}
|
|
start, length := 0, 22
|
|
if strings.Contains(cardType, "小") {
|
|
start = 22
|
|
length = 55
|
|
} else if cardType == "混合" {
|
|
start = 0
|
|
length = 77
|
|
}
|
|
if ok {
|
|
ctx.SendChain(message.Text("少女祈祷中..."))
|
|
var build strings.Builder
|
|
build.WriteString(ctx.CardOrNickName(ctx.Event.UserID))
|
|
build.WriteString("---")
|
|
build.WriteString(match)
|
|
build.WriteString("\n")
|
|
msg := make(message.Message, info.CardsNum+1)
|
|
randomIntMap := make(map[int]int, 30)
|
|
for i := 0; i < info.CardsNum; i++ {
|
|
j := rand.Intn(length)
|
|
_, ok := randomIntMap[j]
|
|
for ok {
|
|
j = rand.Intn(length)
|
|
_, ok = randomIntMap[j]
|
|
}
|
|
randomIntMap[j] = 0
|
|
p := rand.Intn(2)
|
|
card := cardMap[strconv.Itoa(j+start)]
|
|
name := card.Name
|
|
description := card.Description
|
|
if p == 1 {
|
|
description = card.ReverseDescription
|
|
}
|
|
var tarotmsg message.Message
|
|
imgurl := bed + reverse[p] + card.ImgURL
|
|
var imgmsg message.MessageSegment
|
|
var err error
|
|
if p == 1 {
|
|
imgmsg, err = poolimg(ctx, imgurl, reverse[p][:len(reverse[p])-1]+name, cache)
|
|
} else {
|
|
imgmsg, err = poolimg(ctx, imgurl, name, cache)
|
|
}
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
tarotmsg = append(tarotmsg, imgmsg)
|
|
build.WriteString(info.Represent[0][i])
|
|
build.WriteString(":")
|
|
build.WriteString(position[p])
|
|
build.WriteString("的『")
|
|
build.WriteString(name)
|
|
build.WriteString("』\n其释义为: \n")
|
|
build.WriteString(description)
|
|
build.WriteString("\n")
|
|
msg[i] = ctxext.FakeSenderForwardNode(ctx, tarotmsg...)
|
|
}
|
|
txt := build.String()
|
|
formation, err := text.RenderToBase64(txt, text.FontFile, 420, 20)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
msg[info.CardsNum] = ctxext.FakeSenderForwardNode(ctx, message.Message{message.Image("base64://" + binary.BytesToString(formation))}...)
|
|
if id := ctx.Send(msg).ID(); id == 0 {
|
|
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
|
}
|
|
} else {
|
|
ctx.SendChain(message.Text("没有找到", match, "噢~\n现有牌阵列表: \n", strings.Join(formationName, "\n")))
|
|
}
|
|
})
|
|
}
|
|
|
|
func poolimg(ctx *zero.Ctx, imgurl, imgname, cache string) (msg message.MessageSegment, err error) {
|
|
imgfile := cache + "/" + imgname + ".png"
|
|
aimgfile := file.BOTPATH + "/" + imgfile
|
|
m, err := pool.GetImage("pool" + imgname)
|
|
if err == nil {
|
|
msg = message.Image(m.String())
|
|
if ctxext.SendToSelf(ctx)(msg) == 0 {
|
|
msg = msg.Add("cache", "0")
|
|
}
|
|
return
|
|
}
|
|
if file.IsNotExist(aimgfile) {
|
|
var data []byte
|
|
data, err = web.RequestDataWith(web.NewTLS12Client(), imgurl, "GET", "gitcode.net", web.RandUA(), nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
var f *os.File
|
|
f, err = os.Create(imgfile)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer f.Close()
|
|
err = os.WriteFile(f.Name(), data, 0755)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
m.SetFile(aimgfile)
|
|
_, _ = m.Push(ctxext.SendToSelf(ctx), ctxext.GetMessage(ctx))
|
|
msg = message.Image("file:///" + aimgfile)
|
|
return
|
|
}
|