go_seckill/jd_seckill/user.go

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)
	}
}