From 5dca777cafcdaca67e305370f271c311d334a81d Mon Sep 17 00:00:00 2001
From: Xhofe <i@nn.ci>
Date: Sat, 12 Feb 2022 12:04:10 +0800
Subject: [PATCH] :lock: fix baidu direct link

---
 drivers/baidu/driver.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/drivers/baidu/driver.go b/drivers/baidu/driver.go
index 1d322bff..f64c56f8 100644
--- a/drivers/baidu/driver.go
+++ b/drivers/baidu/driver.go
@@ -140,13 +140,13 @@ func (driver Baidu) Link(args base.Args, account *model.Account) (*base.Link, er
 		return nil, err
 	}
 	u := fmt.Sprintf("%s&access_token=%s", resp.List[0].Dlink, account.AccessToken)
-	res, err := base.NoRedirectClient.R().Head(u)
+	res, err := base.NoRedirectClient.R().SetHeader("User-Agent", "pan.baidu.com").Head(u)
 	if err != nil {
 		return nil, err
 	}
-	if res.StatusCode() == 302 {
-		u = res.Header().Get("location")
-	}
+	//if res.StatusCode() == 302 {
+	u = res.Header().Get("location")
+	//}
 	return &base.Link{
 		Url: u,
 		Headers: []base.Header{