dwldutil 3.0.0

A utility for parallel downloading
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use surf::middleware::{Middleware, Next};
use surf::{Client, Request, Response, Result, StatusCode, Url};

pub struct RedirectMiddleware {
    max_redirects: usize,
}

impl RedirectMiddleware {
    pub fn new(max_redirects: usize) -> Self {
        Self { max_redirects }
    }
}

#[surf::utils::async_trait]
impl Middleware for RedirectMiddleware {
    async fn handle(&self, req: Request, client: Client, _: Next<'_>) -> Result<Response> {
        let mut req = req;
        for _ in 0..self.max_redirects {
            let res = client.send(req.clone()).await?;
            if res.status().is_redirection() {
                if let Some(location) = res.header("Location") {
                    req = Request::new(req.method(), Url::parse(location.last().as_str()).unwrap());
                    continue;
                }
            }
            return Ok(res);
        }
        Err(surf::Error::from_str(
            StatusCode::TooManyRequests,
            "Too many redirects, Increase max_redirections in DLStartConfig",
        ))
    }
}