128 lines
3.0 KiB
Go
128 lines
3.0 KiB
Go
package models
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/beego/beego/v2/client/httplib"
|
||
"regexp"
|
||
"time"
|
||
)
|
||
|
||
var codes map[string]chan string
|
||
|
||
type Query struct {
|
||
Ck struct {
|
||
PtPin interface{} `json:"ptPin"`
|
||
PtKey interface{} `json:"ptKey"`
|
||
Empty bool `json:"empty"`
|
||
} `json:"ck"`
|
||
PageStatus string `json:"pageStatus"`
|
||
AuthCodeCountDown int `json:"authCodeCountDown"`
|
||
CanClickLogin bool `json:"canClickLogin"`
|
||
CanSendAuth bool `json:"canSendAuth"`
|
||
SessionTimeOut int `json:"sessionTimeOut"`
|
||
AvailChrome int `json:"availChrome"`
|
||
}
|
||
|
||
type Session struct {
|
||
Value string
|
||
}
|
||
|
||
func (sess *Session) create() error {
|
||
var url = "https://github.com/rubyangxg/jd-qinglong"
|
||
if Config.SMSAddress == "" {
|
||
return errors.New("未配置服务器地址，仓库地址：" + url)
|
||
}
|
||
html, _ := httplib.Get(Config.SMSAddress).String()
|
||
res := regexp.MustCompile(`value="([\d\w]+)"`).FindStringSubmatch(html)
|
||
if len(res) == 0 {
|
||
return errors.New("崩了请找作者，仓库地址：https://github.com/rubyangxg/jd-qinglong")
|
||
}
|
||
sess.Value = res[1]
|
||
return nil
|
||
}
|
||
|
||
func (sess *Session) control(name, value string) error {
|
||
req := httplib.Post(Config.SMSAddress + "/control")
|
||
req.Param("currId", name)
|
||
req.Param("currValue", value)
|
||
req.Param("clientSessionId", sess.String())
|
||
_, err := req.String()
|
||
// fmt.Println("controll", name, value, rt)
|
||
return err
|
||
}
|
||
|
||
func (sess *Session) login(phone, sms_code string) error {
|
||
req := httplib.Post(Config.SMSAddress + "/jdLogin")
|
||
req.Param("phone", phone)
|
||
req.Param("sms_code", sms_code)
|
||
req.Param("clientSessionId", sess.String())
|
||
_, err := req.String()
|
||
// fmt.Println(phone, sms_code, rt)
|
||
return err
|
||
}
|
||
|
||
func (sess *Session) sendAuthCode() error {
|
||
req := httplib.Get(Config.SMSAddress + "/sendAuthCode?clientSessionId=" + sess.String())
|
||
_, err := req.Response()
|
||
return err
|
||
}
|
||
|
||
func (sess *Session) String() string {
|
||
return sess.Value
|
||
}
|
||
|
||
func (sess *Session) query() (*Query, error) {
|
||
query := &Query{}
|
||
// fmt.Println(sess.String(), "+++")
|
||
data, err := httplib.Get(fmt.Sprintf("%s/getScreen?clientSessionId=%s", Config.SMSAddress, sess.String())).Bytes()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
// fmt.Println(string(data))
|
||
err = json.Unmarshal(data, &query)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return query, nil
|
||
}
|
||
|
||
func (sess *Session) Phone(phone string) error {
|
||
err := sess.create()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for {
|
||
query, err := sess.query()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if query.PageStatus == "NORMAL" {
|
||
break
|
||
}
|
||
if query.PageStatus == "SESSION_EXPIRED" {
|
||
return sess.Phone(phone)
|
||
}
|
||
time.Sleep(time.Second)
|
||
}
|
||
err = sess.control("phone", phone)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (sess *Session) SmsCode(sms_code string) error {
|
||
err := sess.control("sms_code", sms_code)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (sess *Session) crackCaptcha() error {
|
||
_, err := httplib.Get(fmt.Sprintf("%s/crackCaptcha?clientSessionId=%s", Config.SMSAddress, sess.String())).Response()
|
||
return err
|
||
}
|