From 601b6e4600f5dbb1f48ee41b21caa119a2f4222f Mon Sep 17 00:00:00 2001
From: eatMoreApple <15055461510@163.com>
Date: Mon, 10 May 2021 21:42:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0client=E8=AF=B7=E6=B1=82?=
 =?UTF-8?q?=E9=92=A9=E5=AD=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client.go | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/client.go b/client.go
index 4099f41..d40dbaf 100644
--- a/client.go
+++ b/client.go
@@ -18,10 +18,27 @@ import (
 	"time"
 )
 
+// 请求上下文钩子
+type HttpHook interface {
+	BeforeRequest(req *http.Request)
+	AfterRequest(response *http.Response, err error)
+}
+
+type HttpHooks []HttpHook
+
+type UserAgentHook struct{}
+
+func (u UserAgentHook) BeforeRequest(req *http.Request) {
+	req.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.114 Safari/537.36")
+}
+
+func (u UserAgentHook) AfterRequest(response *http.Response, err error) {}
+
 // http请求客户端
 // 客户端需要维持Session会话
 // 并且客户端不允许跳转
 type Client struct {
+	HttpHooks HttpHooks
 	*http.Client
 	UrlManager
 	mu      sync.Mutex
@@ -45,12 +62,25 @@ func DefaultClient(urlManager UrlManager) *Client {
 	return NewClient(client, urlManager)
 }
 
+func (c *Client) AddHttpHook(hooks ...HttpHook) {
+	c.HttpHooks = append(c.HttpHooks, hooks...)
+}
+
+func (c *Client) do(req *http.Request) (*http.Response, error) {
+	for _, hook := range c.HttpHooks {
+		hook.BeforeRequest(req)
+	}
+	resp, err := c.Client.Do(req)
+	for _, hook := range c.HttpHooks {
+		hook.AfterRequest(resp, err)
+	}
+	return resp, err
+}
+
 // 抽象Do方法,将所有的有效的cookie存入Client.cookies
 // 方便热登陆时获取
 func (c *Client) Do(req *http.Request) (*http.Response, error) {
-	req.Header.Add("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.114 Safari/537.36")
-	req.URL.RawQuery += "&target=t"
-	resp, err := c.Client.Do(req)
+	resp, err := c.do(req)
 	if err != nil {
 		return resp, err
 	}