package pixiv

import (
	"crypto/tls"
	"errors"
	"fmt"
	"io/ioutil"
	"net"
	"net/http"
	"strings"

	"github.com/tidwall/gjson"
)

// Illust 插画信息
type Illust struct {
	Pid         int64  `db:"pid"`
	Title       string `db:"title"`
	Caption     string `db:"caption"`
	Tags        string `db:"tags"`
	ImageUrls   string `db:"image_urls"`
	AgeLimit    string `db:"age_limit"`
	CreatedTime string `db:"created_time"`
	UserId      int64  `db:"user_id"`
	UserName    string `db:"user_name"`
}

// IllustInfo 根据p站插画id返回插画信息Illust
func (this *Illust) IllustInfo(id int64) (err error) {
	api := fmt.Sprintf("https://pixiv.net/ajax/illust/%d", id)
	transport := http.Transport{
		DisableKeepAlives: true,
		// 绕过sni审查
		TLSClientConfig: &tls.Config{
			ServerName:         "-",
			InsecureSkipVerify: true,
		},
		// 更改dns
		Dial: func(network, addr string) (net.Conn, error) {
			return net.Dial("tcp", "210.140.131.223:443")
		},
	}
	client := &http.Client{
		Transport: &transport,
	}

	// 网络请求
	req, err := http.NewRequest("GET", api, nil)
	if err != nil {
		return err
	}
	req.Header.Set("Host", "pixiv.net")
	req.Header.Set("Referer", "pixiv.net")
	req.Header.Set("Accept", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:6.0) Gecko/20100101 Firefox/6.0")

	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	if code := resp.StatusCode; code != 200 {
		return errors.New(fmt.Sprintf("Search illust's info failed, status %d", code))
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return err
	}
	json := gjson.ParseBytes(body).Get("body")

	// 如果有"R-18"tag则判断为R-18（暂时）
	var ageLimit = "all-age"
	for _, tag := range json.Get("tags.tags.#.tag").Array() {
		if tag.Str == "R-18" {
			ageLimit = "r18"
			break
		}
	}
	// 解决json返回带html格式
	var caption = strings.ReplaceAll(json.Get("illustComment").Str, "<br />", "\n")
	if index := strings.Index(caption, "<"); index != -1 {
		caption = caption[:index]
	}
	// 解析返回插画信息
	this.Pid = json.Get("illustId").Int()
	this.Title = json.Get("illustTitle").Str
	this.Caption = caption
	this.Tags = fmt.Sprintln(json.Get("tags.tags.#.tag").Array())
	this.ImageUrls = json.Get("urls.original").Str
	this.AgeLimit = ageLimit
	this.CreatedTime = json.Get("createDate").Str
	this.UserId = json.Get("userId").Int()
	this.UserName = json.Get("userName").Str
	return nil
}

// BigPic 返回一张XML大图CQ码
func (i *Illust) BigPic(file string) string {
	var hash = PicHash(file)
	return fmt.Sprintf(`[CQ:xml,data=<?xml version='1.0' 
encoding='UTF-8' standalone='yes' ?><msg serviceID="5" 
templateID="12345" action="" brief="不够涩！" 
sourceMsgId="0" url="" flag="0" adverSign="0" multiMsgFlag="0">
<item layout="0" advertiser_id="0" aid="0"><image uuid="%s.jpg" md5="%s" 
GroupFiledid="2235033681" filesize="81322" local_path="%s.jpg" 
minWidth="200" minHeight="200" maxWidth="500" maxHeight="1000" />
</item><source name="%s⭐(id:%d author:%s)" icon="" 
action="" appid="-1" /></msg>]`,
		hash,
		hash,
		hash,
		i.Title,
		i.Pid,
		i.UserName,
	)
}

// NormalPic 返回一张普通图CQ码
func (i *Illust) NormalPic(file string) string {
	return fmt.Sprintf(`[CQ:image,file=file:///%s]`, file)
}

// DetailPic 返回一张带详细信息的图片CQ码
func (i *Illust) DetailPic(file string) string {
	return fmt.Sprintf(`[SetuTime] %s 
标题：%s 
插画ID：%d 
画师：%s 
画师ID：%d 
直链：https://pixivel.moe/detail?id=%d`,
		i.NormalPic(file),
		i.Title,
		i.Pid,
		i.UserName,
		i.UserId,
		i.Pid,
	)
}