Use serde_json to create json response when requested by client.

1 files changed, 18 insertions, 7 deletions

diff --git a/src/main.rs b/src/main.rs
index 03c34df..1794722 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,15 +5,18 @@ extern crate diesel;
 extern crate rocket;
 extern crate rocket_blog;
 #[macro_use] extern crate bart_derive;
+extern crate serde;
+extern crate serde_json;
+#[macro_use] extern crate serde_derive;
 
 use self::diesel::prelude::*;
 use self::rocket_blog::{schema, models};
 use rocket::response::{Response, Responder};
 use rocket::request::Request;
-use rocket::http::{ContentType, Status};
+use rocket::http::{ContentType, MediaType, Status};
 use std::io::Cursor;
 
-#[derive(BartDisplay)]
+#[derive(BartDisplay, Serialize)]
 #[template = "templates/index.html"]
 struct IndexTemplate<'a> {
     title: &'a str,
@@ -21,11 +24,19 @@ struct IndexTemplate<'a> {
 }
 
 impl<'a> Responder<'a> for IndexTemplate<'a> {
-    fn respond_to(self, _: &Request) -> Result<Response<'static>, Status> {
-        Response::build()
-            .header(ContentType::HTML)
-            .sized_body(Cursor::new(format!("{}", &self)))
-            .ok()
+    fn respond_to(self, req: &Request) -> Result<Response<'static>, Status> {
+        if req.format().unwrap().is_json() {
+            Response::build()
+                .header(ContentType::JSON)
+                .sized_body(Cursor::new(format!("{}", serde_json::to_string(&self).unwrap())))
+                .ok()
+        }
+        else {
+            Response::build()
+                .header(ContentType::HTML)
+                .sized_body(Cursor::new(format!("{}", &self)))
+                .ok()
+        }
     }
 }