proksi 0.7.0

A batteries-included reverse proxy with automatic HTTPS using Cloudflare Pingora and Let's Encrypt.
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
65

//! `headers { response { add = { ... }, remove = [...] } }`: changes response headers

use std::str::FromStr;

use anyhow::Result;
use async_trait::async_trait;
use http::{HeaderName, HeaderValue};
use pingora::{http::ResponseHeader, proxy::Session};

use crate::{config::RouteHeader, proxy_server::https_proxy::RouterContext};

use super::Middleware;

pub(super) struct Headers {
    add: Vec<(HeaderName, HeaderValue)>,
    remove: Vec<String>,
}

impl Headers {
    /// `None` when no header is changed
    pub(super) fn new(headers: &RouteHeader) -> Result<Option<Self>, String> {
        let add = headers
            .add
            .iter()
            .flatten()
            .map(|header| {
                let name = HeaderName::from_str(&header.name);
                let value = HeaderValue::from_str(&header.value);
                match (name, value) {
                    (Ok(name), Ok(value)) => Ok((name, value)),
                    _ => Err(format!("headers: {:?} is not a valid header", header.name)),
                }
            })
            .collect::<Result<Vec<_>, _>>()?;
        let remove = headers
            .remove
            .iter()
            .flatten()
            .map(|header| header.name.to_string())
            .collect::<Vec<_>>();

        if add.is_empty() && remove.is_empty() {
            return Ok(None);
        }
        Ok(Some(Self { add, remove }))
    }
}

#[async_trait]
impl Middleware for Headers {
    async fn response(
        &self,
        _session: &mut Session,
        response: &mut ResponseHeader,
        _ctx: &mut RouterContext,
    ) -> Result<()> {
        for (name, value) in &self.add {
            response.insert_header(name, value)?;
        }
        for name in &self.remove {
            response.remove_header(name);
        }
        Ok(())
    }
}