// Package novel 铅笔小说搜索插件
package novel

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"regexp"
	"strings"
	"time"

	"github.com/antchfx/htmlquery"
	log "github.com/sirupsen/logrus"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/extension/rate"
	"github.com/wdvxdr1123/ZeroBot/message"
	"github.com/wdvxdr1123/ZeroBot/utils/helper"

	control "github.com/FloatTech/zbpctrl"
	ub "github.com/FloatTech/zbputils/binary"
	"github.com/FloatTech/zbputils/txt2img"
)

const (
	websiteURL   = "https://www.23qb.net"
	websiteTitle = "铅笔小说"
	errorTitle   = "出现错误！"
	username     = "zerobot"
	password     = "123456"
	submit       = "%26%23160%3B%B5%C7%26%23160%3B%26%23160%3B%C2%BC%26%23160%3B"
	ua           = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"
	loginURL     = websiteURL + "/login.php?do=submit&jumpurl=https%3A%2F%2Fwww.23qb.net%2F"
	searchURL    = websiteURL + "/saerch.php"
	idReg        = `/(\d+)/`
)

var (
	gCurCookieJar *cookiejar.Jar
	engine        = control.Register("novel", &control.Options{
		DisableOnDefault: false,
		Help:             "铅笔小说网搜索\n- 小说[xxx]",
	})
	limit = rate.NewManager(time.Minute, 5)
)

func init() {
	engine.OnRegex("^小说([\u4E00-\u9FA5A-Za-z0-9]{1,25})$").SetBlock(true).
		Handle(func(ctx *zero.Ctx) {
			if !limit.Load(ctx.Event.GroupID).Acquire() {
				ctx.SendChain(message.Text("请稍后重试0x0..."))
				return
			}
			ctx.SendChain(message.Text("少女祈祷中......"))
			login(username, password)
			searchKey := ctx.State["regex_matched"].([]string)[1]
			searchHTML := search(searchKey)
			doc, err := htmlquery.Parse(strings.NewReader(searchHTML))
			if err != nil {
				log.Errorln("[novel]", err)
			}
			htmlTitle := htmlquery.InnerText(htmlquery.FindOne(doc, "/html/head/title"))
			switch htmlTitle {
			case websiteTitle:
				list, err := htmlquery.QueryAll(doc, "//dl[@id='nr']")
				if err != nil {
					log.Errorln("[novel]", err)
				}
				if len(list) != 0 {
					text := ""
					for _, v := range list {
						bookName := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[1]/h3/a[1]"))
						category := htmlquery.InnerText(htmlquery.FindOne(v, "/dt/span[1]"))
						author := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[1]"))
						status := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[2]"))
						wordNumbers := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[3]"))
						description := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[3]"))
						updateTime := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[1]/h3/span[1]"))
						latestChapter := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[4]/a[1]"))

						reg := regexp.MustCompile(idReg)
						id := reg.FindStringSubmatch(htmlquery.SelectAttr(htmlquery.FindOne(v, "/dt/a[1]"), "href"))[1]

						webpageURL := websiteURL + "/book/" + id + "/"
						downloadURL := websiteURL + "/modules/article/txtarticle.php?id=" + id
						text += fmt.Sprintf("书名:%s\n类型:%s\n作者:%s\n状态:%s\n字数:%s\n简介:%s\n更新时间:%s\n最新章节:%s\n网页链接:%s\n下载地址:%s\n\n", bookName, category, author, status, wordNumbers, description, updateTime, latestChapter, webpageURL, downloadURL)
					}
					data, err := txt2img.RenderToBase64(text, 40, 20)
					if err != nil {
						log.Println("err:", err)
					}
					if id := ctx.SendChain(message.Image("base64://" + helper.BytesToString(data))); id == 0 {
						ctx.SendChain(message.Text("ERROR: 可能被风控了"))
					}
				} else {
					text := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@id='tipss']"))
					text = strings.ReplaceAll(text, " ", "")
					text = strings.ReplaceAll(text, "本站", websiteURL)
					ctx.SendChain(message.Text(text))
				}
			case errorTitle:
				ctx.SendChain(message.Text(errorTitle))
				text := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@style='text-align: center;padding:10px']"))
				text = strings.ReplaceAll(text, " ", "")
				ctx.SendChain(message.Text(text))
			default:
				bookName := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:book_name']"), "content")
				category := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:category']"), "content")
				author := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:author']"), "content")
				status := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:status']"), "content")
				description := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@id='bookintro']/p"))
				updateTime := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:update_time']"), "content")
				latestChapter := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:latest_chapter_name']"), "content")

				reg := regexp.MustCompile(idReg)
				id := reg.FindStringSubmatch(htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:read_url']"), "content"))[1]
				webpageURL := websiteURL + "/book/" + id + "/"
				downloadURL := websiteURL + "/modules/article/txtarticle.php?id=" + id
				text := fmt.Sprintf("书名:%s\n类型:%s\n作者:%s\n状态:%s\n简介:%s\n更新时间:%s\n最新章节:%s\n网页链接:%s\n下载地址:%s\n", bookName, category, author, status, description, updateTime, latestChapter, webpageURL, downloadURL)
				data, err := txt2img.RenderToBase64(text, 40, 20)
				if err != nil {
					log.Println("err:", err)
				}
				if id := ctx.SendChain(message.Image("base64://" + helper.BytesToString(data))); id == 0 {
					ctx.SendChain(message.Text("ERROR: 可能被风控了"))
				}
			}
		})
}

