This repository has been archived on 2023-07-04. You can view files and clone it, but cannot push or open issues/pull-requests.
blog/src/main.rs

76 lines
1.8 KiB
Rust
Raw Normal View History

2021-11-23 09:32:08 +01:00
#[macro_use]
extern crate rocket;
#[macro_use]
extern crate diesel_migrations;
use figment::{
providers::{Env, Format, Yaml},
Figment,
};
2021-11-23 17:49:31 +01:00
use rb::auth::JwtConf;
2021-11-23 09:32:08 +01:00
use rocket::{
fairing::AdHoc,
http::Status,
serde::json::{json, Value},
Build, Request, Rocket,
};
use rocket_sync_db_pools::database;
use serde::{Deserialize, Serialize};
pub mod posts;
pub mod sections;
#[database("postgres_rb")]
pub struct RbDbConn(diesel::PgConnection);
#[catch(default)]
fn default_catcher(status: Status, _: &Request) -> Value
{
2021-11-23 18:00:10 +01:00
json!({"status": status.code, "message": "Not found."})
2021-11-23 09:32:08 +01:00
}
embed_migrations!();
async fn run_db_migrations(rocket: Rocket<Build>) -> Result<Rocket<Build>, Rocket<Build>>
{
let conn = RbDbConn::get_one(&rocket)
.await
.expect("database connection");
conn.run(|c| match embedded_migrations::run(c) {
Ok(()) => Ok(rocket),
Err(_) => Err(rocket),
})
.await
}
#[derive(Debug, Deserialize, Serialize)]
pub struct RbConfig
{
2021-11-23 17:49:31 +01:00
jwt: JwtConf,
2021-11-23 09:32:08 +01:00
}
#[launch]
fn rocket() -> _
{
let figment = Figment::from(rocket::config::Config::default())
.merge(Yaml::file("Rb.yaml").nested())
.merge(Env::prefixed("RB_").global());
2021-11-23 18:00:10 +01:00
let rocket = rocket::custom(figment)
2021-11-23 09:32:08 +01:00
.attach(RbDbConn::fairing())
.attach(AdHoc::try_on_ignite(
"Run database migrations",
run_db_migrations,
))
// .attach(AdHoc::try_on_ignite("Create admin user", create_admin_user))
2021-11-23 18:00:10 +01:00
// .attach(AdHoc::config::<JwtConf>())
2021-11-23 09:32:08 +01:00
.register("/", catchers![default_catcher])
.mount("/sections", routes![sections::create_section])
2021-11-23 18:00:10 +01:00
.mount("/posts", routes![posts::get, posts::create]);
let new_figment = rocket.figment();
let jwt_conf: JwtConf = new_figment.extract_inner("jwt").expect("jwt config");
rocket.manage(jwt_conf)
2021-11-23 09:32:08 +01:00
}