Use route decorators to produce different output formats.

This seems to be the more proper way of doing it. Also makes it much
clearer which routes exists, and what formats they accept. Both in the
code and in the status output.

1 files changed, 21 insertions, 22 deletions

diff --git a/src/main.rs b/src/main.rs
index 1794722..1d20cb0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,16 +4,16 @@
 extern crate diesel;
 extern crate rocket;
 extern crate rocket_blog;
+extern crate rocket_contrib;
 #[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_contrib::Json;
 use rocket::response::{Response, Responder};
 use rocket::request::Request;
-use rocket::http::{ContentType, MediaType, Status};
+use rocket::http::{ContentType, Status};
 use std::io::Cursor;
 
 #[derive(BartDisplay, Serialize)]
@@ -24,36 +24,35 @@ struct IndexTemplate<'a> {
 }
 
 impl<'a> Responder<'a> for IndexTemplate<'a> {
-    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()
-        }
+    fn respond_to(self, _: &Request) -> Result<Response<'static>, Status> {
+        Response::build()
+            .header(ContentType::HTML)
+            .sized_body(Cursor::new(format!("{}", &self)))
+            .ok()
     }
 }
 
-#[get("/")]
-fn index<'a>(conn: rocket_blog::DbConn) -> IndexTemplate<'a> {
+fn get_posts(conn: rocket_blog::DbConn) -> Vec<models::Post> {
     use schema::posts::dsl::*;
-    let results = posts.filter(published.eq(false))
+    posts.filter(published.eq(false))
         .limit(5)
         .load::<models::Post>(&*conn)
-        .expect("Error loading posts");
+        .expect("Error loading posts")
+}
+
+#[get("/", format = "text/html")]
+fn index<'a>(conn: rocket_blog::DbConn) -> IndexTemplate<'a> {
+    IndexTemplate { title: "Bloggen", posts: get_posts(conn) }
+}
 
-    IndexTemplate { title: "Bloggen", posts: results }
+#[get("/", format = "application/json")]
+fn index_json(conn: rocket_blog::DbConn) -> Json<Vec<models::Post>> {
+    Json(get_posts(conn))
 }
 
 fn main() {
     rocket::ignite()
         .manage(rocket_blog::init_db_pool())
-        .mount("/", routes![index])
+        .mount("/", routes![index, index_json])
         .launch();
 }