Add handlers for creating new Posts.

2 files changed, 51 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 1d20cb0..a709cd9 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -32,6 +32,15 @@ impl<'a> Responder<'a> for IndexTemplate<'a> {
     }
 }
 
+impl<'a> Responder<'a> for posts::NewPostTemplate<'a> {
+    fn respond_to(self, _: &Request) -> Result<Response<'static>, Status> {
+        Response::build()
+            .header(ContentType::HTML)
+            .sized_body(Cursor::new(format!("{}", &self)))
+            .ok()
+    }
+}
+
 fn get_posts(conn: rocket_blog::DbConn) -> Vec<models::Post> {
     use schema::posts::dsl::*;
     posts.filter(published.eq(false))
@@ -50,9 +59,40 @@ fn index_json(conn: rocket_blog::DbConn) -> Json<Vec<models::Post>> {
     Json(get_posts(conn))
 }
 
+mod posts {
+    use rocket::request::Form;
+    use rocket::response::Redirect;
+    use diesel::{self, ExecuteDsl};
+
+    #[derive(BartDisplay)]
+    #[template = "templates/new_post.html"]
+    pub struct NewPostTemplate<'a> {
+        title: &'a str,
+        post: ::models::NewPost
+    }
+
+    #[get("/new", format = "text/html")]
+    fn new<'a>(_conn: ::rocket_blog::DbConn) -> NewPostTemplate<'a> {
+        NewPostTemplate { title: "Bloggen", post: Default::default() }
+    }
+
+    #[post("/create", data="<post>")]
+    fn create(post: Form<::models::NewPost>, conn: ::rocket_blog::DbConn) -> Redirect {
+        use ::schema::posts;
+
+        diesel::insert(post.get())
+            .into(posts::table)
+            .execute(&*conn)
+            .expect("Error saving post.");
+
+        Redirect::to("/")
+    }
+}
+
 fn main() {
     rocket::ignite()
         .manage(rocket_blog::init_db_pool())
         .mount("/", routes![index, index_json])
+        .mount("/posts", routes![posts::new, posts::create])
         .launch();
 }
diff --git a/src/models.rs b/src/models.rs
index d8aa0aa..67038f1 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -1,7 +1,17 @@
-#[derive(Debug, Default, FromForm, Serialize, Queryable)]
+use super::schema::posts;
+
+#[derive(Serialize, Queryable)]
 pub struct Post {
     pub id: i32,
     pub title: String,
     pub body: String,
     pub published: bool,
 }
+
+#[derive(Default, FromForm, Insertable)]
+#[table_name="posts"]
+pub struct NewPost {
+    pub title: String,
+    pub body: String,
+    pub published: bool,
+}