// Package imagefinder 关键字搜图
package imagefinder

import (
	"encoding/json"
	"errors"
	"math/rand"
	"strings"

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

	"github.com/FloatTech/AnimeAPI/pixiv"

	"github.com/FloatTech/zbputils/control"
	"github.com/FloatTech/zbputils/control/order"
	"github.com/FloatTech/zbputils/ctxext"
	"github.com/FloatTech/zbputils/file"
	"github.com/FloatTech/zbputils/img/pool"
	"github.com/FloatTech/zbputils/web"
)

type resultjson struct {
	Data struct {
		Illusts []struct {
			ID          int64  `json:"id"`
			Title       string `json:"title"`
			AltTitle    string `json:"altTitle"`
			Description string `json:"description"`
			Sanity      int    `json:"sanity"`
		} `json:"illusts"`
	} `json:"data"`
	Error   bool   `json:"error"`
	Message string `json:"message"`
}

func init() {
	control.Register("imgfinder", order.AcquirePrio(), &control.Options{
		DisableOnDefault: false,
		Help: "关键字搜图\n" +
			"- 来张 [xxx]",
	}).OnRegex(`^来张\s?(.*)$`, zero.AdminPermission).SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			keyword := ctx.State["regex_matched"].([]string)[1]
			soutujson, err := soutuapi(keyword)
			if err != nil {
				ctx.SendChain(message.Text("ERROR: ", err))
				return
			}
			rannum := rand.Intn(len(soutujson.Data.Illusts))
			illust, err := pixiv.Works(soutujson.Data.Illusts[rannum].ID)
			if err != nil {
				ctx.SendChain(message.Text("ERROR: ", err))
				return
			}
			u := illust.ImageUrls[0]
			n := u[strings.LastIndex(u, "/")+1 : len(u)-4]
			m, err := pool.GetImage(n)
			if err != nil {
				// 下载图片
				f := ""
				if f, err = illust.DownloadToCache(0, n); err != nil {
					ctx.SendChain(message.Text("ERROR: ", err))
					return
				}
				m.SetFile(file.BOTPATH + "/" + f)
				hassent, err := m.Push(ctxext.Send(ctx), ctxext.GetMessage(ctx))
				if hassent {
					return
				}
				if err != nil {
					ctx.SendChain(message.Text("ERROR: ", err))
					return
				}
			}
			// 发送图片
			id := ctx.SendChain(message.Image(m.String()))
			if id.ID() == 0 {
				id = ctx.SendChain(message.Image(m.String()).Add("cache", "0"))
				if id.ID() == 0 {
					ctx.SendChain(message.Text("图片发送失败，可能被风控了~"))
				}
			}
		})
}

// soutuapi 请求api
func soutuapi(keyword string) (r resultjson, err error) {
	url := "https://api.pixivel.moe/v2/pixiv/illust/search/" + keyword + "?page=0"
	data, err := web.ReqWith(url, "GET", "https://pixivel.moe/", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.61 Safari/537.36")
	if err != nil {
		return
	}
	err = json.Unmarshal(data, &r)
	if err == nil && r.Error {
		err = errors.New(r.Message)
	}
	return
}