src/main.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#![feature(plugin, custom_derive)]
#![plugin(rocket_codegen)]

#[macro_use] extern crate bart_derive;
#[macro_use] extern crate diesel_codegen;
#[macro_use] extern crate diesel;
#[macro_use] extern crate serde_derive;

extern crate dotenv;
extern crate r2d2;
extern crate r2d2_diesel;
extern crate rocket;
extern crate rocket_contrib;

use rocket_contrib::Json;

#[macro_use] mod utils;
mod models;
mod schema;
mod posts;

#[derive(BartDisplay, Serialize)]
#[template = "templates/index.html"]
struct IndexTemplate<'a> {
    title: &'a str,
    posts: Vec<models::Post>
}

implement_responder_for!(IndexTemplate<'a>);

#[get("/", format = "text/html")]
fn index<'a>(conn: utils::DbConn) -> IndexTemplate<'a> {
    IndexTemplate { title: "Bloggen", posts: models::Post::get_all(conn) }
}

#[get("/", format = "application/json")]
fn index_json(conn: utils::DbConn) -> Json<Vec<models::Post>> {
    Json(models::Post::get_all(conn))
}

fn main() {
    rocket::ignite()
        .manage(utils::init_db_pool())
        .mount("/", routes![index, index_json])
        .mount("/posts", routes![posts::new, posts::create, posts::show, posts::edit, posts::update])
        .launch();
}