package cmd

import (
	"github.com/go-resty/resty/v2"
	"github.com/mizuki1412/go-core-kit/class/exception"
	"github.com/mizuki1412/go-core-kit/init/initkit"
	"github.com/mizuki1412/go-core-kit/service/configkit"
	"github.com/mizuki1412/go-core-kit/service/logkit"
	"github.com/spf13/cast"
	"github.com/spf13/cobra"
	"github.com/tidwall/gjson"
	"net/url"
	"strings"
	"time"
)

//法治在沃

func init() {
	rootCmd.AddCommand(lawCmd)
	defFlagsLaw(lawCmd)
}

var lawCmd = &cobra.Command{
	Use:   "law",
	Short: "Batch processing operations of the law platform",
	Run: func(cmd *cobra.Command, args []string) {
		initkit.BindFlags(cmd)
		handleLaw()
	},
}

func defFlagsLaw(cmd *cobra.Command) {
	cmd.Flags().String("cookie", "", "*Specify the cookie")
}

var lawClient = resty.New().SetRetryCount(5).SetRetryWaitTime(10 * time.Second)

func handleLaw() {
	resp, err := lawClient.R().
		SetHeaders(map[string]string{
			"Content-type": "application/x-www-form-urlencoded; charset=UTF-8",
			"Cookie":       cast.ToString(configkit.Get("cookie", "")),
		}).SetFormData(map[string]string{
		"pid": "0",
	}).Post("http://lawplatform.unicom.local/law/tpubTreeMenu/getMenuByRole.do")
	if err != nil || resp.IsError() || gjson.Get(resp.String(), "ok").Bool() != true {
		panic(exception.New("获取菜单失败"))
	}
	data := gjson.Get(resp.String(), "data.menuList").Array()
	for _, v := range data {
		for _, vv := range gjson.Get(v.String(), "chirlds").Array() {
			for _, vvv := range gjson.Get(vv.String(), "chirlds").Array() {
				rawURL := gjson.Get(vvv.String(), "tpubTreeMenuUrl").String()
				resp, err = lawClient.R().
					SetHeaders(map[string]string{
						"Content-type": "application/x-www-form-urlencoded; charset=UTF-8",
						"Cookie":       cast.ToString(configkit.Get("cookie", "")),
					}).SetFormData(map[string]string{
					"page":     "1",
					"pageSize": "1000",
				}).Post(getUrl(rawURL))
				if err != nil || resp.IsError() || resp.StatusCode() != 200 {
					panic(exception.New("获取文章列表失败"))
				}
				articleData := gjson.Get(resp.String(), "data").Array()
				for _, a := range articleData {
					infoId := gjson.Get(a.String(), "infoId").String()
					creatorTime := gjson.Get(a.String(), "creatorTime").String()
					//TODO 以后每年修改这个
					if !strings.Contains(creatorTime, "2024") {
						logkit.Info("非2024文章，跳过...")
						continue
					}
					resp, err = lawClient.R().
						SetHeaders(map[string]string{
							"Content-type": "application/x-www-form-urlencoded; charset=UTF-8",
							"Cookie":       cast.ToString(configkit.Get("cookie", "")),
						}).SetFormData(map[string]string{
						"infoId": infoId,
					}).Post("http://lawplatform.unicom.local/law/tpfLike/toLike.do")
					if err != nil || resp.IsError() || resp.StatusCode() != 200 {
						panic(exception.New("点赞失败"))
					} else {
						time.Sleep(1 * time.Second)
						logkit.Info("点赞成功~")
					}
				}
			}
		}

	}
}

func getUrl(rawURL string) string {
	// 解析 URL
	u, err := url.Parse(rawURL)
	if err != nil {
		panic(exception.New("解析URL失败"))
	}
	// 提取查询参数
	queryParams := u.Query()
	// 获取 pftype 的值
	pfType := queryParams.Get("pftype")
	uri := "http://lawplatform.unicom.local/law/tpfInfoQuery/findPfInfoList.do?pftype=" + pfType + "&pfsortLevel=3&cache_id=" + cast.ToString(time.Now().UnixMilli())
	return uri
}