http-rs
/
surf
mirror of https://github.com/http-rs/surf
1
0
Fork 0
Code Issues Releases Wiki Activity

Update `base_url` with each redirect of a full URL

This commit is contained in:
Kenneth Gitere 2021-02-12 06:12:35 +03:00 committed by Jeremiah Senkpiel
parent 16a9ab948a
commit c32ee64b82
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/middleware/redirect/mod.rs
View File

@ -91,7 +91,7 @@ impl Middleware for Redirect {
// and try sending it until we get some status back that is not a
// redirect.
let base_url = req.url().clone();
let mut base_url = req.url().clone();
while redirect_count < self.attempts {
redirect_count += 1;
@ -101,7 +101,10 @@ impl Middleware for Redirect {
if let Some(location) = res.header(headers::LOCATION) {
let http_req: &mut http::Request = req.as_mut();
*http_req.url_mut() = match Url::parse(location.last().as_str()) {
Ok(valid_url) => valid_url,
Ok(valid_url) => {
base_url = valid_url;
base_url.clone()
}
Err(e) => match e {
http::url::ParseError::RelativeUrlWithoutBase => {
base_url.join(location.last().as_str())?