fav_core 0.1.7

Fav's core crate; A collection of traits.
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
53
54
55
56
57
58
59
60
61
62
63
64

//! Config,
//! helping managing the configuration

use crate::local::ProtoLocal;
use reqwest::header::{HeaderMap, HeaderValue};
use std::collections::HashMap;

/// A HttpConfig, including headers and cookies.
/// # Example
/// ```
/// # #[path = "test_utils/mod.rs"]
/// # mod test_utils;
/// # use test_utils::data::Conf;
/// # use fav_core::config::HttpConfig;
/// # use reqwest::{header::HeaderMap, header};
/// # use std::collections::HashMap;
///
/// impl HttpConfig for Conf {
///     fn headers(&self) -> HeaderMap {
///         let mut hp = HeaderMap::new();
///         hp.insert(header::USER_AGENT, "Mozilla/5.0".parse().unwrap());
///         hp
///     }
///
///     fn cookies(&self) -> &HashMap<String, String> {
///         &self.cookies
///     }
///
///     fn cookies_mut(&mut self) -> &mut HashMap<String, String> {
///         &mut self.cookies
///     }
/// }
/// ```
pub trait HttpConfig {
    /// The default headers
    fn headers(&self) -> HeaderMap;
    /// The cookies. [`HttpConfig::cookie_value`] uses this to generate the `Cookie` header.
    /// # Caution
    /// `HttpConfig::cookie_value` will omit the keys that are not in the cookies
    /// without any warning.
    fn cookies(&self) -> &HashMap<String, String>;
    /// The mutable reference to the cookies
    fn cookies_mut(&mut self) -> &mut HashMap<String, String>;
    /// Set the cookies
    fn extend_cookies(&mut self, cookies: HashMap<String, String>) {
        self.cookies_mut().extend(cookies);
    }
    /// Acquire the cookie value by keys
    fn cookie_value(&self, keys: impl IntoIterator<Item = impl AsRef<str>>) -> HeaderValue {
        let cookies = self.cookies();
        keys.into_iter()
            .filter_map(|k| {
                let k = k.as_ref();
                cookies.get(k).map(|v| format!("{k}={v}; "))
            })
            .collect::<String>()
            .parse()
            .unwrap()
    }
}

/// Mark it able to be a config, which concludes [`HttpConfig`], and can be persisted as protobuf through [`ProtoLocal`].
pub trait Config: HttpConfig + ProtoLocal {}
impl<T> Config for T where T: HttpConfig + ProtoLocal {}