ProjectHentai/alist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(cloudreve): add custom user-agent (close #5020)

Browse Source
This commit is contained in:
Andy Hsu 2023-08-17 19:41:21 +08:00
parent 3b74f8cd9a
commit 3bbae29f93
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
drivers/cloudreve/meta.go
View File

@ -13,6 +13,7 @@ type Addition struct {
Username string `json:"username"` Username string `json:"username"`
Password string `json:"password"` Password string `json:"password"`
Cookie string `json:"cookie"` Cookie string `json:"cookie"`
CustomUA string `json:"custom_ua"`
} }
var config = driver.Config{ var config = driver.Config{

7
drivers/cloudreve/util.go
View File

@ -22,15 +22,18 @@ const loginPath = "/user/session"
func (d *Cloudreve) request(method string, path string, callback base.ReqCallback, out interface{}) error { func (d *Cloudreve) request(method string, path string, callback base.ReqCallback, out interface{}) error {
u := d.Address + "/api/v3" + path u := d.Address + "/api/v3" + path
ua := d.CustomUA
if ua == "" {
ua = base.UserAgent
}
req := base.RestyClient.R() req := base.RestyClient.R()
req.SetHeaders(map[string]string{ req.SetHeaders(map[string]string{
"Cookie": "cloudreve-session=" + d.Cookie, "Cookie": "cloudreve-session=" + d.Cookie,
"Accept": "application/json, text/plain, */*", "Accept": "application/json, text/plain, */*",
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36", "User-Agent": ua,
}) })
var r Resp var r Resp
req.SetResult(&r) req.SetResult(&r)
if callback != nil { if callback != nil {