From 0c9f7a1f8ff87832a4823c2d5d125e19b07b9bc9 Mon Sep 17 00:00:00 2001
From: wdvxdr <wdvxdr1123@gmail.com>
Date: Mon, 20 Feb 2023 14:06:08 +0800
Subject: [PATCH] fix device

---
 cmd/gocq/login.go | 2 ++
 cmd/gocq/main.go  | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/cmd/gocq/login.go b/cmd/gocq/login.go
index 079e9fc..23989df 100644
--- a/cmd/gocq/login.go
+++ b/cmd/gocq/login.go
@@ -52,6 +52,7 @@ func readIfTTY(de string) (str string) {
 }
 
 var cli *client.QQClient
+var device *client.DeviceInfo
 
 // ErrSMSRequestError SMS请求出错
 var ErrSMSRequestError = errors.New("sms request error")
@@ -168,6 +169,7 @@ func loginResponseProcessor(res *client.LoginResponse) error {
 			cli.Disconnect()
 			cli.Release()
 			cli = client.NewClientEmpty()
+			cli.UseDevice(device)
 			return qrcodeLogin()
 		case client.NeedCaptcha:
 			log.Warnf("登录需要验证码.")
diff --git a/cmd/gocq/main.go b/cmd/gocq/main.go
index 9d194f2..fded837 100644
--- a/cmd/gocq/main.go
+++ b/cmd/gocq/main.go
@@ -136,7 +136,6 @@ func Main() {
 		log.SetLevel(log.DebugLevel)
 		log.Warnf("已开启Debug模式.")
 	}
-	var device *client.DeviceInfo
 	if !global.PathExists("device.json") {
 		log.Warn("虚拟设备信息不存在, 将自动生成随机设备.")
 		device = client.GenRandomDevice()
@@ -242,6 +241,7 @@ func Main() {
 				cli.Disconnect()
 				cli.Release()
 				cli = newClient()
+				cli.UseDevice(device)
 			} else {
 				isTokenLogin = true
 			}