2018-08-16 15:05:58 +00:00
|
|
|
use std::collections::HashMap;
|
|
|
|
|
|
2018-08-16 16:49:29 +00:00
|
|
|
use futures::{future, Future, Stream};
|
|
|
|
|
use hyper::{Body, Error, Request, Response, StatusCode};
|
2018-08-16 15:05:58 +00:00
|
|
|
|
2018-08-16 16:49:29 +00:00
|
|
|
use {HOOKS, URIPATTERN};
|
2018-08-16 15:05:58 +00:00
|
|
|
|
2018-08-16 16:49:29 +00:00
|
|
|
pub fn dip_service(req: Request<Body>) -> Box<Future<Item = Response<Body>, Error = Error> + Send> {
|
2018-08-16 15:05:58 +00:00
|
|
|
let path = req.uri().path().to_owned();
|
2018-08-16 16:49:29 +00:00
|
|
|
let captures = match URIPATTERN.captures(path.as_ref()) {
|
|
|
|
|
Some(value) => value,
|
|
|
|
|
None => {
|
|
|
|
|
return Box::new(future::ok(
|
|
|
|
|
Response::builder()
|
|
|
|
|
.status(StatusCode::NOT_FOUND)
|
|
|
|
|
.body(Body::from("not found"))
|
|
|
|
|
.unwrap(),
|
|
|
|
|
))
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
let name = match captures.name("name") {
|
|
|
|
|
Some(value) => value.as_str().to_owned(),
|
|
|
|
|
None => {
|
|
|
|
|
return Box::new(future::ok(
|
|
|
|
|
Response::builder()
|
|
|
|
|
.status(StatusCode::NOT_FOUND)
|
|
|
|
|
.body(Body::from("not found"))
|
|
|
|
|
.unwrap(),
|
|
|
|
|
))
|
|
|
|
|
}
|
2018-08-16 15:05:58 +00:00
|
|
|
};
|
2018-08-16 16:49:29 +00:00
|
|
|
|
|
|
|
|
// TODO: filter by method as well
|
|
|
|
|
|
|
|
|
|
let headers = req.headers()
|
|
|
|
|
.clone()
|
|
|
|
|
.into_iter()
|
|
|
|
|
.filter_map(|(k, v)| {
|
|
|
|
|
let key = k.unwrap().as_str().to_owned();
|
|
|
|
|
v.to_str().map(|value| (key, value.to_owned())).ok()
|
2018-08-16 15:05:58 +00:00
|
|
|
})
|
2018-08-16 16:49:29 +00:00
|
|
|
.collect::<HashMap<_, _>>();
|
|
|
|
|
let method = req.method().as_str().to_owned();
|
|
|
|
|
// probably not idiomatically the best way to do it
|
|
|
|
|
// i was just trying to get something working
|
|
|
|
|
let body = req.into_body();
|
|
|
|
|
Box::new(body.concat2().map(move |body| {
|
|
|
|
|
let req_obj = json!({
|
|
|
|
|
"body": String::from_utf8(body.to_vec()).unwrap(),
|
|
|
|
|
"headers": headers,
|
|
|
|
|
"method": method,
|
|
|
|
|
});
|
|
|
|
|
let hooks = HOOKS.lock().unwrap();
|
|
|
|
|
let hook = hooks.get(&name).unwrap();
|
|
|
|
|
let (code, msg) = hook.iter()
|
|
|
|
|
.fold(Ok(req_obj), |prev, handler| {
|
|
|
|
|
prev.and_then(|val| handler.run(val))
|
|
|
|
|
})
|
2018-08-16 17:51:33 +00:00
|
|
|
.map(|res| {
|
|
|
|
|
(
|
|
|
|
|
StatusCode::ACCEPTED,
|
|
|
|
|
format!(
|
|
|
|
|
"stdout:\n{}\n\nstderr:\n{}",
|
|
|
|
|
res.get("stdout").and_then(|v| v.as_str()).unwrap_or(""),
|
|
|
|
|
res.get("stderr").and_then(|v| v.as_str()).unwrap_or(""),
|
|
|
|
|
),
|
|
|
|
|
)
|
|
|
|
|
})
|
2018-08-16 16:49:29 +00:00
|
|
|
.unwrap_or_else(|err| (StatusCode::BAD_REQUEST, format!("Error: {}", err)));
|
|
|
|
|
Response::builder()
|
|
|
|
|
.status(code)
|
|
|
|
|
.body(Body::from(msg))
|
|
|
|
|
.unwrap_or_else(|err| Response::new(Body::from(format!("{}", err))))
|
|
|
|
|
}))
|
2018-08-16 15:05:58 +00:00
|
|
|
}
|
|
|
|
|
|
2018-08-16 16:49:29 +00:00
|
|
|
/* pub fn dip_service(
|
2018-08-16 15:05:58 +00:00
|
|
|
req: Request<Body>,
|
|
|
|
|
) -> Box<Future<Item = Response<Body>, Error = String> + Send> {
|
|
|
|
|
let uri = req.uri().path().to_owned();
|
|
|
|
|
Box::new(future::ok(service_fn(req).unwrap_or_else(|err| {
|
|
|
|
|
eprintln!("Error from '{}': {}", uri, err);
|
|
|
|
|
Response::builder()
|
|
|
|
|
.status(StatusCode::NOT_FOUND)
|
|
|
|
|
.body(Body::from(NOTFOUND))
|
|
|
|
|
.unwrap()
|
|
|
|
|
})))
|
|
|
|
|
}
|
2018-08-16 16:49:29 +00:00
|
|
|
*/
|
