package codeReceiver

import (
	"io"
	"net/http"
)

type CodeReceiver struct {
	Host     string `json:"host"`
	UserName string `json:"username"`
	Password string `json:"password"`
	Sid      string `json:"sid" name:"项目id"`
	Author   string `json:"kid" name:"作者"`
	Token    string `json:"token" name:"登录凭证"`
}

var client = &http.Client{}

func (c *CodeReceiver) Login() (err error)        { return }
func (c *CodeReceiver) GetMobile() (phone string) { return }

func (c *CodeReceiver) get(inter string, params map[string]string) (res *http.Response, err error) {
	inter = splicingUrl(inter, params)
	req, err := http.NewRequest("GET", inter, nil)
	if err != nil {
		return
	}
	for i := 0; i < RetryTimes; i++ {
		res, err = client.Do(req)
		if err != nil {
			continue
		}
		return res, nil
	}
	return
}

func splicingUrl(inter string, params map[string]string) string {
	// 拼接 inter 和 params
	inter = inter + "?"
	for k, v := range params {
		inter = inter + k + "=" + v + "&"
	}
	// 去除最后一个 &
	inter = inter[:len(inter)-1]
	return inter
}
func closeBody(body io.ReadCloser) {
	err := body.Close()
	if err != nil {
	}
}

package codeReceiver

import (
	"errors"
	"github.com/tidwall/gjson"
	"io"
	"github.com/sirupsen/logrus"
)
const (
	RetryTimes = 3
)

var (
	logger = logrus.New()
)
type Yezi struct {
	CodeReceiver
}

func NewYezi(userName, passWord, sid string) (y *Yezi) {
	return &Yezi{
		CodeReceiver: CodeReceiver{
			Host:     "http://api.sqhyw.net:90",
			Author:   "795226",
			Sid:      sid,
			UserName: userName,
			Password: passWord,
		},
	}

}

func (y *Yezi) Login() (err error) {
	inter := y.Host + "/api/logins"
	params := map[string]string{
		"username": y.UserName,
		"password": y.Password,
	}
	res, err := y.get(inter, params)
	if err != nil {
		return
	}
	defer closeBody(res.Body)
	body, err := io.ReadAll(res.Body)
	if err != nil {
		return
	}
	y.Token = gjson.GetBytes(body, "token").String()
	message := gjson.GetBytes(body, "message").String()
	if y.Token == "" {
		err = errors.New(message)
		return
	}
	return
}

func (y *Yezi) GetMobile() (mobile string, err error) {
	inter := y.Host + "/api/get_mobile"
	params := map[string]string{
		"token":      y.Token,
		"project_id": y.Sid,
		"api_id":     y.Author,
	}
	res, err := y.get(inter, params)
	if err != nil {
		return
	}
	defer closeBody(res.Body)
	body, err := io.ReadAll(res.Body)
	mobile = gjson.GetBytes(body, "mobile").String()
	message := gjson.GetBytes(body, "message").String()
	if mobile == "" {
		return "", errors.New(message)
	}
	return
}
func (y *Yezi) GetSpecificMobile(mobile string) (err error) {
	inter := y.Host + "/api/get_mobile"
	params := map[string]string{
		"token":      y.Token,
		"project_id": y.Sid,
		"api_id":     y.Author,
		"phone_num":  mobile,
	}
	res, err := y.get(inter, params)
	if err != nil {
		return
	}
	defer closeBody(res.Body)
	body, err := io.ReadAll(res.Body)
	mobile = gjson.GetBytes(body, "mobile").String()
	message := gjson.GetBytes(body, "message").String()
	if mobile == "" {
		return errors.New(message)
	}
	return
}

func (y *Yezi) GetMessage(mobile string) (code string, err error) {
	inter := y.Host + "/api/get_message"
	params := map[string]string{
		"token":      y.Token,
		"project_id": y.Sid,
		"phone_num":  mobile,
	}
	res, err := y.get(inter, params)
	if err != nil {
		return
	}
	defer closeBody(res.Body)
	body, err := io.ReadAll(res.Body)
	if err != nil {
		return
	}
	code = gjson.GetBytes(body, "code").String()
	message := gjson.GetBytes(body, "message").String()
	if code == "" && message != "ok" {
		return "", errors.New(message)
	} else if code == "" {
		return "", errors.New("no message")
	}
	return
}

func (y *Yezi) AddBlackList(mobile string) (err error) {
	inter := y.Host + "/api/add_blacklist"
	params := map[string]string{
		"token":      y.Token,
		"project_id": y.Sid,
		"phone_num":  mobile,
	}
	res, err := y.get(inter, params)
	if err != nil {
		return
	}
	defer closeBody(res.Body)
	body, err := io.ReadAll(res.Body)
	message := gjson.GetBytes(body, "message").String()
	if message != "ok" {
		return errors.New(message)
	}
	return
}