go_seckill/jd_seckill/seckill.go

package jd_seckill

import (
	"errors"
	"fmt"
	"github.com/Albert-Zhan/httpc"
	"github.com/PuerkitoBio/goquery"
	"github.com/tidwall/gjson"
	"github.com/unknwon/goconfig"
	"github.com/ztino/jd_seckill/common"
	"github.com/ztino/jd_seckill/service"
	"github.com/ztino/jd_seckill/log"
	"net/http"
	"strconv"
	"strings"
	"time"
)

type Seckill struct {
	client *httpc.HttpClient
	conf   *goconfig.ConfigFile
}

func NewSeckill(client *httpc.HttpClient, conf *goconfig.ConfigFile) *Seckill {
	return &Seckill{client: client, conf: conf}
}

func (this *Seckill) getUserAgent() string {
	return this.conf.MustValue("config", "default_user_agent", "")
}

func (this *Seckill) SkuTitle() (string, error) {
	skuId := this.conf.MustValue("config", "sku_id", "")
	req := httpc.NewRequest(this.client)
	resp, body, err := req.SetUrl(fmt.Sprintf("https://item.jd.com/%s.html", skuId)).SetMethod("get").Send().End()
	if err != nil || resp.StatusCode != http.StatusOK {
		log.Println("访问商品详情失败")
		return "", errors.New("访问商品详情失败")
	}
	html := strings.NewReader(body)
	doc, _ := goquery.NewDocumentFromReader(html)
	return strings.TrimSpace(doc.Find(".sku-name").Text()), nil
}

func (this *Seckill) MakeReserve() {
	user := NewUser(this.client, this.conf)
	userInfo, _ := user.GetUserInfo()
	log.Println("用户:" + userInfo)
	shopTitle, err := this.SkuTitle()
	if err != nil {
		log.Println("获取商品信息失败")
	} else {
		log.Println("商品名称:" + shopTitle)
	}
	skuId := this.conf.MustValue("config", "sku_id", "")
	req := httpc.NewRequest(this.client)
	req.SetHeader("User-Agent", this.getUserAgent())
	req.SetHeader("Referer", fmt.Sprintf("https://item.jd.com/%s.html", skuId))
	resp, body, err := req.SetUrl("https://yushou.jd.com/youshouinfo.action?callback=fetchJSON&sku=" + skuId + "&_=" + strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get").Send().End()
	if err != nil || resp.StatusCode != http.StatusOK {
		log.Println("预约商品失败")
	} else {
		reserveUrl := gjson.Get(body, "url").String()
		req = httpc.NewRequest(this.client)
		_, _, _ = req.SetUrl("https:" + reserveUrl).SetMethod("get").Send().End()
		msg := "预约成功，已获得抢购资格 / 您已成功预约过了，无需重复预约！[我的预约](https://yushou.jd.com/member/qualificationList.action)"
		_ = service.SendMessage(this.conf, "茅台抢购通知", msg)
		log.Println(msg)
	}
}

func (this *Seckill) getSeckillUrl() (string, error) {
	skuId := this.conf.MustValue("config", "sku_id", "")
	req := httpc.NewRequest(this.client)
	req.SetHeader("User-Agent", this.getUserAgent())
	req.SetHeader("Host", "itemko.jd.com")
	req.SetHeader("Referer", fmt.Sprintf("https://item.jd.com/%s.html", skuId))
	req.SetUrl("https://itemko.jd.com/itemShowBtn?callback=jQuery" + strconv.Itoa(common.Rand(1000000, 9999999)) + "&skuId=" + skuId + "&from=pc&_=" + strconv.Itoa(int(time.Now().Unix()*1000))).SetMethod("get")
	url := ""
	for {
		_, body, _ := req.Send().End()
		if gjson.Get(body, "url").Exists() && gjson.Get(body, "url").String() != "" {
			url = gjson.Get(body, "url").String()
			break
		}
		log.Println("抢购链接获取失败，稍后自动重试")
		time.Sleep(300 * time.Millisecond)
	}
	url = "https:"+url
	//https://divide.jd.com/user_routing?skuId=8654289&sn=c3f4ececd8461f0e4d7267e96a91e0e0&from=pc
	url = strings.ReplaceAll(url, "divide", "marathon")
	//https://marathon.jd.com/captcha.html?skuId=8654289&sn=c3f4ececd8461f0e4d7267e96a91e0e0&from=pc
	url = strings.ReplaceAll(url, "user_routing", "captcha.html")
	log.Println("抢购链接获取成功:" + url)
	return url, nil
}

func (this *Seckill) RequestSeckillUrl() {
	user := NewUser(this.client, this.conf)
	userInfo, _ := user.GetUserInfo()
	log.Println("用户:" + userInfo)
	shopTitle, err := this.SkuTitle()
	if err != nil {
		log.Println("获取商品信息失败")
	} else {
		log.Println("商品名称:" + shopTitle)
	}
	url, _ := this.getSeckillUrl()
	skuId := this.conf.MustValue("config", "sku_id", "")
	log.Println("访问商品的抢购连接...")
	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())
	req.SetHeader("Host", "marathon.jd.com")
	req.SetHeader("Referer", fmt.Sprintf("https://item.jd.com/%s.html", skuId))
	_, _, _ = req.SetUrl(url).SetMethod("get").Send().End()
}

