mrps 0.1.1

Mini reverse proxy server
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

use minijinja::Value;
use axum::body::Body;
use std::error::Error;
use serde::Deserialize;
use crate::debug::debug;
use std::collections::HashMap;
use serde_derive::Deserialize;
use axum::http::{StatusCode, HeaderMap};
use reqwest::{Request, RequestBuilder, Client};

#[derive(Deserialize)]
pub struct Proxy {
    url: String,
    method: Option<String>,
    headers: Option<HashMap<String, String>>,
    body: Option<Vec<u8>>
}

impl Proxy {
    pub async fn new (method: &str, headers: &HashMap<String, String>, body: &Vec<u8>,proxy: &Value) -> Result<(StatusCode, HeaderMap, Body), Box<dyn Error>> {
        let proxy = Proxy::deserialize(proxy)?;
        let method = proxy.method.unwrap_or(method.to_string());
        debug(&method, &proxy.url, None, "");
        let mut r = RequestBuilder::from_parts(Client::new(),
            Request::new(method.parse()?, proxy.url.parse()?)
        );
        if let Some(headers) = proxy.headers {
            for (name, value) in headers.iter() {
                r = r.header(name, value);
            }
        }

        for (name, value) in headers.iter() {
            r = r.header(name.clone(), value.clone());
        }
        
        let response = match r.body(
            proxy.body.unwrap_or(body.to_vec())
        ).send().await {
            Ok(response) => {
                debug(
                    &method,
                    &proxy.url,
                    Some(response.status().as_u16()),
                    ""
                );
                response
            },
            Err(err) => {
                debug(&method, &proxy.url, Some(500), &err.to_string());
                return Err(err.into());
            }
        };

        Ok((
            response.status(),
            response.headers().clone(),
            response.bytes().await?.into()
        ))
    }
}