Skip to content

Commit ece1518

Browse files
authored
fix(drivers): add headers in Link methods (#2401)
* fix(drivers): add headers in Link methods Signed-off-by: MadDogOwner <xiaoran@xrgzs.top>
1 parent a5ba6a0 commit ece1518

3 files changed

Lines changed: 9 additions & 1 deletion

File tree

drivers/cloudreve/driver.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,10 @@ func (d *Cloudreve) Link(ctx context.Context, file model.Obj, args model.LinkArg
8989
}
9090
return &model.Link{
9191
URL: dUrl,
92+
Header: http.Header{
93+
"Referer": {d.Address},
94+
"User-Agent": {d.getUA()},
95+
},
9296
}, nil
9397
}
9498

drivers/cloudreve_v4/driver.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,10 @@ func (d *CloudreveV4) Link(ctx context.Context, file model.Obj, args model.LinkA
167167
return &model.Link{
168168
URL: url.Urls[0].URL,
169169
Expiration: &exp,
170+
Header: http.Header{
171+
"Referer": {d.Address},
172+
"User-Agent": {d.getUA()},
173+
},
170174
}, nil
171175
}
172176

drivers/lanzou/driver.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ func (d *LanZou) Link(ctx context.Context, file model.Obj, args model.LinkArgs)
117117
return &model.Link{
118118
URL: dfile.Url,
119119
Header: http.Header{
120-
"User-Agent": []string{base.UserAgent},
120+
"User-Agent": {d.UserAgent},
121121
},
122122
Expiration: &exp,
123123
}, nil

0 commit comments

Comments
 (0)