From c65b9e00d5d37722e7e4887e7165e19931048c83 Mon Sep 17 00:00:00 2001
From: Harald Eilertsen <haraldei@anduin.net>
Date: Mon, 20 Nov 2017 20:53:00 +0100
Subject: 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.
---
 src/main.rs | 43 +++++++++++++++++++++----------------------
 1 file changed, 21 insertions(+), 22 deletions(-)

(limited to 'src')

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();
 }
-- 
cgit v1.2.3