func (this *Seckill) SeckillPage() {
	log.Println("访问抢购订单结算页面...")
	skuId := this.conf.MustValue("config", "sku_id", "")
	seckillNum := this.conf.MustValue("config", "seckill_num", "2")
	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())
	req.SetHeader("Host", "marathon.jd.com")
	req.SetHeader("Referer", fmt.Sprintf("https://item.jd.com/%s.html", skuId))
	_, _, _ = req.SetUrl("https://marathon.jd.com/seckill/seckill.action?skuId=" + skuId + "&num=" + seckillNum + "&rid=" + strconv.Itoa(int(time.Now().Unix()))).SetMethod("get").Send().End()
}

func (this *Seckill) SeckillInitInfo() (string, error) {
	log.Println("获取秒杀初始化信息...")
	skuId := this.conf.MustValue("config", "sku_id", "")
	seckillNum := this.conf.MustValue("config", "seckill_num", "2")
	req := httpc.NewRequest(this.client)
	req.SetHeader("User-Agent", this.getUserAgent())
	req.SetHeader("Host", "marathon.jd.com")
	req.SetData("sku", skuId)
	req.SetData("num", seckillNum)
	req.SetData("isModifyAddress", "false")
	req.SetUrl("https://marathon.jd.com/seckillnew/orderService/pc/init.action").SetMethod("post")
	//尝试获取三次
	errorCount:=3
	errorMsg:=""
	for errorCount > 0 {
		_, body, _ := req.Send().End()
		if body!="null" && gjson.Valid(body) {
			log.Println("获取秒杀初始化信息成功")
			return body,nil
		}else{
			log.Println("获取秒杀初始化信息失败,返回信息:"+body)
			errorMsg=body
		}
		errorCount=errorCount-1
		time.Sleep(300*time.Millisecond)
	}
	return "", errors.New(errorMsg)
}

