:pencli2: 修改mode行为

bot.go

@@ -360,7 +360,7 @@ func NewBot() *Bot {
 // DefaultBot 默认的Bot的构造方法,
 // mode不传入默认为openwechat.Normal,详情见mode
 //     bot := openwechat.DefaultBot(openwechat.Desktop)
-func DefaultBot(modes ...mode) *Bot {
+func DefaultBot(modes ...Mode) *Bot {
 	bot := NewBot()
 	if len(modes) > 0 {
 		bot.Caller.Client.SetMode(modes[0])

caller.go

@@ -402,7 +402,7 @@ func (c *Caller) WebWxRenameChatRoom(request *BaseRequest, info *LoginInfo, newT
 }
 
 // SetMode 设置Client的模式
-func (c *Client) SetMode(mode mode) {
+func (c *Client) SetMode(mode Mode) {
 	c.mode = mode
 }
 

client.go

@@ -41,7 +41,7 @@ type Client struct {
 	HttpHooks HttpHooks
 	*http.Client
 	Domain  WechatDomain
-	mode    mode
+	mode    Mode
 	mu      sync.Mutex
 	cookies map[string][]*http.Cookie
 }
@@ -113,22 +113,7 @@ func (c *Client) GetCookieMap() map[string][]*http.Cookie {
 
 // GetLoginUUID 获取登录的uuid
 func (c *Client) GetLoginUUID() (*http.Response, error) {
-	path, _ := url.Parse(jslogin)
+	return c.mode.GetLoginUUID(c)
-	params := url.Values{}
-	redirectUrl, _ := url.Parse(webwxnewloginpage)
-	if c.mode == Desktop {
-		p := url.Values{"mod": {"desktop"}}
-		redirectUrl.RawQuery = p.Encode()
-	}
-	params.Add("redirect_uri", redirectUrl.String())
-	params.Add("appid", appId)
-	params.Add("fun", "new")
-	params.Add("lang", "zh_CN")
-	params.Add("_", strconv.FormatInt(time.Now().UnixNano() / 1e6, 10))
-
-	path.RawQuery = params.Encode()
-	req, _ := http.NewRequest(http.MethodGet, path.String(), nil)
-	return c.Do(req)
 }
 
 // GetLoginQrcode 获取登录的二维吗
@@ -154,12 +139,7 @@ func (c *Client) CheckLogin(uuid string) (*http.Response, error) {
 
 // GetLoginInfo 请求获取LoginInfo
 func (c *Client) GetLoginInfo(path string) (*http.Response, error) {
-	req, _ := http.NewRequest(http.MethodGet, path, nil)
+	return c.mode.GetLoginInfo(c, path)
-	if c.mode == Desktop {
-		req.Header.Add("client-version", uosPatchClientVersion)
-		req.Header.Add("extspam", uosPatchExtspam)
-	}
-	return c.Do(req)
 }
 
 // WebInit 请求获取初始化信息

mode.go

@@ -0,0 +1,66 @@
+package openwechat
+
+import (
+	"net/http"
+	"net/url"
+	"strconv"
+	"time"
+)
+
+type Mode interface {
+	GetLoginUUID(client *Client) (*http.Response, error)
+	GetLoginInfo(client *Client, path string) (*http.Response, error)
+}
+
+var (
+	Normal  Mode = normalMode{}
+	Desktop Mode = desktopMode{}
+)
+
+type normalMode struct{}
+
+func (n normalMode) GetLoginUUID(client *Client) (*http.Response, error) {
+	path, _ := url.Parse(jslogin)
+	params := url.Values{}
+	redirectUrl, _ := url.Parse(webwxnewloginpage)
+	params.Add("redirect_uri", redirectUrl.String())
+	params.Add("appid", appId)
+	params.Add("fun", "new")
+	params.Add("lang", "zh_CN")
+	params.Add("_", strconv.FormatInt(time.Now().UnixNano()/1e6, 10))
+
+	path.RawQuery = params.Encode()
+	req, _ := http.NewRequest(http.MethodGet, path.String(), nil)
+	return client.Do(req)
+}
+
+func (n normalMode) GetLoginInfo(client *Client, path string) (*http.Response, error) {
+	req, _ := http.NewRequest(http.MethodGet, path, nil)
+	return client.Do(req)
+}
+
+type desktopMode struct{}
+
+func (n desktopMode) GetLoginUUID(client *Client) (*http.Response, error) {
+	path, _ := url.Parse(jslogin)
+	params := url.Values{}
+	redirectUrl, _ := url.Parse(webwxnewloginpage)
+	p := url.Values{"mod": {"desktop"}}
+	redirectUrl.RawQuery = p.Encode()
+	params.Add("redirect_uri", redirectUrl.String())
+	params.Add("appid", appId)
+	params.Add("fun", "new")
+	params.Add("lang", "zh_CN")
+	params.Add("_", strconv.FormatInt(time.Now().UnixNano()/1e6, 10))
+
+	path.RawQuery = params.Encode()
+	req, _ := http.NewRequest(http.MethodGet, path.String(), nil)
+	return client.Do(req)
+}
+
+func (n desktopMode) GetLoginInfo(client *Client, path string) (*http.Response, error) {
+	req, _ := http.NewRequest(http.MethodGet, path, nil)
+	req.Header.Add("client-version", uosPatchClientVersion)
+	req.Header.Add("extspam", uosPatchExtspam)
+	return client.Do(req)
+}

url.go

@@ -1,13 +1,13 @@
 package openwechat
 
-// mode 类型限制
+//// mode 类型限制
-type mode string
+//type mode string
-
+//
-// 向外暴露2种模式
+//// 向外暴露2种模式
-const (
+//const (
-	Normal  mode = "normal"
+//	Normal  mode = "normal"
-	Desktop mode = "desktop" // 突破网页版登录限制
+//	Desktop mode = "desktop" // 突破网页版登录限制
-)
+//)
 
 const (
 	webwxinit            = "/cgi-bin/mmwebwx-bin/webwxinit"