package lib

import (
	"encoding/json"
	"os"

	"github.com/guonaihong/gout"
	log "github.com/sirupsen/logrus"
	"github.com/tidwall/gjson"
)

func init() {
	_, err := os.Stat(`./config/user.json`)
	if err != nil {
		err := os.WriteFile(user_path, []byte("[]"), 0666)
		if err != nil {
			return
		}
		return
	}
}

const (
	user_path = "./config/user.json"
)

type User struct {
	Cookies []Cookie `json:"cookies"`
	Nick    string   `json:"nick"`
	Uid     string   `json:"uid"`
	Time    int64    `json:"time"`
}

// GetUsers
/**
 * @Description: 获取所有cookie有效的用户
 * @return []User
 * @return error
 */
func GetUsers() ([]User, error) {
	file, err := os.ReadFile(user_path)
	if err != nil {
		return nil, err
	}
	var users []User
	err = json.Unmarshal(file, &users)
	if err != nil {
		return nil, err
	}
	var newUsers []User
	for i := 0; i < len(users); i++ {
		if CheckUserCookie(users[i]) {
			newUsers = append(newUsers, users[i])
			continue
		}
		log.Infoln("用户" + users[i].Nick + "cookie已失效")
	}
	return newUsers, err
}

// SaveUser
/**
 * @Description: 保存用户信息到user.json
 * @param user
 * @return error
 */
func SaveUser(user User) error {
	users, err := GetUsers()
	if err != nil {
		log.Errorln("获取用户信息错误")
		return err
	}
	a := false
	for _, u := range users {
		if u.Uid == user.Uid {
			u.Cookies = user.Cookies
			a = true
		}
	}
	if !a {
		users = append(users, user)
	}

	data, err := json.Marshal(&users)
	if err != nil {
		log.Errorln("序列化用户失败")
		return err
	}
	err = os.WriteFile(user_path, data, 0666)
	if err != nil {
		log.Errorln("写入用户信息到文件错误")

		return err
	}
	return err
}

// GetUserInfo
/**
 * @Description: 获取用户信息
 * @param cookies
 * @return string
 * @return string
 * @return error
 */
func GetUserInfo(cookies []Cookie) (string, string, error) {
	var resp []byte
	err := gout.GET(user_Info_url).
		SetCookies(cookieToJar(cookies)...).
		SetHeader(gout.H{
			"Cache-Control": "no-cache",
		}).BindBody(&resp).Do()
	if err != nil {
		log.Errorln("获取用户信息失败")

		return "", "", err
	}
	log.Debugln("[user] 用户信息：", gjson.GetBytes(resp, "@this|@pretty").String())
	uid := gjson.GetBytes(resp, "data.uid").String()
	nick := gjson.GetBytes(resp, "data.nick").String()

	return uid, nick, err
}

// CheckUserCookie
/**
 * @Description: 获取用户成绩
 * @param user
 * @return bool
 */
func CheckUserCookie(user User) bool {
	_, err := GetUserScore(user.Cookies)
	if err != nil && err.Error() == "token check failed" {
		return false
	}
	return true
}