go_seckill/cmd/jdTdudfp.go

package cmd

import (
	"context"
	"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/ztino/jd_seckill/common"
	"github.com/ztino/jd_seckill/jd_seckill"
	"log"
	"net/url"
	"time"
)

func init() {
	rootCmd.AddCommand(jdTdudfpCmd)
}

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),
			//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","")),
		}
		options = append(chromedp.DefaultExecAllocatorOptions[:], options...)

		c, cc := chromedp.NewExecAllocator(context.Background(), options...)
		defer cc()

		ctx, cancel := chromedp.NewContext(c)
		ch := addNewTabListener(ctx)
		defer cancel()

		u, _ := url.Parse("http://jd.com")
		cookies := common.CookieJar.Cookies(u)
		err = chromedp.Run(ctx,
			chromedp.Tasks{
				chromedp.ActionFunc(func(ctx context.Context) error {
					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("https://jd.com"),
				chromedp.Sleep(2*time.Second),
				chromedp.Click(".cate_menu_lk"),
				chromedp.Sleep(2*time.Second),
			},
		)
		if err != nil {
			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),
		)
		if err != nil {
			log.Fatal(err)
		}

		newCtx2, cancel3 := chromedp.NewContext(ctx, chromedp.WithTargetID(<-ch2))
		defer cancel3()

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

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

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