go_seckill/cmd/jdTdudfp.go

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

	}
}