toxblh/podcasts
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added custom redirect policy

Browse Source
This commit is contained in:
James Wykeham-Martin 2018-02-02 09:38:06 +00:00
parent 32296d91b7
commit 82f577c17d
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
hammond-downloader/src/downloader.rs
View File

@ -2,6 +2,7 @@ use glob::glob;
use hyper::header::*; use hyper::header::*;
use mime_guess; use mime_guess;
use reqwest; use reqwest;
use reqwest::RedirectPolicy;
use tempdir::TempDir; use tempdir::TempDir;
use std::fs; use std::fs;
@ -37,7 +38,18 @@ fn download_into(
progress: Option<Arc<Mutex<DownloadProgress>>>, progress: Option<Arc<Mutex<DownloadProgress>>>,
) -> Result<String> { ) -> Result<String> {
info!("GET request to: {}", url); info!("GET request to: {}", url);
let client = reqwest::Client::builder().referer(false).build()?; // Haven't included the loop check as
// Steal the Stars
let policy = RedirectPolicy::custom(|attempt| {
info!("Redirect Attempt URL: {:?}", attempt.url());
if attempt.previous().len() > 10 {
attempt.too_many_redirects()
} else {
attempt.follow()
}
});
let client = reqwest::Client::builder().redirect(policy).referer(false).build()?;
let mut resp = client.get(url).send()?; let mut resp = client.get(url).send()?;
info!("Status Resp: {}", resp.status()); info!("Status Resp: {}", resp.status());