first commit

2026-02-17 13:06:23 +08:00
commit 7cbd3d061d
349 changed files with 126558 additions and 0 deletions
@@ -0,0 +1,156 @@
+package wxrouter
+
+import (
+	"errors"
+	"strings"
+	"time"
+	"xiawan/wx/clientsdk"
+	"xiawan/wx/clientsdk/baseinfo"
+	"xiawan/wx/clientsdk/baseutils"
+	"xiawan/wx/db"
+	"xiawan/wx/protobuf/wechat"
+	"xiawan/wx/srv/wxcore"
+	"xiawan/wx/srv/wxface"
+
+	"github.com/gogo/protobuf/proto"
+)
+
+// WXCheckQrcodeRouter 检测二维码状态响应路由
+type WXCheckQrcodeRouter struct {
+	wxcore.WXBaseRouter
+}
+
+// Handle 处理conn业务的方法
+func (cqr *WXCheckQrcodeRouter) Handle(wxResp wxface.IWXResponse) error {
+	defer wxcore.TryE("WXCheckQrcodeRouter Handle")
+	currentWXConn := wxResp.GetWXConncet()
+	currentAccount := currentWXConn.GetWXAccount()
+	currentInvoker := currentWXConn.GetWXReqInvoker()
+	currentCache := currentWXConn.GetWXCache()
+	currentUserInfo := currentAccount.GetUserInfo()
+
+	// 解析检测二维码响应
+	var checkResp wechat.CheckLoginQRCodeResponse
+	err := clientsdk.ParseResponseData(currentUserInfo, wxResp.GetPackHeader(), &checkResp)
+	if err != nil {
+		// 请求出问题了，判断是否掉线，并重连
+		currentWXConn.Stop()
+		return err
+	}
+
+	// 解密出现问题，说明协议出现了问题
+	qrcodeInfo := currentCache.GetQrcodeInfo()
+	retBytes, err := baseutils.AesDecryptByteKey(checkResp.LoginQrcodeNotifyPkg.NotifyData.Buffer, qrcodeInfo.QrCodeKey)
+
+	if err != nil {
+		// 请求出问题了，应该关闭链接
+		currentWXConn.Stop()
+		return err
+	}
+
+	lgQrNotify := &wechat.LoginQRCodeNotify{}
+	err = proto.Unmarshal(retBytes, lgQrNotify)
+	if err != nil {
+		// 请求出问题了，应该关闭链接
+		currentWXConn.Stop()
+		return err
+	}
+
+	errMsg := ""
+	if lgQrNotify.GetState() == 2 {
+		// 每秒查一次数据库太频繁了，这里改为仅在扫码成功后查一次
+		bind, _ := db.IsLicenseBind(currentUserInfo.UUID)
+		if bind != nil && len(lgQrNotify.GetWxid()) > 0 && lgQrNotify.GetWxid() != bind.WxId {
+			errMsg = "该授权码(KEY)已绑定其他微信号, 请换一个KEY"
+		}
+
+	}
+
+	// 添加扫码状态缓存
+	db.AddCheckStatusCache(currentUserInfo.UUID, &baseinfo.CheckLoginQrCodeResult{
+		LoginQRCodeNotify: lgQrNotify,
+		Msg:               errMsg,
+	})
+
+	if len(errMsg) != 0 {
+		// 当前 License 已绑定其他微信号, 需要更换 License
+		currentWXConn.Stop()
+		return errors.New(errMsg)
+	}
+
+	str := byteArrayToString(retBytes)
+	ticketValue := extractTicketValue(str)
+
+	if ticketValue != "" {
+		// 赋值ticketValue
+		currentUserInfo.Ticket = ticketValue
+		currentUserInfo.LoginDataInfo.Ticket = ticketValue
+		db.AddCheckStatusCache(currentUserInfo.UUID, &baseinfo.CheckLoginQrCodeResult{
+			LoginQRCodeNotify: lgQrNotify,
+			Msg:               "请提交验证码后登录",
+			Ret:               int32(0),
+		})
+	}
+
+	if lgQrNotify.GetState() == 2 {
+		// 扫码成功发送登录包
+		err := currentInvoker.SendManualAuthRequest(lgQrNotify.GetWxnewpass(), lgQrNotify.GetWxid())
+		if err != nil {
+			// 在停止连接前，确保扫码状态被正确持久化保存
+			db.AddCheckStatusCache(currentUserInfo.UUID, &baseinfo.CheckLoginQrCodeResult{
+				LoginQRCodeNotify: lgQrNotify,
+				Msg:               "登录包发送失败：" + err.Error(),
+				Ret:               int32(300),
+			})
+			// 确保状态被保存后再停止连接
+			time.Sleep(time.Millisecond * 200)
+			currentWXConn.Stop()
+			return err
+		}
+		go func() {
+			// 3 秒后发送指令
+			time.Sleep(3 * time.Second)
+			currentCache.SetScanFinish(true)
+
+			// 扫码完成后再次更新状态缓存，确保登录成功的状态被持久化
+			db.AddCheckStatusCache(currentUserInfo.UUID, &baseinfo.CheckLoginQrCodeResult{
+				LoginQRCodeNotify: lgQrNotify,
+				Msg:               "登录成功",
+				Ret:               int32(0),
+			})
+		}()
+	} else if lgQrNotify.GetState() == 4 {
+		// 在停止连接前，确保状态被正确保存
+		db.AddCheckStatusCache(currentUserInfo.UUID, &baseinfo.CheckLoginQrCodeResult{
+			LoginQRCodeNotify: lgQrNotify,
+			Msg:               "二维码失效",
+			Ret:               int32(301),
+		})
+		// 确保状态被保存后再停止连接
+		time.Sleep(time.Millisecond * 200)
+		currentWXConn.Stop()
+		return errors.New("WXCheckQrcodeRouter err: 二维码失效")
+	} else {
+		currentUserInfo.HeadURL = lgQrNotify.GetHeadImgUrl()
+		err := currentWXConn.GetWXReqInvoker().SendCheckLoginQrcodeRequest(qrcodeInfo.QrCodeUUID, qrcodeInfo.QrCodeKey)
+		if err != nil {
+			currentWXConn.Stop()
+			return err
+		}
+		time.Sleep(time.Second)
+	}
+
+	return nil
+}
+
+func byteArrayToString(data []byte) string {
+	return string(data)
+}
+
+func extractTicketValue(str string) string {
+	index := strings.Index(str, "ticket=")
+	if index == -1 {
+		return ""
+	}
+	return str[index+len("ticket="):]
+}