func (this *Seckill) SubmitSeckillOrder() bool {
	eid := this.conf.MustValue("config", "eid", "")
	fp := this.conf.MustValue("config", "fp", "")
	skuId := this.conf.MustValue("config", "sku_id", "")
	seckillNum := this.conf.MustValue("config", "seckill_num", "2")
	paymentPwd := this.conf.MustValue("account", "payment_pwd", "")
	initInfo, err := this.SeckillInitInfo()
	if err != nil {
		log.Println(fmt.Sprintf("抢购失败，无法获取生成订单的基本信息，接口返回:【%s】", err.Error()))
		return false
	}
	address := gjson.Get(initInfo, "addressList").Array()
	if !gjson.Get(initInfo, "addressList").Exists() || len(address)<1 {
		log.Println("抢购失败，可能你还未设置默认收货地址")
		return false
	}
	defaultAddress := address[0]
	isinvoiceInfo := gjson.Get(initInfo, "invoiceInfo").Exists()
	invoiceTitle := "-1"
	invoiceContentType := "-1"
	invoicePhone := ""
	invoicePhoneKey := ""
	invoiceInfo := "false"
	if isinvoiceInfo {
		invoiceTitle = gjson.Get(initInfo, "invoiceInfo.invoiceTitle").String()
		invoiceContentType = gjson.Get(initInfo, "invoiceInfo.invoiceContentType").String()
		invoicePhone = gjson.Get(initInfo, "invoiceInfo.invoicePhone").String()
		invoicePhoneKey = gjson.Get(initInfo, "invoiceInfo.invoicePhoneKey").String()
		invoiceInfo = "true"
	}
	token := gjson.Get(initInfo, "token").String()
	log.Println("提交抢购订单...")
	req := httpc.NewRequest(this.client)
	req.SetHeader("User-Agent", this.getUserAgent())
	req.SetHeader("Host", "marathon.jd.com")
	req.SetHeader("Referer", fmt.Sprintf("https://marathon.jd.com/seckill/seckill.action?skuId=%s&num=%s&rid=%d", skuId, seckillNum, int(time.Now().Unix())))
	req.SetData("skuId", skuId)
	req.SetData("num", seckillNum)
	req.SetData("addressId", defaultAddress.Get("id").String())
	req.SetData("yuShou", "true")
	req.SetData("isModifyAddress", "false")
	req.SetData("name", defaultAddress.Get("name").String())
	req.SetData("provinceId", defaultAddress.Get("provinceId").String())
	req.SetData("cityId", defaultAddress.Get("cityId").String())
	req.SetData("countyId", defaultAddress.Get("countyId").String())
	req.SetData("townId", defaultAddress.Get("townId").String())
	req.SetData("addressDetail", defaultAddress.Get("addressDetail").String())
	req.SetData("mobile", defaultAddress.Get("mobile").String())
	req.SetData("mobileKey", defaultAddress.Get("mobileKey").String())
	req.SetData("email", defaultAddress.Get("email").String())
	req.SetData("postCode", "")
	req.SetData("invoiceTitle", invoiceTitle)
	req.SetData("invoiceCompanyName", "")
	req.SetData("invoiceContent", invoiceContentType)
	req.SetData("invoiceTaxpayerNO", "")
	req.SetData("invoiceEmail", "")
	req.SetData("invoicePhone", invoicePhone)
	req.SetData("invoicePhoneKey", invoicePhoneKey)
	req.SetData("invoice", invoiceInfo)
	req.SetData("password", paymentPwd)
	req.SetData("codTimeType", "3")
	req.SetData("paymentType", "4")
	req.SetData("areaCode", "")
	req.SetData("overseas", "0")
	req.SetData("phone", "")
	req.SetData("eid", eid)
	req.SetData("fp", fp)
	req.SetData("token", token)
	req.SetData("pru", "")
	resp, body, err := req.SetUrl("https://marathon.jd.com/seckillnew/orderService/pc/submitOrder.action?skuId=" + skuId).SetMethod("post").Send().End()
	if err != nil || resp.StatusCode != http.StatusOK {
		log.Println("抢购失败，网络错误")
		_ = service.SendMessage(this.conf, "茅台抢购通知", "抢购失败，网络错误")
		return false
	}

	//临时打印数据
	log.Println("返回信息:"+body)

	if !gjson.Valid(body) {
		log.Println("抢购失败，返回信息:" + body)
		_ = service.SendMessage(this.conf, "茅台抢购通知", "抢购失败，返回信息:"+body)
		return false
	}
	if gjson.Get(body, "success").Bool() {
		orderId := gjson.Get(body, "orderId").String()
		totalMoney := gjson.Get(body, "totalMoney").String()
		payUrl := "https:" + gjson.Get(body, "pcUrl").String()
		log.Println(fmt.Sprintf("抢购成功，订单号:%s, 总价:%s, 电脑端付款链接:%s", orderId, totalMoney, payUrl))
		_ = service.SendMessage(this.conf, "茅台抢购通知", fmt.Sprintf("抢购成功，订单号:%s, 总价:%s, 电脑端付款链接:%s", orderId, totalMoney, payUrl))
		return true
	} else {
		log.Println("抢购失败，返回信息:" + body)
		_ = service.SendMessage(this.conf, "茅台抢购通知", "抢购失败，返回信息:"+body)
		return false
	}
}