mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
// Package cpstory cp短打
|
|
package cpstory
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/math"
|
|
)
|
|
|
|
func init() {
|
|
engine := control.Register("cpstory", &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "cp短打\n- 组cp[@xxx][@xxx]\n- 磕cp大老师 雪乃",
|
|
PublicDataFolder: "CpStory",
|
|
})
|
|
|
|
getdb := ctxext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
|
|
db.DBPath = engine.DataFolder() + "cp.db"
|
|
// os.RemoveAll(dbpath)
|
|
_, _ = engine.GetLazyData("cp.db", true)
|
|
err := db.Create("cp_story", &cpstory{})
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
n, err := db.Count("cp_story")
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
logrus.Printf("[cpstory]读取%d条故事", n)
|
|
return true
|
|
})
|
|
|
|
engine.OnRegex("^组cp.*?(\\d+).*?(\\d+)", zero.OnlyGroup, getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
|
cs := getRandomCpStory()
|
|
gong := ctx.CardOrNickName(math.Str2Int64(ctx.State["regex_matched"].([]string)[1]))
|
|
shou := ctx.CardOrNickName(math.Str2Int64(ctx.State["regex_matched"].([]string)[2]))
|
|
text := strings.ReplaceAll(cs.Story, "<攻>", gong)
|
|
text = strings.ReplaceAll(text, "<受>", shou)
|
|
text = strings.ReplaceAll(text, cs.Gong, gong)
|
|
text = strings.ReplaceAll(text, cs.Shou, gong)
|
|
ctx.SendChain(message.Text(text))
|
|
})
|
|
engine.OnPrefix("磕cp", getdb).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
|
cs := getRandomCpStory()
|
|
params := strings.Split(ctx.State["args"].(string), " ")
|
|
if len(params) < 2 {
|
|
ctx.SendChain(message.Text(ctx.Event.MessageID), message.Text("请用空格分开两个人名"))
|
|
} else {
|
|
gong := params[0]
|
|
shou := params[1]
|
|
text := strings.ReplaceAll(cs.Story, "<攻>", gong)
|
|
text = strings.ReplaceAll(text, "<受>", shou)
|
|
text = strings.ReplaceAll(text, cs.Gong, gong)
|
|
text = strings.ReplaceAll(text, cs.Shou, gong)
|
|
ctx.SendChain(message.Text(text))
|
|
}
|
|
})
|
|
}
|