You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.8 KiB
124 lines
3.8 KiB
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"github.com/chromedp/cdproto/cdp"
|
|
"github.com/chromedp/cdproto/network"
|
|
"github.com/chromedp/chromedp"
|
|
"github.com/spf13/cobra"
|
|
"github.com/tidwall/gjson"
|
|
"github.com/unknwon/goconfig"
|
|
"github.com/ztino/jd_seckill/common"
|
|
"github.com/ztino/jd_seckill/jd_seckill"
|
|
"github.com/ztino/jd_seckill/log"
|
|
"net/url"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(jdTdudfpCmd)
|
|
jdTdudfpCmd.Flags().StringP("good_url","g","","")
|
|
_=jdTdudfpCmd.MarkFlagRequired("good_url")
|
|
}
|
|
|
|
var jdTdudfpCmd = &cobra.Command{
|
|
Use: "jdTdudfp",
|
|
Short: "auto get jd eid and fp",
|
|
Run: startJdTdudfp,
|
|
}
|
|
|
|
func startJdTdudfp(cmd *cobra.Command, args []string) {
|
|
session := jd_seckill.NewSession(common.CookieJar)
|
|
err := session.CheckLoginStatus()
|
|
if err != nil {
|
|
log.Println("自动获取eid和fp失败，请重新登录")
|
|
} else {
|
|
log.Println("开始自动获取eid和fp，如遇卡住请结束进程，重新启动")
|
|
options := []chromedp.ExecAllocatorOption{
|
|
chromedp.Flag("headless", false), //debug使用
|
|
chromedp.Flag("blink-settings", "imagesEnabled=false"), //禁用图片加载
|
|
chromedp.Flag("start-maximized", true), //最大化窗口
|
|
chromedp.Flag("no-sandbox", true), //禁用沙盒, 性能优先
|
|
chromedp.Flag("disable-setuid-sandbox", true), //禁用setuid沙盒, 性能优先
|
|
chromedp.Flag("no-default-browser-check", true), //不检查默认浏览器
|
|
chromedp.Flag("disable-plugins", true), //禁用扩展
|
|
chromedp.UserAgent(common.Config.MustValue("config", "default_user_agent", "")),
|
|
}
|
|
options = append(chromedp.DefaultExecAllocatorOptions[:], options...)
|
|
|
|
c, cc := chromedp.NewExecAllocator(context.Background(), options...)
|
|
defer cc()
|
|
|
|
ctx, cancel := chromedp.NewContext(c)
|
|
defer cancel()
|
|
|
|
//获取cookie
|
|
u, _ := url.Parse("http://jd.com")
|
|
cookies := common.CookieJar.Cookies(u)
|
|
|
|
//商品链接
|
|
good_url,_:=cmd.Flags().GetString("good_url")
|
|
|
|
var res []byte
|
|
err = chromedp.Run(ctx,
|
|
chromedp.Tasks{
|
|
chromedp.ActionFunc(func(ctx context.Context) error {
|
|
//设置cookie
|
|
expr := cdp.TimeSinceEpoch(time.Now().Add(180 * 24 * time.Hour))
|
|
for _, cookie := range cookies {
|
|
network.SetCookie(cookie.Name, cookie.Value).
|
|
WithExpires(&expr).
|
|
WithPath("/").
|
|
WithDomain("." + cookie.Domain).
|
|
Do(ctx)
|
|
}
|
|
return nil
|
|
}),
|
|
},
|
|
chromedp.Navigate(good_url),
|
|
chromedp.WaitVisible("#InitCartUrl"), //加入购物车
|
|
chromedp.Sleep(2 * time.Second),
|
|
chromedp.Click("#InitCartUrl"),
|
|
chromedp.WaitVisible(".btn-addtocart"), //去购车结算
|
|
chromedp.Sleep(2 * time.Second),
|
|
chromedp.Click(".btn-addtocart"),
|
|
chromedp.WaitVisible(".common-submit-btn"), //去结算
|
|
chromedp.Sleep(2 * time.Second),
|
|
chromedp.Click(".common-submit-btn"),
|
|
chromedp.Sleep(3 * time.Second),
|
|
chromedp.Evaluate("_JdTdudfp", &res),
|
|
)
|
|
if err != nil {
|
|
log.Println("chromedp 出错了")
|
|
log.Fatal(err)
|
|
}
|
|
|
|
value := string(res)
|
|
if !gjson.Valid(value) || gjson.Get(value, "eid").String() == "" || gjson.Get(value, "fp").String() == "" {
|
|
log.Println("获取失败，请重新尝试，返回信息:" + value)
|
|
} else {
|
|
eid := gjson.Get(value, "eid").String()
|
|
fp := gjson.Get(value, "fp").String()
|
|
log.Println("eid:" + eid)
|
|
log.Println("fp:" + fp)
|
|
|
|
//修改配置文件
|
|
confFile := "./conf.ini"
|
|
cfg, err := goconfig.LoadConfigFile(confFile)
|
|
if err != nil {
|
|
log.Println("配置文件不存在，程序退出")
|
|
os.Exit(0)
|
|
}
|
|
|
|
cfg.SetValue("config", "eid", eid)
|
|
cfg.SetValue("config", "fp", fp)
|
|
if err := goconfig.SaveConfigFile(cfg, confFile); err != nil {
|
|
log.Println("保存配置文件失败，请手动填入配置文件")
|
|
}
|
|
|
|
log.Println("eid, fp参数已经自动填入配置文件")
|
|
}
|
|
|
|
}
|
|
}
|
|
|