🎨 优化目录结构

plugin/chouxianghua/chouxianghua.go

@@ -0,0 +1,64 @@
+// Package chouxianghua 抽象话转化
+package chouxianghua
+
+import (
+	"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() {
+	en := control.Register("chouxianghua", order.AcquirePrio(), &control.Options{
+		DisableOnDefault: false,
+		Help:             "抽象话\n- 抽象翻译xxx",
+		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))
+		})
+}
+
+func cx(s string) (r string) {
+	h := []rune(s)
+	for i := 0; i < len(h); i++ {
+		if i < len(h)-1 {
+			e := getEmojiByPronun(getPronunByDWord(h[i], h[i+1]))
+			if e != "" {
+				r += e
+				i++
+				continue
+			}
+		}
+		e := getEmojiByPronun(getPinyinByWord(string(h[i])))
+		if e != "" {
+			r += e
+			continue
+		}
+		r += string(h[i])
+	}
+	return
+}

plugin/chouxianghua/model.go

@@ -0,0 +1,30 @@
+package chouxianghua
+
+import sql "github.com/FloatTech/sqlite"
+
+type pinyin struct {
+	Word   string `db:"word"`
+	Pronun string `db:"pronunciation"`
+}
+type emoji struct {
+	Pronun string `db:"pronunciation"`
+	Emoji  string `db:"emoji"`
+}
+
+var db = &sql.Sqlite{}
+
+func getPinyinByWord(word string) string {
+	var p pinyin
+	_ = db.Find("pinyin", &p, "where word = '"+word+"'")
+	return p.Pronun
+}
+
+func getPronunByDWord(w0, w1 rune) string {
+	return getPinyinByWord(string(w0)) + getPinyinByWord(string(w1))
+}
+
+func getEmojiByPronun(pronun string) string {
+	var e emoji
+	_ = db.Find("emoji", &e, "where pronunciation = '"+pronun+"'")
+	return e.Emoji
+}