🔒 fix baidu direct link

This commit is contained in:
Xhofe 2022-02-12 12:04:10 +08:00
parent 0814778a14
commit 5dca777caf

View File

@ -140,13 +140,13 @@ func (driver Baidu) Link(args base.Args, account *model.Account) (*base.Link, er
return nil, err return nil, err
} }
u := fmt.Sprintf("%s&access_token=%s", resp.List[0].Dlink, account.AccessToken) 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 { if err != nil {
return nil, err return nil, err
} }
if res.StatusCode() == 302 { //if res.StatusCode() == 302 {
u = res.Header().Get("location") u = res.Header().Get("location")
} //}
return &base.Link{ return &base.Link{
Url: u, Url: u,
Headers: []base.Header{ Headers: []base.Header{