// Package lolicon 基于 https://api.lolicon.app 随机图片
package lolicon

import (
	"encoding/base64"
	"net/url"
	"strings"
	"time"

	"github.com/tidwall/gjson"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/message"

	"github.com/FloatTech/floatbox/math"
	"github.com/FloatTech/floatbox/process"
	"github.com/FloatTech/floatbox/web"
	ctrl "github.com/FloatTech/zbpctrl"
	"github.com/FloatTech/zbputils/control"
	"github.com/FloatTech/zbputils/ctxext"
	"github.com/FloatTech/zbputils/img/pool"
)

const (
	api      = "https://api.lolicon.app/setu/v2"
	capacity = 10
)

var (
	queue   = make(chan string, capacity)
	custapi = ""
)

func init() {
	en := control.Register("lolicon", &ctrl.Options[*zero.Ctx]{
		DisableOnDefault: false,
		Help: "lolicon\n" +
			"- 随机图片\n" +
			"- 随机图片 萝莉|少女\n" +
			"- 设置随机图片地址[http...]",
	}).ApplySingle(ctxext.DefaultSingle)
	en.OnPrefix("随机图片").Limit(ctxext.LimitByUser).SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			go func() {
				for i := 0; i < math.Min(cap(queue)-len(queue), 2); i++ {
					if custapi != "" {
						data, err := web.GetData(custapi)
						if err != nil {
							ctx.SendChain(message.Text("ERROR: ", err))
							continue
						}
						queue <- "base64://" + base64.StdEncoding.EncodeToString(data)
						continue
					}
					rapi := api
					args := strings.TrimSpace(ctx.State["args"].(string))
					if args != "" {
						rapi += "?tag=" + url.QueryEscape(args)
					}
					data, err := web.GetData(rapi)
					if err != nil {
						ctx.SendChain(message.Text("ERROR: ", err))
						continue
					}
					json := gjson.ParseBytes(data)
					if e := json.Get("error").Str; e != "" {
						ctx.SendChain(message.Text("ERROR: ", e))
						continue
					}
					url := json.Get("data.0.urls.original").Str
					url = strings.ReplaceAll(url, "i.pixiv.cat", "i.pixiv.re")
					name := url[strings.LastIndex(url, "/")+1 : len(url)-4]
					m, err := pool.GetImage(name)
					if err != nil {
						m.SetFile(url)
						_, err = m.Push(ctxext.SendToSelf(ctx), ctxext.GetMessage(ctx))
						process.SleepAbout1sTo2s()
					}
					if err == nil {
						queue <- m.String()
					} else {
						queue <- url
					}
				}
			}()
			select {
			case <-time.After(time.Minute):
				ctx.SendChain(message.Text("ERROR: 等待填充，请稍后再试......"))
			case img := <-queue:
				if id := ctx.Send(message.Message{ctxext.FakeSenderForwardNode(ctx, message.Image(img))}).ID(); id == 0 {
					process.SleepAbout1sTo2s()
					if id := ctx.Send(message.Message{ctxext.FakeSenderForwardNode(ctx, message.Image(img).Add("cache", "0"))}).ID(); id == 0 {
						ctx.SendChain(message.Text("ERROR: 可能被风控或下载图片用时过长，请耐心等待"))
					}
				}
			}
		})
	en.OnPrefix("设置随机图片地址", zero.SuperUserPermission).SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			u := strings.TrimSpace(ctx.State["args"].(string))
			ctx.SendChain(message.Text("成功设置随机图片地址为", u))
			custapi = u
		})
}