⚡️ 🎨 🔥 优化代码结构

plugin_nativesetu/data.go

@@ -10,14 +10,10 @@ import (
 
 	"github.com/corona10/goimagehash"
 	"github.com/sirupsen/logrus"
-	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 	_ "golang.org/x/image/webp" // import webp decoding
 
 	sql "github.com/FloatTech/sqlite"
 	"github.com/FloatTech/zbputils/file"
-	"github.com/FloatTech/zbputils/process"
-
-	"github.com/FloatTech/zbputils/control/order"
 )
 
 // setuclass holds setus in a folder, which is the class name.
@@ -27,25 +23,7 @@ type setuclass struct {
 	Path  string `db:"path"`  // Path 图片路径
 }
 
-var ns = &nsetu{db: &sql.Sqlite{DBPath: dbfile}}
-
-func init() {
-	go func() {
-		defer order.DoneOnExit()()
-		process.SleepAbout1sTo2s()
-		err := os.MkdirAll(datapath, 0755)
-		if err != nil {
-			panic(err)
-		}
-		if file.IsExist(cfgfile) {
-			b, err := os.ReadFile(cfgfile)
-			if err == nil {
-				setupath = helper.BytesToString(b)
-				logrus.Println("[nsetu] set setu dir to", setupath)
-			}
-		}
-	}()
-}
+var ns = &nsetu{db: &sql.Sqlite{}}
 
 type nsetu struct {
 	db *sql.Sqlite

plugin_nativesetu/main.go

@@ -12,16 +12,11 @@ import (
 
 	"github.com/FloatTech/zbputils/control"
 	"github.com/FloatTech/zbputils/ctxext"
+	"github.com/FloatTech/zbputils/file"
 
 	"github.com/FloatTech/zbputils/control/order"
 )
 
-const (
-	datapath = "data/nsetu"
-	dbfile   = datapath + "/data.db"
-	cfgfile  = datapath + "/setupath.txt"
-)
-
 var (
 	setupath = "/tmp" // 绝对路径，图片根目录
 )
@@ -35,7 +30,19 @@ func init() {
 			"- 设置本地setu绝对路径[xxx]\n" +
 			"- 刷新所有本地setu\n" +
 			"- 所有本地setu分类",
+		PrivateDataFolder: "nsetu",
 	})
+
+	ns.db.DBPath = engine.DataFolder() + "data.db"
+	cfgfile := engine.DataFolder() + "setupath.txt"
+	if file.IsExist(cfgfile) {
+		b, err := os.ReadFile(cfgfile)
+		if err == nil {
+			setupath = helper.BytesToString(b)
+			logrus.Println("[nsetu] set setu dir to", setupath)
+		}
+	}
+
 	engine.OnRegex(`^本地(.*)$`, ctxext.FirstValueInList(ns)).SetBlock(true).
 		Handle(func(ctx *zero.Ctx) {
 			imgtype := ctx.State["regex_matched"].([]string)[1]