[update]: 更新微信登录 v2 版本 (#282)

bot.go

@@ -8,7 +8,6 @@ import (
 	"net/url"
 	"os/exec"
 	"runtime"
-	"sync"
 )
 
 type Bot struct {
@@ -22,7 +21,6 @@ type Bot struct {
 	Serializer          Serializer                    // 序列化器, 默认为json
 	Storage             *Storage
 	Caller              *Caller
-	once                sync.Once
 	err                 error
 	context             context.Context
 	cancel              context.CancelFunc
@@ -184,8 +182,7 @@ func (b *Bot) WebInit() error {
 	}
 	// 开启协程，轮询获取是否有新的消息返回
 
-	// FIX: 当bot在线的情况下执行热登录,会开启多次事件监听
-	go b.once.Do(func() {
+	go func() {
 		if b.MessageErrorHandler == nil {
 			b.MessageErrorHandler = defaultSyncCheckErrHandler(b)
 		}
@@ -201,7 +198,7 @@ func (b *Bot) WebInit() error {
 				break
 			}
 		}
-	})
+	}()
 	return nil
 }
 
@@ -293,7 +290,7 @@ func (b *Bot) Block() error {
 		return errors.New("`Block` must be called after user login")
 	}
 	<-b.Context().Done()
-	return nil
+	return b.CrashReason()
 }
 
 // Exit 主动退出，让 Block 不在阻塞

caller.go

@@ -6,7 +6,6 @@ import (
 	"errors"
 	"fmt"
 	"io"
-	"net/http"
 	"net/url"
 	"os"
 )
@@ -67,16 +66,23 @@ func (c *Caller) CheckLogin(uuid, tip string) (CheckLoginResponse, error) {
 // GetLoginInfo 获取登录信息
 func (c *Caller) GetLoginInfo(path *url.URL) (*LoginInfo, error) {
 	// 从响应体里面获取需要跳转的url
+	query := path.Query()
+	query.Set("version", "v2")
+	path.RawQuery = query.Encode()
 	resp, err := c.Client.GetLoginInfo(path)
 	if err != nil {
 		return nil, err
 	}
-	// 判断是否重定向
-	if resp.StatusCode != http.StatusMovedPermanently {
-		return nil, fmt.Errorf("%w: try to login with Desktop Mode", ErrForbidden)
-	}
+	// 微信 v2 版本修复了301 response missing Location header 的问题
 	defer func() { _ = resp.Body.Close() }()
 
+	if _, exists := CookieGroup(resp.Cookies()).GetByName("wxuin"); !exists {
+		err = ErrForbidden
+		if c.Client.mode != desktop {
+			err = fmt.Errorf("%w: try to login with desktop mode", err)
+		}
+		return nil, err
+	}
 	var loginInfo LoginInfo
 	// xml结构体序列化储存
 	if err := scanXml(resp.Body, &loginInfo); err != nil {