diff options
Diffstat (limited to 'src/server.rs')
-rw-r--r-- | src/server.rs | 35 |
1 files changed, 22 insertions, 13 deletions
diff --git a/src/server.rs b/src/server.rs index 12ffedb..eec0e72 100644 --- a/src/server.rs +++ b/src/server.rs @@ -10,6 +10,7 @@ use persistent::Write; use handlebars_iron::handlebars::to_json; use serde_json::Map; use params; +use error::LinoError; #[derive(Debug)] pub struct State { @@ -19,6 +20,14 @@ impl iron::typemap::Key for State { type Value = State; } +fn get_param(r: &mut Request, param: &str) -> IronResult<String> { + let map = itry!(r.get_ref::<params::Params>()); + match map.get(param) { + Some(¶ms::Value::String(ref v)) => Ok(v.to_string()), + _ => Err(From::from(LinoError::NotFound(param.to_string()))), + } +} + fn make_renderer() -> HandlebarsEngine { let mut e = HandlebarsEngine::new(); @@ -41,19 +50,10 @@ fn make_renderer() -> HandlebarsEngine { fn quotes(r: &mut Request) -> IronResult<Response> { let mut result = Map::new(); - - let (quote_id, ordering) = { - let map = itry!(r.get_ref::<params::Params>()); - let quote_id = match map.get("id") { - Some(¶ms::Value::String(ref id)) => Some(itry!(id.parse::<i64>())), - _ => None, - }; - let ordering = match map.get("order") { - Some(¶ms::Value::String(ref ordering)) => ordering.clone(), - _ => "".to_string(), - }; - (quote_id, ordering) - }; + let quote_id = get_param(r, "id").ok().and_then( + |id| id.parse::<i64>().ok(), + ); + let ordering = get_param(r, "order").unwrap_or("".to_string()); let quotes = { let mu = r.get::<Write<State>>().unwrap(); @@ -79,11 +79,20 @@ fn add_get(r: &mut Request) -> IronResult<Response> { ))) } +fn add_post(r: &mut Request) -> IronResult<Response> { + Ok(Response::with(( + status::Ok, + Header(ContentType::html()), + Template::new("add", Map::new()), + ))) +} + pub fn serve(state: State, port: u16) { let router = router!( index: get "/" => quotes, add_get: get "/add.jsp" => add_get, + add_post: post "/add.jsp" => add_post, quotes_jsp: get "/quotes.jsp" => quotes, view_quote: get "/view_quote" => quotes, ); |