合入主仓库

main.go

@@ -1,67 +1,59 @@
-// Package wife 抽老婆
-package wife
+package main
 
 import (
 	"encoding/json"
-	"strconv"
+	"os"
+	"strings"
 
-	fcext "github.com/FloatTech/floatbox/ctxext"
-	"github.com/FloatTech/floatbox/file"
-	ctrl "github.com/FloatTech/zbpctrl"
-	"github.com/FloatTech/zbputils/control"
-	"github.com/FloatTech/zbputils/ctxext"
-	"github.com/sirupsen/logrus"
-	zero "github.com/wdvxdr1123/ZeroBot"
-	"github.com/wdvxdr1123/ZeroBot/message"
+	"github.com/fumiama/imgsz"
 )
 
-type cardInfo struct {
-	Name  string   `json:"name"`
-	URL   string   `json:"url"`
-	Lines []string `json:"lines"`
-}
-type cardSet = map[string]cardInfo
+//go:generate go run main.go
 
-var cardMap = make(cardSet, 50)
-var datapath string
+const wifvesdir = "./wives/"
+const jsonfile = "wife.json"
 
-func init() {
-	engine := control.Register("wife", &ctrl.Options[*zero.Ctx]{
-		DisableOnDefault: false,
-		Help: "抽老婆\n" +
-			"- 抽老婆",
-		PublicDataFolder: "Wives",
-	}).ApplySingle(ctxext.DefaultSingle)
-	engine.OnFullMatchGroup([]string{"抽老婆"}, fcext.DoOnceOnSuccess(
-		func(ctx *zero.Ctx) bool {
-			data, err := engine.GetLazyData("wife.json", true)
+func main() {
+	ent, err := os.ReadDir(wifvesdir)
+	if err != nil {
+		panic(err)
+	}
+	cards := make([]string, 0, len(ent))
+	for _, en := range ent {
+		if en.IsDir() {
+			continue
+		}
+		name := en.Name()
+		fn := wifvesdir + name
+		f, err := os.Open(fn)
+		if err != nil {
+			continue
+		}
+		_, format, err := imgsz.DecodeSize(f)
+		_ = f.Close()
+		if err != nil {
+			continue
+		}
+		i := strings.LastIndex(name, ".")
+		if i <= 0 {
+			continue
+		}
+		name = name[:i] + "." + format
+		nfn := wifvesdir + name
+		if fn != nfn {
+			err = os.Rename(fn, nfn)
 			if err != nil {
-				ctx.SendChain(message.Text("ERROR:", err))
-				return false
+				continue
 			}
-			err = json.Unmarshal(data, &cardMap)
-			if err != nil {
-				ctx.SendChain(message.Text("ERROR:", err))
-				return false
-			}
-			datapath = file.BOTPATH + "/" + engine.DataFolder()
-			logrus.Infof("[wife]读取%d个老婆", len(cardMap))
-			return true
-		},
-	)).SetBlock(true).
-		Handle(func(ctx *zero.Ctx) {
-			i := fcext.RandSenderPerDayN(ctx.Event.UserID, len(cardMap))
-			card := cardMap[(strconv.Itoa(i))]
-
-			if id := ctx.SendChain(
-				message.At(ctx.Event.UserID),
-				message.Text("今天的二次元老婆是~【", card.Name, "】哒"),
-				message.Image("file:///"+datapath+card.URL),
-				// message.Text("\n【", card.Lines[rand.Intn(len(card.Lines))], "】"),
-			); id.ID() == 0 {
-				ctx.SendChain(
-					message.At(ctx.Event.UserID),
-					message.Text("今天的二次元老婆是~【", card.Name, "】哒\n【图片发送失败，请联系维护者~】"))
-			}
-		})
+		}
+		cards = append(cards, name)
+	}
+	f, err := os.Create(jsonfile)
+	if err != nil {
+		panic(err)
+	}
+	err = json.NewEncoder(f).Encode(cards)
+	if err != nil {
+		panic(err)
+	}
 }