shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-05 03:23:49 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1629 from shigma/login-ticket

Browse Source 
feat: support manually input ticket
This commit is contained in:
Mrs4s 2022-08-13 15:43:40 +08:00 committed by GitHub
commit a73d27f40b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 31 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
cmd/gocq/login.go
View File

@ -151,7 +151,7 @@ func loginResponseProcessor(res *client.LoginResponse) error {
log.Warn("请输入(1 - 2) (将在10秒后自动选择1)：") log.Warn("请输入(1 - 2) (将在10秒后自动选择1)：")
text = readLineTimeout(time.Second*10, "1") text = readLineTimeout(time.Second*10, "1")
if strings.Contains(text, "1") { if strings.Contains(text, "1") {
ticket := sliderCaptchaProcessor(res.VerifyUrl) ticket := getTicket(res.VerifyUrl)
if ticket == "" { if ticket == "" {
os.Exit(0) os.Exit(0)
} }
@ -218,22 +218,39 @@ func loginResponseProcessor(res *client.LoginResponse) error {
} }
} }
func sliderCaptchaProcessor(u string) string { func getTicket(u string) (str string) {
id := utils.RandomString(8) id := utils.RandomString(8)
log.Warnf("请前往该地址验证 -> %v", strings.ReplaceAll(u, "https://ssl.captcha.qq.com/template/wireless_mqq_captcha.html?", fmt.Sprintf("https://captcha.go-cqhttp.org/captcha?id=%v&", id))) log.Warnf("请前往该地址验证 -> %v <- 或输入手动抓取的 ticket：（Enter 提交）", strings.ReplaceAll(u, "https://ssl.captcha.qq.com/template/wireless_mqq_captcha.html?", fmt.Sprintf("https://captcha.go-cqhttp.org/captcha?id=%v&", id)))
start := time.Now() manual := make(chan string, 1)
for time.Since(start).Minutes() < 2 { go func() {
time.Sleep(time.Second) manual <- readLine()
data, err := global.GetBytes("https://captcha.go-cqhttp.org/captcha/ticket?id=" + id) }()
if err != nil { ticker := time.NewTicker(time.Second)
log.Warnf("获取 Ticket 时出现错误: %v", err) defer ticker.Stop()
return "" for count := 120; count > 0; count-- {
} select {
g := gjson.ParseBytes(data) case <-ticker.C:
if g.Get("ticket").Exists() { str = fetchCaptcha(id)
return g.Get("ticket").String() if str != "" {
return
}
case str = <-manual:
return
} }
} }
log.Warnf("验证超时") log.Warnf("验证超时")
return "" return ""
} }
func fetchCaptcha(id string) string {
data, err := global.GetBytes("https://captcha.go-cqhttp.org/captcha/ticket?id=" + id)
if err != nil {
log.Warnf("获取 Ticket 时出现错误: %v", err)
return ""
}
g := gjson.ParseBytes(data)
if g.Get("ticket").Exists() {
return g.Get("ticket").String()
}
return ""
}