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

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参数已经自动填入配置文件")
}
}
}