kumo 0.3.16

An async web crawling framework for Rust - Scrapy for Rust
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use super::{FetchRequest, Middleware};
use crate::error::KumoError;
use reqwest::header::{HeaderName, HeaderValue};
use std::collections::HashMap;

/// Injects a fixed set of HTTP headers into every request.
///
/// # Example
/// ```rust,ignore
/// DefaultHeaders::new().user_agent("kumo/0.2 (+https://github.com/you/kumo)")
/// ```
pub struct DefaultHeaders {
    headers: HashMap<String, String>,
}

impl DefaultHeaders {
    pub fn new() -> Self {
        Self {
            headers: HashMap::new(),
        }
    }

    pub fn header(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
        self.headers.insert(name.into(), value.into());
        self
    }

    pub fn user_agent(self, ua: impl Into<String>) -> Self {
        self.header("User-Agent", ua)
    }
}

impl Default for DefaultHeaders {
    fn default() -> Self {
        Self::new()
    }
}

#[async_trait::async_trait]
impl Middleware for DefaultHeaders {
    async fn before_request(&self, request: &mut FetchRequest) -> Result<(), KumoError> {
        for (name, value) in &self.headers {
            if let (Ok(n), Ok(v)) = (
                HeaderName::from_bytes(name.as_bytes()),
                HeaderValue::from_str(value),
            ) {
                request.headers.insert(n, v);
            }
        }
        Ok(())
    }
}