70 lines
1.9 KiB
Rust
70 lines
1.9 KiB
Rust
#![recursion_limit = "256"]
|
|
#![feature(proc_macro_hygiene, decl_macro)]
|
|
|
|
#[macro_use]
|
|
extern crate diesel;
|
|
#[macro_use]
|
|
extern crate diesel_migrations;
|
|
|
|
use rocket::fairing::AdHoc;
|
|
use rocket::{catchers, routes, Rocket};
|
|
use rocket_contrib::database;
|
|
use rocket_contrib::serve::StaticFiles;
|
|
|
|
#[macro_use]
|
|
pub mod helpers;
|
|
|
|
pub mod catchers;
|
|
pub mod graphql;
|
|
pub mod handlers;
|
|
pub mod models;
|
|
/// The application database schema.
|
|
pub mod schema;
|
|
|
|
// This macro from `diesel_migrations` defines an `embedded_migrations` module containing a
|
|
// function named `run`. This allows the example to be run and tested without any outside setup
|
|
// of the database.
|
|
embed_migrations!();
|
|
|
|
#[database("stoptime_db")]
|
|
pub struct DbConn(diesel::SqliteConnection);
|
|
|
|
/// Runs the database migrations.
|
|
fn run_db_migrations(rocket: Rocket) -> Result<Rocket, Rocket> {
|
|
let conn = DbConn::get_one(&rocket).expect("Failed to get a database connection");
|
|
match embedded_migrations::run(&*conn) {
|
|
Ok(()) => Ok(rocket),
|
|
Err(e) => {
|
|
eprintln!("Failed to run database migrations: {:?}", e);
|
|
Err(rocket)
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Sets up the Rocket application.
|
|
fn rocket() -> Rocket {
|
|
let static_dir = concat!(env!("CARGO_MANIFEST_DIR"), "/static");
|
|
let static_files = StaticFiles::from(static_dir);
|
|
|
|
rocket::ignite()
|
|
.attach(DbConn::fairing())
|
|
.attach(AdHoc::on_attach("Database Migrations", run_db_migrations))
|
|
.manage(graphql::Schema::new(graphql::Query, graphql::Mutation))
|
|
.mount(
|
|
"/",
|
|
routes![
|
|
handlers::index,
|
|
handlers::graphql_playground,
|
|
handlers::graphql_get,
|
|
handlers::graphql_post,
|
|
],
|
|
)
|
|
.mount("/static", static_files)
|
|
.register(catchers![catchers::not_found])
|
|
}
|
|
|
|
/// Runs the Rocket application.
|
|
fn main() {
|
|
rocket().launch();
|
|
}
|