⚡️ 🎨 🔥 优化代码结构

plugin_chouxianghua/chouxianghua.go

@@ -2,18 +2,40 @@
 package chouxianghua
 
 import (
-	control "github.com/FloatTech/zbputils/control"
+	"github.com/sirupsen/logrus"
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
 
+	"github.com/FloatTech/zbputils/control"
 	"github.com/FloatTech/zbputils/control/order"
+	"github.com/FloatTech/zbputils/file"
 )
 
 func init() {
-	control.Register("chouxianghua", order.AcquirePrio(), &control.Options{
+	en := control.Register("chouxianghua", order.AcquirePrio(), &control.Options{
 		DisableOnDefault: false,
 		Help:             "抽象话\n- 抽象翻译xxx",
-	}).OnRegex("^抽象翻译((\\s|[\\r\\n]|[\\p{Han}\\p{P}A-Za-z0-9])+)$").SetBlock(true).
+		PublicDataFolder: "ChouXiangHua",
+	})
+
+	go func() {
+		dbpath := en.DataFolder()
+		db.DBPath = dbpath + "cxh.db"
+		defer order.DoneOnExit()()
+		// os.RemoveAll(dbpath)
+		_, _ = file.GetLazyData(db.DBPath, false, true)
+		err := db.Create("pinyin", &pinyin{})
+		if err != nil {
+			panic(err)
+		}
+		n, err := db.Count("pinyin")
+		if err != nil {
+			panic(err)
+		}
+		logrus.Printf("[chouxianghua]读取%d条拼音", n)
+	}()
+
+	en.OnRegex("^抽象翻译((\\s|[\\r\\n]|[\\p{Han}\\p{P}A-Za-z0-9])+)$").SetBlock(true).
 		Handle(func(ctx *zero.Ctx) {
 			r := cx(ctx.State["regex_matched"].([]string)[1])
 			ctx.SendChain(message.Text(r))

plugin_chouxianghua/data.go

@@ -1,40 +0,0 @@
-package chouxianghua
-
-import (
-	"os"
-
-	log "github.com/sirupsen/logrus"
-
-	sql "github.com/FloatTech/sqlite"
-	"github.com/FloatTech/zbputils/file"
-	"github.com/FloatTech/zbputils/process"
-
-	"github.com/FloatTech/zbputils/control/order"
-)
-
-const (
-	dbpath = "data/ChouXiangHua/"
-	dbfile = dbpath + "cxh.db"
-)
-
-var db = &sql.Sqlite{DBPath: dbfile}
-
-// 加载数据库
-func init() {
-	go func() {
-		defer order.DoneOnExit()()
-		process.SleepAbout1sTo2s()
-		// os.RemoveAll(dbpath)
-		_ = os.MkdirAll(dbpath, 0755)
-		_, _ = file.GetLazyData(dbfile, false, true)
-		err := db.Create("pinyin", &pinyin{})
-		if err != nil {
-			panic(err)
-		}
-		n, err := db.Count("pinyin")
-		if err != nil {
-			panic(err)
-		}
-		log.Printf("[chouxianghua]读取%d条拼音", n)
-	}()
-}

plugin_chouxianghua/model.go

@@ -1,5 +1,7 @@
 package chouxianghua
 
+import sql "github.com/FloatTech/sqlite"
+
 type pinyin struct {
 	Word   string `db:"word"`
 	Pronun string `db:"pronunciation"`
@@ -9,6 +11,8 @@ type emoji struct {
 	Emoji  string `db:"emoji"`
 }
 
+var db = &sql.Sqlite{}
+
 func getPinyinByWord(word string) string {
 	var p pinyin
 	_ = db.Find("pinyin", &p, "where word = '"+word+"'")