apisdk 0.0.4-beta.3

A highlevel API client framework 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

use async_trait::async_trait;
use hyper::header::{HeaderValue, HOST};
use reqwest::{Request, Response};
use reqwest_middleware::{Middleware, Next};
use task_local_extensions::Extensions;

pub(crate) struct RewriteHost {
    host: String,
}

impl RewriteHost {
    pub fn new(host: impl ToString) -> Self {
        Self {
            host: host.to_string(),
        }
    }
}

pub(crate) struct RewriteHostMiddleware;

#[async_trait]
impl Middleware for RewriteHostMiddleware {
    async fn handle(
        &self,
        req: Request,
        extensions: &mut Extensions,
        next: Next<'_>,
    ) -> Result<Response, reqwest_middleware::Error> {
        let mut req = req;

        // Rewrite host
        if let Some(rewriter_host) = extensions.get::<RewriteHost>() {
            let headers = req.headers_mut();
            headers.insert(
                HOST,
                HeaderValue::from_str(rewriter_host.host.as_str()).unwrap(),
            );
        }

        next.run(req, extensions).await
    }
}