优化自动获取eid和fp，并写入conf.ini

.gitignore

@@ -15,3 +15,8 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+/*.log
+/.idea
+/qr_code.png
+/cookie.txt
+*.bak

README.md

@@ -44,7 +44,8 @@ jd_seckill login
 
 ### 自动获取eid,fp
 
-> ⚠依赖谷歌浏览器，请安装谷歌浏览器，获取到的eid和fp请手动填入配置文件
+> ⚠依赖谷歌浏览器，请安装谷歌浏览器，windows下请将安全目录加入系统变量Path
+> ⚠【重要提醒】自动获取eid和fp期间，建议鼠标跟随页面跳转，滑动到【加入购物车】【去购车结算】【去结算】按钮，但不要点击，可以提升获取成功率！(经验之谈！)
 
 执行以下命令按照提示操作:
 ```shell

cmd/jdTdudfp.go

@@ -2,16 +2,20 @@ package cmd
 
 import (
 	"context"
+	"fmt"
 	"github.com/chromedp/cdproto/cdp"
 	"github.com/chromedp/cdproto/network"
 	"github.com/chromedp/cdproto/target"
 	"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"
 	"log"
 	"net/url"
+	"os"
+	"strconv"
 	"time"
 )
 
@@ -22,26 +26,26 @@ func init() {
 var jdTdudfpCmd = &cobra.Command{
 	Use:   "jdTdudfp",
 	Short: "auto get jd eid and fp",
-	Run: startJdTdudfp,
+	Run:   startJdTdudfp,
 }
 
-func startJdTdudfp(cmd *cobra.Command, args []string)  {
-	session:=jd_seckill.NewSession(common.CookieJar)
-	err:=session.CheckLoginStatus()
-	if err!=nil {
+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，如遇卡住请结束进程，重新启动")
+	} else {
+		retryTimes, _ := strconv.Atoi(common.Config.MustValue("config", "retry_times", "5"))
+
 		options := []chromedp.ExecAllocatorOption{
-			chromedp.Flag("headless", false),
-			//chromedp.Flag("blink-settings", "imagesEnabled=false"),
-			chromedp.Flag("start-maximized", true),
-			chromedp.Flag("no-sandbox",true),
-			chromedp.Flag("disable-setuid-sandbox",true),
-			chromedp.Flag("no-default-browser-check",true),
-			chromedp.Flag("disable-plugins",true),
-			chromedp.WindowSize(1920,1080),
-			chromedp.UserAgent(common.Config.MustValue("config","default_user_agent","")),
+			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...)
 
@@ -49,9 +53,10 @@ func startJdTdudfp(cmd *cobra.Command, args []string)  {
 		defer cc()
 
 		ctx, cancel := chromedp.NewContext(c)
-		ch := addNewTabListener(ctx)
+		_ = addNewTabListener(ctx)
 		defer cancel()
 
+		//设置cookie
 		u, _ := url.Parse("http://jd.com")
 		cookies := common.CookieJar.Cookies(u)
 		err = chromedp.Run(ctx,
@@ -67,61 +72,74 @@ func startJdTdudfp(cmd *cobra.Command, args []string)  {
 					}
 					return nil
 				}),
-				chromedp.Navigate("https://jd.com"),
-				chromedp.Sleep(2*time.Second),
-				chromedp.Click(".cate_menu_lk"),
-				chromedp.Sleep(2*time.Second),
 			},
 		)
 		if err != nil {
+			log.Println("设置cookie出错了")
 			log.Fatal(err)
 		}
 
-		newCtx, cancel2 := chromedp.NewContext(ctx, chromedp.WithTargetID(<-ch))
-		ch2 := addNewTabListener(newCtx)
-		defer cancel2()
-
-		err = chromedp.Run(newCtx,
-			chromedp.Click(".goods_item_link"),
-			chromedp.Sleep(4*time.Second),
+	RETRY:
+		retryTimes--
+		log.Println("【重要提醒】自动获取eid和fp期间，建议鼠标跟随页面跳转，滑动到【加入购物车】【去购车结算】【去结算】按钮，但不要点击，可以提升获取成功率！")
+		log.Println(fmt.Sprintf("开始自动获取eid和fp，如遇卡住请耐心等待，重试次数剩余： %v 次", retryTimes))
+		var res []byte
+		testSkuId := common.Config.MustValue("config", "test_sku_id", "")
+		err = chromedp.Run(ctx,
+			chromedp.Tasks{
+				chromedp.Navigate(fmt.Sprintf("http://item.jd.com/%s.html", testSkuId)),
+				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)
 		}
 
-		newCtx2, cancel3 := chromedp.NewContext(ctx, chromedp.WithTargetID(<-ch2))
-		defer cancel3()
+		value := string(res)
+		if !gjson.Valid(value) || gjson.Get(value, "eid").String() == "" || gjson.Get(value, "fp").String() == "" {
+			log.Println("获取失败，请重新尝试，返回信息:" + value)
+			if retryTimes > 0 {
+				goto RETRY
+			}
+		} else {
+			eid := gjson.Get(value, "eid").String()
+			fp := gjson.Get(value, "fp").String()
+			log.Println("eid:" + eid)
+			log.Println("fp:" + fp)
 
-		var res []byte
-		err = chromedp.Run(newCtx2,
-			chromedp.Click("#InitCartUrl"),
-			chromedp.Sleep(2*time.Second),
-			chromedp.Click(".btn-addtocart"),
-			chromedp.Sleep(2*time.Second),
-			chromedp.Click(".common-submit-btn"),
-			chromedp.Sleep(3*time.Second),
-			chromedp.Evaluate("_JdTdudfp", &res),
-		)
-		if err != nil {
-			log.Fatal(err)
-		}
+			//修改配置文件
+			confFile := "./conf.ini"
+			cfg, err := goconfig.LoadConfigFile(confFile)
+			if err != nil {
+				log.Println("配置文件不存在，程序退出")
+				os.Exit(0)
+			}
 
-		value:=string(res)
-		if !gjson.Valid(value) || gjson.Get(value,"eid").String()=="" || gjson.Get(value,"fp").String()=="" {
-			log.Println("获取失败，请重新尝试，返回信息:"+value)
-		}else{
-			log.Println("获取成功，请手动填入配置文件")
-			log.Println("eid:"+gjson.Get(value,"eid").String())
-			log.Println("fp:"+gjson.Get(value,"fp").String())
+			cfg.SetValue("config", "eid", eid)
+			cfg.SetValue("config", "fp", fp)
+			if err := goconfig.SaveConfigFile(cfg, confFile); err != nil {
+				log.Println("保存配置文件失败，请手动填入配置文件")
+			}
+
+			log.Println("eid, fp参数已经自动填入配置文件")
 		}
+
 	}
 }
 
 func addNewTabListener(ctx context.Context) <-chan target.ID {
-/*	mux := http.NewServeMux()
-	ts := httptest.NewServer(mux)
-	defer ts.Close()*/
 	return chromedp.WaitNewTarget(ctx, func(info *target.Info) bool {
 		return true
 	})
-}
+}

conf.ini

@@ -4,6 +4,9 @@
 # 随意填写可能导致订单无法提交等问题
 eid =
 fp =
+# 自动获取eid, fp的重试次数，测试商品sku_id
+retry_times = 5
+test_sku_id = 100016034372
 # 商品id
 # 已经是茅台的sku_id了
 sku_id = 100012043978