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

#[macro_use]
extern crate rocket;
#[macro_use]
extern crate rocket_contrib;
#[macro_use]
extern crate diesel;
#[macro_use]
extern crate diesel_migrations;

use rocket::Rocket;

mod api;
mod backend;
mod models;
mod schema;
mod web;

embed_migrations!();

/// Automatically create/update our schema at launch
async fn run_db_migrations(rocket: Rocket) -> Result<Rocket, Rocket> {
    web::ToolsDb::get_one(&rocket)
        .await
        .expect("database connection")
        .run(|c| match embedded_migrations::run(c) {
            Ok(()) => Ok(rocket),
            Err(e) => {
                error!("Failed to run database migrations: {:?}", e);
                Err(rocket)
            }
        })
        .await
}

#[launch]
fn launch() -> Rocket {
    web::rocket()
}