147 wiersze
4.9 KiB
147 wiersze
4.9 KiB
package jd_seckill
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/Albert-Zhan/httpc"
|
|
"github.com/tidwall/gjson"
|
|
"github.com/unknwon/goconfig"
|
|
"github.com/ztino/jd_seckill/common"
|
|
"github.com/ztino/jd_seckill/log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type User struct {
|
|
client *httpc.HttpClient
|
|
conf *goconfig.ConfigFile
|
|
}
|
|
|
|
func NewUser(client *httpc.HttpClient,conf *goconfig.ConfigFile) *User {
|
|
return &User{client: client,conf:conf }
|
|
}
|
|
|
|
func (this *User) getUserAgent() string {
|
|
return this.conf.MustValue("config","default_user_agent","")
|
|
}
|
|
|
|
func (this *User) loginPage() {
|
|
req:=httpc.NewRequest(this.client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
req.SetHeader("Connection","keep-alive")
|
|
req.SetHeader("Accept","text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3")
|
|
_,_,_=req.SetUrl("https://passport.jd.com/new/login.aspx").SetMethod("get").Send().End()
|
|
}
|
|
|
|
func (this *User) QrLogin() (string,error) {
|
|
//登录页面
|
|
this.loginPage()
|
|
//二维码登录
|
|
req:=httpc.NewRequest(this.client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
req.SetHeader("Referer","https://passport.jd.com/new/login.aspx")
|
|
resp,err:=req.SetUrl("https://qr.m.jd.com/show?appid=133&size=300&t="+strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get").Send().EndFile("./","qr_code.png")
|
|
if err!=nil || resp.StatusCode!=http.StatusOK {
|
|
log.Println("获取二维码失败")
|
|
return "",errors.New("获取二维码失败")
|
|
}
|
|
cookies:=resp.Cookies()
|
|
wlfstkSmdl:=""
|
|
for _,cookie:= range cookies {
|
|
if cookie.Name=="wlfstk_smdl" {
|
|
wlfstkSmdl=cookie.Value
|
|
break
|
|
}
|
|
}
|
|
log.Println("二维码获取成功,请打开京东APP扫描")
|
|
dir,_:=os.Getwd()
|
|
qrPath := filepath.Join(dir, `./qr_code.png`)
|
|
common.OpenImage(qrPath)
|
|
return wlfstkSmdl,nil
|
|
}
|
|
|
|
func (this *User) QrcodeTicket(wlfstkSmdl string) (string,error) {
|
|
req:=httpc.NewRequest(this.client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
req.SetHeader("Referer","https://passport.jd.com/new/login.aspx")
|
|
resp,body,err:=req.SetUrl("https://qr.m.jd.com/check?appid=133&callback=jQuery"+strconv.Itoa(common.Rand(1000000,9999999))+"&token="+wlfstkSmdl+"&_="+strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get").Send().End()
|
|
if err!=nil || resp.StatusCode!=http.StatusOK {
|
|
log.Println("获取二维码扫描结果异常")
|
|
return "",errors.New("获取二维码扫描结果异常")
|
|
}
|
|
if gjson.Get(body,"code").Int()!=200 {
|
|
log.Printf("Code: %s, Message: %s",gjson.Get(body,"code").String(),gjson.Get(body,"msg").String())
|
|
return "",errors.New(fmt.Sprintf("Code: %s, Message: %s",gjson.Get(body,"code").String(),gjson.Get(body,"msg").String()))
|
|
}
|
|
log.Println("已完成手机客户端确认")
|
|
return gjson.Get(body,"ticket").String(),nil
|
|
}
|
|
|
|
func (this *User) TicketInfo(ticket string) (string,error) {
|
|
req:=httpc.NewRequest(this.client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
req.SetHeader("Referer","https://passport.jd.com/uc/login?ltype=logout")
|
|
resp,body,err:=req.SetUrl("https://passport.jd.com/uc/qrCodeTicketValidation?t="+ticket).SetMethod("get").Send().End()
|
|
defer this.DelQrCode()
|
|
if err!=nil || resp.StatusCode!=http.StatusOK {
|
|
log.Println("二维码信息校验失败")
|
|
return "",errors.New("二维码信息校验失败")
|
|
}
|
|
if gjson.Get(body,"returnCode").Int()==0 {
|
|
log.Println("二维码信息校验成功")
|
|
return "",nil
|
|
}else{
|
|
log.Println("二维码信息校验失败")
|
|
return "",errors.New("二维码信息校验失败")
|
|
}
|
|
}
|
|
|
|
func (this *User) RefreshStatus() error {
|
|
client:=httpc.NewHttpClient()
|
|
client.SetCookieJar(common.CookieJar)
|
|
client.SetRedirect(func(req *http.Request, via []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
})
|
|
req:=httpc.NewRequest(client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
resp,_,err:=req.SetUrl("https://order.jd.com/center/list.action?rid="+strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get").Send().End()
|
|
if err==nil && resp.StatusCode==http.StatusOK {
|
|
return nil
|
|
}else{
|
|
return errors.New("登录失效")
|
|
}
|
|
}
|
|
|
|
func (this *User) GetUserInfo() (string,error) {
|
|
req:=httpc.NewRequest(this.client)
|
|
req.SetHeader("User-Agent",this.getUserAgent())
|
|
req.SetHeader("Referer","https://order.jd.com/center/list.action")
|
|
errorCount:=5
|
|
nickName:=""
|
|
for {
|
|
if errorCount>0 {
|
|
_,body,_:=req.SetUrl("https://passport.jd.com/user/petName/getUserInfoForMiniJd.action?callback="+strconv.Itoa(common.Rand(1000000,9999999))+"&_="+strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get").Send().End()
|
|
if gjson.Get(body,"nickName").Exists() {
|
|
nickName=gjson.Get(body,"nickName").String()
|
|
break
|
|
}
|
|
errorCount=errorCount-1
|
|
time.Sleep(300*time.Millisecond)
|
|
}else{
|
|
break
|
|
}
|
|
}
|
|
b,_:=common.GbkToUtf8([]byte(nickName))
|
|
return string(b), nil
|
|
}
|
|
|
|
func (this *User) DelQrCode() {
|
|
dir, _ := os.Getwd()
|
|
qrPath := filepath.Join(dir, `./qr_code.png`)
|
|
if common.Exists(qrPath) {
|
|
_=os.Remove(qrPath)
|
|
}
|
|
}
|
|
|