aboutsummaryrefslogtreecommitdiffstats
path: root/rust/src
diff options
context:
space:
mode:
Diffstat (limited to 'rust/src')
-rw-r--r--rust/src/bin/fatcat-iron.rs8
-rw-r--r--rust/src/lib.rs21
2 files changed, 22 insertions, 7 deletions
diff --git a/rust/src/bin/fatcat-iron.rs b/rust/src/bin/fatcat-iron.rs
index eed98b1d..0bed0fa2 100644
--- a/rust/src/bin/fatcat-iron.rs
+++ b/rust/src/bin/fatcat-iron.rs
@@ -8,10 +8,8 @@ extern crate futures;
extern crate iron;
extern crate iron_slog;
extern crate swagger;
-#[macro_use]
-extern crate error_chain;
-#[macro_use]
-extern crate slog;
+#[macro_use] extern crate error_chain;
+#[macro_use] extern crate slog;
extern crate slog_term;
extern crate slog_async;
@@ -50,6 +48,8 @@ fn main() {
// for the purpose of this example, pretend we have authenticated a user
chain.link_before(AllowAllMiddleware::new("cosmo"));
+ chain.link_after(fatcat::XClacksOverheadMiddleware);
+
if matches.is_present("https") {
unimplemented!()
} else {
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index fbff6ab2..55040f0c 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -4,10 +4,10 @@ extern crate chrono;
extern crate diesel;
extern crate dotenv;
extern crate futures;
-extern crate hyper;
+#[macro_use] extern crate hyper;
extern crate swagger;
-#[macro_use]
-extern crate error_chain;
+#[macro_use] extern crate error_chain;
+extern crate iron;
pub mod api_server;
@@ -20,6 +20,9 @@ use diesel::pg::PgConnection;
use diesel::prelude::*;
use dotenv::dotenv;
use std::env;
+use hyper::header::Headers;
+use iron::{Request, Response};
+use iron::middleware::AfterMiddleware;
pub fn establish_connection() -> PgConnection {
dotenv().ok();
@@ -32,3 +35,15 @@ pub fn establish_connection() -> PgConnection {
pub fn server() -> Result<api_server::Server> {
Ok(api_server::Server {})
}
+
+/// HTTP header middleware
+header! { (XClacksOverhead, "X-Clacks-Overhead") => [String] }
+
+pub struct XClacksOverheadMiddleware;
+
+impl AfterMiddleware for XClacksOverheadMiddleware {
+ fn after(&self, _req: &mut Request, mut res: Response) -> iron::IronResult<Response> {
+ res.headers.set(XClacksOverhead("GNU aaronsw, jpb".to_owned()));
+ Ok(res)
+ }
+}