login.go

package baidupcscmd

import (
	"fmt"
	"github.com/iikira/Baidu-Login"
	"github.com/peterh/liner"
)

// RunLogin 登录百度帐号
func RunLogin() (bduss, ptoken, stoken string, err error) {
	line := liner.NewLiner()
	line.SetCtrlCAborts(true)
	defer line.Close()

	bc := baidulogin.NewBaiduClinet()

	username, err := line.Prompt("请输入百度用户名(手机号/邮箱/用户名), 回车键提交 > ")
	if err != nil {
		return
	}

	password, err := line.PasswordPrompt("请输入密码(输入的密码无回显, 确认输入完成, 回车提交即可) > ")
	if err != nil {
		return
	}

	var vcode, vcodestr string
for_1:
	for i := 0; i < 10; i++ {
		lj := bc.BaiduLogin(username, password, vcode, vcodestr)

		switch lj.ErrInfo.No {
		case "0": // 登录成功, 退出循环
			return lj.Data.BDUSS, lj.Data.PToken, lj.Data.SToken, nil
		case "400023", "400101": // 需要验证手机或邮箱
			fmt.Printf("\n需要验证手机或邮箱才能登录\n选择一种验证方式\n")
			fmt.Printf("1: 手机: %s\n", lj.Data.Phone)
			fmt.Printf("2: 邮箱: %s\n", lj.Data.Email)
			fmt.Printf("\n")

			var verifyType string
			for et := 0; et < 3; et++ {
				verifyType, err = line.Prompt("请输入验证方式 (1 或 2) > ")
				if err != nil {
					return
				}

				switch verifyType {
				case "1":
					verifyType = "mobile"
				case "2":
					verifyType = "email"
				default:
					fmt.Printf("[%d/3] 验证方式不合法\n", et+1)
					continue
				}
				break
			}
			if verifyType != "mobile" && verifyType != "email" {
				err = fmt.Errorf("验证方式不合法")
				return
			}

			msg := bc.SendCodeToUser(verifyType, lj.Data.Token) // 发送验证码
			fmt.Printf("消息: %s\n\n", msg)

			for et := 0; et < 5; et++ {
				vcode, err := line.Prompt("请输入接收到的验证码 > ")
				if err != nil {
					return "", "", "", err
				}

				nlj := bc.VerifyCode(verifyType, lj.Data.Token, vcode, lj.Data.U)
				if nlj.ErrInfo.No != "0" {
					fmt.Printf("[%d/5] 错误消息: %s\n\n", et+1, nlj.ErrInfo.Msg)
					continue
				}
				// 登录成功
				return nlj.Data.BDUSS, nlj.Data.PToken, nlj.Data.SToken, nil
			}
			break for_1
		case "500001", "500002": // 验证码
			fmt.Printf("\n%s\n", lj.ErrInfo.Msg)
			vcodestr = lj.Data.CodeString
			if vcodestr == "" {
				err = fmt.Errorf("未找到codeString")
				return
			}
			verifyImgURL := "https://wappass.baidu.com/cgi-bin/genimage?" + vcodestr
			fmt.Printf("打开以下的网址, 以查看验证码\n")
			fmt.Printf("%s\n\n", verifyImgURL)

			vcode, err = line.Prompt("请输入验证码 > ")
			if err != nil {
				return
			}
			continue
		default:
			err = fmt.Errorf("错误代码: %s, 消息: %s", lj.ErrInfo.No, lj.ErrInfo.Msg)
			return
		}
	}
	return
}