mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
// Package funny 冷笑话
|
|
package funny
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
|
|
sql "github.com/FloatTech/sqlite"
|
|
control "github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/file"
|
|
)
|
|
|
|
type joke struct {
|
|
ID uint32 `db:"id"`
|
|
Text string `db:"text"`
|
|
}
|
|
|
|
var db = &sql.Sqlite{}
|
|
|
|
func init() {
|
|
en := control.Register("funny", &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "讲个笑话\n" +
|
|
"- 讲个笑话[@xxx|qq号|人名] | 夸夸[@xxx|qq号|人名] ",
|
|
PublicDataFolder: "Funny",
|
|
})
|
|
|
|
en.OnPrefixGroup([]string{"讲个笑话", "夸夸"}, ctxext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
|
|
dbpath := en.DataFolder()
|
|
db.DBPath = dbpath + "jokes.db"
|
|
_, err := file.GetLazyData(db.DBPath, false, true)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
err = db.Create("jokes", &joke{})
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
c, err := db.Count("jokes")
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return false
|
|
}
|
|
logrus.Infoln("[funny]加载", c, "个笑话")
|
|
return true
|
|
})).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
|
// 获取名字
|
|
name := ctx.NickName()
|
|
var j joke
|
|
err := db.Pick("jokes", &j)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR:", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Text(strings.ReplaceAll(j.Text, "%name", name)))
|
|
})
|
|
}
|