go_seckill/cmd/login.go

package cmd

import (
	"github.com/spf13/cobra"
	"github.com/ztino/jd_seckill/common"
	"github.com/ztino/jd_seckill/jd_seckill"
	"log"
	"os"
	"time"
)

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

var loginCmd = &cobra.Command{
	Use:   "login",
	Short: "Open JD’s simulated login",
	Run: startLogin,
}

func startLogin(cmd *cobra.Command, args []string)  {
	session:=jd_seckill.NewSession(common.CookieJar)
	//检测是否登录过
	if common.Exists("./cookie.txt") {
		//已登录，检测登录状态
		err:=session.CheckLoginStatus()
		if err!=nil {
			log.Println("登录失效，请重新登录")
			return
		}
		user:=jd_seckill.NewUser(common.Client,common.Config)
		log.Println("登录成功")
		userInfo,_:=user.GetUserInfo()
		log.Println("用户:"+userInfo)
	}else{
		//未登录
		user:=jd_seckill.NewUser(common.Client,common.Config)
		wlfstkSmdl,err:=user.QrLogin()
		if err!=nil{
			os.Exit(0)
		}
		ticket:=""
		for  {
			ticket,err=user.QrcodeTicket(wlfstkSmdl)
			if err==nil && ticket!=""{
				break
			}
			time.Sleep(2*time.Second)
		}
		_,err=user.TicketInfo(ticket)
		if err==nil {
			if status:=user.RefreshStatus();status==nil {
				//保存cookie
				_=session.SaveCookieToFile("./cookie.txt")
				log.Println("登录成功")
				userInfo,_:=user.GetUserInfo()
				log.Println("用户:"+userInfo)
			}else{
				log.Println("登录失效")
			}
		}else{
			log.Println("登录失败")
		}
	}
}