// Package nativesetu 本地setu
package nativesetu

import (
	"fmt"
	"os"

	"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"
	ctrl "github.com/FloatTech/zbpctrl"
	"github.com/FloatTech/zbputils/control"
	"github.com/FloatTech/zbputils/ctxext"
)

var (
	setupath = "/tmp" // 绝对路径，图片根目录
)

func init() {
	engine := control.Register("nativesetu", &ctrl.Options[*zero.Ctx]{
		DisableOnDefault: false,
		Brief:            "本地涩图",
		Help: "- 本地[xxx]\n" +
			"- 刷新本地[xxx]\n" +
			"- 设置本地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.Infoln("[nsetu] set setu dir to", setupath)
		}
	}

	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))
		})
}