1 files changed, 16 insertions, 2 deletions
diff --git a/src/server.rs b/src/server.rs
index 404c3b2..5d30cf7 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -8,7 +8,8 @@ use rusqlite::Connection;
 use std::collections::BTreeMap;
 use persistent::Write;
 use handlebars_iron::handlebars::to_json;
-use serde_json::{Value, Map};
+use serde_json::Map;
+use params;
 
 #[derive(Debug)]
 pub struct State {
@@ -36,10 +37,22 @@ fn make_renderer() -> HandlebarsEngine {
 
 fn quotes(r: &mut Request) -> IronResult<Response> {
     let mut result = Map::new();
+
+    let quote_id = {
+      let map = itry!(r.get_ref::<params::Params>());
+      match map.get("id") {
+        Some(&params::Value::String(ref id)) => Some(itry!(id.parse::<i64>())),
+        _ => None,
+      }
+    };
+
     let quotes = {
         let mu = r.get::<Write<State>>().unwrap();
         let state = mu.lock().unwrap();
-        try!(data::get_quotes(&state.connection))
+        match quote_id {
+          Some(id) => vec!(data::get_quote(&state.connection, id)?),
+          None => data::get_quotes(&state.connection)?,
+        }
     };
     result.insert("quotes".to_string(), to_json(&quotes));
     Ok(Response::with((
@@ -53,6 +66,7 @@ pub fn serve(state: State, port: u16) {
     let router =
         router!(
         index: get "/" => quotes,
+        view_quote: get "/view_quote" => quotes,
     );
     let mut chain = Chain::new(router);
     chain.link_after(make_renderer());