aboutsummaryrefslogtreecommitdiffstats
path: root/src/lib.rs
blob: 975308acd638b6567def9c8ae575cd5731de5850 (plain)
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
extern crate fastcgi;
extern crate http;

use std::io::Write;

use http::{Request, Response};
use http::request;
use inflector::cases::traincase::to_train_case;


pub fn run<F, T>(handler: F)
where F: Fn(Request<()>) -> Response<T> + Send + Sync + 'static
{
    fastcgi::run(move |mut req| {
        // build request
        let r = request::Builder::new()
            .method("GET")
            .body(())
            .unwrap();

        handler(r);

        let params = req.params()
            .map(|(k, v)| k + ": " + &v)
            .collect::<Vec<String>>()
            .join("\n");

        write!(
            &mut req.stdout(),
            "Content-Type: text/plain\n\n{}",
            params
        )
            .unwrap_or(());
    });
}

trait From<T>: Sized {
    fn from(_: T) -> Self;
}

impl From<fastcgi::Request> for http::request::Builder {
    fn from(request: fastcgi::Request) -> Self {
        let method = request.param("REQUEST_METHOD")
            .unwrap_or("".to_owned());

        let uri = format!(
            "{}://{}{}",
            request.param("REQUEST_SCHEME").unwrap_or("".to_owned()),
            request.param("HTTP_HOST").unwrap_or("".to_owned()),
            request.param("REQUEST_URI").unwrap_or("".to_owned()),
        );

        let mut http_request = http::request::Builder::new()
            .method(&*method)
            .uri(&uri);

        let headers = headers_from_params(request.params());
        for (k, v) in headers {
            http_request = http_request.header(&k, &v);
        }

        http_request

        // HTTP_* params become headers
    }
}

fn headers_from_params(params: fastcgi::Params) -> Vec<(String, String)> {
    return params
        .filter(|(key, _)| key.starts_with("HTTP_"))
        .map(|(key, value)| {
            let mut key = key.get(5..).unwrap_or("").to_owned();
            key = key.replace("_", "-");
            key = to_train_case(&key);

            // Change _ to -
            // Uppercase each word

            (key, value)
        })
        .collect()
}