mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-18 20:50:12 +08:00
126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
// Package nativesetu 本地setu
|
|
package nativesetu
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
"github.com/wdvxdr1123/ZeroBot/utils/helper"
|
|
|
|
fcext "github.com/FloatTech/floatbox/ctxext"
|
|
"github.com/FloatTech/floatbox/file"
|
|
sql "github.com/FloatTech/sqlite"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
)
|
|
|
|
var (
|
|
setupath = "/tmp" // 绝对路径，图片根目录
|
|
dbpath = ""
|
|
)
|
|
|
|
func init() {
|
|
engine := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "本地涩图",
|
|
Help: "- 本地[xxx]\n" +
|
|
"- 刷新本地[xxx]\n" +
|
|
"- 设置本地setu绝对路径[xxx]\n" +
|
|
"- 刷新所有本地setu\n" +
|
|
"- 所有本地setu分类",
|
|
PrivateDataFolder: "nsetu",
|
|
})
|
|
|
|
dbpath = engine.DataFolder() + "data.db"
|
|
ns.db = sql.New(dbpath)
|
|
cfgfile := engine.DataFolder() + "setupath.txt"
|
|
if file.IsExist(cfgfile) {
|
|
b, err := os.ReadFile(cfgfile)
|
|
if err == nil {
|
|
setupath = helper.BytesToString(b)
|
|
logrus.Infoln("[nsetu] set setu dir to", setupath)
|
|
}
|
|
}
|
|
err := ns.db.Open(time.Hour)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
engine.OnRegex(`^本地(.*)$`, fcext.ValueInList(func(ctx *zero.Ctx) string { return ctx.State["regex_matched"].([]string)[1] }, &ns)).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
imgtype := ctx.State["regex_matched"].([]string)[1]
|
|
sc := new(setuclass)
|
|
ns.mu.RLock()
|
|
err := ns.db.Pick(imgtype, sc)
|
|
ns.mu.RUnlock()
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
} else {
|
|
p := "file:///" + setupath + "/" + sc.Path
|
|
if ctx.Event.GroupID != 0 {
|
|
ctx.SendGroupForwardMessage(ctx.Event.GroupID, message.Message{
|
|
ctxext.FakeSenderForwardNode(ctx,
|
|
message.Text(imgtype, ": ", sc.Name, "\n"), message.Image(p),
|
|
)})
|
|
return
|
|
}
|
|
ctx.SendChain(message.Text(imgtype, ": ", sc.Name, "\n"), message.Image(p))
|
|
}
|
|
})
|
|
engine.OnRegex(`^刷新本地(.*)$`, fcext.ValueInList(func(ctx *zero.Ctx) string { return ctx.State["regex_matched"].([]string)[1] }, &ns), zero.SuperUserPermission).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
imgtype := ctx.State["regex_matched"].([]string)[1]
|
|
err := ns.scanclass(os.DirFS(setupath), imgtype, imgtype)
|
|
if err == nil {
|
|
ctx.SendChain(message.Text("成功！"))
|
|
} else {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
}
|
|
})
|
|
engine.OnRegex(`^设置本地setu绝对路径(.*)$`, zero.SuperUserPermission).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
setupath = ctx.State["regex_matched"].([]string)[1]
|
|
err := os.WriteFile(cfgfile, helper.StringToBytes(setupath), 0644)
|
|
if err == nil {
|
|
ctx.SendChain(message.Text("成功！"))
|
|
} else {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
}
|
|
})
|
|
engine.OnFullMatch("刷新所有本地setu", zero.SuperUserPermission).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
err := ns.scanall(setupath)
|
|
if err == nil {
|
|
ctx.SendChain(message.Text("成功！"))
|
|
} else {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
}
|
|
})
|
|
engine.OnFullMatch("所有本地setu分类").SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
msg := "本地setu分类一览"
|
|
hasnotchange := true
|
|
ns.mu.RLock()
|
|
for i, c := range ns.List() {
|
|
n, err := ns.db.Count(c)
|
|
if err == nil {
|
|
msg += fmt.Sprintf("\n%02d. %s(%d)", i, c, n)
|
|
} else {
|
|
msg += fmt.Sprintf("\n%02d. %s(error)", i, c)
|
|
logrus.Errorln("[nsetu]", err)
|
|
}
|
|
hasnotchange = false
|
|
}
|
|
ns.mu.RUnlock()
|
|
if hasnotchange {
|
|
msg += "\n空"
|
|
}
|
|
ctx.SendChain(message.Text(msg))
|
|
})
|
|
}
|