func login(username, password string) {
	gCurCookieJar, _ = cookiejar.New(nil)
	client := &http.Client{
		Jar: gCurCookieJar,
	}
	usernameData, err := ub.UTF82GBK(helper.StringToBytes(username))
	if err != nil {
		log.Errorln("[novel]", err)
	}
	usernameGbk := helper.BytesToString(usernameData)
	passwordData, err := ub.UTF82GBK(helper.StringToBytes(password))
	if err != nil {
		log.Errorln("[novel]", err)
	}
	passwordGbk := helper.BytesToString(passwordData)
	loginReq, err := http.NewRequest("POST", loginURL, strings.NewReader(fmt.Sprintf("username=%s&password=%s&usecookie=315360000&action=login&submit=%s", url.QueryEscape(usernameGbk), url.QueryEscape(passwordGbk), submit)))
	if err != nil {
		log.Errorln("[novel]", err)
	}
	loginReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	loginReq.Header.Set("User-Agent", ua)
	loginResp, err := client.Do(loginReq)
	if err != nil {
		log.Errorln("[novel]", err)
	}
	defer loginResp.Body.Close()
}

func search(searchKey string) (searchHTML string) {
	searchKeyData, err := ub.UTF82GBK(helper.StringToBytes(searchKey))
	if err != nil {
		log.Errorln("[novel]", err)
	}
	searchKeyGbk := helper.BytesToString(searchKeyData)
	client := &http.Client{
		Jar: gCurCookieJar,
	}
	searchReq, err := http.NewRequest("POST", searchURL, strings.NewReader(fmt.Sprintf("searchkey=%s&searchtype=all", url.QueryEscape(searchKeyGbk))))
	if err != nil {
		log.Errorln("[novel]", err)
	}
	searchReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	searchReq.Header.Set("User-Agent", ua)
	searchResp, err := client.Do(searchReq)
	if err != nil {
		log.Errorln("[novel]", err)
	}
	defer searchResp.Body.Close()
	searchData, err := ioutil.ReadAll(searchResp.Body)
	if err != nil {
		log.Errorf("[novel] get response for url=%s got error=%s\n", searchURL, err.Error())
	}
	searchData, err = ub.GBK2UTF8(searchData)
	if err != nil {
		log.Errorln("[novel]", err)
	}
	searchHTML = helper.BytesToString(searchData)
	return searchHTML
}